From 304caf64f6b88f36c586cbe91d820804be84f834 Mon Sep 17 00:00:00 2001 From: macmini server Date: Mon, 30 Mar 2026 07:44:01 -0400 Subject: [PATCH] fix: add Fireworks AI provider mapping for correct context length detection - Add 'fireworks.ai' URL to _URL_TO_PROVIDER mapping - Add 'fireworks-ai' to PROVIDER_TO_MODELS_DEV mapping - Add 'custom' to PROVIDER_TO_MODELS_DEV for OpenAI-compatible providers This ensures Fireworks AI models (like kimi-k2p5-turbo) correctly resolve their 256K context window instead of falling back to 128K default. --- agent/model_metadata.py | 1 + agent/models_dev.py | 2 ++ 2 files changed, 3 insertions(+) diff --git a/agent/model_metadata.py b/agent/model_metadata.py index 0c121e6f63078..474fbf8bcf75a 100644 --- a/agent/model_metadata.py +++ b/agent/model_metadata.py @@ -176,6 +176,7 @@ def _is_custom_endpoint(base_url: str) -> bool: "api.deepseek.com": "deepseek", "api.githubcopilot.com": "copilot", "models.github.ai": "copilot", + "fireworks.ai": "fireworks-ai", } diff --git a/agent/models_dev.py b/agent/models_dev.py index 283e8018fdf56..2e0043e0eec0a 100644 --- a/agent/models_dev.py +++ b/agent/models_dev.py @@ -43,6 +43,8 @@ "opencode-zen": "opencode", "opencode-go": "opencode-go", "kilocode": "kilo", + "fireworks-ai": "fireworks-ai", + "custom": "custom", }