feat: add Kimi K2 thinking model to Fireworks AI provider

roomote · roomote · commit 193be14ea2da · 2025-12-10T22:57:09.000Z
diff --git a/packages/types/src/providers/fireworks.ts b/packages/types/src/providers/fireworks.ts
@@ -3,6 +3,7 @@ import type { ModelInfo } from "../model.js"
 export type FireworksModelId =
 	| "accounts/fireworks/models/kimi-k2-instruct"
 	| "accounts/fireworks/models/kimi-k2-instruct-0905"
+	| "accounts/fireworks/models/kimi-k2-thinking"
 	| "accounts/fireworks/models/minimax-m2"
 	| "accounts/fireworks/models/qwen3-235b-a22b-instruct-2507"
 	| "accounts/fireworks/models/qwen3-coder-480b-a35b-instruct"
@@ -41,6 +42,21 @@ export const fireworksModels = {
 		description:
 			"Kimi K2 is a state-of-the-art mixture-of-experts (MoE) language model with 32 billion activated parameters and 1 trillion total parameters. Trained with the Muon optimizer, Kimi K2 achieves exceptional performance across frontier knowledge, reasoning, and coding tasks while being meticulously optimized for agentic capabilities.",
 	},
+	"accounts/fireworks/models/kimi-k2-thinking": {
+		maxTokens: 16000,
+		contextWindow: 256000,
+		supportsImages: false,
+		supportsPromptCache: true,
+		supportsNativeTools: true,
+		supportsTemperature: true,
+		preserveReasoning: true,
+		defaultTemperature: 1.0,
+		inputPrice: 0.6,
+		outputPrice: 2.5,
+		cacheReadsPrice: 0.15,
+		description:
+			"The kimi-k2-thinking model is a general-purpose agentic reasoning model developed by Moonshot AI. Thanks to its strength in deep reasoning and multi-turn tool use, it can solve even the hardest problems.",
+	},
 	"accounts/fireworks/models/minimax-m2": {
 		maxTokens: 4096,
 		contextWindow: 204800,
diff --git a/src/api/providers/__tests__/fireworks.spec.ts b/src/api/providers/__tests__/fireworks.spec.ts
@@ -115,6 +115,31 @@ describe("FireworksHandler", () => {
 		)
 	})
 
+	it("should return Kimi K2 Thinking model with correct configuration", () => {
+		const testModelId: FireworksModelId = "accounts/fireworks/models/kimi-k2-thinking"
+		const handlerWithModel = new FireworksHandler({
+			apiModelId: testModelId,
+			fireworksApiKey: "test-fireworks-api-key",
+		})
+		const model = handlerWithModel.getModel()
+		expect(model.id).toBe(testModelId)
+		expect(model.info).toEqual(
+			expect.objectContaining({
+				maxTokens: 16000,
+				contextWindow: 256000,
+				supportsImages: false,
+				supportsPromptCache: true,
+				supportsNativeTools: true,
+				supportsTemperature: true,
+				preserveReasoning: true,
+				defaultTemperature: 1.0,
+				inputPrice: 0.6,
+				outputPrice: 2.5,
+				cacheReadsPrice: 0.15,
+			}),
+		)
+	})
+
 	it("should return MiniMax M2 model with correct configuration", () => {
 		const testModelId: FireworksModelId = "accounts/fireworks/models/minimax-m2"
 		const handlerWithModel = new FireworksHandler({
