File tree 1 file changed +8
-8
lines changed 1 file changed +8
-8
lines changed Original file line number Diff line number Diff line change @@ -337,11 +337,11 @@ func TestVertexAILive(t *testing.T) {
337
337
m := googlegenai .VertexAIModel (g , "gemini-2.5-flash-preview-04-17" )
338
338
resp , err := genkit .Generate (ctx , g ,
339
339
ai .WithConfig (
340
- googlegenai. GeminiConfig {
341
- Temperature : 1 ,
342
- ThinkingConfig : & googlegenai .ThinkingConfig {
340
+ genai. GenerateContentConfig {
341
+ Temperature : genai. Ptr [ float32 ]( 1 ) ,
342
+ ThinkingConfig : & genai .ThinkingConfig {
343
343
IncludeThoughts : true ,
344
- ThinkingBudget : 1024 ,
344
+ ThinkingBudget : genai. Ptr [ int32 ]( 1024 ) ,
345
345
},
346
346
},
347
347
),
@@ -369,11 +369,11 @@ func TestVertexAILive(t *testing.T) {
369
369
m := googlegenai .VertexAIModel (g , "gemini-2.5-flash-preview-04-17" )
370
370
resp , err := genkit .Generate (ctx , g ,
371
371
ai .WithConfig (
372
- googlegenai. GeminiConfig {
373
- Temperature : 1 ,
374
- ThinkingConfig : & googlegenai .ThinkingConfig {
372
+ genai. GenerateContentConfig {
373
+ Temperature : genai. Ptr [ float32 ]( 1 ) ,
374
+ ThinkingConfig : & genai .ThinkingConfig {
375
375
IncludeThoughts : false ,
376
- ThinkingBudget : 0 ,
376
+ ThinkingBudget : genai. Ptr [ int32 ]( 0 ) ,
377
377
},
378
378
},
379
379
),
You can’t perform that action at this time.
0 commit comments