synth-ai 0.2.4.dev5__py3-none-any.whl → 0.2.4.dev7__py3-none-any.whl

synth_ai/__init__.py

@@ -2,27 +2,28 @@
 Synth AI - Software for aiding the best and multiplying the will.
 """
 
+# Environment exports - moved from synth-env
+from synth_ai.environments import *  # noqa
+import synth_ai.environments as environments  # expose module name for __all__
 from synth_ai.lm.core.main import LM  # Moved from zyk to lm for better organization
+from synth_ai.lm.provider_support.anthropic import Anthropic, AsyncAnthropic
+
+# Provider support exports - moved from synth-sdk to synth_ai/lm
+from synth_ai.lm.provider_support.openai import AsyncOpenAI, OpenAI
 
 # Tracing exports - moved from synth-sdk (deprecated v1)
 from synth_ai.tracing_v1 import *  # noqa
+import synth_ai.tracing_v1 as tracing  # expose module name for __all__
 from synth_ai.tracing_v1.abstractions import (
     EventPartitionElement,
+    RewardSignal,
     SystemTrace,
     TrainingQuestion,
-    RewardSignal,
 )
 from synth_ai.tracing_v1.decorators import trace_event_async, trace_event_sync
 from synth_ai.tracing_v1.upload import upload
 
-# Provider support exports - moved from synth-sdk to synth_ai/lm
-from synth_ai.lm.provider_support.openai import OpenAI, AsyncOpenAI
-from synth_ai.lm.provider_support.anthropic import Anthropic, AsyncAnthropic
-
-# Environment exports - moved from synth-env
-from synth_ai.environments import *  # noqa
-
-__version__ = "0.2.1.dev0"
+__version__ = "0.2.4.dev7"
 __all__ = [
     "LM",
     "tracing",
@@ -31,4 +32,12 @@ __all__ = [
     "Anthropic",
     "AsyncAnthropic",
     "environments",
+    # v1 tracing legacy API re-exports
+    "EventPartitionElement",
+    "RewardSignal",
+    "SystemTrace",
+    "TrainingQuestion",
+    "trace_event_async",
+    "trace_event_sync",
+    "upload",
 ]  # Explicitly define public API

synth_ai/cli/__init__.py

@@ -7,13 +7,10 @@ pyproject entry point `synth_ai.cli:cli`.
 
 from __future__ import annotations
 
-import importlib.util
-import pathlib
-import sys
-
 # Load environment variables from a local .env if present (repo root)
 try:
-    from dotenv import load_dotenv, find_dotenv
+    from dotenv import find_dotenv, load_dotenv
+
     # Source .env early so CLI subcommands inherit config; do not override shell
     load_dotenv(find_dotenv(usecwd=True), override=False)
 except Exception:
@@ -26,41 +23,49 @@ from .root import cli  # new canonical CLI entrypoint
 # Register subcommands from this package onto the group
 try:
     from . import watch as _watch
+
     _watch.register(cli)
 except Exception:
     pass
 try:
     from . import balance as _balance
+
     _balance.register(cli)
 except Exception:
     pass
 try:
     from . import man as _man
+
     _man.register(cli)
 except Exception:
     pass
 try:
     from . import traces as _traces
+
     _traces.register(cli)
 except Exception:
     pass
 try:
     from . import recent as _recent
+
     _recent.register(cli)
 except Exception:
     pass
 try:
     from . import calc as _calc
+
     _calc.register(cli)
 except Exception:
     pass
 try:
     from . import status as _status
+
     _status.register(cli)
 except Exception:
     pass
 try:
     from . import demo as _demo
+
     _demo.register(cli)
 except Exception:
     pass

synth_ai/cli/balance.py

@@ -6,15 +6,14 @@ CLI: check remaining credit balance from Synth backend.
 from __future__ import annotations
 
 import os
+from urllib.parse import urlparse
+
 import click
 import requests
 from requests import Response
-from urllib.parse import urlparse
+from rich import box
 from rich.console import Console
-from rich.panel import Panel
 from rich.table import Table
-from rich import box
-
 
 PROD_BACKEND_BASE = "https://agent-learning.onrender.com/api/v1"
 
@@ -83,9 +82,7 @@ def register(cli):
 
         key_val, key_src = _resolve_api_key(api_key)
         if not key_val:
-            console.print(
-                "[red]Missing API key.[/red] Set via --api-key or SYNTH_API_KEY env var."
-            )
+            console.print("[red]Missing API key.[/red] Set via --api-key or SYNTH_API_KEY env var.")
             return
 
         base = _ensure_api_v1_prefix(base_url)
@@ -127,19 +124,27 @@ def register(cli):
                 if u.ok:
                     uj = u.json()
                     rows = uj.get("windows", [])
-                    windows = {int(r.get("window_hours")): r for r in rows if isinstance(r.get("window_hours"), int)}
+                    windows = {
+                        int(r.get("window_hours")): r
+                        for r in rows
+                        if isinstance(r.get("window_hours"), int)
+                    }
+
                     def _usd(c):
                         try:
-                            return f"${(int(c)/100):,.2f}"
+                            return f"${(int(c) / 100):,.2f}"
                         except Exception:
                             return "$0.00"
+
                     if 24 in windows or 168 in windows:
-                        t = Table(title="Spend (Tokens vs GPU)", box=box.SIMPLE, header_style="bold")
+                        t = Table(
+                            title="Spend (Tokens vs GPU)", box=box.SIMPLE, header_style="bold"
+                        )
                         t.add_column("Window")
                         t.add_column("Tokens", justify="right")
                         t.add_column("GPU", justify="right")
                         t.add_column("Total", justify="right")
-                        for h,label in ((24,"24h"),(168,"7d")):
+                        for h, label in ((24, "24h"), (168, "7d")):
                             if h in windows:
                                 w = windows[h]
                                 t.add_row(
@@ -167,15 +172,15 @@ def register(cli):
                         t.add_column("Total", justify="right")
                         t.add_row(
                             "Current Month",
-                            f"${(cm.get('token_spend_cents',0)/100):,.2f}",
-                            f"${(cm.get('gpu_spend_cents',0)/100):,.2f}",
-                            f"${(cm.get('total_spend_cents',0)/100):,.2f}",
+                            f"${(cm.get('token_spend_cents', 0) / 100):,.2f}",
+                            f"${(cm.get('gpu_spend_cents', 0) / 100):,.2f}",
+                            f"${(cm.get('total_spend_cents', 0) / 100):,.2f}",
                         )
                         t.add_row(
                             "Last 30 days",
-                            f"${(l30.get('token_spend_cents',0)/100):,.2f}",
-                            f"${(l30.get('gpu_spend_cents',0)/100):,.2f}",
-                            f"${(l30.get('total_spend_cents',0)/100):,.2f}",
+                            f"${(l30.get('token_spend_cents', 0) / 100):,.2f}",
+                            f"${(l30.get('gpu_spend_cents', 0) / 100):,.2f}",
+                            f"${(l30.get('total_spend_cents', 0) / 100):,.2f}",
                         )
                         console.print(t)
             except Exception:

synth_ai/cli/calc.py

@@ -6,10 +6,10 @@ Safe evaluation of arithmetic expressions.
 
 import ast
 import operator as op
+
 import click
 from rich.console import Console
 
-
 # Supported operators
 _OPS = {
     ast.Add: op.add,
@@ -33,7 +33,7 @@ def _safe_eval(expr: str) -> float:
         if isinstance(n, ast.Num):  # 3.8 and earlier
             return n.n
         if isinstance(n, ast.Constant):  # 3.8+
-            if isinstance(n.value, (int, float)):
+            if isinstance(n.value, int | float):
                 return n.value
             raise ValueError("Only numeric constants are allowed")
         if isinstance(n, ast.BinOp) and type(n.op) in _OPS:
@@ -67,4 +67,3 @@ def register(cli):
             console.print(f"= [bold]{result}[/bold]")
         except Exception as e:
             console.print(f"[red]Error:[/red] {e}")
-

synth_ai/cli/demo.py

@@ -11,12 +11,11 @@ from __future__ import annotations
 import os
 import subprocess
 from pathlib import Path
-from typing import List
 
 import click
 
 
-def _find_demo_scripts(root: Path) -> List[Path]:
+def _find_demo_scripts(root: Path) -> list[Path]:
     if not root.exists():
         return []
     return sorted([p for p in root.rglob("run_demo.sh") if p.is_file()])
@@ -52,8 +51,8 @@ def register(cli):
         def _validate_choice(val: str) -> int:
             try:
                 i = int(val)
-            except Exception:
-                raise click.BadParameter("Enter a number from the list")
+            except Exception as err:
+                raise click.BadParameter("Enter a number from the list") from err
             if i < 1 or i > len(demos):
                 raise click.BadParameter(f"Choose a number between 1 and {len(demos)}")
             return i
@@ -71,4 +70,3 @@ def register(cli):
             click.echo(f"❌ Demo exited with non-zero status: {e.returncode}")
         except KeyboardInterrupt:
             click.echo("\n🛑 Demo interrupted by user")
-

synth_ai/cli/legacy_root_backup.py

@@ -3,22 +3,20 @@
 Synth AI CLI - Command line interface for Synth AI services.
 """
 
-import sys
+import logging
 import os
-import subprocess
+import shutil
 import signal
+import subprocess
+import sys
 import time
-import shutil
-from pathlib import Path
-from typing import Optional
-import logging
 
 import click
 
 logger = logging.getLogger(__name__)
 
 
-def find_sqld_binary() -> Optional[str]:
+def find_sqld_binary() -> str | None:
     """Find the sqld binary in common locations."""
     # Check if sqld is in PATH
     sqld_path = shutil.which("sqld")
@@ -132,14 +130,18 @@ def env():
 
 @env.command("register")
 @click.option("--name", required=True, help="Environment name (e.g., 'MyEnv-v1')")
-@click.option("--module", "module_path", required=True, help="Python module path (e.g., 'my_package.env')")
+@click.option(
+    "--module", "module_path", required=True, help="Python module path (e.g., 'my_package.env')"
+)
 @click.option("--class-name", "class_name", required=True, help="Environment class name")
 @click.option("--description", help="Optional description")
 @click.option("--service-url", default="http://localhost:8901", help="Environment service URL")
-def register_env(name: str, module_path: str, class_name: str, description: Optional[str], service_url: str):
+def register_env(
+    name: str, module_path: str, class_name: str, description: str | None, service_url: str
+):
     """Register a new environment with the service."""
     import requests
-    
+
     payload = {
         "name": name,
         "module_path": module_path,
@@ -147,14 +149,14 @@ def register_env(name: str, module_path: str, class_name: str, description: Opti
     }
     if description:
         payload["description"] = description
-    
+
     try:
         response = requests.post(f"{service_url}/registry/environments", json=payload, timeout=10)
         response.raise_for_status()
-        
+
         result = response.json()
         click.echo(f"✅ {result['message']}")
-        
+
     except requests.exceptions.ConnectionError:
         click.echo(f"❌ Could not connect to environment service at {service_url}")
         click.echo("💡 Make sure the service is running: synth-ai serve")
@@ -170,29 +172,29 @@ def register_env(name: str, module_path: str, class_name: str, description: Opti
 def list_envs(service_url: str):
     """List all registered environments."""
     import requests
-    
+
     try:
         response = requests.get(f"{service_url}/registry/environments", timeout=10)
         response.raise_for_status()
-        
+
         result = response.json()
         environments = result["environments"]
-        
+
         if not environments:
             click.echo("No environments registered.")
             return
-        
+
         click.echo(f"\n📦 Registered Environments ({result['total_count']}):")
         click.echo("=" * 60)
-        
+
         for env in environments:
             click.echo(f"🌍 {env['name']}")
             click.echo(f"   Class: {env['class_name']}")
             click.echo(f"   Module: {env['module']}")
-            if env['description']:
+            if env["description"]:
                 click.echo(f"   Description: {env['description']}")
             click.echo()
-            
+
     except requests.exceptions.ConnectionError:
         click.echo(f"❌ Could not connect to environment service at {service_url}")
         click.echo("💡 Make sure the service is running: synth-ai serve")
@@ -206,14 +208,14 @@ def list_envs(service_url: str):
 def unregister_env(name: str, service_url: str):
     """Unregister an environment from the service."""
     import requests
-    
+
     try:
         response = requests.delete(f"{service_url}/registry/environments/{name}", timeout=10)
         response.raise_for_status()
-        
+
         result = response.json()
         click.echo(f"✅ {result['message']}")
-        
+
     except requests.exceptions.ConnectionError:
         click.echo(f"❌ Could not connect to environment service at {service_url}")
         click.echo("💡 Make sure the service is running: synth-ai serve")
@@ -228,11 +230,14 @@ def unregister_env(name: str, service_url: str):
 
 
 @cli.command()
-@click.option("--url", default="sqlite+aiosqlite:///./synth_ai.db/dbs/default/data", help="Database URL")
+@click.option(
+    "--url", default="sqlite+aiosqlite:///./synth_ai.db/dbs/default/data", help="Database URL"
+)
 def view(url: str):
     """Launch the interactive TUI dashboard."""
     try:
         from .tui.dashboard import SynthDashboard
+
         app = SynthDashboard(db_url=url)
         app.run()
     except ImportError:
@@ -241,6 +246,7 @@ def view(url: str):
     except KeyboardInterrupt:
         click.echo("\n👋 Dashboard closed", err=True)
 
+
 # Note: subcommands (watch, experiments, experiment, usage, traces, status, recent, calc)
 # are registered from the package module synth_ai.cli at import time.
 
@@ -251,9 +257,23 @@ def view(url: str):
 @click.option("--env-port", default=8901, type=int, help="Port for environment service")
 @click.option("--no-sqld", is_flag=True, help="Skip starting sqld daemon")
 @click.option("--no-env", is_flag=True, help="Skip starting environment service")
-@click.option("--reload/--no-reload", default=False, help="Enable auto-reload (default: off). Or set SYNTH_RELOAD=1")
-@click.option("--force", is_flag=True, help="Kill any process already bound to --env-port without prompting")
-def serve(db_file: str, sqld_port: int, env_port: int, no_sqld: bool, no_env: bool, reload: bool, force: bool):
+@click.option(
+    "--reload/--no-reload",
+    default=False,
+    help="Enable auto-reload (default: off). Or set SYNTH_RELOAD=1",
+)
+@click.option(
+    "--force", is_flag=True, help="Kill any process already bound to --env-port without prompting"
+)
+def serve(
+    db_file: str,
+    sqld_port: int,
+    env_port: int,
+    no_sqld: bool,
+    no_env: bool,
+    reload: bool,
+    force: bool,
+):
     """Start Synth AI services (sqld daemon and environment service)."""
 
     # Configure logging
@@ -307,7 +327,7 @@ def serve(db_file: str, sqld_port: int, env_port: int, no_sqld: bool, no_env: bo
                 ]
 
                 # Create log file
-                sqld_log = open("sqld.log", "w")
+                sqld_log = open("sqld.log", "w")  # noqa: SIM115
                 proc = subprocess.Popen(sqld_cmd, stdout=sqld_log, stderr=subprocess.STDOUT)
                 processes.append(proc)
 
@@ -317,7 +337,7 @@ def serve(db_file: str, sqld_port: int, env_port: int, no_sqld: bool, no_env: bo
                 # Verify it started
                 if proc.poll() is not None:
                     click.echo("❌ Failed to start sqld. Check sqld.log for details.", err=True)
-                    with open("sqld.log", "r") as f:
+                    with open("sqld.log") as f:
                         click.echo("\nLast 10 lines of sqld.log:")
                         lines = f.readlines()
                         for line in lines[-10:]:
@@ -362,7 +382,9 @@ def serve(db_file: str, sqld_port: int, env_port: int, no_sqld: bool, no_env: bo
             # Try to find PIDs using lsof (macOS/Linux)
             pids: list[str] = []
             try:
-                out = subprocess.run(["lsof", "-ti", f":{env_port}"], capture_output=True, text=True)
+                out = subprocess.run(
+                    ["lsof", "-ti", f":{env_port}"], capture_output=True, text=True
+                )
                 if out.returncode == 0 and out.stdout.strip():
                     pids = [p for p in out.stdout.strip().splitlines() if p]
             except FileNotFoundError:
@@ -374,12 +396,16 @@ def serve(db_file: str, sqld_port: int, env_port: int, no_sqld: bool, no_env: bo
                     time.sleep(0.5)
             else:
                 pid_info = f" PIDs: {', '.join(pids)}" if pids else ""
-                if click.confirm(f"⚠️  Port {env_port} is in use.{pid_info} Kill and continue?", default=True):
+                if click.confirm(
+                    f"⚠️  Port {env_port} is in use.{pid_info} Kill and continue?", default=True
+                ):
                     if pids:
                         subprocess.run(["kill", "-9", *pids], check=False)
                         time.sleep(0.5)
                 else:
-                    click.echo("❌ Aborting. Re-run with --force to auto-kill or choose a different --env-port.")
+                    click.echo(
+                        "❌ Aborting. Re-run with --force to auto-kill or choose a different --env-port."
+                    )
                     sys.exit(1)
 
         # Set environment variables

synth_ai/cli/man.py

@@ -5,11 +5,10 @@ CLI: human-friendly manual for Synth AI commands and options.
 
 from __future__ import annotations
 
-import click
+from rich import box
 from rich.console import Console
-from rich.table import Table
 from rich.panel import Panel
-from rich import box
+from rich.table import Table
 
 
 def _commands_table() -> Table:
@@ -23,28 +22,23 @@ def _commands_table() -> Table:
     )
     t.add_row(
         "traces",
-        "List local trace DBs, trace counts, experiments, and per-system counts.\n"
-        "Options: --root",
+        "List local trace DBs, trace counts, experiments, and per-system counts.\nOptions: --root",
     )
     t.add_row(
         "experiments",
-        "Snapshot table of experiments from the local traces DB.\n"
-        "Options: --url, --limit",
+        "Snapshot table of experiments from the local traces DB.\nOptions: --url, --limit",
     )
     t.add_row(
         "experiment <id>",
-        "Details and sessions for an experiment (accepts partial ID).\n"
-        "Options: --url",
+        "Details and sessions for an experiment (accepts partial ID).\nOptions: --url",
     )
     t.add_row(
         "usage",
-        "Model usage statistics (tokens, cost).\n"
-        "Options: --url, --model",
+        "Model usage statistics (tokens, cost).\nOptions: --url, --model",
     )
     t.add_row(
         "status",
-        "DB stats, systems, and environment service health.\n"
-        "Options: --url, --service-url",
+        "DB stats, systems, and environment service health.\nOptions: --url, --service-url",
     )
     t.add_row(
         "calc '<expr>'",
@@ -52,8 +46,7 @@ def _commands_table() -> Table:
     )
     t.add_row(
         "env list | env register | env unregister",
-        "Manage environment registry via the service.\n"
-        "Options vary; see examples.",
+        "Manage environment registry via the service.\nOptions vary; see examples.",
     )
     return t
 
@@ -63,14 +56,22 @@ def _env_table() -> Table:
     t.add_column("Variable")
     t.add_column("Used By")
     t.add_column("Purpose")
-    t.add_row("SYNTH_BACKEND_BASE_URL", "balance", "Backend base URL (preferred) e.g. http://localhost:8000/api/v1")
+    t.add_row(
+        "SYNTH_BACKEND_BASE_URL",
+        "balance",
+        "Backend base URL (preferred) e.g. http://localhost:8000/api/v1",
+    )
     t.add_row("BACKEND_BASE_URL", "balance", "Fallback backend base URL")
     t.add_row("LOCAL_BACKEND_URL", "balance", "Another fallback backend base URL")
     t.add_row("SYNTH_BASE_URL", "balance", "Generic base URL (may point to Modal, guarded)")
     t.add_row("SYNTH_BACKEND_API_KEY", "balance", "Backend API key (preferred)")
     t.add_row("SYNTH_API_KEY", "balance, env*", "API key used if backend-specific key not set")
     t.add_row("DEFAULT_DEV_API_KEY", "balance", "Dev fallback key for local testing")
-    t.add_row("SYNTH_TRACES_ROOT", "traces", "Root directory of local trace DBs (default ./synth_ai.db/dbs)")
+    t.add_row(
+        "SYNTH_TRACES_ROOT",
+        "traces",
+        "Root directory of local trace DBs (default ./synth_ai.db/dbs)",
+    )
     return t
 
 
@@ -79,7 +80,10 @@ def _examples_table() -> Table:
     t.add_column("Command")
     t.add_column("Example")
     t.add_row("Balance (local backend)", "uvx . balance")
-    t.add_row("Balance with URL+key", "uvx . balance --base-url http://localhost:8000 --api-key $SYNTH_API_KEY")
+    t.add_row(
+        "Balance with URL+key",
+        "uvx . balance --base-url http://localhost:8000 --api-key $SYNTH_API_KEY",
+    )
     t.add_row("Traces (default root)", "uvx . traces")
     t.add_row("Traces (custom root)", "uvx . traces --root /path/to/dbs")
     t.add_row("Experiments", "uvx . experiments --limit 20")
@@ -100,4 +104,3 @@ def register(cli):
         console.print(_commands_table())
         console.print(_env_table())
         console.print(_examples_table())
-

synth_ai/cli/recent.py

@@ -5,12 +5,11 @@ CLI: experiments active in the last K hours with summary stats.
 
 import asyncio
 from datetime import datetime, timedelta
-from typing import Optional
 
 import click
+from rich import box
 from rich.console import Console
 from rich.table import Table
-from rich import box
 
 
 def _fmt_int(v) -> str:
@@ -90,18 +89,20 @@ def register(cli):
         async def _run():
             df = await _fetch_recent(db_url, hours)
 
-            table = Table(title=f"Experiments in last {hours:g}h", header_style="bold", box=box.SIMPLE)
+            table = Table(
+                title=f"Experiments in last {hours:g}h", header_style="bold", box=box.SIMPLE
+            )
             for col in ["Experiment", "Runs", "First", "Last", "Events", "Msgs", "Cost", "Tokens"]:
-                table.add_column(col, justify="right" if col in {"Runs","Events","Msgs","Tokens"} else "left")
+                table.add_column(
+                    col, justify="right" if col in {"Runs", "Events", "Msgs", "Tokens"} else "left"
+                )
 
             if df is None or df.empty:
                 table.add_row("-", "0", "-", "-", "-", "-", "-", "-")
             else:
-                count = 0
-                for _, r in df.iterrows():
-                    if count >= limit:
+                for count, (_, r) in enumerate(df.iterrows(), start=1):
+                    if count > limit:
                         break
-                    count += 1
                     name = r.get("name") or "Unnamed"
                     exp_disp = f"{name[:28]} [dim]({_short(r.get('experiment_id'))})[/dim]"
                     table.add_row(