From ecda6ad375970b2b4ea80397851f3a4de8e8e9d7 Mon Sep 17 00:00:00 2001
From: Saumyaranjan Nayak <107764221+Sdnsoumy@users.noreply.github.com>
Date: Fri, 31 Jul 2026 23:27:53 +0530
Subject: [PATCH] feat(fireworks_ai): add Kimi K3 model registry entries

Fixes missing supports_function_calling and supports_tool_choice metadata for fireworks_ai/accounts/fireworks/models/kimi-k3, which caused UnsupportedParamsError when using tool_choice with that model despite Fireworks supporting it upstream. Adds both the full-path entry and short-form alias, plus a regression test.
---
 ...odel_prices_and_context_window_backup.json | 32 +++++++++++++++++++
 model_prices_and_context_window.json          | 32 +++++++++++++++++++
 tests/test_litellm/test_utils.py              | 11 +++++++
 3 files changed, 75 insertions(+)

diff --git a/litellm/model_prices_and_context_window_backup.json b/litellm/model_prices_and_context_window_backup.json
index 02b3cde217a..e2a62777a00 100644
--- a/litellm/model_prices_and_context_window_backup.json
+++ b/litellm/model_prices_and_context_window_backup.json
@@ -16726,6 +16726,22 @@
         "supports_tool_choice": true,
         "supports_vision": true
     },
+    "fireworks_ai/accounts/fireworks/models/kimi-k3": {
+        "cache_read_input_token_cost": 3e-07,
+        "input_cost_per_token": 3e-06,
+        "litellm_provider": "fireworks_ai",
+        "max_input_tokens": 1048576,
+        "max_output_tokens": 131072,
+        "max_tokens": 131072,
+        "mode": "chat",
+        "output_cost_per_token": 1.5e-05,
+        "source": "https://fireworks.ai/models/fireworks/kimi-k3",
+        "supports_function_calling": true,
+        "supports_reasoning": true,
+        "supports_response_schema": true,
+        "supports_tool_choice": true,
+        "supports_vision": true
+    },
     "fireworks_ai/accounts/fireworks/models/llama-v3p1-405b-instruct": {
         "input_cost_per_token": 3e-06,
         "litellm_provider": "fireworks_ai",
@@ -17132,6 +17148,22 @@
         "supports_tool_choice": true,
         "supports_vision": true
     },
+    "fireworks_ai/kimi-k3": {
+        "cache_read_input_token_cost": 3e-07,
+        "input_cost_per_token": 3e-06,
+        "litellm_provider": "fireworks_ai",
+        "max_input_tokens": 1048576,
+        "max_output_tokens": 131072,
+        "max_tokens": 131072,
+        "mode": "chat",
+        "output_cost_per_token": 1.5e-05,
+        "source": "https://fireworks.ai/models/fireworks/kimi-k3",
+        "supports_function_calling": true,
+        "supports_reasoning": true,
+        "supports_response_schema": true,
+        "supports_tool_choice": true,
+        "supports_vision": true
+    },
     "fireworks_ai/minimax-m2p1": {
         "cache_read_input_token_cost": 3e-08,
         "input_cost_per_token": 3e-07,
diff --git a/model_prices_and_context_window.json b/model_prices_and_context_window.json
index bc7330ec99c..1b39800d528 100644
--- a/model_prices_and_context_window.json
+++ b/model_prices_and_context_window.json
@@ -16726,6 +16726,22 @@
         "supports_tool_choice": true,
         "supports_vision": true
     },
+    "fireworks_ai/accounts/fireworks/models/kimi-k3": {
+        "cache_read_input_token_cost": 3e-07,
+        "input_cost_per_token": 3e-06,
+        "litellm_provider": "fireworks_ai",
+        "max_input_tokens": 1048576,
+        "max_output_tokens": 131072,
+        "max_tokens": 131072,
+        "mode": "chat",
+        "output_cost_per_token": 1.5e-05,
+        "source": "https://fireworks.ai/models/fireworks/kimi-k3",
+        "supports_function_calling": true,
+        "supports_reasoning": true,
+        "supports_response_schema": true,
+        "supports_tool_choice": true,
+        "supports_vision": true
+    },
     "fireworks_ai/accounts/fireworks/models/llama-v3p1-405b-instruct": {
         "input_cost_per_token": 3e-06,
         "litellm_provider": "fireworks_ai",
@@ -17146,6 +17162,22 @@
         "supports_response_schema": true,
         "supports_tool_choice": true
     },
+    "fireworks_ai/kimi-k3": {
+        "cache_read_input_token_cost": 3e-07,
+        "input_cost_per_token": 3e-06,
+        "litellm_provider": "fireworks_ai",
+        "max_input_tokens": 1048576,
+        "max_output_tokens": 131072,
+        "max_tokens": 131072,
+        "mode": "chat",
+        "output_cost_per_token": 1.5e-05,
+        "source": "https://fireworks.ai/models/fireworks/kimi-k3",
+        "supports_function_calling": true,
+        "supports_reasoning": true,
+        "supports_response_schema": true,
+        "supports_tool_choice": true,
+        "supports_vision": true
+    },
     "fireworks_ai/minimax-m2p7": {
         "cache_read_input_token_cost": 6e-08,
         "input_cost_per_token": 3e-07,
diff --git a/tests/test_litellm/test_utils.py b/tests/test_litellm/test_utils.py
index b23d3333ea7..3c02eb36905 100644
--- a/tests/test_litellm/test_utils.py
+++ b/tests/test_litellm/test_utils.py
@@ -4514,6 +4514,16 @@ _FIREWORKS_MODELS = [
         False,
         True,
     ),
+    (
+        "accounts/fireworks/models/kimi-k3",
+        3e-06,
+        1.5e-05,
+        3e-07,
+        1048576,
+        131072,
+        True,
+        True,
+    ),
 ]
 
 _FIREWORKS_SHORT_FORMS = [
@@ -4524,6 +4534,7 @@ _FIREWORKS_SHORT_FORMS = [
     "minimax-m2p7",
     "kimi-k2p7-code",
     "kimi-k2p6",
+    "kimi-k3",
     "gpt-oss-120b",
     "gpt-oss-20b",
     "deepseek-v4-pro",