Skip to content

Commit cbf0fe3

Browse files
committed
fix(openrouter): correct reserved tokens for google/gemini-2.5-pro-preview to 66k and add test coverage
1 parent 8729027 commit cbf0fe3

File tree

2 files changed

+13
-1
lines changed

2 files changed

+13
-1
lines changed

src/api/providers/__tests__/openrouter.test.ts

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -114,6 +114,14 @@ describe("OpenRouterHandler", () => {
114114
expect(result.reasoningBudget).toBeUndefined()
115115
expect(result.temperature).toBe(0)
116116
})
117+
it("sets maxTokens to 66000 for google/gemini-2.5-pro-preview", async () => {
118+
const handler = new OpenRouterHandler({
119+
openRouterApiKey: "test-key",
120+
openRouterModelId: "google/gemini-2.5-pro-preview",
121+
})
122+
const result = await handler.fetchModel()
123+
expect(result.maxTokens).toBe(66000)
124+
})
117125
})
118126

119127
describe("createMessage", () => {

src/api/providers/fetchers/openrouter.ts

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -106,7 +106,11 @@ export async function getOpenRouterModels(options?: ApiHandlerOptions): Promise<
106106
id,
107107
model,
108108
modality: architecture?.modality,
109-
maxTokens: id.startsWith("anthropic/") ? top_provider?.max_completion_tokens : 0,
109+
maxTokens: id.startsWith("anthropic/")
110+
? top_provider?.max_completion_tokens
111+
: id === "google/gemini-2.5-pro-preview"
112+
? 66000
113+
: 0,
110114
supportedParameters: supported_parameters,
111115
})
112116
}

0 commit comments

Comments
 (0)