synth-ai 0.2.9.dev5__py3-none-any.whl → 0.2.9.dev6__py3-none-any.whl

synth_ai/{lm → v0/lm}/vendors/openai_standard.py

@@ -9,20 +9,20 @@ import openai
 import pydantic_core
 from pydantic import BaseModel
 
-from synth_ai.lm.caching.initialize import (
+from synth_ai.v0.lm.caching.initialize import (
     get_cache_handler,
 )
-from synth_ai.lm.constants import SPECIAL_BASE_TEMPS
-from synth_ai.lm.injection import apply_injection
-from synth_ai.lm.overrides import (
+from synth_ai.v0.lm.constants import SPECIAL_BASE_TEMPS
+from synth_ai.v0.lm.injection import apply_injection
+from synth_ai.v0.lm.overrides import (
     apply_param_overrides,
     apply_tool_overrides,
     use_overrides_for_messages,
 )
-from synth_ai.lm.tools.base import BaseTool
-from synth_ai.lm.vendors.base import BaseLMResponse, VendorBase
-from synth_ai.lm.vendors.openai_standard_responses import OpenAIResponsesAPIMixin
-from synth_ai.lm.vendors.retries import MAX_BACKOFF
+from synth_ai.v0.lm.tools.base import BaseTool
+from synth_ai.v0.lm.vendors.base import BaseLMResponse, VendorBase
+from synth_ai.v0.lm.vendors.openai_standard_responses import OpenAIResponsesAPIMixin
+from synth_ai.v0.lm.vendors.retries import MAX_BACKOFF
 
 DEFAULT_EXCEPTIONS_TO_RETRY = (
     pydantic_core._pydantic_core.ValidationError,
@@ -427,7 +427,9 @@ class OpenAIStandard(VendorBase, OpenAIResponsesAPIMixin):
                             "type": getattr(tc, "type", "function"),
                             "function": {
                                 "name": getattr(getattr(tc, "function", None), "name", None),
-                                "arguments": getattr(getattr(tc, "function", None), "arguments", None),
+                                "arguments": getattr(
+                                    getattr(tc, "function", None), "arguments", None
+                                ),
                             },
                         }
                     )

synth_ai/{lm → v0/lm}/vendors/openai_standard_responses.py

@@ -8,8 +8,8 @@ that extend the OpenAIStandard class functionality.
 import uuid
 from typing import Any
 
-from synth_ai.lm.tools.base import BaseTool
-from synth_ai.lm.vendors.base import BaseLMResponse
+from synth_ai.v0.lm.tools.base import BaseTool
+from synth_ai.v0.lm.vendors.base import BaseLMResponse
 
 
 def _silent_backoff_handler(_details):
@@ -223,7 +223,9 @@ class OpenAIResponsesAPIMixin:
         if not synth_gpu_endpoint:
             raise ValueError("SYNTH_GPU_HARMONY_ENDPOINT environment variable not set")
 
-        async with aiohttp.ClientSession() as session, session.post(
+        async with (
+            aiohttp.ClientSession() as session,
+            session.post(
                 f"{synth_gpu_endpoint}/v1/completions",
                 json={
                     "model": model,
@@ -231,8 +233,9 @@ class OpenAIResponsesAPIMixin:
                     "max_tokens": lm_config.get("max_tokens", 4096),
                     "temperature": lm_config.get("temperature", 0.8),
                 },
-            ) as resp:
-                result = await resp.json()
+            ) as resp,
+        ):
+            result = await resp.json()
 
         # Parse response using Harmony
         response_tokens = result.get("choices", [{}])[0].get("text", "")

synth_ai/{lm → v0/lm}/vendors/supported/custom_endpoint.py

@@ -11,9 +11,9 @@ import requests
 from requests.adapters import HTTPAdapter
 from urllib3.util.retry import Retry
 
-from synth_ai.lm.caching.initialize import get_cache_handler
-from synth_ai.lm.tools.base import BaseTool
-from synth_ai.lm.vendors.base import BaseLMResponse, VendorBase
+from synth_ai.v0.lm.caching.initialize import get_cache_handler
+from synth_ai.v0.lm.tools.base import BaseTool
+from synth_ai.v0.lm.vendors.base import BaseLMResponse, VendorBase
 
 # Exception types for retry
 CUSTOM_ENDPOINT_EXCEPTIONS_TO_RETRY: tuple[type[Exception], ...] = (
@@ -194,9 +194,7 @@ IMPORTANT: To use a tool, respond with JSON wrapped in ```json fences:
 
 For regular responses, just respond normally without JSON fences."""
 
-    def _extract_tool_calls(
-        self, content: str, tools: list[BaseTool]
-    ) -> tuple[list | None, str]:
+    def _extract_tool_calls(self, content: str, tools: list[BaseTool]) -> tuple[list | None, str]:
         """Extract and validate tool calls from response."""
         # Look for JSON fenced blocks
         json_pattern = r"```json\s*(\{.*?\})\s*```"

synth_ai/{lm → v0/lm}/vendors/supported/deepseek.py

@@ -3,8 +3,8 @@ from typing import Any
 
 from openai import AsyncOpenAI, OpenAI
 
-from synth_ai.lm.tools.base import BaseTool
-from synth_ai.lm.vendors.openai_standard import OpenAIStandard
+from synth_ai.v0.lm.tools.base import BaseTool
+from synth_ai.v0.lm.vendors.openai_standard import OpenAIStandard
 
 
 class DeepSeekAPI(OpenAIStandard):

synth_ai/{lm → v0/lm}/vendors/supported/grok.py

@@ -3,8 +3,8 @@ from typing import Any
 
 from openai import AsyncOpenAI, OpenAI
 
-from synth_ai.lm.tools.base import BaseTool
-from synth_ai.lm.vendors.openai_standard import OpenAIStandard
+from synth_ai.v0.lm.tools.base import BaseTool
+from synth_ai.v0.lm.vendors.openai_standard import OpenAIStandard
 
 
 class GrokAPI(OpenAIStandard):

synth_ai/{lm → v0/lm}/vendors/supported/groq.py

@@ -3,7 +3,7 @@ import os
 from dotenv import load_dotenv
 from groq import AsyncGroq, Groq
 
-from synth_ai.lm.vendors.openai_standard import OpenAIStandard
+from synth_ai.v0.lm.vendors.openai_standard import OpenAIStandard
 
 load_dotenv()
 

synth_ai/{lm → v0/lm}/vendors/supported/ollama.py

@@ -1,6 +1,6 @@
 from openai import AsyncOpenAI, OpenAI
 
-from synth_ai.lm.vendors.openai_standard import OpenAIStandard
+from synth_ai.v0.lm.vendors.openai_standard import OpenAIStandard
 
 
 class OllamaAPI(OpenAIStandard):

synth_ai/{lm → v0/lm}/vendors/supported/openrouter.py

@@ -3,9 +3,9 @@ from typing import Any
 
 from openai import AsyncOpenAI, OpenAI
 
-from synth_ai.lm.tools.base import BaseTool
-from synth_ai.lm.vendors.base import BaseLMResponse
-from synth_ai.lm.vendors.openai_standard import OpenAIStandard
+from synth_ai.v0.lm.tools.base import BaseTool
+from synth_ai.v0.lm.vendors.base import BaseLMResponse
+from synth_ai.v0.lm.vendors.openai_standard import OpenAIStandard
 
 
 class OpenRouterAPI(OpenAIStandard):

synth_ai/{lm → v0/lm}/vendors/supported/together.py

@@ -2,7 +2,7 @@ import os
 
 from together import AsyncTogether, Together
 
-from synth_ai.lm.vendors.openai_standard import OpenAIStandard
+from synth_ai.v0.lm.vendors.openai_standard import OpenAIStandard
 
 
 class TogetherAPI(OpenAIStandard):

synth_ai/{lm → v0/lm}/vendors/synth_client.py

@@ -276,8 +276,20 @@ class AsyncSynthClient:
         This method provides the OpenAI client interface structure.
         """
         return await self._chat_completions_create(
-            model, messages, temperature, max_tokens, top_p, frequency_penalty,
-            presence_penalty, stop, stream, tools, tool_choice, response_format, seed, **kwargs
+            model,
+            messages,
+            temperature,
+            max_tokens,
+            top_p,
+            frequency_penalty,
+            presence_penalty,
+            stop,
+            stream,
+            tools,
+            tool_choice,
+            response_format,
+            seed,
+            **kwargs,
         )
 
     async def _chat_completions_create(
@@ -368,7 +380,8 @@ class AsyncSynthClient:
             if isinstance(bt, int) and isinstance(mt, int) and bt > mt:
                 logger.warning(
                     "thinking_budget (%s) exceeds max_tokens (%s) – forwarding as-is",
-                    str(bt), str(mt)
+                    str(bt),
+                    str(mt),
                 )
         except Exception:
             pass
@@ -387,6 +400,7 @@ class AsyncSynthClient:
 
                 # If streaming requested, return an async stream adapter
                 if stream:
+
                     async def _astream():
                         await self._ensure_client()
                         async with self._client.stream("POST", url, json=payload) as r:  # type: ignore
@@ -678,6 +692,7 @@ def create_sync_client(config: SynthConfig | None = None) -> SyncSynthClient:
 # Drop-in replacements for OpenAI clients
 # These allow Synth to be used as a complete replacement for OpenAI
 
+
 class AsyncOpenAI(AsyncSynthClient):
     """
     Drop-in replacement for openai.AsyncOpenAI.
@@ -685,7 +700,7 @@ class AsyncOpenAI(AsyncSynthClient):
     Use Synth backend instead of OpenAI while maintaining the same API.
 
     Example:
-        from synth_ai.lm.vendors.synth_client import AsyncOpenAI
+        from synth_ai.v0.lm.vendors.synth_client import AsyncOpenAI
 
         client = AsyncOpenAI(
             api_key="sk_live_...",
@@ -710,16 +725,22 @@ class AsyncOpenAI(AsyncSynthClient):
         """
         # Handle OpenAI-style initialization
         from ..config import SynthConfig
+
         if api_key or base_url:
             config = SynthConfig(
-                base_url=base_url or os.getenv("OPENAI_API_BASE", "https://synth-backend-dev-docker.onrender.com/api"),
-                api_key=api_key or os.getenv("OPENAI_API_KEY", "")
+                base_url=base_url
+                or os.getenv(
+                    "OPENAI_API_BASE", "https://synth-backend-dev-docker.onrender.com/api"
+                ),
+                api_key=api_key or os.getenv("OPENAI_API_KEY", ""),
             )
         else:
             # Fallback to environment variables (OPENAI_* first, then SYNTH_*)
             env_base = os.getenv("OPENAI_API_BASE") or os.getenv("SYNTH_BASE_URL")
             env_key = os.getenv("OPENAI_API_KEY") or os.getenv("SYNTH_API_KEY")
-            config = SynthConfig(base_url=env_base, api_key=env_key) if env_base and env_key else None
+            config = (
+                SynthConfig(base_url=env_base, api_key=env_key) if env_base and env_key else None
+            )
 
         super().__init__(config, **kwargs)
 
@@ -742,15 +763,21 @@ class OpenAI(SyncSynthClient):
         """
         # Handle OpenAI-style initialization
         from ..config import SynthConfig
+
         if api_key or base_url:
             config = SynthConfig(
-                base_url=base_url or os.getenv("OPENAI_API_BASE", "https://synth-backend-dev-docker.onrender.com/api"),
-                api_key=api_key or os.getenv("OPENAI_API_KEY", "")
+                base_url=base_url
+                or os.getenv(
+                    "OPENAI_API_BASE", "https://synth-backend-dev-docker.onrender.com/api"
+                ),
+                api_key=api_key or os.getenv("OPENAI_API_KEY", ""),
             )
         else:
             env_base = os.getenv("OPENAI_API_BASE") or os.getenv("SYNTH_BASE_URL")
             env_key = os.getenv("OPENAI_API_KEY") or os.getenv("SYNTH_API_KEY")
-            config = SynthConfig(base_url=env_base, api_key=env_key) if env_base and env_key else None
+            config = (
+                SynthConfig(base_url=env_base, api_key=env_key) if env_base and env_key else None
+            )
 
         super().__init__(config, **kwargs)
 
@@ -760,7 +787,7 @@ __all__ = [
     "AsyncSynthClient",
     "SyncSynthClient",
     "AsyncOpenAI",  # Drop-in replacement for openai.AsyncOpenAI
-    "OpenAI",       # Drop-in replacement for openai.OpenAI
+    "OpenAI",  # Drop-in replacement for openai.OpenAI
     "create_async_client",
     "create_sync_client",
     "create_chat_completion_async",

synth_ai/v0/tracing/upload.py

@@ -373,140 +373,37 @@ def upload_helper(
         raise ValueError("SYNTH_API_KEY environment variable not set")
     base_url = os.getenv("SYNTH_ENDPOINT_OVERRIDE", PROD_BASE_URL_DEFAULT)
 
-"""Legacy block below retained for reference and disabled for linting/parsing.
-Start disabled block.
-"""
-"""
-from .decorators import _local, active_events_var
-from .trackers import synth_tracker_async, synth_tracker_sync
-
-    # First close any tracker events
-    if hasattr(synth_tracker_async, "active_events"):
-        for event_type, event in list(synth_tracker_async.active_events.items()):
-            if event and event.closed is None:
-                event.closed = time.time()
-                try:
-                    event_store.add_event(
-                        event.system_name,
-                        event.system_id,
-                        event.system_instance_id,
-                        event,
-                    )
-                    if verbose:
-                        print(f"Closed and stored tracker async event: {event_type}")
-                except Exception as e:
-                    logging.error(f"Failed to store tracker event {event_type}: {str(e)}")
-        synth_tracker_async.active_events.clear()
-
-    # End all active events before uploading
-    if hasattr(_local, "active_events"):
-        for event_type, event in _local.active_events.items():
-            if event and event.closed is None:
-                event.closed = time.time()
-                if hasattr(_local, "system_instance_id"):
-                    try:
-                        event_store.add_event(
-                            _local.system_name,
-                            _local.system_id,
-                            _local.system_instance_id,
-                            event,
-                        )
-                        if verbose:
-                            print(f"Closed and stored active event: {event_type}")
-                    except Exception as e:
-                        logging.error(f"Failed to store event {event_type}: {str(e)}")
-        _local.active_events.clear()
-
-    # NEW: Close all open asynchronous events
-    active_events_async = active_events_var.get()
-    if active_events_async:
-        current_time = time.time()
-        for event_type, event in list(active_events_async.items()):
-            if event and event.closed is None:
-                event.closed = current_time
-                try:
-                    event_store.add_event(
-                        event.system_name,
-                        event.system_id,
-                        event.system_instance_id,
-                        event,
-                    )
-                    if verbose:
-                        print(f"Closed and stored async event: {event_type}")
-                except Exception as e:
-                    logging.error(f"Failed to store async event {event_type}: {str(e)}")
-        active_events_var.set({})
-
-    # Also close any unclosed events in existing traces
     logged_traces = event_store.get_system_traces()
-    traces = logged_traces + traces
-    # traces = event_store.get_system_traces() if len(traces) == 0 else traces
-    current_time = time.time()
-    for trace in traces:
-        for partition in trace.partition:
-            for event in partition.events:
-                if event.closed is None:
-                    event.closed = current_time
-                    event_store.add_event(
-                        trace.system_name,
-                        trace.system_id,
-                        trace.system_instance_id,
-                        event,
-                    )
-                    if verbose:
-                        print(f"Closed existing unclosed event: {event.event_type}")
-
-    try:
-        # Get traces and convert to dict format
-        if len(traces) == 0:
-            raise ValueError("No system traces found")
-        traces_dict = [trace.to_dict() for trace in traces]
-        dataset_dict = dataset.to_dict()
-
-        # Validate upload format
-        if verbose:
-            print("Validating upload format...")
-        validate_upload(traces_dict, dataset_dict)
-        if verbose:
-            print("Upload format validation successful")
-
-        # Send to server
-        upload_id, signed_url = send_system_traces_s3(
-            dataset=dataset,
-            traces=traces,
-            base_url=base_url,
-            api_key=api_key,
-            system_id=traces[0].system_id,
-            system_name=traces[0].system_name,
-            verbose=verbose,
-        )
-
-        questions_json, reward_signals_json, traces_json = format_upload_output(dataset, traces)
-        return (
-            {
-                "status": "success",
-                "upload_id": upload_id,
-                "signed_url": signed_url,
-            },
-            questions_json,
-            reward_signals_json,
-            traces_json,
-        )
+    combined_traces: List[SystemTrace] = list(logged_traces) + list(traces)
+
+    if not combined_traces:
+        raise ValueError("No system traces found")
+
+    traces_dict = [trace.to_dict() for trace in combined_traces]
+    dataset_dict = dataset.to_dict()
+
+    if verbose:
+        print("Validating upload format...")
+    validate_upload(traces_dict, dataset_dict)
+
+    upload_id, signed_url = send_system_traces_s3(
+        dataset=dataset,
+        traces=combined_traces,
+        base_url=base_url,
+        api_key=api_key,
+        system_id=combined_traces[0].system_id,
+        system_name=combined_traces[0].system_name,
+        verbose=verbose,
+    )
 
-    except ValueError as e:
-        if verbose:
-            print("Validation error:", str(e))
-            print("\nTraces:")
-            print(json.dumps(traces_dict, indent=2))
-            print("\nDataset:")
-            print(json.dumps(dataset_dict, indent=2))
-        raise
-    except requests.exceptions.HTTPError as e:
-        if verbose:
-            print("HTTP error occurred:", e)
-            print("\nTraces:")
-            print(json.dumps(traces_dict, indent=2))
-            print("\nDataset:")
-            print(json.dumps(dataset_dict, indent=2))
-        raise
-"""
+    questions_json, reward_signals_json, traces_json = format_upload_output(dataset, combined_traces)
+    return (
+        {
+            "status": "success",
+            "upload_id": upload_id,
+            "signed_url": signed_url,
+        },
+        questions_json,
+        reward_signals_json,
+        traces_json,
+    )

synth_ai/v0/tracing_v3/__init__.py

@@ -0,0 +1,10 @@
+from __future__ import annotations
+
+# Backward-compat shim for legacy imports: `synth_ai.v0.tracing_v3.*`
+# Re-export from the canonical `synth_ai.tracing_v3` package.
+
+from synth_ai.tracing_v3 import *  # type: ignore[F401,F403]
+
+__all__ = []  # names are provided by the upstream module
+
+

synth_ai/v0/tracing_v3/abstractions.py

@@ -0,0 +1,3 @@
+from __future__ import annotations
+
+from synth_ai.tracing_v3.abstractions import *  # type: ignore[F401,F403]

synth_ai/v0/tracing_v3/decorators.py

@@ -0,0 +1,3 @@
+from __future__ import annotations
+
+from synth_ai.tracing_v3.decorators import *  # type: ignore[F401,F403]

synth_ai/v0/tracing_v3/llm_call_record_helpers.py

@@ -0,0 +1,3 @@
+from __future__ import annotations
+
+from synth_ai.tracing_v3.llm_call_record_helpers import *  # type: ignore[F401,F403]

synth_ai/v0/tracing_v3/session_tracer.py

@@ -0,0 +1,3 @@
+from __future__ import annotations
+
+from synth_ai.tracing_v3.session_tracer import *  # type: ignore[F401,F403]