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

synth_ai/api/train/task_app.py

@@ -1,195 +0,0 @@
-from __future__ import annotations
-
-from collections.abc import Iterable
-from dataclasses import dataclass
-
-import click
-import requests
-
-from .utils import CLIResult, http_get, run_cli
-
-
-@dataclass(slots=True)
-class TaskAppHealth:
-    ok: bool
-    health_status: int | None
-    task_info_status: int | None
-    detail: str | None = None
-
-
-def _health_response_ok(resp: requests.Response | None) -> tuple[bool, str]:
-    if resp is None:
-        return False, ""
-    status = resp.status_code
-    if status == 200:
-        return True, ""
-    if status in {401, 403}:
-        try:
-            payload = resp.json()
-        except ValueError:
-            payload = {}
-        prefix = payload.get("expected_api_key_prefix")
-        detail = str(payload.get("detail", ""))
-        if prefix or "expected prefix" in detail.lower():
-            note = "auth-optional"
-            if prefix:
-                note += f" (expected-prefix={prefix})"
-            return True, note
-    return False, ""
-
-
-def check_task_app_health(base_url: str, api_key: str, *, timeout: float = 10.0) -> TaskAppHealth:
-    # Send ALL known environment keys so the server can authorize any valid one
-    import os
-
-    headers = {"X-API-Key": api_key}
-    aliases = (os.getenv("ENVIRONMENT_API_KEY_ALIASES") or "").strip()
-    keys: list[str] = [api_key]
-    if aliases:
-        keys.extend([p.strip() for p in aliases.split(",") if p.strip()])
-    if keys:
-        headers["X-API-Keys"] = ",".join(keys)
-        headers.setdefault("Authorization", f"Bearer {api_key}")
-    base = base_url.rstrip("/")
-    detail_parts: list[str] = []
-
-    health_resp: requests.Response | None = None
-    health_ok = False
-    try:
-        health_resp = http_get(f"{base}/health", headers=headers, timeout=timeout)
-        health_ok, note = _health_response_ok(health_resp)
-        suffix = f" ({note})" if note else ""
-        # On non-200, include brief JSON detail if present
-        if not health_ok and health_resp is not None:
-            try:
-                hjs = health_resp.json()
-                # pull a few helpful fields without dumping everything
-                expected = hjs.get("expected_api_key_prefix")
-                authorized = hjs.get("authorized")
-                detail = hjs.get("detail")
-                extras = []
-                if authorized is not None:
-                    extras.append(f"authorized={authorized}")
-                if expected:
-                    extras.append(f"expected_prefix={expected}")
-                if detail:
-                    extras.append(f"detail={str(detail)[:80]}")
-                if extras:
-                    suffix += " [" + ", ".join(extras) + "]"
-            except Exception:
-                pass
-        detail_parts.append(f"/health={health_resp.status_code}{suffix}")
-    except requests.RequestException as exc:
-        detail_parts.append(f"/health_error={exc}")
-
-    task_resp: requests.Response | None = None
-    task_ok = False
-    try:
-        task_resp = http_get(f"{base}/task_info", headers=headers, timeout=timeout)
-        task_ok = bool(task_resp.status_code == 200)
-        if not task_ok and task_resp is not None:
-            try:
-                tjs = task_resp.json()
-                msg = tjs.get("detail") or tjs.get("status")
-                detail_parts.append(f"/task_info={task_resp.status_code} ({str(msg)[:80]})")
-            except Exception:
-                detail_parts.append(f"/task_info={task_resp.status_code}")
-        else:
-            detail_parts.append(f"/task_info={task_resp.status_code}")
-    except requests.RequestException as exc:
-        detail_parts.append(f"/task_info_error={exc}")
-
-    ok = bool(health_ok and task_ok)
-    detail = ", ".join(detail_parts)
-    return TaskAppHealth(
-        ok=ok,
-        health_status=None if health_resp is None else health_resp.status_code,
-        task_info_status=None if task_resp is None else task_resp.status_code,
-        detail=detail,
-    )
-
-
-@dataclass(slots=True)
-class ModalSecret:
-    name: str
-    value: str
-
-
-@dataclass(slots=True)
-class ModalApp:
-    app_id: str
-    label: str
-    url: str
-
-
-def _run_modal(args: Iterable[str]) -> CLIResult:
-    return run_cli(["modal", *args], timeout=30.0)
-
-
-def list_modal_secrets(pattern: str | None = None) -> list[str]:
-    result = _run_modal(["secret", "list"])
-    if result.code != 0:
-        raise click.ClickException(f"modal secret list failed: {result.stderr or result.stdout}")
-    names: list[str] = []
-    for line in result.stdout.splitlines():
-        line = line.strip()
-        if not line or line.startswith("NAME"):
-            continue
-        parts = line.split()
-        name = parts[0]
-        if pattern and pattern.lower() not in name.lower():
-            continue
-        names.append(name)
-    return names
-
-
-def get_modal_secret_value(name: str) -> str:
-    result = _run_modal(["secret", "get", name])
-    if result.code != 0:
-        raise click.ClickException(
-            f"modal secret get {name} failed: {result.stderr or result.stdout}"
-        )
-    value = result.stdout.strip()
-    if not value:
-        raise click.ClickException(f"Secret {name} is empty")
-    return value
-
-
-def list_modal_apps(pattern: str | None = None) -> list[ModalApp]:
-    result = _run_modal(["app", "list"])
-    if result.code != 0:
-        raise click.ClickException(f"modal app list failed: {result.stderr or result.stdout}")
-    apps: list[ModalApp] = []
-    for line in result.stdout.splitlines():
-        line = line.strip()
-        if not line or line.startswith("APP"):
-            continue
-        parts = line.split()
-        if len(parts) < 3:
-            continue
-        app_id, label, url = parts[0], parts[1], parts[-1]
-        if pattern and pattern.lower() not in (label.lower() + url.lower() + app_id.lower()):
-            continue
-        apps.append(ModalApp(app_id=app_id, label=label, url=url))
-    return apps
-
-
-def format_modal_apps(apps: list[ModalApp]) -> str:
-    rows = [f"{idx}) {app.label} {app.url}" for idx, app in enumerate(apps, start=1)]
-    return "\n".join(rows)
-
-
-def format_modal_secrets(names: list[str]) -> str:
-    return "\n".join(f"{idx}) {name}" for idx, name in enumerate(names, start=1))
-
-
-__all__ = [
-    "ModalApp",
-    "ModalSecret",
-    "check_task_app_health",
-    "format_modal_apps",
-    "format_modal_secrets",
-    "get_modal_secret_value",
-    "list_modal_apps",
-    "list_modal_secrets",
-]

synth_ai/api/train/utils.py

@@ -1,225 +0,0 @@
-from __future__ import annotations
-
-import importlib
-import json
-import os
-import re
-import subprocess
-import tempfile
-import time
-import tomllib
-from collections.abc import Callable, Iterable, Mapping
-from dataclasses import dataclass
-from pathlib import Path
-from typing import Any, cast
-
-import requests
-
-try:
-    sft_module = cast(Any, importlib.import_module("synth_ai.learning.sft"))
-    collect_sft_jsonl_errors = cast(
-        Callable[..., list[dict[str, Any]]], sft_module.collect_sft_jsonl_errors
-    )
-except Exception as exc:  # pragma: no cover - critical dependency
-    raise RuntimeError("Unable to load SFT JSONL helpers") from exc
-
-REPO_ROOT = Path(__file__).resolve().parents[3]
-
-
-class TrainError(RuntimeError):
-    """Raised for interactive CLI failures."""
-
-
-def load_toml(path: Path) -> dict[str, Any]:
-    try:
-        with path.open("rb") as fh:
-            return tomllib.load(fh)
-    except FileNotFoundError as exc:  # pragma: no cover - guarded by CLI
-        raise TrainError(f"Config not found: {path}") from exc
-    except tomllib.TOMLDecodeError as exc:  # pragma: no cover - malformed input
-        raise TrainError(f"Failed to parse TOML: {path}\n{exc}") from exc
-
-
-def mask_value(value: str | None) -> str:
-    if not value:
-        return "<unset>"
-    value = str(value)
-    if len(value) <= 6:
-        return "****"
-    return f"{value[:4]}…{value[-2:]}"
-
-
-_ENV_LINE = re.compile(r"^\s*(?:export\s+)?(?P<key>[A-Za-z0-9_]+)\s*=\s*(?P<value>.*)$")
-
-
-def read_env_file(path: Path) -> dict[str, str]:
-    if not path.exists():
-        return {}
-    data: dict[str, str] = {}
-    for line in path.read_text(encoding="utf-8", errors="ignore").splitlines():
-        m = _ENV_LINE.match(line)
-        if not m:
-            continue
-        raw = m.group("value").strip()
-        if raw and raw[0] == raw[-1] and raw[0] in {'"', "'"} and len(raw) >= 2:
-            raw = raw[1:-1]
-        data[m.group("key")] = raw
-    return data
-
-
-def write_env_value(path: Path, key: str, value: str) -> None:
-    existing = []
-    if path.exists():
-        existing = path.read_text(encoding="utf-8", errors="ignore").splitlines()
-    updated = False
-    new_lines: list[str] = []
-    for line in existing:
-        m = _ENV_LINE.match(line)
-        if m and m.group("key") == key:
-            new_lines.append(f"{key}={value}")
-            updated = True
-        else:
-            new_lines.append(line)
-    if not updated:
-        new_lines.append(f"{key}={value}")
-    path.write_text("\n".join(new_lines) + "\n", encoding="utf-8")
-
-
-@dataclass(slots=True)
-class CLIResult:
-    code: int
-    stdout: str
-    stderr: str
-
-
-def run_cli(
-    args: Iterable[str],
-    *,
-    cwd: Path | None = None,
-    env: Mapping[str, str] | None = None,
-    timeout: float | None = None,
-) -> CLIResult:
-    proc = subprocess.run(
-        list(args),
-        cwd=cwd,
-        env=dict(os.environ, **(env or {})),
-        capture_output=True,
-        text=True,
-        timeout=timeout,
-    )
-    return CLIResult(code=proc.returncode, stdout=proc.stdout.strip(), stderr=proc.stderr.strip())
-
-
-def http_post(
-    url: str,
-    *,
-    headers: Mapping[str, str] | None = None,
-    json_body: Any | None = None,
-    timeout: float = 60.0,
-) -> requests.Response:
-    resp = requests.post(url, headers=dict(headers or {}), json=json_body, timeout=timeout)
-    return resp
-
-
-def http_get(
-    url: str, *, headers: Mapping[str, str] | None = None, timeout: float = 30.0
-) -> requests.Response:
-    resp = requests.get(url, headers=dict(headers or {}), timeout=timeout)
-    return resp
-
-
-def post_multipart(
-    url: str, *, api_key: str, file_field: str, file_path: Path, purpose: str = "fine-tune"
-) -> requests.Response:
-    headers = {"Authorization": f"Bearer {api_key}"}
-    files = {file_field: (file_path.name, file_path.read_bytes(), "application/jsonl")}
-    data = {"purpose": purpose}
-    return requests.post(url, headers=headers, files=files, data=data, timeout=300)
-
-
-def fmt_duration(seconds: float) -> str:
-    if seconds < 60:
-        return f"{seconds:.1f}s"
-    minutes, secs = divmod(seconds, 60)
-    if minutes < 60:
-        return f"{int(minutes)}m{int(secs):02d}s"
-    hours, mins = divmod(minutes, 60)
-    return f"{int(hours)}h{int(mins):02d}m"
-
-
-def validate_sft_jsonl(path: Path, *, max_errors: int = 20) -> None:
-    if not path.exists():
-        raise TrainError(f"Dataset not found: {path}")
-
-    issues = collect_sft_jsonl_errors(path, min_messages=1, max_errors=max_errors)
-    if not issues:
-        return
-
-    truncated = max_errors is not None and len(issues) >= max_errors
-    suffix = "" if not truncated else f" (showing first {max_errors} issues)"
-    details = "\n - ".join(cast("list[str]", issues))
-    raise TrainError(f"{path}: Dataset validation failed{suffix}:\n - {details}")
-
-
-def limit_jsonl_examples(src: Path, limit: int) -> Path:
-    if limit <= 0:
-        raise TrainError("Example limit must be positive")
-    if not src.exists():
-        raise TrainError(f"Dataset not found: {src}")
-
-    tmp_dir = Path(tempfile.mkdtemp(prefix="sft_subset_"))
-    dest = tmp_dir / f"{src.stem}.head{limit}{src.suffix}"
-
-    written = 0
-    with src.open("r", encoding="utf-8") as fin, dest.open("w", encoding="utf-8") as fout:
-        for line in fin:
-            if not line.strip():
-                continue
-            fout.write(line)
-            written += 1
-            if written >= limit:
-                break
-
-    if written == 0:
-        raise TrainError("Subset dataset is empty; check limit value")
-
-    return dest
-
-
-def ensure_api_base(base: str) -> str:
-    base = base.rstrip("/")
-    if not base.endswith("/api"):
-        base = f"{base}/api"
-    return base
-
-
-def preview_json(data: Any, limit: int = 600) -> str:
-    try:
-        return json.dumps(data, indent=2)[:limit]
-    except Exception:
-        return str(data)[:limit]
-
-
-def sleep(seconds: float) -> None:
-    time.sleep(seconds)
-
-
-__all__ = [
-    "CLIResult",
-    "REPO_ROOT",
-    "TrainError",
-    "ensure_api_base",
-    "fmt_duration",
-    "http_get",
-    "http_post",
-    "load_toml",
-    "mask_value",
-    "post_multipart",
-    "preview_json",
-    "read_env_file",
-    "run_cli",
-    "sleep",
-    "limit_jsonl_examples",
-    "validate_sft_jsonl",
-    "write_env_value",
-]

synth_ai/cli/_modal_wrapper.py

@@ -1,29 +0,0 @@
-from __future__ import annotations
-
-import sys
-
-
-def main() -> int:
-    # Apply Typer compatibility patch before Modal CLI bootstraps Click/Typer internals.
-    try:
-        from synth_ai.cli._typer_patch import patch_typer_make_metavar
-
-        patch_typer_make_metavar()
-    except Exception:
-        pass
-
-    from modal.__main__ import main as modal_main
-
-    # Present ourselves as the upstream `modal` CLI so Typer/Click parsing stays intact.
-    if sys.argv:
-        sys.argv[0] = "modal"
-    else:
-        sys.argv = ["modal"]
-
-    result = modal_main()
-    return result if result is not None else 0
-
-
-if __name__ == "__main__":
-    sys.exit(main())
-

synth_ai/cli/_storage.py

@@ -1,20 +0,0 @@
-"""Dynamic loader for tracing storage components.
-
-This avoids hard dependencies on the tracing_v3 storage package during import
-time, which keeps CLI modules usable in constrained environments while still
-allowing type checkers to resolve the symbols dynamically.
-"""
-
-from __future__ import annotations
-
-import importlib
-from collections.abc import Callable
-from typing import Any, cast
-
-
-def load_storage() -> tuple[Any, Any]:
-    """Return (create_storage, StorageConfig) from tracing_v3.storage."""
-    storage_module = cast(Any, importlib.import_module("synth_ai.tracing_v3.storage"))
-    create_storage = cast(Callable[..., Any], storage_module.create_storage)
-    storage_config = storage_module.StorageConfig
-    return create_storage, storage_config

synth_ai/cli/_typer_patch.py

@@ -1,49 +0,0 @@
-from __future__ import annotations
-
-from click import Parameter
-
-
-def patch_typer_make_metavar() -> None:
-    try:
-        from typer.main import TyperArgument, TyperOption
-    except Exception:
-        return
-
-    def _call_get_metavar(param_type: object, param: Parameter) -> str | None:
-        getter = getattr(param_type, "get_metavar", None)
-        if getter is None:
-            return None
-        try:
-            return getter(param)
-        except TypeError:
-            try:
-                return getter(param, None)
-            except TypeError:
-                return None
-
-    def _patched_argument_make_metavar(self, ctx=None) -> str:
-        if self.metavar is not None:
-            return self.metavar
-        var = (self.name or "").upper()
-        if not self.required:
-            var = f"[{var}]"
-        type_var = _call_get_metavar(self.type, self)
-        if type_var:
-            var += f":{type_var}"
-        if self.nargs != 1:
-            var += "..."
-        return var
-
-    def _patched_option_make_metavar(self, ctx=None) -> str:
-        if self.metavar is not None:
-            return self.metavar
-        metavar = _call_get_metavar(self.type, self)
-        if not metavar:
-            name = getattr(self.type, "name", "") or ""
-            metavar = name.upper()
-        if self.nargs != 1:
-            metavar += "..."
-        return metavar
-
-    TyperArgument.make_metavar = _patched_argument_make_metavar  # type: ignore[assignment]
-    TyperOption.make_metavar = _patched_option_make_metavar  # type: ignore[assignment]

synth_ai/cli/_validate_task_app.py

@@ -1,11 +0,0 @@
-"""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"]
-