synth-ai 0.2.4.dev6__py3-none-any.whl → 0.2.4.dev8__py3-none-any.whl

synth_ai/v0/tracing_v1/local.py

@@ -1,7 +1,7 @@
 import logging
 import threading
 from contextvars import ContextVar
-from typing import Any, Dict
+from typing import Any
 
 logger = logging.getLogger(__name__)
 
@@ -12,7 +12,7 @@ _local = threading.local()
 system_name_var: ContextVar[str] = ContextVar("system_name", default=None)
 system_id_var: ContextVar[str] = ContextVar("system_id", default=None)
 system_instance_id_var: ContextVar[str] = ContextVar("system_instance_id", default=None)
-system_instance_metadata_var: ContextVar[Dict[str, Any]] = ContextVar(
+system_instance_metadata_var: ContextVar[dict[str, Any]] = ContextVar(
     "system_instance_metadata", default={}
 )
-active_events_var: ContextVar[Dict[str, Any]] = ContextVar("active_events", default={})
+active_events_var: ContextVar[dict[str, Any]] = ContextVar("active_events", default={})

synth_ai/v0/tracing_v1/log_client_base.py

@@ -1,4 +1,3 @@
-from typing import Dict
 
 from .abstractions import Event
 from .config import TracingConfig
@@ -11,7 +10,7 @@ class BaseLogClient:
         self.config = config
         self.client_manager = None
 
-    def _prepare_payload(self, event: Event, system_info: Dict[str, str]) -> Dict:
+    def _prepare_payload(self, event: Event, system_info: dict[str, str]) -> dict:
         """Prepare the payload for sending."""
         return {
             "event": event.to_dict(),
@@ -33,7 +32,7 @@ class BaseLogClient:
         pass
 
     def _handle_failure(
-        self, event: Event, system_info: Dict[str, str], exception: Exception
+        self, event: Event, system_info: dict[str, str], exception: Exception
     ) -> None:
         """Handle failed event sending."""
         pass
@@ -46,7 +45,7 @@ class BaseAsyncLogClient:
         self.config = config
         self.client_manager = None
 
-    def _prepare_payload(self, event: Event, system_info: Dict[str, str]) -> Dict:
+    def _prepare_payload(self, event: Event, system_info: dict[str, str]) -> dict:
         """Prepare the payload for sending."""
         return {
             "event": event.to_dict(),
@@ -68,7 +67,7 @@ class BaseAsyncLogClient:
         pass
 
     def _handle_failure(
-        self, event: Event, system_info: Dict[str, str], exception: Exception
+        self, event: Event, system_info: dict[str, str], exception: Exception
     ) -> None:
         """Handle failed event sending."""
         pass

synth_ai/v0/tracing_v1/retry_queue.py

@@ -3,7 +3,6 @@ import threading
 import time
 from collections import deque
 from dataclasses import dataclass
-from typing import Dict, List, Optional, Tuple
 
 from .abstractions import Event
 from .config import TracingConfig
@@ -16,7 +15,7 @@ class QueuedEvent:
     """Represents an event that failed to upload and needs to be retried."""
 
     event: Event
-    system_info: Dict[str, str]
+    system_info: dict[str, str]
     attempt_count: int = 0
     last_attempt: float = 0
 
@@ -31,7 +30,7 @@ class RetryQueue:
         self._is_processing = False
         self._batch_size = config.batch_size
 
-    def add_failed_event(self, event: Event, system_info: Dict[str, str]) -> None:
+    def add_failed_event(self, event: Event, system_info: dict[str, str]) -> None:
         """Add a failed event to the retry queue."""
         with self._lock:
             # Check if event is already in queue to avoid duplicates
@@ -53,7 +52,7 @@ class RetryQueue:
             )
             logger.debug(f"Added event to retry queue. Queue size: {len(self.queue)}")
 
-    def get_retryable_events(self, max_events: Optional[int] = None) -> List[QueuedEvent]:
+    def get_retryable_events(self, max_events: int | None = None) -> list[QueuedEvent]:
         """Get events that are ready to be retried."""
         now = time.time()
         retryable = []
@@ -74,7 +73,7 @@ class RetryQueue:
 
         return retryable
 
-    def process_sync(self) -> Tuple[int, int]:
+    def process_sync(self) -> tuple[int, int]:
         """Process the retry queue synchronously.
 
         Returns:
@@ -125,7 +124,7 @@ class RetryQueue:
 
         return success_count, failure_count
 
-    async def process_async(self) -> Tuple[int, int]:
+    async def process_async(self) -> tuple[int, int]:
         """Process the retry queue asynchronously.
 
         Returns:

synth_ai/v0/tracing_v1/trackers.py

@@ -1,6 +1,6 @@
 import asyncio
 import contextvars
-from typing import Any, Dict, List, Literal, Optional, Tuple, Union
+from typing import Any, Literal
 
 from pydantic import BaseModel
 
@@ -29,9 +29,9 @@ class SynthTrackerSync:
     @classmethod
     def track_lm(
         cls,
-        messages: List[Dict[str, str]],
+        messages: list[dict[str, str]],
         model_name: str,
-        model_params: Optional[Dict[str, Union[str, int, float]]] = None,
+        model_params: dict[str, str | int | float] | None = None,
         finetune: bool = False,
     ):
         # print("Tracking LM call in sync context - ",messages)  # Added logging
@@ -52,9 +52,9 @@ class SynthTrackerSync:
     def track_state(
         cls,
         variable_name: str,
-        variable_value: Union[BaseModel, str, dict, int, float, bool, list, None],
+        variable_value: BaseModel | str | dict | int | float | bool | list | None,
         origin: Literal["agent", "environment"],
-        annotation: Optional[str] = None,
+        annotation: str | None = None,
     ):
         # Skip if value is not a trackable type instead of raising error
         if not isinstance(variable_value, VALID_TYPES):
@@ -79,7 +79,7 @@ class SynthTrackerSync:
             # )
 
     @classmethod
-    def get_traced_data(cls) -> Tuple[List[Dict[str, Any]], List[Dict[str, Any]]]:
+    def get_traced_data(cls) -> tuple[list[dict[str, Any]], list[dict[str, Any]]]:
         return getattr(cls._local, "inputs", []), getattr(cls._local, "outputs", [])
 
     @classmethod
@@ -93,7 +93,7 @@ class SynthTrackerSync:
     @classmethod
     def track_lm_output(
         cls,
-        messages: List[Dict[str, str]],
+        messages: list[dict[str, str]],
         model_name: str,
         finetune: bool = False,
     ):
@@ -137,9 +137,9 @@ class SynthTrackerAsync:
     @classmethod
     def track_lm(
         cls,
-        messages: List[Dict[str, str]],
+        messages: list[dict[str, str]],
         model_name: str,
-        model_params: Optional[Dict[str, Union[str, int, float]]] = None,
+        model_params: dict[str, str | int | float] | None = None,
         finetune: bool = False,
     ):
         # print("Tracking LM call in async context")  # Added logging
@@ -165,9 +165,9 @@ class SynthTrackerAsync:
     def track_state(
         cls,
         variable_name: str,
-        variable_value: Union[BaseModel, str, dict, int, float, bool, list, None],
+        variable_value: BaseModel | str | dict | int | float | bool | list | None,
         origin: Literal["agent", "environment"],
-        annotation: Optional[str] = None,
+        annotation: str | None = None,
         io_type: Literal["input", "output"] = "output",
     ):
         # Skip if value is not a trackable type instead of raising error
@@ -207,7 +207,7 @@ class SynthTrackerAsync:
             # )
 
     @classmethod
-    def get_traced_data(cls) -> Tuple[List[Dict[str, Any]], List[Dict[str, Any]]]:
+    def get_traced_data(cls) -> tuple[list[dict[str, Any]], list[dict[str, Any]]]:
         traced_inputs = trace_inputs_var.get()
         traced_outputs = trace_outputs_var.get()
         return traced_inputs, traced_outputs
@@ -222,7 +222,7 @@ class SynthTrackerAsync:
     @classmethod
     def track_lm_output(
         cls,
-        messages: List[Dict[str, str]],
+        messages: list[dict[str, str]],
         model_name: str,
         finetune: bool = False,
     ):
@@ -270,9 +270,9 @@ class SynthTracker:
     @classmethod
     def track_lm(
         cls,
-        messages: List[Dict[str, str]],
+        messages: list[dict[str, str]],
         model_name: str,
-        model_params: Optional[Dict[str, Union[str, int, float]]] = None,
+        model_params: dict[str, str | int | float] | None = None,
         finetune: bool = False,
     ):
         # Debug logging disabled: print("DEBUG: Tracking LM call")
@@ -334,9 +334,9 @@ class SynthTracker:
     def track_state(
         cls,
         variable_name: str,
-        variable_value: Union[BaseModel, str, dict, int, float, bool, list, None],
+        variable_value: BaseModel | str | dict | int | float | bool | list | None,
         origin: Literal["agent", "environment"],
-        annotation: Optional[str] = None,
+        annotation: str | None = None,
     ):
         """
         Track a state change or variable value within the current trace.
@@ -392,7 +392,7 @@ class SynthTracker:
         async_sync: Literal[
             "async", "sync", ""
         ] = "",  # Force only async or sync data to be returned
-    ) -> Tuple[list, list]:
+    ) -> tuple[list, list]:
         traced_inputs, traced_outputs = [], []
         # Debug logging disabled: print(
         #     f"\nDEBUG: Getting traced data with async_sync='{async_sync}'"
@@ -424,7 +424,7 @@ class SynthTracker:
     @classmethod
     def track_lm_output(
         cls,
-        messages: List[Dict[str, str]],
+        messages: list[dict[str, str]],
         model_name: str,
         finetune: bool = False,
     ):
@@ -452,10 +452,10 @@ class SynthTracker:
 
 
 def track_messages_sync(
-    input_messages: List[Message],
-    output_messages: List[Message],
+    input_messages: list[Message],
+    output_messages: list[Message],
     model_name: str,
-    model_params: Optional[ModelParams] = None,
+    model_params: ModelParams | None = None,
     finetune: bool = False,
 ) -> None:
     """Track both input and output messages in a conversation synchronously.
@@ -484,10 +484,10 @@ def track_messages_sync(
 
 
 async def track_messages_async(
-    input_messages: List[Message],
-    output_messages: List[Message],
+    input_messages: list[Message],
+    output_messages: list[Message],
     model_name: str,
-    model_params: Optional[ModelParams] = None,
+    model_params: ModelParams | None = None,
     finetune: bool = False,
 ) -> None:
     """Track both input and output messages in a conversation asynchronously.

synth_ai/v0/tracing_v1/upload.py

@@ -5,7 +5,7 @@ import os
 import ssl
 import time
 import warnings
-from typing import Any, Dict, List, Tuple, TypedDict
+from typing import Any, TypedDict
 
 import requests
 from dotenv import load_dotenv
@@ -35,7 +35,7 @@ def _raise_not_supported_error():
 
 
 # NOTE: This may cause memory issues in the future
-def validate_json(data: Dict[str, Any]) -> None:
+def validate_json(data: dict[str, Any]) -> None:
     """Validate that a dictionary contains only JSON-serializable values.
 
     Args:
@@ -51,7 +51,7 @@ def validate_json(data: Dict[str, Any]) -> None:
         raise ValueError(f"Contains non-JSON-serializable values: {e}. {data}")
 
 
-def createPayload(dataset: Dataset, traces: List[SystemTrace]) -> Dict[str, Any]:
+def createPayload(dataset: Dataset, traces: list[SystemTrace]) -> dict[str, Any]:
     payload = {
         "traces": [trace.to_dict() for trace in traces],  # Convert SystemTrace objects to dicts
         "dataset": dataset.to_dict(),
@@ -73,7 +73,7 @@ class TLSAdapter(HTTPAdapter):
         )
 
 
-def load_signed_url(signed_url: str, dataset: Dataset, traces: List[SystemTrace]) -> None:
+def load_signed_url(signed_url: str, dataset: Dataset, traces: list[SystemTrace]) -> None:
     payload = createPayload(dataset, traces)
     validate_json(payload)
 
@@ -102,7 +102,7 @@ def load_signed_url(signed_url: str, dataset: Dataset, traces: List[SystemTrace]
 
 def send_system_traces_s3(
     dataset: Dataset,
-    traces: List[SystemTrace],
+    traces: list[SystemTrace],
     base_url: str,
     api_key: str,
     system_id: str,
@@ -146,7 +146,7 @@ def send_system_traces_s3(
 
 def get_upload_id(
     base_url: str, api_key: str, system_id: str, system_name: str, verbose: bool = False
-) -> Tuple[str, str]:
+) -> tuple[str, str]:
     """
     Modified client-side function to send both system_id and system_name.
     """
@@ -188,12 +188,12 @@ class UploadValidator(BaseModel):
         extra="forbid",  # Prevent additional fields
     )
 
-    traces: List[Dict[str, Any]]
-    dataset: Dict[str, Any]
+    traces: list[dict[str, Any]]
+    dataset: dict[str, Any]
 
     @field_validator("traces")
     @classmethod
-    def validate_traces(cls, traces: List[Dict[str, Any]]) -> List[Dict[str, Any]]:
+    def validate_traces(cls, traces: list[dict[str, Any]]) -> list[dict[str, Any]]:
         if not traces:
             raise ValueError("Traces list cannot be empty")
 
@@ -240,7 +240,7 @@ class UploadValidator(BaseModel):
 
     @field_validator("dataset")
     @classmethod
-    def validate_dataset(cls, dataset: Dict[str, Any]) -> Dict[str, Any]:
+    def validate_dataset(cls, dataset: dict[str, Any]) -> dict[str, Any]:
         required_fields = ["questions", "reward_signals"]
         missing_fields = [f for f in required_fields if f not in dataset]
         if missing_fields:
@@ -263,7 +263,7 @@ class UploadValidator(BaseModel):
         return dataset
 
 
-def validate_upload(traces: List[Dict[str, Any]], dataset: Dict[str, Any]):
+def validate_upload(traces: list[dict[str, Any]], dataset: dict[str, Any]):
     # Validate the upload format before sending to server.
     # Raises ValueError if validation fails.
     try:
@@ -283,8 +283,8 @@ def is_event_loop_running():
 
 
 def format_upload_output(
-    dataset: Dataset, traces: List[SystemTrace]
-) -> Tuple[List[Dict[str, Any]], List[Dict[str, Any]], List[Dict[str, Any]]]:
+    dataset: Dataset, traces: list[SystemTrace]
+) -> tuple[list[dict[str, Any]], list[dict[str, Any]], list[dict[str, Any]]]:
     # Format questions array
     questions_data = [
         {"intent": q.intent, "criteria": q.criteria, "id": q.id} for q in dataset.questions
@@ -334,14 +334,14 @@ class ProcessUploadResponse(TypedDict):
 
 def upload(
     dataset: Dataset,
-    traces: List[SystemTrace] = [],
+    traces: list[SystemTrace] = [],
     verbose: bool = False,
     show_payload: bool = False,
-) -> Tuple[
+) -> tuple[
     ProcessUploadResponse,
-    List[Dict[str, Any]],
-    List[Dict[str, Any]],
-    List[Dict[str, Any]],
+    list[dict[str, Any]],
+    list[dict[str, Any]],
+    list[dict[str, Any]],
 ]:
     """Upload all system traces and dataset to the server.
 
@@ -371,14 +371,14 @@ def upload(
 
 def upload_helper(
     dataset: Dataset,
-    traces: List[SystemTrace] = [],
+    traces: list[SystemTrace] = [],
     verbose: bool = False,
     show_payload: bool = False,
-) -> Tuple[
+) -> tuple[
     ProcessUploadResponse,
-    List[Dict[str, Any]],
-    List[Dict[str, Any]],
-    List[Dict[str, Any]],
+    list[dict[str, Any]],
+    list[dict[str, Any]],
+    list[dict[str, Any]],
 ]:
     """Helper function to handle the upload process.
 
@@ -391,7 +391,7 @@ def upload_helper(
     base_url = os.getenv("SYNTH_ENDPOINT_OVERRIDE", "https://agent-learning.onrender.com")
 
     from .decorators import _local, active_events_var
-    from .trackers import synth_tracker_async, synth_tracker_sync
+    from .trackers import synth_tracker_async
 
     # First close any tracker events
     if hasattr(synth_tracker_async, "active_events"):
@@ -522,3 +522,4 @@ def upload_helper(
             print("\nDataset:")
             print(json.dumps(dataset_dict, indent=2))
         raise
+# ruff: noqa

synth_ai/zyk/__init__.py

@@ -12,6 +12,7 @@ or
 NEW: from synth_ai import LM  # (recommended)
 """
 
+# ruff: noqa: E402
 import warnings
 
 # Issue deprecation warning