Skip to content

Commit 2202be9

Browse files
committed
fix rebase issues
1 parent 5ac7e70 commit 2202be9

File tree

1 file changed

+8
-8
lines changed

1 file changed

+8
-8
lines changed

go/plugins/googlegenai/vertexai_live_test.go

Lines changed: 8 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -337,11 +337,11 @@ func TestVertexAILive(t *testing.T) {
337337
m := googlegenai.VertexAIModel(g, "gemini-2.5-flash-preview-04-17")
338338
resp, err := genkit.Generate(ctx, g,
339339
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{
343343
IncludeThoughts: true,
344-
ThinkingBudget: 1024,
344+
ThinkingBudget: genai.Ptr[int32](1024),
345345
},
346346
},
347347
),
@@ -369,11 +369,11 @@ func TestVertexAILive(t *testing.T) {
369369
m := googlegenai.VertexAIModel(g, "gemini-2.5-flash-preview-04-17")
370370
resp, err := genkit.Generate(ctx, g,
371371
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{
375375
IncludeThoughts: false,
376-
ThinkingBudget: 0,
376+
ThinkingBudget: genai.Ptr[int32](0),
377377
},
378378
},
379379
),

0 commit comments

Comments
 (0)