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

synth_ai/cli/traces.py

@@ -51,6 +51,7 @@ def register(cli):
                     for fn in files:
                         fp = os.path.join(dp, fn)
                         import contextlib
+
                         with contextlib.suppress(OSError):
                             total += os.path.getsize(fp)
                 return total

synth_ai/cli/turso.py

@@ -0,0 +1,73 @@
+"""Utility CLI command for managing Turso sqld binaries."""
+
+from __future__ import annotations
+
+import subprocess
+
+import click
+
+from .root import SQLD_VERSION, find_sqld_binary, install_sqld
+
+
+def register(cli: click.Group) -> None:
+    """Register the turso command on the main CLI group."""
+
+    cli.add_command(turso)
+
+
+def _get_sqld_version(binary: str) -> str | None:
+    """Return the version string reported by the sqld binary."""
+
+    try:
+        result = subprocess.run(
+            [binary, "--version"],
+            capture_output=True,
+            text=True,
+            check=True,
+            timeout=5,
+        )
+    except (OSError, subprocess.CalledProcessError, ValueError):
+        return None
+
+    output = result.stdout.strip() or result.stderr.strip()
+    return output or None
+
+
+@click.command()
+@click.option(
+    "--force",
+    is_flag=True,
+    help="Reinstall the pinned sqld build even if one is already available.",
+)
+def turso(force: bool) -> None:
+    """Ensure the Turso sqld binary required for tracing v3 is installed."""
+
+    existing_path = find_sqld_binary()
+
+    if existing_path and not force:
+        version_info = _get_sqld_version(existing_path)
+        click.echo(f"✅ Turso sqld detected at {existing_path}.")
+        if version_info:
+            click.echo(f"   Reported version: {version_info}")
+        if version_info and SQLD_VERSION not in version_info:
+            click.echo(
+                f"⚠️ Pinned version is {SQLD_VERSION}. Run with --force to install the supported build."
+            )
+        else:
+            click.echo("No action taken. Use --force to reinstall the pinned build.")
+        return
+
+    if existing_path and force:
+        click.echo(f"♻️ Reinstalling Turso sqld {SQLD_VERSION} (previously at {existing_path}).")
+    else:
+        click.echo(f"📦 Installing Turso sqld {SQLD_VERSION}…")
+
+    try:
+        installed_path = install_sqld()
+    except subprocess.CalledProcessError as exc:  # pragma: no cover - surfaced as Click error
+        raise click.ClickException(
+            f"sqld installation failed (exit code {exc.returncode})."
+        ) from exc
+
+    click.echo(f"✅ sqld installed to {installed_path}")
+    click.echo("Ensure ~/.local/bin is on your PATH before running Synth AI services.")

synth_ai/core/experiment.py

@@ -1,5 +1,3 @@
-
-
 class ExperimentalSystem:
     system_id: str
     system_version_id: str

synth_ai/demo_registry.py

@@ -85,7 +85,7 @@ DEMO_TEMPLATES: tuple[DemoTemplate, ...] = (
     DemoTemplate(
         template_id="crafter-local",
         name="Crafter GRPO (local FastAPI)",
-        description="Lightweight wrapper around synth_ai.task.apps.grpo_crafter for local experimentation.",
+        description="Lightweight wrapper around examples/warming_up_to_rl/task_app/grpo_crafter for local experimentation.",
         copy_specs=(
             CopySpec(
                 "synth_ai/demos/demo_task_apps/crafter/grpo_crafter_task_app.py",
@@ -103,6 +103,34 @@ DEMO_TEMPLATES: tuple[DemoTemplate, ...] = (
                 "synth_ai/demos/demo_task_apps/crafter/configs/crafter_fft_4b.toml",
                 "configs/crafter_fft_4b.toml",
             ),
+            CopySpec(
+                "examples/warming_up_to_rl/task_app/grpo_crafter.py",
+                "grpo_crafter.py",
+            ),
+            CopySpec(
+                "examples/warming_up_to_rl/task_app/synth_envs_hosted",
+                "synth_envs_hosted",
+            ),
+            CopySpec(
+                "examples/warming_up_to_rl/run_local_rollout.py",
+                "run_local_rollout.py",
+            ),
+            CopySpec(
+                "examples/warming_up_to_rl/run_local_rollout_traced.py",
+                "run_local_rollout_traced.py",
+            ),
+            CopySpec(
+                "examples/warming_up_to_rl/export_trace_sft.py",
+                "export_trace_sft.py",
+            ),
+            CopySpec(
+                "examples/warming_up_to_rl/run_fft_and_save.py",
+                "run_fft_and_save.py",
+            ),
+            CopySpec(
+                "examples/warming_up_to_rl/run_local_rollout_modal.py",
+                "run_local_rollout_modal.py",
+            ),
         ),
         default_subdir="crafter_demo",
         env_lines=(
@@ -137,12 +165,16 @@ def _postprocess_math_modal(root: Path) -> None:
         return
     text = task_path.read_text(encoding="utf-8")
     text = text.replace('App("hendrycks-math-task-app")', 'App("hendrycks-math-task-app-demo")')
-    text = text.replace('DEFAULT_TASK_APP_SECRET_NAME = "hendrycks-math-task-app-secret"', 'DEFAULT_TASK_APP_SECRET_NAME = "hendrycks-math-task-app-demo-secret"')
+    text = text.replace(
+        'DEFAULT_TASK_APP_SECRET_NAME = "hendrycks-math-task-app-secret"',
+        'DEFAULT_TASK_APP_SECRET_NAME = "hendrycks-math-task-app-demo-secret"',
+    )
     task_path.write_text(text, encoding="utf-8")
 
 
-CRAFT_DEMO_TEMPLATE = textwrap.dedent(
-    '''
+CRAFT_DEMO_TEMPLATE = (
+    textwrap.dedent(
+        '''
 """Demo-friendly wrapper for the GRPO Crafter task app."""
 
 from __future__ import annotations
@@ -151,7 +183,12 @@ import argparse
 from pathlib import Path
 
 from synth_ai.task.apps import ModalDeploymentConfig, registry
-from synth_ai.task.apps.grpo_crafter import build_config
+import sys
+from pathlib import Path
+_EXAMPLES_TASK_APP = Path(__file__).resolve().parents[4] / "examples" / "warming_up_to_rl" / "task_app"
+if str(_EXAMPLES_TASK_APP) not in sys.path:
+    sys.path.insert(0, str(_EXAMPLES_TASK_APP))
+from grpo_crafter import build_config
 from synth_ai.task.server import TaskAppConfig, create_task_app, run_task_app
 
 
@@ -159,28 +196,6 @@ APP_ID = "grpo-crafter-demo"
 BASE_APP_ID = "grpo-crafter"
 
 
-_BASE_CONFIG = build_config()
-TASK_APP_CONFIG = TaskAppConfig(
-    app_id="grpo-crafter-demo",
-    name=_BASE_CONFIG.name,
-    description=_BASE_CONFIG.description,
-    base_task_info=_BASE_CONFIG.base_task_info,
-    describe_taskset=_BASE_CONFIG.describe_taskset,
-    provide_task_instances=_BASE_CONFIG.provide_task_instances,
-    rollout=_BASE_CONFIG.rollout,
-    dataset_registry=_BASE_CONFIG.dataset_registry,
-    rubrics=_BASE_CONFIG.rubrics,
-    proxy=_BASE_CONFIG.proxy,
-    routers=_BASE_CONFIG.routers,
-    middleware=_BASE_CONFIG.middleware,
-    app_state=_BASE_CONFIG.app_state,
-    require_api_key=_BASE_CONFIG.require_api_key,
-    expose_debug_env=_BASE_CONFIG.expose_debug_env,
-    cors_origins=_BASE_CONFIG.cors_origins,
-    startup_hooks=_BASE_CONFIG.startup_hooks,
-    shutdown_hooks=_BASE_CONFIG.shutdown_hooks,
-)
-
 try:
     _BASE_ENTRY = registry.get(BASE_APP_ID)
 except Exception:  # pragma: no cover - registry may be unavailable
@@ -213,7 +228,28 @@ ENV_FILES: tuple[str, ...] = ()
 def build_task_app_config() -> TaskAppConfig:
     """Return a fresh TaskAppConfig for the demo wrapper."""
 
-    return TASK_APP_CONFIG.clone()
+    # Build config dynamically so environment variables are read at runtime
+    _base_config = build_config()
+    return TaskAppConfig(
+        app_id="grpo-crafter-demo",
+        name=_base_config.name,
+        description=_base_config.description,
+        base_task_info=_base_config.base_task_info,
+        describe_taskset=_base_config.describe_taskset,
+        provide_task_instances=_base_config.provide_task_instances,
+        rollout=_base_config.rollout,
+        dataset_registry=_base_config.dataset_registry,
+        rubrics=_base_config.rubrics,
+        proxy=_base_config.proxy,
+        routers=_base_config.routers,
+        middleware=_base_config.middleware,
+        app_state=_base_config.app_state,
+        require_api_key=_base_config.require_api_key,
+        expose_debug_env=_base_config.expose_debug_env,
+        cors_origins=_base_config.cors_origins,
+        startup_hooks=_base_config.startup_hooks,
+        shutdown_hooks=_base_config.shutdown_hooks,
+    )
 
 
 def fastapi_app():
@@ -247,7 +283,9 @@ if __name__ == "__main__":
         env_files=env_files,
     )
     '''
-).strip() + "\n"
+    ).strip()
+    + "\n"
+)
 
 
 def _postprocess_crafter_local(root: Path) -> None:
@@ -255,4 +293,3 @@ def _postprocess_crafter_local(root: Path) -> None:
     if not task_path.exists():
         return
     task_path.write_text(CRAFT_DEMO_TEMPLATE, encoding="utf-8")
-