scriptoriumadmin/Roo-Code
1
0
Fork 0
mirror of https://github.com/RooVetGit/Roo-Code.git synced 2026-09-07 08:26:51 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix test error for wrong arguments after additional base url added

Browse source
This commit is contained in:
dongqing 2025-03-11 17:37:07 +08:00
parent a3a5592654
commit 9d0b824b90
1 changed files with 17 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

24
src/api/providers/__tests__/gemini.test.ts
View file

@ -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: {