diff --git a/container/entrypoint.py b/container/entrypoint.py index 05345ec..b5394ea 100644 --- a/container/entrypoint.py +++ b/container/entrypoint.py @@ -81,12 +81,8 @@ def fetch_role_secrets(bao_token: str, role: str) -> dict: timeout=10, ) resp.raise_for_status() - data = resp.json()["data"]["data"] - secrets["litellm-key"] = data["key"] - # Use first allowed model; fall back to a sensible default - models = data.get("models", []) - secrets["litellm-model"] = models[0] if models else "copilot/claude-sonnet-4.5" - log.info("Fetched litellm-key; model=%s", secrets["litellm-model"]) + secrets["litellm-key"] = resp.json()["data"]["data"]["key"] + log.info("Fetched litellm-key") except Exception as e: log.warning("Could not fetch litellm-key: %s", e) return secrets @@ -97,7 +93,6 @@ def write_opencode_config(secrets: dict, role: str) -> None: CONFIG_DIR.mkdir(parents=True, exist_ok=True) litellm_key = secrets.get("litellm-key", "") - litellm_model = f"litellm/{secrets.get('litellm-model', 'copilot/claude-sonnet-4.5')}" # Set the LiteLLM API key as env var — opencode reads OPENAI_API_KEY for # openai-compatible providers @@ -105,14 +100,17 @@ def write_opencode_config(secrets: dict, role: str) -> None: config = { "$schema": "https://opencode.ai/config.json", - "model": litellm_model, + "model": "litellm/copilot/claude-sonnet-4.6", "provider": { "litellm": { "npm": "@ai-sdk/openai-compatible", "name": "LiteLLM", "options": { "baseURL": f"{LITELLM_BASE_URL}/v1", - } + }, + "models": { + "copilot/claude-sonnet-4.6": {"name": "copilot/claude-sonnet-4.6"}, + }, } }, "mcp": {