synth-ai 0.2.16__py3-none-any.whl → 0.2.19__py3-none-any.whl

synth_ai/cli/claude.py

@@ -0,0 +1,70 @@
+import os
+import subprocess
+
+import click
+from synth_ai.types import MODEL_NAMES, ModelName
+from synth_ai.urls import BACKEND_URL_SYNTH_RESEARCH_ANTHROPIC
+from synth_ai.utils import find_bin_path, install_bin, resolve_env_var, verify_bin, write_agents_md
+
+
+@click.command("claude")
+@click.option(
+    "--model",
+    "model_name",
+    type=str,
+    default=None
+)
+@click.option(
+    "--force",
+    is_flag=True,
+    help="Prompt for API keys even if cached values exist."
+)
+@click.option(
+    "--url",
+    "override_url",
+    type=str,
+    default=None,
+)
+def claude_cmd(
+    model_name: ModelName | None = None,
+    force: bool = False,
+    override_url: str | None = None
+) -> None:
+
+    while True:
+        bin_path = find_bin_path("claude")
+        if bin_path:
+            break
+        if not install_bin(
+            "Claude Code",
+            ["curl -fsSL https://claude.ai/install.sh | bash"]
+        ):
+            print("Failed to find your installed Claude Code")
+            print("Please install from: https://claude.com/claude-code")
+            return
+    print(f"Using Claude at {bin_path}")
+
+    if not verify_bin(bin_path):
+        print("Failed to verify Claude Code is runnable")
+        return
+
+    write_agents_md()
+    env = os.environ.copy()
+
+    if model_name is not None:
+        if model_name not in MODEL_NAMES:
+            raise ValueError(f"model_name={model_name} is invalid. Valid values for model_name: {MODEL_NAMES}")
+        if override_url:
+            url = f"{override_url.rstrip('/')}/{model_name}"
+            print(f"Using override URL with model: {url}")
+        else:
+            url = f"{BACKEND_URL_SYNTH_RESEARCH_ANTHROPIC}/{model_name}"
+        env["ANTHROPIC_BASE_URL"] = url
+        api_key = resolve_env_var("SYNTH_API_KEY", override_process_env=force)
+        env["ANTHROPIC_AUTH_TOKEN"] = api_key
+        env["SYNTH_API_KEY"] = api_key
+
+    try:
+        subprocess.run(["claude"], check=True, env=env)
+    except subprocess.CalledProcessError:
+        print("Failed to launch Claude Code")

synth_ai/cli/codex.py

@@ -0,0 +1,84 @@
+import os
+import subprocess
+
+import click
+from synth_ai.types import MODEL_NAMES, ModelName
+from synth_ai.urls import BACKEND_URL_SYNTH_RESEARCH_OPENAI
+from synth_ai.utils import find_bin_path, install_bin, resolve_env_var, verify_bin, write_agents_md
+
+
+@click.command("codex")
+@click.option(
+    "--model",
+    "model_name",
+    type=str,
+    default=None
+)
+@click.option(
+    "--force",
+    is_flag=True,
+    help="Prompt for API keys even if cached values exist."
+)
+@click.option(
+    "--url",
+    "override_url",
+    type=str,
+    default=None,
+)
+def codex_cmd(
+    model_name: ModelName | None = None,
+    force: bool = False,
+    override_url: str | None = None
+)-> None:
+
+    while True:
+        bin_path = find_bin_path("codex")
+        if bin_path:
+            break
+        if not install_bin(
+            "Codex",
+            [
+                "brew install codex",
+                "npm install -g @openai/codex"
+            ]
+        ):
+            print("Failed to find your installed Codex")
+            print("Please install from: https://developers.openai.com/codex/cli/")
+            return
+    print(f"Using Codex at {bin_path}")
+
+    if not verify_bin(bin_path):
+        print("Failed to verify Codex is runnable")
+        return
+    
+    write_agents_md()
+    env = os.environ.copy()
+    override_args = []
+    
+    if model_name is not None:
+        if model_name not in MODEL_NAMES:
+            raise ValueError(f"model_name={model_name} is invalid. Valid values for model_name: {MODEL_NAMES}")
+        if override_url:
+            url = override_url
+            print("Using override URL:", url)
+        else:
+            url = BACKEND_URL_SYNTH_RESEARCH_OPENAI
+        provider_config = f'{{name="Synth",base_url="{url}",env_key="OPENAI_API_KEY"}}'
+        config_overrides = [
+            f"model_providers.synth={provider_config}",
+            'model_provider="synth"',
+            f'default_model="{model_name}"'
+        ]
+        override_args = [arg for override in config_overrides for arg in ("-c", override)]
+        env["OPENAI_API_KEY"] = resolve_env_var("SYNTH_API_KEY", override_process_env=force)
+        env["SYNTH_API_KEY"] = env["OPENAI_API_KEY"]
+    
+    try:
+        cmd = ["codex"]
+        if model_name is not None:
+            cmd.extend(["-m", model_name])
+        cmd.extend(override_args)
+        print(" ".join(cmd))
+        subprocess.run(cmd, check=True, env=env)
+    except subprocess.CalledProcessError:
+        print("Failed to run Codex")

synth_ai/cli/commands/__init__.py

@@ -0,0 +1,18 @@
+"""
+Structured CLI command implementations.
+
+Each subpackage under this namespace provides the core command entrypoints,
+validation helpers, and error types for a top-level CLI command (e.g. train,
+eval, deploy).
+"""
+
+from __future__ import annotations
+
+__all__ = [
+    "train",
+    "eval",
+    "filter",
+    "deploy",
+    "status",
+    "smoke",
+]

synth_ai/cli/commands/baseline/__init__.py

@@ -0,0 +1,12 @@
+"""CLI command for baseline evaluation."""
+
+from __future__ import annotations
+
+from .core import command
+from .list import list_command
+
+__all__ = ["command"]
+
+# Register list subcommand
+command.add_command(list_command, name="list")
+