1"""
2Example 1: Basic Trace Capture
3 
4Demonstrates capturing reasoning traces from M2.1 for a simple task.
5This shows how interleaved thinking provides visibility into agent decisions.
6"""
7 
8import os
9from pathlib import Path
10from dotenv import load_dotenv
11 
12from reasoning_trace_optimizer import TraceCapture
13from reasoning_trace_optimizer.capture import format_trace_for_display
14 
15# Load environment variables from the project root
16env_path = Path(__file__).parent.parent / ".env"
17load_dotenv(env_path)
18 
19 
20def main():
21    """Run a simple task and capture the reasoning trace."""
22 
23    # Initialize capture with M2.1
24    capture = TraceCapture(
25        api_key=os.getenv("ANTHROPIC_API_KEY"),
26        base_url="https://api.minimax.io/anthropic",
27        model="MiniMax-M2.1",
28    )
29 
30    # Define a simple task
31    task = "Explain what interleaved thinking is and why it matters for AI agents."
32    system_prompt = "You are an AI researcher explaining concepts clearly."
33 
34    print("=" * 60)
35    print("BASIC TRACE CAPTURE EXAMPLE")
36    print("=" * 60)
37    print(f"\nTask: {task}")
38    print(f"System Prompt: {system_prompt}")
39    print("\nCapturing reasoning trace...\n")
40 
41    # Capture the trace
42    trace = capture.run(
43        task=task,
44        system_prompt=system_prompt,
45        max_turns=5,
46    )
47 
48    # Display the trace
49    print(format_trace_for_display(trace))
50 
51    # Summary statistics
52    print("\n" + "=" * 60)
53    print("TRACE STATISTICS")
54    print("=" * 60)
55    print(f"Session ID: {trace.session_id}")
56    print(f"Model: {trace.model}")
57    print(f"Success: {trace.success}")
58    print(f"Total Turns: {trace.total_turns}")
59    print(f"Thinking Blocks: {len(trace.thinking_blocks)}")
60    print(f"Tool Calls: {len(trace.tool_calls)}")
61    print(f"Total Tokens: {trace.total_tokens}")
62 
63    # Show each thinking block summary
64    if trace.thinking_blocks:
65        print("\n" + "=" * 60)
66        print("THINKING BLOCK SUMMARIES")
67        print("=" * 60)
68        for i, thinking in enumerate(trace.thinking_blocks):
69            print(f"\n[Turn {thinking.turn_index}] ({len(thinking.content)} chars)")
70            # Show first 200 chars
71            preview = thinking.content[:200].replace("\n", " ")
72            print(f"  Preview: {preview}...")
73 
74 
75if __name__ == "__main__":
76    main()
77