Browse Source

fix test error for wrong arguments after additional base url added

dongqing 11 months ago
parent
commit
9d0b824b90
1 changed files with 17 additions and 7 deletions
  1. 17 7
      src/api/providers/__tests__/gemini.test.ts

+ 17 - 7
src/api/providers/__tests__/gemini.test.ts

@@ -101,10 +101,15 @@ describe("GeminiHandler", () => {
 			})
 
 			// Verify the model configuration
-			expect(mockGetGenerativeModel).toHaveBeenCalledWith({
-				model: "gemini-2.0-flash-thinking-exp-1219",
-				systemInstruction: systemPrompt,
-			})
+			expect(mockGetGenerativeModel).toHaveBeenCalledWith(
+				{
+					model: "gemini-2.0-flash-thinking-exp-1219",
+					systemInstruction: systemPrompt,
+				},
+				{
+					baseUrl: undefined,
+				},
+			)
 
 			// Verify generation config
 			expect(mockGenerateContentStream).toHaveBeenCalledWith(
@@ -149,9 +154,14 @@ describe("GeminiHandler", () => {
 
 			const result = await handler.completePrompt("Test prompt")
 			expect(result).toBe("Test response")
-			expect(mockGetGenerativeModel).toHaveBeenCalledWith({
-				model: "gemini-2.0-flash-thinking-exp-1219",
-			})
+			expect(mockGetGenerativeModel).toHaveBeenCalledWith(
+				{
+					model: "gemini-2.0-flash-thinking-exp-1219",
+				},
+				{
+					baseUrl: undefined,
+				},
+			)
 			expect(mockGenerateContent).toHaveBeenCalledWith({
 				contents: [{ role: "user", parts: [{ text: "Test prompt" }] }],
 				generationConfig: {