synth-ai 0.2.6.dev5__py3-none-any.whl → 0.2.7__py3-none-any.whl

synth_ai/cli/balance.py

@@ -15,7 +15,9 @@ from rich import box
 from rich.console import Console
 from rich.table import Table
 
-PROD_BACKEND_BASE = "https://agent-learning.onrender.com/api/v1"
+from synth_ai.config.base_url import get_backend_from_env, PROD_BASE_URL_DEFAULT
+
+PROD_BACKEND_BASE = f"{PROD_BASE_URL_DEFAULT.rstrip('/')}/api/v1"
 
 
 def _get_default_base_url() -> str:
@@ -24,7 +26,11 @@ def _get_default_base_url() -> str:
         val = os.getenv(var)
         if val:
             return val
-    return PROD_BACKEND_BASE
+    base, _ = get_backend_from_env()
+    base = base.rstrip("/")
+    if base.endswith("/api"):
+        base = base[:-len("/api")]
+    return f"{base}/api/v1"
 
 
 def _ensure_api_v1_prefix(base_url: str) -> str:

synth_ai/cli/demo.py

@@ -92,7 +92,11 @@ def register(cli):
 
     # (prepare command removed; configure now prepares baseline TOML)
 
-    @demo.command("deploy")
+    # Help pyright understand dynamic Click group attributes
+    from typing import Any, cast as _cast
+    _dg = _cast(Any, demo)
+
+    @_dg.command("deploy")
     @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")
@@ -109,11 +113,15 @@ def register(cli):
             args.extend(["--script", script])
         _forward_to_new(args)
 
-    @demo.command("configure")
+    @_dg.command("configure")
     def demo_configure():
         _forward_to_new(["rl_demo.configure"]) 
 
-    @demo.command("run")
+    @_dg.command("setup")
+    def demo_setup():
+        _forward_to_new(["rl_demo.setup"]) 
+
+    @_dg.command("run")
     @click.option("--batch-size", type=int, default=None)
     @click.option("--group-size", type=int, default=None)
     @click.option("--model", type=str, default=None)

synth_ai/cli/rl_demo.py

@@ -7,6 +7,7 @@ Usage examples:
   uvx synth-ai rl_demo deploy --app /path/to/math_task_app.py --name synth-math-demo
   uvx synth-ai rl_demo configure
   uvx synth-ai rl_demo run --batch-size 4 --group-size 16 --model Qwen/Qwen3-0.6B
+  uvx synth-ai run --config demo_config.toml
 
 For convenience, dotted aliases are also exposed:
   uvx synth-ai rl_demo.check
@@ -38,9 +39,9 @@ def register(cli):
     from typing import Any, cast as _cast
     _rlg = _cast(Any, rl_demo)
 
-    @_rlg.command("check")
-    def rl_check():
-        _forward(["rl_demo.check"])  # reuse same implementation
+    @_rlg.command("setup")
+    def rl_setup():
+        _forward(["rl_demo.setup"])  # primary setup command
 
     # (prepare command removed; consolidated into configure)
 
@@ -96,10 +97,14 @@ def register(cli):
             args.append("--dry-run")
         _forward(args)
 
-    # Dotted aliases (top-level) for convenience: rl_demo.check etc.
+    # Dotted aliases (top-level): legacy check → setup
     @cli.command("rl_demo.check")
     def rl_check_alias():
-        _forward(["rl_demo.check"]) 
+        _forward(["rl_demo.setup"]) 
+
+    @cli.command("rl_demo.setup")
+    def rl_setup_alias():
+        _forward(["rl_demo.setup"]) 
 
     # (prepare alias removed)
 
@@ -154,3 +159,44 @@ def register(cli):
         if dry_run:
             args.append("--dry-run")
         _forward(args)
+
+    # Top-level convenience alias: `synth-ai deploy`
+    @cli.command("deploy")
+    @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_top(local: bool, app: str | None, name: str, script: str | None):
+        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(args)
+
+    @cli.command("run")
+    @click.option("--config", type=click.Path(), default=None, help="Path to TOML config (skip prompt)")
+    @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)
+    @click.option("--dry-run", is_flag=True, help="Print request body and exit")
+    def run_top(config: str | None, batch_size: int | None, group_size: int | None, model: str | None, timeout: int, dry_run: bool):
+        args = ["run"]
+        if config:
+            args.extend(["--config", config])
+        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 is not None:
+            args.extend(["--timeout", str(timeout)])
+        if dry_run:
+            args.append("--dry-run")
+        _forward(args)

synth_ai/cli/root.py

@@ -117,6 +117,12 @@ def configure():
     _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("--batch-size", type=int, default=None)
 @click.option("--group-size", type=int, default=None)

synth_ai/config/base_url.py

@@ -66,6 +66,7 @@ def get_backend_from_env() -> tuple[str, str]:
     """Resolve (base_url, api_key) using a simple LOCAL/DEV/PROD override scheme.
 
     Env vars consulted:
+    - BACKEND_OVERRIDE = full URL (with or without /api)
     - SYNTH_BACKEND_URL_OVERRIDE = local|dev|prod (case-insensitive)
     - LOCAL_BACKEND_URL, TESTING_LOCAL_SYNTH_API_KEY
     - DEV_BACKEND_URL, DEV_SYNTH_API_KEY
@@ -74,6 +75,14 @@ def get_backend_from_env() -> tuple[str, str]:
     Base URL is normalized to end with '/api'.
     Defaults: prod base URL → https://agent-learning.onrender.com/api
     """
+    direct_override = (os.getenv("BACKEND_OVERRIDE") or "").strip()
+    if direct_override:
+        base = direct_override.rstrip("/")
+        if base.endswith("/api"):
+            base = base[: -len("/api")]
+        api_key = os.getenv("SYNTH_API_KEY", "").strip()
+        return base, api_key
+
     mode = _resolve_override_mode()
     if mode == "local":
         base = os.getenv("LOCAL_BACKEND_URL", "http://localhost:8000")