synth-ai 0.2.4.dev3__py3-none-any.whl → 0.2.4.dev4__py3-none-any.whl

synth_ai/tracing_v3/examples/basic_usage.py

@@ -0,0 +1,188 @@
+"""Basic usage example for tracing v3."""
+
+import asyncio
+import time
+from datetime import datetime
+
+from synth_ai.tracing_v3 import SessionTracer
+from synth_ai.tracing_v3.abstractions import LMCAISEvent, EnvironmentEvent, RuntimeEvent, TimeRecord
+from synth_ai.tracing_v3.turso.daemon import SqldDaemon
+
+
+async def simulate_llm_call(model: str, prompt: str) -> dict:
+    """Simulate an LLM API call."""
+    await asyncio.sleep(0.1)  # Simulate network latency
+
+    # Simulate response
+    tokens = len(prompt.split()) * 3
+    return {
+        "model": model,
+        "response": f"Response to: {prompt[:50]}...",
+        "usage": {
+            "prompt_tokens": len(prompt.split()) * 2,
+            "completion_tokens": tokens,
+            "total_tokens": len(prompt.split()) * 2 + tokens,
+        },
+    }
+
+
+async def main():
+    """Demonstrate basic tracing v3 usage."""
+    print("Starting tracing v3 example...")
+
+    # Option 1: Start sqld daemon programmatically
+    with SqldDaemon() as daemon:
+        print("✓ Started sqld daemon")
+
+        # Wait for daemon to be ready
+        await asyncio.sleep(1)
+
+        # Create tracer
+        tracer = SessionTracer()
+        await tracer.initialize()
+        print("✓ Initialized tracer")
+
+        # Example 1: Basic session with events
+        print("\n--- Example 1: Basic Session ---")
+        async with tracer.session(metadata={"example": "basic"}) as session_id:
+            print(f"Started session: {session_id}")
+
+            # Timestep 1: LLM interaction
+            async with tracer.timestep("llm_step", turn_number=1):
+                # Simulate LLM call
+                result = await simulate_llm_call("gpt-4", "What is the capital of France?")
+
+                # Record LLM event
+                event = LMCAISEvent(
+                    system_instance_id="llm_system",
+                    time_record=TimeRecord(event_time=time.time()),
+                    model_name=result["model"],
+                    input_tokens=result["usage"]["prompt_tokens"],
+                    output_tokens=result["usage"]["completion_tokens"],
+                    total_tokens=result["usage"]["total_tokens"],
+                    cost_usd=0.003,  # $0.003
+                    latency_ms=100,
+                    metadata={"prompt": "What is the capital of France?"},
+                )
+                await tracer.record_event(event)
+
+                # Record messages
+                await tracer.record_message(
+                    content="What is the capital of France?", message_type="user"
+                )
+                await tracer.record_message(content=result["response"], message_type="assistant")
+                print("✓ Recorded LLM interaction")
+
+            # Timestep 2: Environment interaction
+            async with tracer.timestep("env_step", turn_number=2):
+                # Record environment event
+                env_event = EnvironmentEvent(
+                    system_instance_id="environment",
+                    time_record=TimeRecord(event_time=time.time()),
+                    reward=0.8,
+                    terminated=False,
+                    system_state_before={"position": [0, 0]},
+                    system_state_after={"position": [1, 0]},
+                )
+                await tracer.record_event(env_event)
+                print("✓ Recorded environment event")
+
+            # Timestep 3: Runtime action
+            async with tracer.timestep("runtime_step", turn_number=3):
+                # Record runtime event
+                runtime_event = RuntimeEvent(
+                    system_instance_id="agent",
+                    time_record=TimeRecord(event_time=time.time()),
+                    actions=[1, 0, 0, 1],  # Example action vector
+                    metadata={"action_type": "move_right"},
+                )
+                await tracer.record_event(runtime_event)
+                print("✓ Recorded runtime event")
+
+        print(f"✓ Session {session_id} saved\n")
+
+        # Example 2: Concurrent sessions
+        print("--- Example 2: Concurrent Sessions ---")
+
+        async def run_concurrent_session(session_num: int):
+            """Run a session concurrently."""
+            async with tracer.session(
+                metadata={"example": "concurrent", "session_num": session_num}
+            ) as sid:
+                for i in range(3):
+                    async with tracer.timestep(f"step_{i}", turn_number=i):
+                        # Simulate some work
+                        await asyncio.sleep(0.05)
+
+                        # Record event
+                        event = RuntimeEvent(
+                            system_instance_id=f"worker_{session_num}",
+                            time_record=TimeRecord(event_time=time.time()),
+                            actions=[i],
+                            metadata={"iteration": i},
+                        )
+                        await tracer.record_event(event)
+
+                return sid
+
+        # Run 5 concurrent sessions
+        tasks = [run_concurrent_session(i) for i in range(5)]
+        session_ids = await asyncio.gather(*tasks)
+        print(f"✓ Completed {len(session_ids)} concurrent sessions")
+
+        # Example 3: Query stored data
+        print("\n--- Example 3: Querying Data ---")
+
+        # Get model usage statistics
+        model_usage = await tracer.db.get_model_usage()
+        print("\nModel Usage:")
+        print(model_usage)
+
+        # Query recent sessions
+        recent_sessions = await tracer.get_session_history(limit=5)
+        print(f"\nRecent Sessions: {len(recent_sessions)} found")
+        for session in recent_sessions:
+            print(
+                f"  - {session['session_id']}: "
+                f"{session['num_events']} events, "
+                f"{session['num_messages']} messages"
+            )
+
+        # Get specific session details
+        if recent_sessions:
+            session_detail = await tracer.db.get_session_trace(recent_sessions[0]["session_id"])
+            print(f"\nSession Detail for {session_detail['session_id']}:")
+            print(f"  Created: {session_detail['created_at']}")
+            print(f"  Timesteps: {len(session_detail['timesteps'])}")
+
+        # Example 4: Using hooks
+        print("\n--- Example 4: Hooks ---")
+
+        # Add a custom hook
+        call_count = {"count": 0}
+
+        async def count_events(event, **kwargs):
+            call_count["count"] += 1
+            print(f"  Hook: Event #{call_count['count']} recorded")
+
+        tracer.hooks.register("event_recorded", count_events, name="event_counter")
+
+        async with tracer.session(metadata={"example": "hooks"}) as session_id:
+            async with tracer.timestep("hook_test"):
+                for i in range(3):
+                    event = RuntimeEvent(
+                        system_instance_id="hook_test",
+                        time_record=TimeRecord(event_time=time.time()),
+                        actions=[i],
+                    )
+                    await tracer.record_event(event)
+
+        print(f"✓ Hook called {call_count['count']} times")
+
+        # Cleanup
+        await tracer.close()
+        print("\n✓ Example completed successfully!")
+
+
+if __name__ == "__main__":
+    asyncio.run(main())

{synth_ai-0.2.4.dev3.dist-info → synth_ai-0.2.4.dev4.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: synth-ai
-Version: 0.2.4.dev3
+Version: 0.2.4.dev4
 Summary: Software for aiding the best and multiplying the will - Core AI functionality and tracing
 Author-email: Synth AI <josh@usesynth.ai>
 License-Expression: MIT

{synth_ai-0.2.4.dev3.dist-info → synth_ai-0.2.4.dev4.dist-info}/RECORD

@@ -126,6 +126,7 @@ synth_ai/tracing_v3/migration_helper.py,sha256=q6G4v3UKJAU4nMytz7znIK1M3xeLGObxl
 synth_ai/tracing_v3/replica_sync.py,sha256=CP7hPCR2oETrdsInwQl79mdInFA5xL1Q4Nzp3KGLu5A,8879
 synth_ai/tracing_v3/session_tracer.py,sha256=zPCtO73oXVsHN5N64_pleporKiMuAmue9dStAZSFESA,11379
 synth_ai/tracing_v3/utils.py,sha256=RAFZTf8JyRCQsR9DwJ1ykelKkoY3OSrCBUStCLvpn8w,3431
+synth_ai/tracing_v3/examples/basic_usage.py,sha256=zbl9bJaZawToMEV0skzvP4sIwLnmBeo7Pa_f1gJlkf8,7265
 synth_ai/tracing_v3/storage/__init__.py,sha256=S_g7h34JZ6v1hoCcrlP7TaAAU-f0owKUx78QB8TUGkQ,332
 synth_ai/tracing_v3/storage/base.py,sha256=FlV3NMXsXK7MZaUqc7uXCv2YFD021C6LvFaPppyYr0A,3611
 synth_ai/tracing_v3/storage/config.py,sha256=ZvBum3wkDtXjm_p5mcEMjBtPNSUHjDh-fhZqVxXC4JQ,2096
@@ -180,9 +181,9 @@ synth_ai/v0/tracing_v1/events/manage.py,sha256=dgIlH43aEz7yA1_rj4IisJP6ROiROmLwD
 synth_ai/v0/tracing_v1/events/scope.py,sha256=BuBkhSpVHUJt8iGT9HJZF82rbb88mQcd2vM2shg-w2I,2550
 synth_ai/v0/tracing_v1/events/store.py,sha256=gVM3L_bV_PAEh9GD9eg2AQqNfANu-zH9ied7bAKk_Eg,9040
 synth_ai/zyk/__init__.py,sha256=6HFa13sdHCouhzKOUPEWcJeVxKlzjWO7ERsrrwtTugs,752
-synth_ai-0.2.4.dev3.dist-info/licenses/LICENSE,sha256=ynhjRQUfqA_RdGRATApfFA_fBAy9cno04sLtLUqxVFM,1069
-synth_ai-0.2.4.dev3.dist-info/METADATA,sha256=ohKQPWyVxScWAB8xBzTvSW8hOrkEt_ttXEAa-LBIQ5I,7150
-synth_ai-0.2.4.dev3.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-synth_ai-0.2.4.dev3.dist-info/entry_points.txt,sha256=GSFXaJreq4PJXbixkUI0GHZwGh2dZDG5pYaoVmqr_KE,46
-synth_ai-0.2.4.dev3.dist-info/top_level.txt,sha256=fBmtZyVHuKaGa29oHBaaUkrUIWTqSpoVMPiVdCDP3k8,9
-synth_ai-0.2.4.dev3.dist-info/RECORD,,
+synth_ai-0.2.4.dev4.dist-info/licenses/LICENSE,sha256=ynhjRQUfqA_RdGRATApfFA_fBAy9cno04sLtLUqxVFM,1069
+synth_ai-0.2.4.dev4.dist-info/METADATA,sha256=rQqUijPG-9CeDt33ihFOSoDyKrQpF_3obxoNG-tCCJg,7150
+synth_ai-0.2.4.dev4.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+synth_ai-0.2.4.dev4.dist-info/entry_points.txt,sha256=GSFXaJreq4PJXbixkUI0GHZwGh2dZDG5pYaoVmqr_KE,46
+synth_ai-0.2.4.dev4.dist-info/top_level.txt,sha256=fBmtZyVHuKaGa29oHBaaUkrUIWTqSpoVMPiVdCDP3k8,9
+synth_ai-0.2.4.dev4.dist-info/RECORD,,