synth-ai 0.2.14__py3-none-any.whl → 0.2.16__py3-none-any.whl

synth_ai/tracing_v3/storage/base.py

@@ -2,7 +2,7 @@
 
 from abc import ABC, abstractmethod
 from datetime import datetime
-from typing import Any, Optional
+from typing import Any
 
 from ..abstractions import SessionTrace
 
@@ -28,7 +28,7 @@ class TraceStorage(ABC):
         pass
 
     @abstractmethod
-    async def get_session_trace(self, session_id: str) -> Optional[dict[str, Any]]:
+    async def get_session_trace(self, session_id: str) -> dict[str, Any] | None:
         """Retrieve a session trace by ID.
 
         Args:
@@ -40,7 +40,7 @@ class TraceStorage(ABC):
         pass
 
     @abstractmethod
-    async def query_traces(self, query: str, params: Optional[dict[str, Any]] = None) -> Any:
+    async def query_traces(self, query: str, params: dict[str, Any] | None = None) -> Any:
         """Execute a query and return results.
 
         Args:
@@ -55,9 +55,9 @@ class TraceStorage(ABC):
     @abstractmethod
     async def get_model_usage(
         self,
-        start_date: Optional[datetime] = None,
-        end_date: Optional[datetime] = None,
-        model_name: Optional[str] = None,
+        start_date: datetime | None = None,
+        end_date: datetime | None = None,
+        model_name: str | None = None,
     ) -> Any:
         """Get model usage statistics.
 
@@ -95,8 +95,8 @@ class TraceStorage(ABC):
         self,
         session_id: str,
         *,
-        created_at: Optional[datetime] = None,
-        metadata: Optional[dict[str, Any]] = None,
+        created_at: datetime | None = None,
+        metadata: dict[str, Any] | None = None,
     ) -> None:
         """Ensure a session row exists for the given session id."""
         pass
@@ -108,10 +108,10 @@ class TraceStorage(ABC):
         *,
         step_id: str,
         step_index: int,
-        turn_number: Optional[int] = None,
-        started_at: Optional[datetime] = None,
-        completed_at: Optional[datetime] = None,
-        metadata: Optional[dict[str, Any]] = None,
+        turn_number: int | None = None,
+        started_at: datetime | None = None,
+        completed_at: datetime | None = None,
+        metadata: dict[str, Any] | None = None,
     ) -> int:
         """Ensure a timestep row exists and return its database id."""
         pass
@@ -121,9 +121,9 @@ class TraceStorage(ABC):
         self,
         session_id: str,
         *,
-        timestep_db_id: Optional[int],
+        timestep_db_id: int | None,
         event: Any,
-        metadata_override: Optional[dict[str, Any]] = None,
+        metadata_override: dict[str, Any] | None = None,
     ) -> int:
         """Insert an event and return its database id."""
         pass
@@ -133,12 +133,12 @@ class TraceStorage(ABC):
         self,
         session_id: str,
         *,
-        timestep_db_id: Optional[int],
+        timestep_db_id: int | None,
         message_type: str,
         content: Any,
-        event_time: Optional[float] = None,
-        message_time: Optional[int] = None,
-        metadata: Optional[dict[str, Any]] = None,
+        event_time: float | None = None,
+        message_time: int | None = None,
+        metadata: dict[str, Any] | None = None,
     ) -> int:
         """Insert a message row linked to a session/timestep."""
         pass
@@ -151,7 +151,7 @@ class TraceStorage(ABC):
         total_reward: int,
         achievements_count: int,
         total_steps: int,
-        reward_metadata: Optional[dict] = None,
+        reward_metadata: dict | None = None,
     ) -> int:
         """Record an outcome reward for a session."""
         pass
@@ -162,13 +162,13 @@ class TraceStorage(ABC):
         session_id: str,
         *,
         event_id: int,
-        message_id: Optional[int] = None,
-        turn_number: Optional[int] = None,
+        message_id: int | None = None,
+        turn_number: int | None = None,
         reward_value: float = 0.0,
-        reward_type: Optional[str] = None,
-        key: Optional[str] = None,
-        annotation: Optional[dict[str, Any]] = None,
-        source: Optional[str] = None,
+        reward_type: str | None = None,
+        key: str | None = None,
+        annotation: dict[str, Any] | None = None,
+        source: str | None = None,
     ) -> int:
         """Record a reward tied to a specific event."""
         pass
@@ -178,8 +178,8 @@ class TraceStorage(ABC):
         self,
         experiment_id: str,
         name: str,
-        description: Optional[str] = None,
-        configuration: Optional[dict[str, Any]] = None,
+        description: str | None = None,
+        configuration: dict[str, Any] | None = None,
     ) -> str:
         """Create a new experiment."""
         raise NotImplementedError("Experiment management not supported by this backend")
@@ -189,14 +189,14 @@ class TraceStorage(ABC):
         raise NotImplementedError("Experiment management not supported by this backend")
 
     async def get_sessions_by_experiment(
-        self, experiment_id: str, limit: Optional[int] = None
+        self, experiment_id: str, limit: int | None = None
     ) -> list[dict[str, Any]]:
         """Get all sessions for an experiment."""
         raise NotImplementedError("Experiment management not supported by this backend")
 
     # Batch operations
     async def batch_insert_sessions(
-        self, traces: list[SessionTrace], batch_size: Optional[int] = 1000
+        self, traces: list[SessionTrace], batch_size: int | None = 1000
     ) -> list[str]:
         """Batch insert multiple session traces.
 

synth_ai/tracing_v3/storage/config.py

@@ -3,7 +3,7 @@
 import os
 from dataclasses import dataclass
 from enum import Enum
-from typing import Any, Optional
+from typing import Any
 
 
 class StorageBackend(str, Enum):
@@ -14,7 +14,7 @@ class StorageBackend(str, Enum):
     POSTGRES = "postgres"  # Future support
 
 
-def _is_enabled(value: Optional[str]) -> bool:
+def _is_enabled(value: str | None) -> bool:
     if value is None:
         return False
     return value.lower() in {"1", "true", "yes", "on"}
@@ -25,7 +25,7 @@ class StorageConfig:
     """Configuration for storage backend."""
 
     backend: StorageBackend = StorageBackend.TURSO_NATIVE
-    connection_string: Optional[str] = None
+    connection_string: str | None = None
 
     # Turso-specific settings
     turso_url: str = os.getenv("TURSO_DATABASE_URL", "sqlite+libsql://http://127.0.0.1:8080")

synth_ai/tracing_v3/turso/daemon.py

@@ -7,7 +7,6 @@ import time
 
 import requests
 from requests import RequestException
-from typing import Any, Optional
 
 from ..config import CONFIG
 
@@ -17,9 +16,9 @@ class SqldDaemon:
 
     def __init__(
         self,
-        db_path: Optional[str] = None,
-        http_port: Optional[int] = None,
-        binary_path: Optional[str] = None,
+        db_path: str | None = None,
+        http_port: int | None = None,
+        binary_path: str | None = None,
     ):
         """Initialize sqld daemon manager.
 
@@ -31,7 +30,7 @@ class SqldDaemon:
         self.db_path = db_path or CONFIG.sqld_db_path
         self.http_port = http_port or CONFIG.sqld_http_port
         self.binary_path = binary_path or self._find_binary()
-        self.process: Optional[Any] = None
+        self.process: subprocess.Popen[str] | None = None
 
     def _find_binary(self) -> str:
         """Find sqld binary in PATH."""
@@ -85,7 +84,7 @@ class SqldDaemon:
                 pass
 
             # Check if process crashed
-            if self.process.poll() is not None:
+            if self.process and self.process.poll() is not None:
                 stdout, stderr = self.process.communicate()
                 raise RuntimeError(
                     f"sqld daemon failed to start:\nstdout: {stdout}\nstderr: {stderr}"
@@ -124,10 +123,10 @@ class SqldDaemon:
 
 
 # Convenience functions
-_daemon: Optional[SqldDaemon] = None
+_daemon: SqldDaemon | None = None
 
 
-def start_sqld(db_path: Optional[str] = None, port: Optional[int] = None) -> SqldDaemon:
+def start_sqld(db_path: str | None = None, port: int | None = None) -> SqldDaemon:
     """Start a global sqld daemon instance."""
     global _daemon
     if _daemon and _daemon.is_running():
@@ -146,6 +145,6 @@ def stop_sqld():
         _daemon = None
 
 
-def get_daemon() -> Optional[SqldDaemon]:
+def get_daemon() -> SqldDaemon | None:
     """Get the global daemon instance."""
     return _daemon

synth_ai/tracing_v3/turso/native_manager.py

@@ -13,7 +13,7 @@ import logging
 import re
 from collections.abc import Callable
 from dataclasses import asdict, dataclass
-from datetime import datetime, timezone
+from datetime import UTC, datetime
 from typing import TYPE_CHECKING, Any, cast
 
 import libsql
@@ -378,8 +378,10 @@ class NativeLibsqlTraceManager(TraceStorage):
         session_exists = await self._session_exists(trace.session_id)
         _logger.info(f"[TRACE_DEBUG] Session exists: {session_exists}")
         
+        step_id_map: dict[str, int] = {}
+        
         if session_exists:
-            _logger.warning(f"[TRACE_DEBUG] Session {trace.session_id} already exists, need to save messages anyway!")
+            _logger.warning(f"[TRACE_DEBUG] Session {trace.session_id} already exists, skipping events/timesteps, only updating messages!")
             # Don't return early - we need to save messages!
             # Just update metadata
             async with self._op_lock:
@@ -390,10 +392,9 @@ class NativeLibsqlTraceManager(TraceStorage):
                     (_json_dumps(trace.metadata or {}), trace.session_id),
                 )
                 conn.commit()
-            # Continue to save messages instead of returning
-
-        if not session_exists:
-            created_at = trace.created_at or datetime.now(timezone.utc)
+            # Skip events and timesteps to ensure idempotency
+        else:
+            created_at = trace.created_at or datetime.now(UTC)
 
             async with self._op_lock:
                 conn = self._conn
@@ -417,73 +418,76 @@ class NativeLibsqlTraceManager(TraceStorage):
                     ),
                 )
                 conn.commit()
-                _logger.info(f"[TRACE_DEBUG] Session row inserted")
-
-        step_id_map: dict[str, int] = {}
+                _logger.info("[TRACE_DEBUG] Session row inserted")
 
-        for step in trace.session_time_steps:
-            step_db_id = await self.ensure_timestep(
-                trace.session_id,
-                step_id=step.step_id,
-                step_index=step.step_index,
-                turn_number=step.turn_number,
-                started_at=step.timestamp,
-                completed_at=step.completed_at,
-                metadata=step.step_metadata or {},
-            )
-            step_id_map[step.step_id] = step_db_id
-
-        for event in trace.event_history:
-            step_ref = None
-            metadata = event.metadata or {}
-            if isinstance(metadata, dict):
-                step_ref = metadata.get("step_id")
-            timestep_db_id = step_id_map.get(step_ref) if step_ref else None
-            await self.insert_event_row(
-                trace.session_id,
-                timestep_db_id=timestep_db_id,
-                event=event,
-                metadata_override=event.metadata or {},
-            )
+            # Only insert timesteps and events if this is a new session
+            for step in trace.session_time_steps:
+                step_db_id = await self.ensure_timestep(
+                    trace.session_id,
+                    step_id=step.step_id,
+                    step_index=step.step_index,
+                    turn_number=step.turn_number,
+                    started_at=step.timestamp,
+                    completed_at=step.completed_at,
+                    metadata=step.step_metadata or {},
+                )
+                step_id_map[step.step_id] = step_db_id
+
+            for event in trace.event_history:
+                step_ref = None
+                metadata = event.metadata or {}
+                if isinstance(metadata, dict):
+                    step_ref = metadata.get("step_id")
+                timestep_db_id = step_id_map.get(step_ref) if step_ref else None
+                await self.insert_event_row(
+                    trace.session_id,
+                    timestep_db_id=timestep_db_id,
+                    event=event,
+                    metadata_override=event.metadata or {},
+                )
 
         import logging as _logging
         _logger = _logging.getLogger(__name__)
-        _logger.info(f"[TRACE_DEBUG] insert_session_trace: saving {len(trace.markov_blanket_message_history)} messages")
+        _logger.info(f"[TRACE_DEBUG] insert_session_trace: saving {len(trace.markov_blanket_message_history)} messages (session_exists={session_exists})")
         
-        for idx, msg in enumerate(trace.markov_blanket_message_history):
-            metadata = dict(getattr(msg, "metadata", {}) or {})
-            step_ref = metadata.get("step_id")
-            content_value = msg.content
-            if isinstance(msg.content, SessionMessageContent):
-                if msg.content.json_payload:
-                    metadata.setdefault("json_payload", msg.content.json_payload)
-                    content_value = msg.content.json_payload
-                else:
-                    content_value = msg.content.as_text()
-                    if msg.content.text:
-                        metadata.setdefault("text", msg.content.text)
-            elif not isinstance(content_value, str):
+        # Only insert messages if this is a new session (for idempotency)
+        if not session_exists:
+            for idx, msg in enumerate(trace.markov_blanket_message_history):
+                metadata = dict(getattr(msg, "metadata", {}) or {})
+                step_ref = metadata.get("step_id")
+                content_value = msg.content
+                if isinstance(msg.content, SessionMessageContent):
+                    if msg.content.json_payload:
+                        metadata.setdefault("json_payload", msg.content.json_payload)
+                        content_value = msg.content.json_payload
+                    else:
+                        content_value = msg.content.as_text()
+                        if msg.content.text:
+                            metadata.setdefault("text", msg.content.text)
+                elif not isinstance(content_value, str):
+                    try:
+                        content_value = json.dumps(content_value, ensure_ascii=False)
+                    except (TypeError, ValueError):
+                        content_value = str(content_value)
+
+                _logger.info(f"[TRACE_DEBUG]   Message {idx+1}: type={msg.message_type}, content_len={len(str(content_value))}")
+                
                 try:
-                    content_value = json.dumps(content_value, ensure_ascii=False)
-                except (TypeError, ValueError):
-                    content_value = str(content_value)
-
-            _logger.info(f"[TRACE_DEBUG]   Message {idx+1}: type={msg.message_type}, content_len={len(str(content_value))}")
-            
-            try:
-                await self.insert_message_row(
-                    trace.session_id,
-                    timestep_db_id=step_id_map.get(step_ref) if step_ref else None,
-                    message_type=msg.message_type,
-                    content=content_value,
-                    event_time=msg.time_record.event_time,
-                    message_time=msg.time_record.message_time,
-                    metadata=metadata,
-                )
-                _logger.info(f"[TRACE_DEBUG]   Message {idx+1}: saved successfully")
-            except Exception as exc:
-                _logger.error(f"[TRACE_DEBUG]   Message {idx+1}: FAILED TO SAVE: {exc}", exc_info=True)
-                raise
+                    await self.insert_message_row(
+                        trace.session_id,
+                        timestep_db_id=step_id_map.get(step_ref) if step_ref else None,
+                        message_type=msg.message_type,
+                        content=content_value,
+                        event_time=msg.time_record.event_time,
+                        message_time=msg.time_record.message_time,
+                        metadata=metadata,
+                    )
+                    _logger.info(f"[TRACE_DEBUG]   Message {idx+1}: saved successfully")
+                except Exception as exc:
+                    _logger.error(f"[TRACE_DEBUG]   Message {idx+1}: FAILED TO SAVE: {exc}", exc_info=True)
+                    raise
+        else:
+            _logger.info("[TRACE_DEBUG] Skipping message insertion for existing session (idempotency)")
 
         async with self._op_lock:
             conn = self._conn
@@ -806,7 +810,7 @@ class NativeLibsqlTraceManager(TraceStorage):
     ) -> None:
         await self.initialize()
 
-        created_at_val = (created_at or datetime.now(timezone.utc)).isoformat()
+        created_at_val = (created_at or datetime.now(UTC)).isoformat()
         metadata_json = _json_dumps(metadata or {})
 
         async with self._op_lock:
@@ -838,7 +842,7 @@ class NativeLibsqlTraceManager(TraceStorage):
     ) -> int:
         await self.initialize()
 
-        started_at_val = (started_at or datetime.now(timezone.utc)).isoformat()
+        started_at_val = (started_at or datetime.now(UTC)).isoformat()
         completed_at_val = completed_at.isoformat() if completed_at else None
         metadata_json = _json_dumps(metadata or {})
 
@@ -927,7 +931,11 @@ class NativeLibsqlTraceManager(TraceStorage):
         if isinstance(event, LMCAISEvent):
             call_records = None
             if getattr(event, "call_records", None):
-                call_records = [asdict(record) for record in event.call_records]
+                # Handle both dataclass instances and dicts (from deserialization)
+                call_records = [
+                    asdict(record) if not isinstance(record, dict) else record
+                    for record in event.call_records
+                ]
             payload.update(
                 {
                     "event_type": "cais",
@@ -1150,7 +1158,7 @@ class NativeLibsqlTraceManager(TraceStorage):
                     total_reward,
                     achievements_count,
                     total_steps,
-                    datetime.now(timezone.utc).isoformat(),
+                    datetime.now(UTC).isoformat(),
                     _json_dumps(reward_metadata),
                 ),
             )
@@ -1202,7 +1210,7 @@ class NativeLibsqlTraceManager(TraceStorage):
                     key,
                     _json_dumps(annotation),
                     source,
-                    datetime.now(timezone.utc).isoformat(),
+                    datetime.now(UTC).isoformat(),
                 ),
             )
             conn.commit()

synth_ai/tracing_v3/utils.py

@@ -5,13 +5,13 @@ from __future__ import annotations
 import hashlib
 import json
 import uuid
-from datetime import datetime, timezone
+from datetime import UTC, datetime
 from typing import Any
 
 
 def iso_now() -> str:
     """Get current timezone.utc time as ISO format string."""
-    return datetime.now(timezone.utc).isoformat()
+    return datetime.now(UTC).isoformat()
 
 
 def json_dumps(obj: Any) -> str:

synth_ai/tui/cli/query_experiments.py

@@ -6,12 +6,12 @@ Query experiments and sessions from Turso/sqld using v3 tracing.
 import argparse
 import asyncio
 
-from synth_ai.tracing_v3.turso.manager import AsyncSQLTraceManager
+from synth_ai.tracing_v3.turso import NativeLibsqlTraceManager
 
 
 async def list_experiments(db_url: str):
     """List all experiments in the database."""
-    db = AsyncSQLTraceManager(db_url)
+    db = NativeLibsqlTraceManager(db_url)
     await db.initialize()
 
     try:
@@ -57,7 +57,7 @@ async def list_experiments(db_url: str):
 
 async def show_experiment_details(db_url: str, experiment_id: str):
     """Show detailed information about a specific experiment."""
-    db = AsyncSQLTraceManager(db_url)
+    db = NativeLibsqlTraceManager(db_url)
     await db.initialize()
 
     try:
@@ -120,7 +120,7 @@ async def show_experiment_details(db_url: str, experiment_id: str):
 
 async def show_model_usage(db_url: str, model_name: str | None = None):
     """Show model usage statistics."""
-    db = AsyncSQLTraceManager(db_url)
+    db = NativeLibsqlTraceManager(db_url)
     await db.initialize()
 
     try:

synth_ai/tui/cli/query_experiments_v3.py

@@ -6,12 +6,12 @@ Query experiments and sessions from Turso/sqld using v3 tracing.
 import argparse
 import asyncio
 
-from synth_ai.tracing_v3.turso.manager import AsyncSQLTraceManager
+from synth_ai.tracing_v3.turso import NativeLibsqlTraceManager
 
 
 async def list_experiments(db_url: str):
     """List all experiments in the database."""
-    db = AsyncSQLTraceManager(db_url)
+    db = NativeLibsqlTraceManager(db_url)
     await db.initialize()
 
     try:
@@ -57,7 +57,7 @@ async def list_experiments(db_url: str):
 
 async def show_experiment_details(db_url: str, experiment_id: str):
     """Show detailed information about a specific experiment."""
-    db = AsyncSQLTraceManager(db_url)
+    db = NativeLibsqlTraceManager(db_url)
     await db.initialize()
 
     try:
@@ -120,7 +120,7 @@ async def show_experiment_details(db_url: str, experiment_id: str):
 
 async def show_model_usage(db_url: str, model_name: str | None = None):
     """Show model usage statistics."""
-    db = AsyncSQLTraceManager(db_url)
+    db = NativeLibsqlTraceManager(db_url)
     await db.initialize()
 
     try:

synth_ai/tui/dashboard.py

@@ -32,7 +32,10 @@ except (ImportError, ModuleNotFoundError):
     ComposeResult = object  # type: ignore
     Binding = object  # type: ignore
     Container = object  # type: ignore
-    reactive = lambda x: x  # type: ignore
+
+    def reactive(value, *_, **__):
+        return value
+
     Timer = object  # type: ignore
     DataTable = object  # type: ignore
     Footer = object  # type: ignore
@@ -42,17 +45,19 @@ except (ImportError, ModuleNotFoundError):
 
 # Import database manager with graceful fallback
 try:
-    from synth_ai.tracing_v3.turso.native_manager import NativeLibsqlTraceManager  # type: ignore[import-untyped]
+    from synth_ai.tracing_v3.turso.native_manager import (
+        NativeLibsqlTraceManager,  # type: ignore[import-untyped]
+    )
     _DB_AVAILABLE = True
 except (ImportError, ModuleNotFoundError, TypeError):
     # Database manager not available - provide dummy class
     NativeLibsqlTraceManager = object  # type: ignore
     _DB_AVAILABLE = False
 
-import asyncio
-import requests
 from datetime import timedelta
 
+import requests
+
 
 class ExperimentRow:
     """Data structure for experiment display."""
@@ -218,8 +223,8 @@ class DatabaseStatus(Static):
                 path_part = url.split("///")[-1]
                 filename = Path(path_part).name
                 self.connection_status = f"🟢 {filename}"
-            except:
-                self.connection_status = f"🟢 Connected"
+            except Exception:
+                self.connection_status = "🟢 Connected"
         else:
             host_info = f"{parsed.hostname}:{parsed.port}" if parsed.port else str(parsed.hostname)
             self.connection_status = f"🟢 {host_info}"
@@ -317,10 +322,10 @@ class BalanceStatus(Static):
 
     def set_global_error(self, error: str):
         """Show error state for global data."""
-        self.global_balance = f"Error"
+        self.global_balance = "Error"
         self.global_spend_24h = "-"
         self.global_spend_7d = "-"
-        self.global_status = f"❌"
+        self.global_status = "❌"
         
     def set_local_error(self, error: str):
         """Show error state for local data."""
@@ -328,7 +333,7 @@ class BalanceStatus(Static):
         self.local_cost = "Error"
         self.local_tokens = 0
         self.local_tasks = []
-        self.local_status = f"❌"
+        self.local_status = "❌"
     
     def set_global_unavailable(self):
         """Mark global data as unavailable (no API key)."""