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

examples/rl/run_eval.py

@@ -0,0 +1,435 @@
+#!/usr/bin/env python3
+"""Evaluate math single-step task policies against the task app."""
+
+from __future__ import annotations
+
+import argparse
+import asyncio
+import json
+import os
+from pathlib import Path
+from typing import Any, Dict, List, Optional, Tuple
+
+import httpx
+import tomllib
+
+
+class TaskAppClient:
+    """Minimal async client for math single-step task app."""
+
+    def __init__(self, base_url: str, api_key: Optional[str] = None) -> None:
+        self.base_url = base_url.rstrip("/")
+        self.api_key = api_key
+        self._client: Optional[httpx.AsyncClient] = None
+
+    async def __aenter__(self) -> "TaskAppClient":
+        headers = {"X-API-Key": self.api_key} if self.api_key else {}
+        self._client = httpx.AsyncClient(
+            base_url=self.base_url, headers=headers, timeout=httpx.Timeout(120.0), follow_redirects=True
+        )
+        return self
+
+    async def __aexit__(self, exc_type, exc, tb) -> None:
+        if self._client:
+            await self._client.aclose()
+            self._client = None
+
+    @property
+    def client(self) -> httpx.AsyncClient:
+        if self._client is None:
+            headers = {"X-API-Key": self.api_key} if self.api_key else {}
+            self._client = httpx.AsyncClient(
+                base_url=self.base_url, headers=headers, timeout=httpx.Timeout(120.0), follow_redirects=True
+            )
+        return self._client
+
+    async def initialize(self, split: str, seed: int | None) -> Dict[str, Any]:
+        payload: Dict[str, Any] = {"config": {"split": split}}
+        if seed is not None:
+            payload["seed"] = seed
+        resp = await self.client.post("/env/math/initialize", json=payload)
+        resp.raise_for_status()
+        return resp.json()
+
+    async def step(self, env_id: str, tool_calls: List[Dict[str, Any]]) -> Dict[str, Any]:
+        payload = {"env_id": env_id, "action": {"tool_calls": tool_calls}}
+        resp = await self.client.post("/env/math/step", json=payload)
+        resp.raise_for_status()
+        return resp.json()
+
+    async def terminate(self, env_id: str) -> None:
+        try:
+            await self.client.post("/env/math/terminate", json={"env_id": env_id})
+        except Exception:
+            pass
+
+    async def get_info(self) -> Dict[str, Any]:
+        resp = await self.client.get("/info")
+        resp.raise_for_status()
+        return resp.json()
+
+    async def rollout(self, payload: Dict[str, Any]) -> Dict[str, Any]:
+        resp = await self.client.post("/rollout", json=payload)
+        resp.raise_for_status()
+        return resp.json()
+
+    async def post_inference(
+        self,
+        url: str,
+        payload: Dict[str, Any],
+        *,
+        headers: Dict[str, str] | None = None,
+    ) -> Dict[str, Any]:
+        async with httpx.AsyncClient(timeout=httpx.Timeout(60.0)) as c:
+            resp = await c.post(url, json=payload, headers=headers)
+        resp.raise_for_status()
+        return resp.json()
+
+
+TOOL_NAME = "math_submit"
+DEFAULT_SPLIT = os.getenv("MATH_EVAL_DEFAULT_SPLIT", "validation")
+
+
+def _math_tool_schema() -> List[Dict[str, Any]]:
+    return [
+        {
+            "type": "function",
+            "function": {
+                "name": TOOL_NAME,
+                "description": "Submit the final answer for the math problem.",
+                "parameters": {
+                    "type": "object",
+                    "properties": {
+                        "answer": {
+                            "type": "string",
+                            "description": "Final answer in simplest form",
+                        }
+                        ,
+                        "explanation": {
+                            "type": "string",
+                            "description": "Optional explanation of reasoning",
+                        },
+                    },
+                    "required": ["answer"],
+                    "additionalProperties": False,
+                },
+            },
+        }
+    ]
+
+
+def _build_messages(problem: str) -> List[Dict[str, Any]]:
+    return [
+        {
+            "role": "system",
+            "content": (
+                "You solve math problems. Always respond with a single math_submit tool call "
+                "containing only the final answer."
+            ),
+        },
+        {
+            "role": "user",
+            "content": f"Problem:\n{problem}\nReturn the final answer via math_submit.",
+        },
+    ]
+
+
+def _parse_tool_calls(data: Dict[str, Any]) -> List[Dict[str, Any]]:
+    choices = data.get("choices") or []
+    if not choices:
+        return []
+    message = choices[0].get("message") or {}
+    raw_calls = message.get("tool_calls") or []
+    tool_calls: List[Dict[str, Any]] = []
+    for call in raw_calls:
+        function = call.get("function") or {}
+        name = function.get("name")
+        arguments = function.get("arguments")
+        parsed_args: Dict[str, Any]
+        if isinstance(arguments, str):
+            try:
+                parsed_args = json.loads(arguments)
+            except Exception:
+                parsed_args = {}
+        elif isinstance(arguments, dict):
+            parsed_args = dict(arguments)
+        else:
+            parsed_args = {}
+        tool_calls.append({"tool": name, "args": parsed_args})
+    return tool_calls
+
+
+def _detect_provider(model: str, hint: Optional[str]) -> str:
+    if hint:
+        return hint.lower()
+    lowered = (model or "").lower()
+    if lowered.startswith("groq:"):
+        return "groq"
+    return "generic"
+
+
+def _resolve_inference_url(base_url: str) -> str:
+    normalized = (base_url or "").rstrip("/")
+    if not normalized:
+        raise RuntimeError("inference_url cannot be empty")
+    if normalized.endswith("/v1/chat/completions"):
+        return normalized
+    if normalized.endswith("/chat/completions"):
+        return normalized
+    if normalized.endswith("/v1"):
+        return f"{normalized}/chat/completions"
+    if "/v1/" in normalized:
+        return f"{normalized}/chat/completions"
+    return f"{normalized}/v1/chat/completions"
+
+
+async def _choose_actions(
+    client: TaskAppClient,
+    provider: str,
+    model: str,
+    problem: str,
+    policy_cfg: Dict[str, Any],
+) -> Tuple[List[Dict[str, Any]], Dict[str, Any]]:
+    messages = _build_messages(problem)
+    payload: Dict[str, Any] = {
+        "model": model,
+        "messages": messages,
+        "tools": _math_tool_schema(),
+        "tool_choice": {"type": "function", "function": {"name": TOOL_NAME}},
+        "temperature": policy_cfg.get("temperature", 0.0),
+        "top_p": policy_cfg.get("top_p", 1.0),
+        "max_tokens": policy_cfg.get("max_tokens", 256),
+    }
+
+    if provider == "groq":
+        # Task app proxies Groq requests; reuse existing headers on the client
+        response = await client.client.post(
+            "/proxy/groq/v1/chat/completions", json=payload
+        )
+        response.raise_for_status()
+        body = response.json()
+    else:
+        inference_url = policy_cfg.get("inference_url")
+        if not inference_url:
+            raise RuntimeError("inference_url required for non-groq evaluations")
+        headers = dict(policy_cfg.get("headers") or {})
+        for key, value in (policy_cfg.get("extra_headers") or {}).items():
+            headers.setdefault(key, value)
+        final_url = _resolve_inference_url(inference_url)
+        try:
+            response = await client.client.post(
+                final_url,
+                json=payload,
+                headers=headers or None,
+            )
+        except httpx.ReadTimeout as exc:
+            raise RuntimeError(
+                "Inference request timed out. Check the inference service." ) from exc
+        try:
+            body = response.json()
+        except Exception:
+            body = {"raw": response.text[:800]}
+        if response.status_code >= 500:
+            raise RuntimeError(
+                f"Inference server error {response.status_code}: {body}")
+        if response.status_code >= 400:
+            raise RuntimeError(
+                f"Inference request invalid ({response.status_code}): {body}")
+    tool_calls = _parse_tool_calls(body)
+    return tool_calls, body
+
+
+def _tool_to_answer(tool_calls: List[Dict[str, Any]]) -> str:
+    if not tool_calls:
+        return ""
+    args = tool_calls[0].get("args") or {}
+    answer = str(args.get("answer") or "")
+    return answer.strip()
+
+
+async def eval_episode(
+    client: TaskAppClient,
+    *,
+    split: str,
+    seed: Optional[int],
+    model: str,
+    provider: str,
+    policy_cfg: Dict[str, Any],
+) -> Dict[str, Any]:
+    created = await client.initialize(split, seed)
+    env_id = created["env_id"]
+    observation = created.get("observation") or {}
+    problem = observation.get("problem") or ""
+
+    tool_calls, raw_response = await _choose_actions(client, provider, model, problem, policy_cfg)
+    answer = _tool_to_answer(tool_calls)
+    result = await client.step(env_id, tool_calls)
+    await client.terminate(env_id)
+
+    info = result.get("info") or {}
+    reward = result.get("reward") or 0.0
+    status = info.get("status") or ("correct" if reward > 0 else "incorrect")
+    return {
+        "seed": seed,
+        "split": split,
+        "problem": problem,
+        "answer": answer,
+        "expected": info.get("expected_answer"),
+        "reward": reward,
+        "status": status,
+        "correct": bool(info.get("correct")),
+        "raw_response": raw_response,
+        "tool_calls": tool_calls,
+    }
+
+
+async def eval_via_rollout(
+    client: TaskAppClient,
+    *,
+    run_id: str,
+    split: str,
+    seed: Optional[int],
+    model: str,
+    policy_cfg: Dict[str, Any],
+) -> Dict[str, Any]:
+    payload = {
+        "run_id": run_id,
+        "env": {
+            "env_name": "math",
+            "config": {"split": split},
+            "seed": seed,
+        },
+        "policy": {
+            "policy_name": "math-single-step",
+            "config": policy_cfg,
+        },
+        "ops": ["agent", "env"],
+        "on_done": "terminate",
+    }
+    resp = await client.rollout(payload)
+    trajs = resp.get("trajectories") or []
+    if not trajs:
+        return {"reward": 0.0, "correct": False, "status": "missing"}
+    traj = trajs[0]
+    steps = traj.get("steps") or []
+    step = steps[0] if steps else {}
+    info = step.get("info") or {}
+    return {
+        "seed": seed,
+        "split": split,
+        "problem": observation.get("problem"),
+        "answer": _tool_to_answer(step.get("tool_calls") or []),
+        "expected": info.get("expected_answer"),
+        "reward": step.get("reward") or 0.0,
+        "status": info.get("status"),
+        "correct": bool(info.get("correct")),
+        "raw_response": resp,
+        "tool_calls": step.get("tool_calls") or [],
+    }
+
+
+def _load_config(path: Optional[str]) -> Dict[str, Any]:
+    if not path:
+        return {}
+    with open(path, "rb") as fh:
+        return tomllib.load(fh)
+
+
+def _default_policy_cfg(cfg: Dict[str, Any]) -> Dict[str, Any]:
+    policy = dict(cfg.get("policy") or {})
+    if "inference_url" not in policy:
+        env_url = os.getenv("INFERENCE_URL")
+        if env_url:
+            policy["inference_url"] = env_url
+    for key in ("max_tokens", "temperature", "top_p", "headers", "extra_headers"):
+        if key not in policy and key in cfg:
+            policy[key] = cfg[key]
+    extra_headers = dict(policy.get("extra_headers") or {})
+    headers = dict(policy.get("headers") or {})
+    if "Authorization" not in headers and "Authorization" not in extra_headers:
+        synth_key = os.getenv("SYNTH_API_KEY")
+        if synth_key:
+            extra_headers["Authorization"] = f"Bearer {synth_key}"
+    if extra_headers:
+        policy["extra_headers"] = extra_headers
+    return policy
+
+
+async def main() -> None:
+    parser = argparse.ArgumentParser(description="Evaluate math task app policies")
+    parser.add_argument("--toml", help="Path to TOML config", default=None)
+    parser.add_argument("--use-rollout", action="store_true", help="Use server-side rollout")
+    args = parser.parse_args()
+
+    cfg = _load_config(args.toml)
+    task_app_url = (cfg.get("task_app_url") or os.getenv("TASK_APP_URL") or "").rstrip("/")
+    if not task_app_url:
+        raise RuntimeError("task_app_url missing; set in TOML or export TASK_APP_URL")
+    model = cfg.get("model") or os.getenv("EVAL_MODEL") or "groq:qwen-2.5-7b"
+    split = cfg.get("split") or os.getenv("EVAL_SPLIT") or DEFAULT_SPLIT
+    episodes = int(cfg.get("num_episodes") or os.getenv("NUM_EPISODES") or 50)
+    seed_start = int(cfg.get("seed_start") or 0)
+
+    policy_cfg = _default_policy_cfg(cfg)
+    provider_hint = cfg.get("provider") or cfg.get("policy", {}).get("provider") or policy_cfg.get("provider")
+    provider = _detect_provider(model, provider_hint)
+    policy_cfg.pop("provider", None)
+
+    api_key = os.getenv("ENVIRONMENT_API_KEY")
+
+    successes = 0
+    failures: Dict[str, int] = {}
+    results: List[Dict[str, Any]] = []
+
+    async with TaskAppClient(task_app_url, api_key=api_key) as client:
+        for episode in range(episodes):
+            seed = seed_start + episode
+            if args.use_rollout:
+                data = await eval_via_rollout(
+                    client,
+                    run_id=f"eval-{seed}",
+                    split=split,
+                    seed=seed,
+                    model=model,
+                    policy_cfg={"model": model, **policy_cfg},
+                )
+            else:
+                data = await eval_episode(
+                    client,
+                    split=split,
+                    seed=seed,
+                    model=model,
+                    provider=provider,
+                    policy_cfg={"model": model, **policy_cfg},
+                )
+            results.append(data)
+            if data.get("correct"):
+                successes += 1
+            status = data.get("status") or "unknown"
+            failures[status] = failures.get(status, 0) + (0 if data.get("correct") else 1)
+            answer = data.get("answer")
+            expected = data.get("expected")
+            problem = data.get("problem")
+            tool_calls = data.get("tool_calls") or []
+            print(
+                f"Episode {episode+1}/{episodes} seed={seed} status={status} reward={data.get('reward')}\n"
+                f"  problem: {problem!r}\n"
+                f"  tool   : {tool_calls!r}\n"
+                f"  answer : {answer!r}\n  expected: {expected!r}",
+                flush=True,
+            )
+
+    accuracy = successes / max(episodes, 1)
+    print("=== Evaluation Summary ===")
+    print(f"Task App: {task_app_url}")
+    print(f"Model: {model}")
+    print(f"Split: {split}")
+    print(f"Episodes: {episodes}")
+    print(f"Accuracy: {accuracy:.3f}")
+    print("Failure breakdown:")
+    for status, count in sorted(failures.items(), key=lambda kv: (-kv[1], kv[0])):
+        print(f"  {status}: {count}")
+
+
+if __name__ == "__main__":
+    asyncio.run(main())

examples/rl/run_rl_and_save.py

@@ -0,0 +1,94 @@
+#!/usr/bin/env python3
+"""Submit math RL training jobs via Synth backend."""
+
+from __future__ import annotations
+
+import argparse
+import json
+import os
+import sys
+from pathlib import Path
+from typing import Any, Dict
+
+import requests
+import tomllib
+
+
+def _load_toml(path: Path) -> Dict[str, Any]:
+    if not path.exists():
+        print(f"config not found: {path}", file=sys.stderr)
+        sys.exit(2)
+    with path.open("rb") as fh:
+        return tomllib.load(fh)
+
+
+def main() -> None:
+    parser = argparse.ArgumentParser(description="Create math RL job via backend RL endpoint")
+    parser.add_argument("--backend", default=os.getenv("BACKEND_BASE_URL", "http://localhost:8000/api"))
+    parser.add_argument("--config", required=True, help="Path to RL TOML config")
+    parser.add_argument("--task-url", default=os.getenv("TASK_APP_URL", ""), help="Override task service URL")
+    parser.add_argument("--idempotency", default=os.getenv("RL_IDEMPOTENCY_KEY", ""), help="Optional Idempotency-Key header")
+    args = parser.parse_args()
+
+    cfg_path = Path(args.config).expanduser()
+    cfg = _load_toml(cfg_path)
+
+    services = cfg.get("services") if isinstance(cfg.get("services"), dict) else {}
+
+    task_url = (args.task_url or "").strip() or (os.getenv("TASK_APP_URL") or "").strip() or (services.get("task_url") or "").strip()
+    if not task_url:
+        print("Missing task service URL. Provide --task-url or set TASK_APP_URL or services.task_url in TOML", file=sys.stderr)
+        sys.exit(2)
+
+    model_cfg = cfg.get("model") if isinstance(cfg.get("model"), dict) else {}
+    has_source = bool((model_cfg.get("source") or "").strip())
+    has_base = bool((model_cfg.get("base") or "").strip())
+    if has_source == has_base:
+        print("Model section must specify exactly one of [model].source or [model].base", file=sys.stderr)
+        sys.exit(2)
+
+    payload: Dict[str, Any] = {
+        "job_type": "rl",
+        "compute": cfg.get("compute", {}),
+        "data": {
+            "endpoint_base_url": task_url,
+            "config": cfg,
+        },
+        "tags": cfg.get("tags", {}),
+    }
+
+    backend = str(args.backend).rstrip("/")
+    url = f"{backend}/rl/jobs"
+    api_key = (os.getenv("SYNTH_API_KEY") or os.getenv("synth_key") or "").strip()
+    if not api_key:
+        print("Missing SYNTH_API_KEY in env", file=sys.stderr)
+        sys.exit(2)
+
+    headers = {
+        "content-type": "application/json",
+        "authorization": f"Bearer {api_key}",
+    }
+    idem = (args.idempotency or "").strip()
+    if idem:
+        headers["Idempotency-Key"] = idem
+
+    print(f"[INFO] POST {url}")
+    try:
+        preview = {"job_type": payload["job_type"], "data": {"config_keys": list(cfg.keys())}}
+        print(f"[INFO] Payload preview: {json.dumps(preview)}")
+    except Exception:
+        pass
+
+    resp = requests.post(url, headers=headers, json=payload, timeout=120)
+    ok = resp.status_code in (200, 201)
+    try:
+        snippet = resp.json()
+    except Exception:
+        snippet = resp.text[:300]
+    print(f"[INFO] Response: {resp.status_code} {snippet}")
+    sys.exit(0 if ok else 1)
+
+
+if __name__ == "__main__":
+    main()
+

examples/rl/task_app/README.md

@@ -0,0 +1,22 @@
+# Math Single-Step Task App
+
+This directory hosts the legacy entrypoint for the math single-step task app. Prefer starting the app via:
+
+```bash
+uvx synth-ai serve math-single-step --env-file examples/rl/.env --port 8101
+```
+
+If you need to run it directly (e.g., for Modal `modal deploy` compatibility), use:
+
+```bash
+python examples/rl/task_app/math_task_app.py --env-file examples/rl/.env --port 8101
+```
+
+Environment variables:
+
+- `MATH_DATASET_NAME` – defaults to `EleutherAI/math`
+- `MATH_DATASET_CONFIG` – defaults to `algebra__linear_1d`
+- `MATH_DATASET_DEFAULT_SPLIT`, `MATH_DATASET_VALIDATION_SPLIT`, `MATH_DATASET_TEST_SPLIT`
+
+The task app enforces a single `math_submit` tool call per episode, enabling RL to reward correct final answers and penalise missing or malformed submissions.
+

{synth_ai/task/apps → examples/rl/task_app}/math_single_step.py

@@ -16,7 +16,7 @@ from datasets import load_dataset
 from fastapi import APIRouter, HTTPException, Request
 from pydantic import BaseModel, Field
 
-from ..contracts import (
+from synth_ai.task.contracts import (
     RolloutMetrics,
     RolloutRequest,
     RolloutResponse,
@@ -24,18 +24,18 @@ from ..contracts import (
     RolloutTrajectory,
     TaskInfo,
 )
-from ..datasets import TaskDatasetRegistry, TaskDatasetSpec
-from ..rubrics import Rubric, load_rubric
-from ..server import ProxyConfig, RubricBundle, TaskAppConfig
-from ..errors import http_exception
-from ..tracing_utils import (
+from synth_ai.task.datasets import TaskDatasetRegistry, TaskDatasetSpec
+from synth_ai.task.rubrics import Rubric, load_rubric
+from synth_ai.task.server import ProxyConfig, RubricBundle, TaskAppConfig
+from synth_ai.task.errors import http_exception
+from synth_ai.task.tracing_utils import (
     build_tracer_factory,
     resolve_sft_output_dir,
     resolve_tracing_db_url,
     tracing_env_enabled,
 )
-from ..vendors import normalize_vendor_keys
-from . import ModalDeploymentConfig, TaskAppEntry, register_task_app
+from synth_ai.task.vendors import normalize_vendor_keys
+from synth_ai.task.apps import ModalDeploymentConfig, TaskAppEntry, register_task_app
 from synth_ai.tracing_v3.session_tracer import SessionTracer
 
 REPO_ROOT = Path(__file__).resolve().parents[3]

examples/rl/task_app/math_task_app.py

@@ -0,0 +1,107 @@
+"""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.server import create_task_app, run_task_app
+from .math_single_step import build_config
+from synth_ai.task.auth import is_api_key_header_authorized, normalize_environment_api_key
+
+
+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(getattr(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,
+    )