Skip to content

Commit 9d0b824

Browse files
author
dongqing
committed
fix test error for wrong arguments after additional base url added
1 parent a3a5592 commit 9d0b824

File tree

1 file changed

+17
-7
lines changed

1 file changed

+17
-7
lines changed

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

Lines changed: 17 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -101,10 +101,15 @@ describe("GeminiHandler", () => {
101101
})
102102

103103
// Verify the model configuration
104-
expect(mockGetGenerativeModel).toHaveBeenCalledWith({
105-
model: "gemini-2.0-flash-thinking-exp-1219",
106-
systemInstruction: systemPrompt,
107-
})
104+
expect(mockGetGenerativeModel).toHaveBeenCalledWith(
105+
{
106+
model: "gemini-2.0-flash-thinking-exp-1219",
107+
systemInstruction: systemPrompt,
108+
},
109+
{
110+
baseUrl: undefined,
111+
},
112+
)
108113

109114
// Verify generation config
110115
expect(mockGenerateContentStream).toHaveBeenCalledWith(
@@ -149,9 +154,14 @@ describe("GeminiHandler", () => {
149154

150155
const result = await handler.completePrompt("Test prompt")
151156
expect(result).toBe("Test response")
152-
expect(mockGetGenerativeModel).toHaveBeenCalledWith({
153-
model: "gemini-2.0-flash-thinking-exp-1219",
154-
})
157+
expect(mockGetGenerativeModel).toHaveBeenCalledWith(
158+
{
159+
model: "gemini-2.0-flash-thinking-exp-1219",
160+
},
161+
{
162+
baseUrl: undefined,
163+
},
164+
)
155165
expect(mockGenerateContent).toHaveBeenCalledWith({
156166
contents: [{ role: "user", parts: [{ text: "Test prompt" }] }],
157167
generationConfig: {

0 commit comments

Comments
 (0)