File tree Expand file tree Collapse file tree 1 file changed +17
-7
lines changed
src/api/providers/__tests__ Expand file tree Collapse file tree 1 file changed +17
-7
lines changed Original file line number Diff line number Diff 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 : {
You can’t perform that action at this time.
0 commit comments