synth-ai 0.2.13.dev1__py3-none-any.whl → 0.2.14__py3-none-any.whl

synth_ai/cli/_validate_task_app.py

@@ -0,0 +1,11 @@
+"""Task app validation utilities - imported by task_apps.py"""
+
+# This module provides the validate_task_app function for CLI use
+# The actual implementation is imported from the task module
+
+from synth_ai.task.validators import (
+    validate_task_app_endpoint as validate_task_app,  # type: ignore[attr-defined]
+)
+
+__all__ = ["validate_task_app"]
+

synth_ai/cli/balance.py

@@ -8,6 +8,7 @@ from __future__ import annotations
 import importlib
 import os
 from collections.abc import Callable
+from typing import Any, cast
 
 import click
 import requests
@@ -19,9 +20,9 @@ from rich.table import Table
 
 def _load_base_url_module() -> tuple[str, Callable[[], tuple[str, str]]]:
     try:
-        module = importlib.import_module("synth_ai.config.base_url")
-        default = module.PROD_BASE_URL_DEFAULT
-        getter = module.get_backend_from_env
+        module = cast(Any, importlib.import_module("synth_ai.config.base_url"))
+        default = cast(str, module.PROD_BASE_URL_DEFAULT)
+        getter = cast(Callable[[], tuple[str, str]], module.get_backend_from_env)
         return str(default), getter
     except Exception:
         return "https://agent-learning.onrender.com", lambda: ("https://agent-learning.onrender.com", "")

synth_ai/cli/calc.py

@@ -35,13 +35,13 @@ def _safe_eval(expr: str) -> float:
         if isinstance(n, ast.Expression):
             return _eval(n.body)
         if isinstance(n, ast.Constant):
-            if isinstance(n.value, (int, float)):
+            if isinstance(n.value, int | float):
                 return float(n.value)
             raise ValueError("Only numeric constants are allowed")
         num_node = getattr(ast, "Num", None)
         if num_node is not None and isinstance(n, num_node):  # pragma: no cover
             numeric_value = getattr(n, "n", None)
-            if isinstance(numeric_value, (int, float)):
+            if isinstance(numeric_value, int | float):
                 return float(numeric_value)
             raise ValueError("Only numeric constants are allowed")
         if isinstance(n, ast.BinOp):

synth_ai/cli/demo.py

@@ -8,13 +8,18 @@ CLI: interactive launcher for example demos and RL demo helpers.
 
 from __future__ import annotations
 
+import importlib
 import os
 import subprocess
 from pathlib import Path
+from typing import Any, cast
 
 import click
+from click.exceptions import Exit
 
-from synth_ai.demos.core import cli as demo_commands
+demo_commands = cast(
+    Any, importlib.import_module("synth_ai.demos.core.cli")
+)
 
 
 def _find_demo_scripts(root: Path) -> list[Path]:
@@ -29,7 +34,7 @@ def _run_demo_command(func, *args, **kwargs) -> None:
     try:
         result = func(*args, **kwargs)
     except SystemExit as exc:  # pragma: no cover - defensive
-        raise click.exceptions.Exit(exc.code or 1) from exc
+        raise Exit(exc.code or 1) from exc
 
     if result is None:
         return
@@ -39,7 +44,7 @@ def _run_demo_command(func, *args, **kwargs) -> None:
     except (TypeError, ValueError):
         return
     if code != 0:
-        raise click.exceptions.Exit(code)
+        raise Exit(code)
 
 
 def register(cli):
@@ -106,10 +111,7 @@ def register(cli):
     # (prepare command removed; configure now prepares baseline TOML)
 
     # Help pyright understand dynamic Click group attributes
-    from typing import Any
-    from typing import cast as _cast
-
-    _dg = _cast(Any, demo)
+    _dg = cast(Any, demo)
 
     @_dg.command("deploy")
     @click.option("--local", is_flag=True, help="Run local FastAPI instead of Modal deploy")
@@ -156,3 +158,8 @@ def register(cli):
             model=model,
             timeout=timeout,
         )
+
+    @cli.command("setup")
+    def setup_alias():
+        """Perform SDK handshake and write keys to .env."""
+        _run_demo_command(demo_commands.setup)

synth_ai/cli/legacy_root_backup.py

@@ -235,7 +235,7 @@ def view(url: str):
     """Launch the interactive TUI dashboard."""
     try:
         module = importlib.import_module(".tui.dashboard", __package__)
-        synth_dashboard_cls = module.SynthDashboard
+        synth_dashboard_cls = getattr(module, "SynthDashboard")
         app = synth_dashboard_cls(db_url=url)
         app.run()
     except ImportError:

synth_ai/cli/recent.py

@@ -12,7 +12,7 @@ from rich import box
 from rich.console import Console
 from rich.table import Table
 
-from ._storage import load_storage
+from synth_ai.cli._storage import load_storage
 
 if TYPE_CHECKING:  # pragma: no cover - typing only
     import pandas as pd

synth_ai/cli/rl_demo.py

@@ -15,9 +15,13 @@ For convenience, dotted aliases are also exposed:
 
 from __future__ import annotations
 
+import importlib
+from typing import Any, cast
+
 import click
+from click.exceptions import Exit
 
-from synth_ai.demos.core import cli as demo_commands
+demo_commands = cast(Any, importlib.import_module("synth_ai.demos.core.cli"))
 
 
 def _run_demo_command(func, *args, **kwargs) -> None:
@@ -26,7 +30,7 @@ def _run_demo_command(func, *args, **kwargs) -> None:
     try:
         result = func(*args, **kwargs)
     except SystemExit as exc:  # pragma: no cover - defensive
-        raise click.exceptions.Exit(exc.code or 1) from exc
+        raise Exit(exc.code or 1) from exc
 
     if result is None:
         return
@@ -35,7 +39,7 @@ def _run_demo_command(func, *args, **kwargs) -> None:
     except (TypeError, ValueError):
         return
     if code != 0:
-        raise click.exceptions.Exit(code)
+        raise Exit(code)
 
 
 def register(cli):
@@ -44,10 +48,7 @@ def register(cli):
         """RL Demo commands (separate from legacy demo)."""
 
     # Help pyright understand dynamic Click group attributes
-    from typing import Any
-    from typing import cast as _cast
-
-    _rlg = _cast(Any, rl_demo)
+    _rlg = cast(Any, rl_demo)
 
     @_rlg.command("setup")
     def rl_setup():

synth_ai/cli/root.py

@@ -156,103 +156,6 @@ def cli():
     """Top-level command group for Synth AI."""
 
 
-# === Legacy demo command group (aliases new rl_demo implementation) ===
-@cli.group()
-def demo():
-    """Demo helpers (deploy, configure, run)."""
-
-
-def _forward_to_demo(args: list[str]) -> None:
-    # Lazy import to avoid loading demo deps unless needed
-    try:
-        from synth_ai.demos.core import cli as demo_cli  # type: ignore
-    except Exception as e:  # pragma: no cover
-        click.echo(f"Failed to import demo CLI: {e}")
-        sys.exit(1)
-    rc = int(demo_cli.main(args) or 0)  # type: ignore[attr-defined]
-    if rc != 0:
-        sys.exit(rc)
-
-
-# (prepare command removed; handled by configure)
-
-
-@demo.command()
-@click.option("--local", is_flag=True, help="Run local FastAPI instead of Modal deploy")
-@click.option(
-    "--app", type=click.Path(), default=None, help="Path to Modal app.py for uv run modal deploy"
-)
-@click.option("--name", type=str, default="synth-math-demo", help="Modal app name")
-@click.option(
-    "--script", type=click.Path(), default=None, help="Path to deploy_task_app.sh (optional legacy)"
-)
-def deploy(local: bool, app: str | None, name: str, script: str | None):
-    """Deploy the Math Task App (Modal by default)."""
-    args: list[str] = ["rl_demo.deploy"]
-    if local:
-        args.append("--local")
-    if app:
-        args.extend(["--app", app])
-    if name:
-        args.extend(["--name", name])
-    if script:
-        args.extend(["--script", script])
-    _forward_to_demo(args)
-
-
-@demo.command()
-def configure():
-    """Print resolved environment and config path."""
-    _forward_to_demo(["rl_demo.configure"])
-
-
-@demo.command()
-def setup():
-    """Perform SDK handshake and write keys to .env."""
-    _forward_to_demo(["rl_demo.setup"])
-
-
-@demo.command()
-@click.option("--template", type=str, default=None, help="Template id to instantiate")
-@click.option("--dest", type=str, default=None, help="Destination directory for files")
-@click.option("--force", is_flag=True, help="Overwrite existing files in destination")
-def init(template: str | None, dest: str | None, force: bool):
-    """Copy demo task app template into the current directory."""
-    args: list[str] = ["demo.init"]
-    if template:
-        args.extend(["--template", template])
-    if dest:
-        args.extend(["--dest", dest])
-    if force:
-        args.append("--force")
-    _forward_to_demo(args)
-
-
-@demo.command()
-@click.option("--batch-size", type=int, default=None)
-@click.option("--group-size", type=int, default=None)
-@click.option("--model", type=str, default=None)
-@click.option("--timeout", type=int, default=600)
-def run(batch_size: int | None, group_size: int | None, model: str | None, timeout: int):
-    """Kick off a short RL job using the prepared TOML."""
-    args = ["rl_demo.run"]
-    if batch_size is not None:
-        args.extend(["--batch-size", str(batch_size)])
-    if group_size is not None:
-        args.extend(["--group-size", str(group_size)])
-    if model:
-        args.extend(["--model", model])
-    if timeout:
-        args.extend(["--timeout", str(timeout)])
-    _forward_to_demo(args)
-
-
-@cli.command(name="setup")
-def setup_command():
-    """Perform SDK handshake and write keys to .env."""
-    _forward_to_demo(["rl_demo.setup"])
-
-
 @cli.command()
 @click.option("--db-file", default="traces/v3/synth_ai.db", help="Database file path")
 @click.option("--sqld-port", default=8080, type=int, help="Port for sqld HTTP interface")

synth_ai/cli/status.py

@@ -12,7 +12,7 @@ from rich.console import Console
 from rich.panel import Panel
 from rich.table import Table
 
-from ._storage import load_storage
+from synth_ai.cli._storage import load_storage
 
 
 async def _db_stats(db_url: str) -> dict: