mirror of
https://github.com/harivansh-afk/sandbox-agent.git
synced 2026-04-17 16:02:42 +00:00
29 lines
947 B
Python
29 lines
947 B
Python
"""Agent detection and credential helpers for sandbox-agent examples."""
|
|
|
|
import os
|
|
import sys
|
|
|
|
|
|
def detect_agent() -> str:
|
|
"""Pick an agent based on env vars. Exits if no credentials are found."""
|
|
if os.environ.get("SANDBOX_AGENT"):
|
|
return os.environ["SANDBOX_AGENT"]
|
|
has_anthropic = bool(
|
|
os.environ.get("ANTHROPIC_API_KEY")
|
|
or os.environ.get("CLAUDE_API_KEY")
|
|
)
|
|
has_openai = bool(os.environ.get("OPENAI_API_KEY"))
|
|
if has_anthropic or has_openai:
|
|
return "opencode"
|
|
print("No API keys found. Set ANTHROPIC_API_KEY or OPENAI_API_KEY.")
|
|
sys.exit(1)
|
|
|
|
|
|
def build_container_env() -> dict[str, str]:
|
|
"""Collect credential env vars to forward into the Docker container."""
|
|
env: dict[str, str] = {}
|
|
for key in ("ANTHROPIC_API_KEY", "CLAUDE_API_KEY", "OPENAI_API_KEY", "CODEX_API_KEY"):
|
|
val = os.environ.get(key)
|
|
if val:
|
|
env[key] = val
|
|
return env
|