synth-ai 0.2.14__py3-none-any.whl → 0.4.4__py3-none-any.whl

examples/task_apps/math/math_task_app.py

@@ -1,115 +0,0 @@
-"""Legacy entrypoint for the math single-step task app."""
-
-from __future__ import annotations
-
-import argparse
-from pathlib import Path
-
-from fastapi.exceptions import RequestValidationError
-from fastapi.responses import JSONResponse
-from starlette.requests import Request
-from synth_ai.task.auth import is_api_key_header_authorized, normalize_environment_api_key
-from synth_ai.task.server import create_task_app, run_task_app
-
-from .math_single_step import build_config
-
-
-def fastapi_app():
-    """Return a FastAPI application for hosting the math task app."""
-
-    app = create_task_app(build_config())
-
-    # Replace default health endpoints with auth-tolerant handlers.
-    filtered_routes = []
-    for route in app.router.routes:
-        path = getattr(route, "path", None)
-        methods = getattr(route, "methods", set()) or set()
-        if path in {"/health", "/health/rollout"} and "GET" in methods:
-            continue
-        filtered_routes.append(route)
-    app.router.routes = filtered_routes
-
-    def _log_env_key_prefix(source: str, env_key: str | None) -> str | None:
-        if not env_key:
-            return None
-        prefix = env_key[: max(1, len(env_key) // 2)]
-        print(f"[{source}] expected ENVIRONMENT_API_KEY prefix: {prefix}")
-        return prefix
-
-    @app.get("/health")
-    async def health(request: Request):
-        env_key = normalize_environment_api_key()
-        if not env_key:
-            return JSONResponse(
-                status_code=503,
-                content={"status": "unhealthy", "detail": "Missing ENVIRONMENT_API_KEY"},
-            )
-        if not is_api_key_header_authorized(request):
-            prefix = _log_env_key_prefix("health", env_key)
-            content = {"status": "healthy", "authorized": False}
-            if prefix:
-                content["expected_api_key_prefix"] = prefix
-            return JSONResponse(status_code=200, content=content)
-        return {"status": "healthy", "authorized": True}
-
-    @app.get("/health/rollout")
-    async def health_rollout(request: Request):
-        env_key = normalize_environment_api_key()
-        if not env_key:
-            return JSONResponse(
-                status_code=503,
-                content={"status": "unhealthy", "detail": "Missing ENVIRONMENT_API_KEY"},
-            )
-        if not is_api_key_header_authorized(request):
-            prefix = _log_env_key_prefix("health/rollout", env_key)
-            content = {"status": "healthy", "authorized": False}
-            if prefix:
-                content["expected_api_key_prefix"] = prefix
-            return JSONResponse(status_code=200, content=content)
-        return {"ok": True, "authorized": True}
-
-    @app.exception_handler(RequestValidationError)
-    async def _on_validation_error(request: Request, exc: RequestValidationError):
-        try:
-            hdr = request.headers
-            snapshot = {
-                "path": str(request.url.path),
-                "have_x_api_key": bool(hdr.get("x-api-key")),
-                "have_x_api_keys": bool(hdr.get("x-api-keys")),
-                "have_authorization": bool(hdr.get("authorization")),
-                "errors": exc.errors()[:5],
-            }
-            print("[422] validation", snapshot, flush=True)
-        except Exception:
-            pass
-        return JSONResponse(
-            status_code=422, content={"status": "invalid", "detail": exc.errors()[:5]}
-        )
-
-    return app
-
-
-if __name__ == "__main__":
-    parser = argparse.ArgumentParser(description="Run the math single-step task app locally")
-    parser.add_argument("--host", default="0.0.0.0")
-    parser.add_argument("--port", type=int, default=8101)
-    parser.add_argument("--reload", action="store_true", help="Enable uvicorn autoreload")
-    parser.add_argument(
-        "--env-file",
-        action="append",
-        default=[],
-        help="Additional .env files to load before startup",
-    )
-    args = parser.parse_args()
-
-    default_env = Path(__file__).resolve().parents[2] / ".env"
-    env_files = [str(default_env)] if default_env.exists() else []
-    env_files.extend(args.env_file or [])
-
-    run_task_app(
-        build_config,
-        host=args.host,
-        port=args.port,
-        reload=args.reload,
-        env_files=env_files,
-    )

examples/task_apps/pokemon_battle/__init__.py

@@ -1,2 +0,0 @@
-"""Pokemon competitive battle task app examples."""
-

examples/task_apps/pokemon_battle/modal_app.py

@@ -1,104 +0,0 @@
-"""Modal deployment helper for the Pokémon Showdown task app example.
-
-This file mirrors the manual setup steps documented in the README:
-
-- Clone `pokechamp` and install its Python dependencies.
-- Clone the reference Pokémon Showdown server and install Node dependencies.
-- Mount the local `synth-ai` repository so the task app code is available.
-
-Deploy with:
-
-```
-modal deploy examples/task_apps/pokemon_battle/modal_app.py
-```
-
-After deployment the FastAPI service will be reachable at a URL similar to
-`https://<org>--pokemon-showdown-task-app-example.modal.run`.
-"""
-
-from __future__ import annotations
-
-import subprocess
-import sys
-from pathlib import Path
-
-import modal
-
-REPO_ROOT = Path(__file__).resolve().parents[3]
-POKECHAMP_REPO = "https://github.com/sethkarten/pokechamp.git"
-SHOWDOWN_REPO = "https://github.com/jakegrigsby/pokemon-showdown.git"
-
-app = modal.App("pokemon-showdown-task-app-example")
-
-BASE_IMAGE = (
-    modal.Image.debian_slim(python_version="3.11")
-    .apt_install("git", "nodejs", "npm")
-    .pip_install(["uvicorn[standard]", "fastapi", "httpx", "horizons-ai"])
-    .run_commands(
-        [
-            "mkdir -p /external",
-            f"git clone --depth 1 {POKECHAMP_REPO} /external/pokechamp || true",
-            "pip install --no-cache-dir -r /external/pokechamp/requirements.txt",
-            f"git clone --depth 1 {SHOWDOWN_REPO} /external/pokemon-showdown || true",
-            "cd /external/pokemon-showdown && npm ci --no-optional",
-        ]
-    )
-)
-
-REPO_MOUNT = modal.Mount.from_local_dir(REPO_ROOT, remote_path="/workspace/synth-ai")
-
-
-@app.function(
-    image=BASE_IMAGE,
-    mounts=[REPO_MOUNT],
-    timeout=900,
-    memory=8192,
-    cpu=4.0,
-    secrets=[modal.Secret.from_name("environment-api-key")],
-    keep_warm=1,
-)
-@modal.asgi_app()
-def fastapi_app():
-    """Serve the Synth task app via Modal."""
-
-    import os
-    from fastapi import APIRouter
-
-    repo_path = Path("/workspace/synth-ai").resolve()
-    if str(repo_path) not in sys.path:
-        sys.path.insert(0, str(repo_path))
-
-    marker = Path("/tmp/.synth_ai_editable")
-    if not marker.exists():
-        subprocess.check_call([sys.executable, "-m", "pip", "install", "-e", str(repo_path)])
-        marker.touch()
-
-    os.environ.setdefault("PYTHONHASHSEED", "0")
-    os.environ.setdefault("OPENBLAS_NUM_THREADS", "1")
-    os.environ.setdefault("OMP_NUM_THREADS", "1")
-    os.environ.setdefault("POKECHAMP_ROOT", "/external/pokechamp")
-    os.environ.setdefault("POKEMON_SHOWDOWN_ROOT", "/external/pokemon-showdown")
-
-    from examples.task_apps.pokemon_battle.task_app.pokemon_showdown import build_config
-    from synth_ai.task.server import create_task_app
-
-    app = create_task_app(build_config())
-
-    health_router = APIRouter()
-
-    @health_router.get("/healthz")
-    def healthz():
-        return {"status": "ok"}
-
-    app.include_router(health_router)
-    return app
-
-
-@app.local_entrypoint()
-def main():
-    """Print handy commands for local testing."""
-
-    print("Pokémon Showdown task app Modal helper")
-    print("Deploy with: modal deploy examples/task_apps/pokemon_battle/modal_app.py")
-    print("Test locally: modal serve examples/task_apps/pokemon_battle/modal_app.py")
-    print("Once deployed, set TASK_APP_URL to the issued modal.run domain.")

examples/task_apps/pokemon_battle/task_app/README.md

@@ -1,68 +0,0 @@
-# Pokemon Battle Task App
-
-This example shows how to expose a Horizons-compatible Pokémon Showdown battle
-environment through the Synth AI task app scaffolding. The adapter runs fully
-locally by driving pokechamp’s deterministic `LocalSim`, so battles can be
-snapshotted and restored without a live Showdown server.
-
-## Local setup (Track 1)
-
-1. Clone and install **PokeChamp** together with its `poke-env` fork:
-
-   ```bash
-   git clone https://github.com/sethkarten/pokechamp.git
-   cd pokechamp
-   pip install -r requirements.txt
-   ```
-
-2. Export environment variables so the task app can locate the cloned repo:
-
-   ```bash
-   export POKECHAMP_ROOT=/path/to/pokechamp
-   export POKEMON_SHOWDOWN_ROOT=/path/to/pokemon-showdown
-   ```
-
-3. Run a rollout to sanity-check the wiring:
-
-   ```bash
-   uv run python -m synth_ai.task.describe pokemon_showdown
-   uv run python -m synth_ai.task.rollout pokemon_showdown --seed 1001
-   ```
-
-The adapter uses the pokechamp dataset teams bundled with the repository to
-instantiate deterministic Gen 9 OU battles. You can point `POKECHAMP_ROOT` at a
-fork with custom teams to experiment with other formats.
-
-## Modal deployment
-
-A ready-to-use deployment helper is available at
-`examples/task_apps/pokemon_battle/modal_app.py`. It mirrors the above manual
-steps (cloning `pokechamp`, installing requirements, and mounting the Synth AI
-repo). Deploy with:
-
-```bash
-modal deploy examples/task_apps/pokemon_battle/modal_app.py
-```
-
-The resulting URL can be plugged into Synth AI workflows via `TASK_APP_URL`.
-
-## Notes
-
-- The dataset catalog resolves team files from the PokeChamp repo when available
-  (`POKECHAMP_ROOT`). If the assets are missing, `/info` marks the scenario as
-  unavailable.
-- Snapshots serialise the entire deterministic battle state, allowing training
-  algorithms to branch or reset mid-match.
-- Deterministic RNG seeding (Python, NumPy, PyTorch) keeps rollouts reproducible
-  across Modal replicas and local runs.
-- The opponent policy now favours super-effective moves to provide a stronger
-  baseline; swap it out with a pokechamp minimax bot for ladder-level play.
-- A `/healthz` endpoint is exposed in the Modal service for liveness probes.
-
-## Status & Next Steps
-
-- **Observation polish**: expose richer per-turn summaries (hazards, stat boosts, tera states) and compact text strings tailored for language agents.
-- **Action helpers**: surface explicit target slots/tera/mega toggles so higher formats (doubles, VGC) can plug in with minimal code.
-- **Benchmark opponent**: replace the heuristic opponent with a pokechamp bot (e.g. minimax) or hook into the official PokéAgent ladder for eval parity.
-- **Integration tests**: add pytest smoke tests covering `/snapshot` → `/restore` loops and multi-step rollouts.
-- **Agent wiring**: ship a reference RL/LLM policy config (Synth CLI or Modal job) that exercises the adapter end-to-end and logs battle traces.

examples/task_apps/pokemon_battle/task_app/__init__.py

@@ -1,6 +0,0 @@
-"""Pokemon Showdown task app configuration."""
-
-from .pokemon_showdown import build_config
-
-__all__ = ["build_config"]
-