synth-ai 0.2.8.dev2__py3-none-any.whl → 0.4.3__py3-none-any.whl

synth_ai/cli/demo_apps/demo_task_apps/math/task_app_entry.py

@@ -0,0 +1,39 @@
+"""Task app registry entry for the math demo Modal deployment."""
+
+from __future__ import annotations
+
+from synth_ai.sdk.task.apps import ModalDeploymentConfig, TaskAppEntry, register_task_app
+from synth_ai.sdk.task.apps.math_single_step import (  # type: ignore[unresolved-import]
+    build_config as base_build_config,
+)
+
+DEMO_MODAL_CONFIG = ModalDeploymentConfig(
+    app_name="hendrycks-math-task-app",
+    pip_packages=(
+        "fastapi>=0.110.0",
+        "uvicorn>=0.23.0",
+        "pydantic>=2.6.0",
+        "httpx>=0.24.0",
+        "numpy>=1.24.0",
+        "aiohttp>=3.8.0",
+        "datasets>=2.16.0",
+        "synth-ai",
+    ),
+)
+
+
+def build_config():
+    """Reuse the shared math single-step TaskAppConfig."""
+
+    return base_build_config()
+
+
+register_task_app(
+    entry=TaskAppEntry(
+        app_id="hendrycks-math-demo",
+        description="Demo math task app (Modal-focused) shipping with synth-ai demos.",
+        config_factory=build_config,
+        env_files=("examples/rl/.env",),
+        modal=DEMO_MODAL_CONFIG,
+    )
+)

synth_ai/cli/demo_apps/math/__init__.py

@@ -0,0 +1 @@
+# Package namespace for Math demo task app

synth_ai/cli/demo_apps/math/_common.py

@@ -0,0 +1,16 @@
+from __future__ import annotations
+
+"""Minimal helpers for the math task app.
+
+This module provides a local fallback for install_problem_bank_into_shared so
+the modal task app can import it without requiring an external math_rl package.
+"""
+
+
+def install_problem_bank_into_shared() -> None:
+    """No-op placeholder for installing the Hendrycks MATH problem bank.
+
+    In production deployments, this can download or unpack the problem bank
+    into a shared directory. For the demo scaffold, it is a no-op.
+    """
+    return None

synth_ai/cli/demo_apps/math/app.py

@@ -0,0 +1,38 @@
+from __future__ import annotations
+
+import os
+
+from fastapi import FastAPI
+from starlette.middleware.cors import CORSMiddleware
+
+# Reuse the examples/rl task_app routes if available
+try:
+    from synth_ai.examples.rl.task_app import make_app as make_rl_app  # type: ignore
+except Exception:  # fallback path when imported from repo root
+    try:
+        from examples.rl.task_app import make_app as make_rl_app  # type: ignore
+    except Exception as e:  # pragma: no cover
+        raise ImportError(f"Unable to import RL task app: {e}") from e
+
+
+def create_app() -> FastAPI:
+    # Configure math defaults via env (consumed by RL task_app helpers)
+    os.environ.setdefault("DEMO_ENV_NAME", "math")
+    os.environ.setdefault("DEMO_POLICY_NAME", "math-react")
+    # Build base app
+    app = make_rl_app()
+    # CORS for local demo
+    app.add_middleware(
+        CORSMiddleware,
+        allow_origins=["*"],
+        allow_credentials=True,
+        allow_methods=["*"],
+        allow_headers=["*"],
+    )
+    return app
+
+
+def run(host: str = "127.0.0.1", port: int = 8080):
+    import uvicorn
+
+    uvicorn.run(create_app(), host=host, port=int(os.getenv("PORT", port)))

synth_ai/cli/demo_apps/math/config.toml

@@ -0,0 +1,75 @@
+type = "rl"
+
+[algorithm]
+type = "online"
+method = "policy_gradient"
+variety = "gspo"
+
+[services]
+task_url = "http://localhost:8101"
+
+[model]
+base = "Qwen/Qwen3-1.7B"
+
+[policy]
+model = "Qwen/Qwen3-1.7B"
+inference_url = "http://localhost:8000/api/inference"
+max_tokens = 1028
+temperature = 0.2
+
+[data]
+split = "train"
+seed_start = 0
+episodes_per_iteration = 1280  # 8 per group * 4 groups per batch * 2 batches per step * 20 steps
+evaluation_split = "validation"
+evaluation_episodes = 50
+
+[training]
+max_turns = 1
+batch_size = 2
+group_size = 16
+reward_positive = 1.0
+reward_negative_no_tool = -1.0
+reward_negative_no_answer = -0.5
+learning_rate = 5e-6
+log_interval = 1
+weight_sync_interval = 1
+
+[training.weight_sync]
+enable = true
+targets = ["policy"]
+
+[compute]
+gpu_type = "H100"
+gpu_count = 4
+
+[topology]
+type = "single_node_split"
+gpus_for_vllm = 2
+gpus_for_training = 1
+gpus_for_ref = 1
+tensor_parallel = 1
+
+[vllm]
+tensor_parallel_size = 1
+max_model_len = 4096
+
+[reference]
+placement = "dedicated"
+port = 8002
+tp = 1
+health_max_wait_s = 180
+health_interval_ms = 300
+
+[rollout]
+policy_name = "math-single-step"
+max_turns = 1
+episodes_per_batch = 32  # group_size * batch_size
+
+[evaluation]
+instances = 32
+every_n_iters = 10
+seeds = [0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15]
+
+[tags]
+experiment = "math_single_step_qwen17"

synth_ai/cli/demo_apps/math/deploy_modal.py

@@ -0,0 +1,54 @@
+from __future__ import annotations
+
+import os
+import subprocess
+
+
+def _parse_public_url_from_log(log_path: str) -> str | None:
+    try:
+        with open(log_path) as fh:
+            for line in fh:
+                if "modal.run" in line:
+                    return line.strip().split()[-1].rstrip("/")
+    except Exception:
+        return None
+    return None
+
+
+def deploy(script_path: str | None = None, *, env_api_key: str | None = None) -> str:
+    """
+    Deploy the Math Task App to Modal and return the public URL.
+
+    - If script_path is provided, run it (bash) and parse .last_deploy*.log for URL.
+    - Otherwise, try to call a built-in deploy() in examples.rl.task_app if available.
+    """
+    envp = os.environ.copy()
+    if env_api_key:
+        envp["ENVIRONMENT_API_KEY"] = env_api_key
+
+    if script_path:
+        script_path = os.path.abspath(script_path)
+        if not os.path.isfile(script_path):
+            raise FileNotFoundError(f"Deploy script not found: {script_path}")
+        subprocess.check_call(["bash", script_path], cwd=os.path.dirname(script_path), env=envp)
+        for name in (".last_deploy.log", ".last_deploy.dev.log", ".last_deploy.manual.log"):
+            url = _parse_public_url_from_log(os.path.join(os.path.dirname(script_path), name))
+            if url:
+                return url
+        raise RuntimeError("Deployed, but failed to extract Modal public URL from deploy logs.")
+
+    try:
+        import importlib
+
+        mod = importlib.import_module("examples.rl.task_app")
+        if hasattr(mod, "deploy"):
+            url = mod.deploy(env_api_key=env_api_key)
+            if not url:
+                raise RuntimeError("examples.rl.task_app.deploy() returned empty URL")
+            return str(url).rstrip("/")
+        raise RuntimeError("examples.rl.task_app.deploy() not found")
+    except Exception as exc:
+        raise RuntimeError(
+            f"No deploy script provided and Python-based deploy failed: {exc}. "
+            "Pass --script /path/to/deploy_task_app.sh to demo.deploy."
+        ) from exc