fix: tests

Author: daniel-lxs

packages/types/src/providers/openrouter.ts

@@ -40,8 +40,6 @@ export const OPEN_ROUTER_PROMPT_CACHING_MODELS = new Set([
 	"anthropic/claude-sonnet-4",
 	"anthropic/claude-opus-4",
 	"google/gemini-2.5-pro-preview",
-	"google/gemini-2.5-pro-preview:thinking",
-	"google/gemini-2.5-pro-preview-05-06",
 	"google/gemini-2.5-flash-preview",
 	"google/gemini-2.5-flash-preview:thinking",
 	"google/gemini-2.5-flash-preview-05-20",
@@ -68,13 +66,11 @@ export const OPEN_ROUTER_REASONING_BUDGET_MODELS = new Set([
 	"anthropic/claude-opus-4",
 	"anthropic/claude-sonnet-4",
 	"google/gemini-2.5-pro-preview",
-	"google/gemini-2.5-pro-preview:thinking",
 	"google/gemini-2.5-flash-preview-05-20",
 	"google/gemini-2.5-flash-preview-05-20:thinking",
 ])
 
 export const OPEN_ROUTER_REQUIRED_REASONING_BUDGET_MODELS = new Set([
 	"anthropic/claude-3.7-sonnet:thinking",
-	"google/gemini-2.5-pro-preview:thinking",
 	"google/gemini-2.5-flash-preview-05-20:thinking",
 ])

src/api/providers/fetchers/__tests__/openrouter.spec.ts

@@ -185,10 +185,11 @@ describe("OpenRouter API", () => {
 
 			expect(endpoints).toEqual({
 				Google: {
-					maxTokens: 0,
+					maxTokens: 65535,
 					contextWindow: 1048576,
 					supportsImages: true,
 					supportsPromptCache: true,
+					supportsReasoningBudget: true,
 					inputPrice: 1.25,
 					outputPrice: 10,
 					cacheWritesPrice: 1.625,
@@ -198,10 +199,11 @@ describe("OpenRouter API", () => {
 					supportedParameters: undefined,
 				},
 				"Google AI Studio": {
-					maxTokens: 0,
+					maxTokens: 65536,
 					contextWindow: 1048576,
 					supportsImages: true,
 					supportsPromptCache: true,
+					supportsReasoningBudget: true,
 					inputPrice: 1.25,
 					outputPrice: 10,
 					cacheWritesPrice: 1.625,