From d4718e0166079cc1f48057f79ef971c0dc302afb Mon Sep 17 00:00:00 2001 From: Hugo Aguirre Parra Date: Thu, 30 Oct 2025 21:20:45 +0000 Subject: [PATCH 1/2] chore(go): update gemini models in docs and samples --- go/genkit/genkit.go | 6 +++--- go/plugins/googlecloud/generate_test.go | 16 ++++++++-------- go/plugins/googlegenai/cache_test.go | 6 +++--- go/plugins/googlegenai/googleai_live_test.go | 8 ++++---- go/plugins/googlegenai/vertexai_live_test.go | 4 ++-- go/samples/basic-gemini-with-context/main.go | 2 +- go/samples/cache-gemini/main.go | 3 +-- go/samples/code-execution-gemini/main.go | 3 +-- go/samples/coffee-shop/main.go | 4 ++-- go/samples/files-api-vision/main.go | 2 +- go/samples/imagen-gemini/main.go | 2 +- go/samples/mcp-ception/mcp_ception.go | 2 +- go/samples/mcp-client/main.go | 4 ++-- go/samples/mcp-server/client.go | 2 +- go/samples/menu/main.go | 2 +- go/samples/partials-and-helpers/main.go | 2 +- go/samples/prompts-dir/prompts/example.prompt | 2 +- go/samples/prompts/prompts/countries.prompt | 2 +- go/samples/rag/main.go | 2 +- go/samples/vectorsearch-biqguery/main.go | 2 +- go/samples/vectorsearch-firestore/main.go | 2 +- 21 files changed, 38 insertions(+), 40 deletions(-) diff --git a/go/genkit/genkit.go b/go/genkit/genkit.go index 83cd33afeb..2022d48a70 100644 --- a/go/genkit/genkit.go +++ b/go/genkit/genkit.go @@ -138,7 +138,7 @@ func WithPromptDir(dir string) GenkitOption { // // Assumes a prompt file at ./prompts/jokePrompt.prompt // g := genkit.Init(ctx, // genkit.WithPlugins(&googlegenai.GoogleAI{}), -// genkit.WithDefaultModel("googleai/gemini-2.0-flash"), +// genkit.WithDefaultModel("googleai/gemini-2.5-flash"), // genkit.WithPromptDir("./prompts"), // ) // @@ -556,7 +556,7 @@ func LookupTool(g *Genkit, name string) ai.Tool { // // Define the prompt // capitalPrompt := genkit.DefinePrompt(g, "findCapital", // ai.WithDescription("Finds the capital of a country."), -// ai.WithModelName("googleai/gemini-1.5-flash"), // Specify the model +// ai.WithModelName("googleai/gemini-2.5-flash"), // Specify the model // ai.WithSystem("You are a helpful geography assistant."), // ai.WithPrompt("What is the capital of {{country}}?"), // ai.WithInputType(GeoInput{Country: "USA"}), @@ -637,7 +637,7 @@ func GenerateWithRequest(ctx context.Context, g *Genkit, actionOpts *ai.Generate // Example: // // resp, err := genkit.Generate(ctx, g, -// ai.WithModelName("googleai/gemini-2.0-flash"), +// ai.WithModelName("googleai/gemini-2.5-flash"), // ai.WithPrompt("Write a short poem about clouds."), // ai.WithConfig(&genai.GenerateContentConfig{MaxOutputTokens: 50}), // ) diff --git a/go/plugins/googlecloud/generate_test.go b/go/plugins/googlecloud/generate_test.go index 301088d4b3..694365f153 100644 --- a/go/plugins/googlecloud/generate_test.go +++ b/go/plugins/googlecloud/generate_test.go @@ -29,10 +29,10 @@ func TestGenerateTelemetry_PipelineIntegration(t *testing.T) { // Create span using the TracerProvider - this triggers the full pipeline ctx := context.Background() - _, span := f.tracer.Start(ctx, "googleai/gemini-2.0-pro-flash") + _, span := f.tracer.Start(ctx, "googleai/gemini-2.5-pro") span.SetAttributes( - attribute.String("genkit:name", "googleai/gemini-2.0-pro-flash"), + attribute.String("genkit:name", "googleai/gemini-2.5-pro"), attribute.String("genkit:metadata:subtype", "model"), attribute.String("genkit:path", "/{chatFlow,t:flow}/{generate,t:action}"), attribute.String("genkit:input", inputJSON), @@ -74,11 +74,11 @@ func TestGenerateTelemetry_MetricCapture(t *testing.T) { name: "successful generation captures all metrics", attrs: map[string]string{ "genkit:type": "action", - "genkit:name": "googleai/gemini-2.0-pro-flash", + "genkit:name": "googleai/gemini-2.5-pro", "genkit:metadata:subtype": "model", "genkit:path": "/{chatFlow,t:flow}/{generate,t:action}", }, - inputJSON: `{"model":"googleai/gemini-2.5-flash","messages":[{"content":[{"text":"Hello world"}],"role":"user"}]}`, + inputJSON: `{"model":"googleai/gemini-2.5-pro","messages":[{"content":[{"text":"Hello world"}],"role":"user"}]}`, outputJSON: `{"message":{"content":[{"text":"Hello! How can I help you today?"}],"role":"model"},"usage":{"inputTokens":12,"outputTokens":8,"inputCharacters":11,"outputCharacters":33,"inputImages":1,"outputImages":0},"latencyMs":342.5}`, expectMetrics: true, expectedStatus: "success", @@ -98,11 +98,11 @@ func TestGenerateTelemetry_MetricCapture(t *testing.T) { name: "failed generation captures error metrics", attrs: map[string]string{ "genkit:type": "action", - "genkit:name": "googleai/gemini-2.0-pro-flash", + "genkit:name": "googleai/gemini-2.5-pro", "genkit:metadata:subtype": "model", "genkit:path": "/{errorFlow,t:flow}/{generate,t:action}", }, - inputJSON: `{"model":"googleai/gemini-2.5-flash","messages":[{"content":[{"text":"Invalid prompt"}],"role":"user"}]}`, + inputJSON: `{"model":"googleai/gemini-2.5-pro","messages":[{"content":[{"text":"Invalid prompt"}],"role":"user"}]}`, outputJSON: `{"usage":{"inputCharacters":14}}`, expectMetrics: true, expectedStatus: "failure", @@ -116,11 +116,11 @@ func TestGenerateTelemetry_MetricCapture(t *testing.T) { name: "generation with video and audio captures multimedia metrics", attrs: map[string]string{ "genkit:type": "action", - "genkit:name": "googleai/gemini-2.0-pro-flash", + "genkit:name": "googleai/gemini-2.5-pro-flash", "genkit:metadata:subtype": "model", "genkit:path": "/{multimediaFlow,t:flow}/{generate,t:action}", }, - inputJSON: `{"model":"googleai/gemini-2.5-flash","messages":[{"content":[{"text":"Analyze this video and audio"}],"role":"user"}]}`, + inputJSON: `{"model":"googleai/gemini-2.5-pro","messages":[{"content":[{"text":"Analyze this video and audio"}],"role":"user"}]}`, outputJSON: `{"message":{"content":[{"text":"Analysis complete"}],"role":"model"},"usage":{"inputTokens":5,"outputTokens":3,"inputCharacters":26,"outputCharacters":17,"inputImages":0,"outputImages":1,"inputVideos":2,"outputVideos":0,"inputAudioFiles":1,"outputAudioFiles":0},"latencyMs":1250.3}`, expectMetrics: true, expectedStatus: "success", diff --git a/go/plugins/googlegenai/cache_test.go b/go/plugins/googlegenai/cache_test.go index 4591566806..031def5315 100644 --- a/go/plugins/googlegenai/cache_test.go +++ b/go/plugins/googlegenai/cache_test.go @@ -79,7 +79,7 @@ func TestGetContentForCache_Invalid(t *testing.T) { }, }, } - err := validateContextCacheRequest(req, "gemini-1.5-fash-001") + err := validateContextCacheRequest(req, "gemini-2.5-fash") if err == nil { t.Fatal("expecting error, system instructions are not supported with Context Cache") } @@ -89,7 +89,7 @@ func TestValidateContextCacheRequest_HasTools(t *testing.T) { req := &ai.ModelRequest{ Tools: []*ai.ToolDefinition{{Name: "someTool"}}, } - err := validateContextCacheRequest(req, "gemini-1.5-flash-001") + err := validateContextCacheRequest(req, "gemini-2.5-flash") if err == nil { t.Fatal("expected error if Tools are present") } @@ -100,7 +100,7 @@ func TestValidateContextCacheRequest_HasTools(t *testing.T) { func TestValidateContextCacheRequest_Valid(t *testing.T) { req := &ai.ModelRequest{} - err := validateContextCacheRequest(req, "gemini-1.5-flash-001") + err := validateContextCacheRequest(req, "gemini-2.5-flash") if err != nil { t.Fatalf("did not expect error, got: %v", err) } diff --git a/go/plugins/googlegenai/googleai_live_test.go b/go/plugins/googlegenai/googleai_live_test.go index 41939755a7..e8cb09e92d 100644 --- a/go/plugins/googlegenai/googleai_live_test.go +++ b/go/plugins/googlegenai/googleai_live_test.go @@ -66,7 +66,7 @@ func TestGoogleAILive(t *testing.T) { ctx := context.Background() g := genkit.Init(ctx, - genkit.WithDefaultModel("googleai/gemini-2.0-flash"), + genkit.WithDefaultModel("googleai/gemini-2.5-flash"), genkit.WithPlugins(&googlegenai.GoogleAI{APIKey: apiKey}), ) @@ -408,7 +408,7 @@ func TestGoogleAILive(t *testing.T) { } }) t.Run("image generation", func(t *testing.T) { - m := googlegenai.GoogleAIModel(g, "gemini-2.0-flash-preview-image-generation") + m := googlegenai.GoogleAIModel(g, "gemini-2.5-flash-image") resp, err := genkit.Generate(ctx, g, ai.WithConfig(genai.GenerateContentConfig{ ResponseModalities: []string{"IMAGE", "TEXT"}, @@ -602,8 +602,8 @@ func TestCacheHelper(t *testing.T) { func fetchImgAsBase64() (string, error) { // CC0 license image - imgUrl := "https://pd.w.org/2025/07/896686fbbcd9990c9.84605288-2048x1365.jpg" - resp, err := http.Get(imgUrl) + imgURL := "https://pd.w.org/2025/07/896686fbbcd9990c9.84605288-2048x1365.jpg" + resp, err := http.Get(imgURL) if err != nil { return "", err } diff --git a/go/plugins/googlegenai/vertexai_live_test.go b/go/plugins/googlegenai/vertexai_live_test.go index 0618879fee..6f4c7c2b21 100644 --- a/go/plugins/googlegenai/vertexai_live_test.go +++ b/go/plugins/googlegenai/vertexai_live_test.go @@ -44,7 +44,7 @@ func TestVertexAILive(t *testing.T) { ctx := context.Background() g := genkit.Init(ctx, - genkit.WithDefaultModel("vertexai/gemini-2.0-flash"), + genkit.WithDefaultModel("vertexai/gemini-2.5-flash"), genkit.WithPlugins(&googlegenai.VertexAI{ProjectID: projectID, Location: location}), ) @@ -267,7 +267,7 @@ func TestVertexAILive(t *testing.T) { if location != "global" { t.Skipf("image generation in Vertex AI is only supported in region: global, got: %s", location) } - m := googlegenai.VertexAIModel(g, "gemini-2.0-flash-preview-image-generation") + m := googlegenai.VertexAIModel(g, "gemini-2.5-flash-image") resp, err := genkit.Generate(ctx, g, ai.WithConfig(genai.GenerateContentConfig{ ResponseModalities: []string{"IMAGE", "TEXT"}, diff --git a/go/samples/basic-gemini-with-context/main.go b/go/samples/basic-gemini-with-context/main.go index 47c9f01af7..f971ecc9bc 100644 --- a/go/samples/basic-gemini-with-context/main.go +++ b/go/samples/basic-gemini-with-context/main.go @@ -36,7 +36,7 @@ func main() { // Define a simple flow that generates jokes about a given topic with a context of bananas genkit.DefineFlow(g, "contextFlow", func(ctx context.Context, input string) (string, error) { resp, err := genkit.Generate(ctx, g, - ai.WithModelName("googleai/gemini-2.0-flash"), + ai.WithModelName("googleai/gemini-2.5-flash"), ai.WithConfig(&genai.GenerateContentConfig{ Temperature: genai.Ptr[float32](1.0), }), diff --git a/go/samples/cache-gemini/main.go b/go/samples/cache-gemini/main.go index 69e3b71b16..4a862a2ee8 100644 --- a/go/samples/cache-gemini/main.go +++ b/go/samples/cache-gemini/main.go @@ -19,7 +19,6 @@ package main import ( "context" "errors" - "os" "github.com/firebase/genkit/go/ai" @@ -37,7 +36,7 @@ type duneQuestionInput struct { func main() { ctx := context.Background() g := genkit.Init(ctx, - genkit.WithDefaultModel("googleai/gemini-2.5-flash-preview-04-17"), + genkit.WithDefaultModel("googleai/gemini-2.5-flash"), genkit.WithPlugins(&googlegenai.GoogleAI{}), ) diff --git a/go/samples/code-execution-gemini/main.go b/go/samples/code-execution-gemini/main.go index 0162e23276..a8ee2ff264 100644 --- a/go/samples/code-execution-gemini/main.go +++ b/go/samples/code-execution-gemini/main.go @@ -18,7 +18,6 @@ import ( "context" "errors" "fmt" - "strings" "github.com/firebase/genkit/go/ai" @@ -35,7 +34,7 @@ func main() { // Define a flow to demonstrate code execution genkit.DefineFlow(g, "codeExecutionFlow", func(ctx context.Context, _ any) (string, error) { - m := googlegenai.GoogleAIModel(g, "gemini-2.5-flash-preview-04-17") + m := googlegenai.GoogleAIModel(g, "gemini-2.5-flash") if m == nil { return "", errors.New("failed to find model") } diff --git a/go/samples/coffee-shop/main.go b/go/samples/coffee-shop/main.go index cd146ebdd3..43a6259c4d 100755 --- a/go/samples/coffee-shop/main.go +++ b/go/samples/coffee-shop/main.go @@ -99,11 +99,11 @@ type testAllCoffeeFlowsOutput struct { func main() { ctx := context.Background() g := genkit.Init(ctx, - genkit.WithDefaultModel("googleai/gemini-2.0-flash"), + genkit.WithDefaultModel("googleai/gemini-2.5-flash"), genkit.WithPlugins(&googlegenai.GoogleAI{}), ) - m := googlegenai.GoogleAIModel(g, "gemini-2.0-flash") + m := googlegenai.GoogleAIModel(g, "gemini-2.5-flash") simpleGreetingPrompt := genkit.DefinePrompt(g, "simpleGreeting2", ai.WithPrompt(simpleGreetingPromptTemplate), ai.WithModel(m), diff --git a/go/samples/files-api-vision/main.go b/go/samples/files-api-vision/main.go index 2c1fe07a1e..f194fbabcd 100644 --- a/go/samples/files-api-vision/main.go +++ b/go/samples/files-api-vision/main.go @@ -63,7 +63,7 @@ func main() { // Use Files API URI directly with Genkit (now supported!) fmt.Println("Analyzing image with Genkit using Files API URI...") resp, err := genkit.Generate(ctx, g, - ai.WithModelName("googleai/gemini-2.0-flash"), + ai.WithModelName("googleai/gemini-2.5-flash"), ai.WithMessages( ai.NewUserMessage( ai.NewTextPart("What do you see in this image?"), diff --git a/go/samples/imagen-gemini/main.go b/go/samples/imagen-gemini/main.go index 3f803c5f52..dc36b6464c 100644 --- a/go/samples/imagen-gemini/main.go +++ b/go/samples/imagen-gemini/main.go @@ -36,7 +36,7 @@ func main() { // Define a simple flow that generates an image of a given topic genkit.DefineFlow(g, "imageFlow", func(ctx context.Context, input string) ([]string, error) { - m := googlegenai.GoogleAIModel(g, "gemini-2.0-flash-exp") + m := googlegenai.GoogleAIModel(g, "gemini-2.5-flash-image") if m == nil { return nil, errors.New("imageFlow: failed to find model") } diff --git a/go/samples/mcp-ception/mcp_ception.go b/go/samples/mcp-ception/mcp_ception.go index 9ffcf1cc30..b39bf0d31c 100644 --- a/go/samples/mcp-ception/mcp_ception.go +++ b/go/samples/mcp-ception/mcp_ception.go @@ -137,7 +137,7 @@ func mcpSelfConnection() { // Initialize Genkit with Google AI for the client g := genkit.Init(ctx, genkit.WithPlugins(&googlegenai.GoogleAI{}), - genkit.WithDefaultModel("googleai/gemini-2.0-flash"), + genkit.WithDefaultModel("googleai/gemini-2.5-flash"), ) // Server process is spawned automatically via stdio diff --git a/go/samples/mcp-client/main.go b/go/samples/mcp-client/main.go index 1864a27c08..b6f9149e26 100644 --- a/go/samples/mcp-client/main.go +++ b/go/samples/mcp-client/main.go @@ -57,7 +57,7 @@ func clientExample() { } response, err := genkit.Generate(ctx, g, - ai.WithModelName("googleai/gemini-2.5-pro-preview-05-06"), + ai.WithModelName("googleai/gemini-2.5-pro"), ai.WithPrompt("Convert the current time from New York to London timezone."), ai.WithTools(toolRefs...), ai.WithToolChoice(ai.ToolChoiceAuto), @@ -109,7 +109,7 @@ func managerExample() { } response, err := genkit.Generate(ctx, g, - ai.WithModelName("googleai/gemini-2.5-pro-preview-05-06"), + ai.WithModelName("googleai/gemini-2.5-pro"), ai.WithPrompt("What time is it in New York and Tokyo?"), ai.WithTools(toolRefs...), ai.WithToolChoice(ai.ToolChoiceAuto), diff --git a/go/samples/mcp-server/client.go b/go/samples/mcp-server/client.go index 7189a35e38..13896cb731 100644 --- a/go/samples/mcp-server/client.go +++ b/go/samples/mcp-server/client.go @@ -66,7 +66,7 @@ func client() { logger.FromContext(ctx).Info("Starting demo: Fetch and summarize content") response, err := genkit.Generate(ctx, g, - ai.WithModelName("googleai/gemini-2.5-pro-preview-05-06"), + ai.WithModelName("googleai/gemini-2.5-pro"), ai.WithPrompt("Fetch content from https://httpbin.org/json and give me a summary of what you find"), ai.WithTools(toolRefs...), ai.WithToolChoice(ai.ToolChoiceAuto), diff --git a/go/samples/menu/main.go b/go/samples/menu/main.go index 343e1707fb..3e8e5216fd 100644 --- a/go/samples/menu/main.go +++ b/go/samples/menu/main.go @@ -62,7 +62,7 @@ func main() { ctx := context.Background() g := genkit.Init(ctx, genkit.WithPlugins(&googlegenai.VertexAI{})) - model := googlegenai.VertexAIModel(g, "gemini-2.0-flash") + model := googlegenai.VertexAIModel(g, "gemini-2.5-flash") embedder := googlegenai.VertexAIEmbedder(g, "text-embedding-004") if err := setup01(g, model); err != nil { diff --git a/go/samples/partials-and-helpers/main.go b/go/samples/partials-and-helpers/main.go index 99a7400e50..cabcb263e2 100644 --- a/go/samples/partials-and-helpers/main.go +++ b/go/samples/partials-and-helpers/main.go @@ -35,7 +35,7 @@ func main() { }) g := genkit.Init(ctx, - genkit.WithDefaultModel("googleai/gemini-2.0-flash"), + genkit.WithDefaultModel("googleai/gemini-2.5-flash"), genkit.WithPlugins(&googlegenai.GoogleAI{}), ) diff --git a/go/samples/prompts-dir/prompts/example.prompt b/go/samples/prompts-dir/prompts/example.prompt index 3d5930865e..0492cfd326 100644 --- a/go/samples/prompts-dir/prompts/example.prompt +++ b/go/samples/prompts-dir/prompts/example.prompt @@ -1,5 +1,5 @@ --- -model: googleai/gemini-1.5-flash +model: googleai/gemini-2.5-flash config: temperature: 0.9 input: diff --git a/go/samples/prompts/prompts/countries.prompt b/go/samples/prompts/prompts/countries.prompt index fad2a41c25..93889ea48e 100644 --- a/go/samples/prompts/prompts/countries.prompt +++ b/go/samples/prompts/prompts/countries.prompt @@ -1,5 +1,5 @@ --- -model: googleai/gemini-2.0-flash +model: googleai/gemini-2.5-flash config: temperature: 0.5 output: diff --git a/go/samples/rag/main.go b/go/samples/rag/main.go index 1d67537c3d..34e5ebbfc3 100644 --- a/go/samples/rag/main.go +++ b/go/samples/rag/main.go @@ -106,7 +106,7 @@ func main() { } simpleQaPrompt := genkit.DefinePrompt(g, "simpleQaPrompt", - ai.WithModelName("googleai/gemini-2.0-flash"), + ai.WithModelName("googleai/gemini-2.5-flash"), ai.WithPrompt(simpleQaPromptTemplate), ai.WithInputType(simpleQaPromptInput{}), ai.WithOutputFormat(ai.OutputFormatText), diff --git a/go/samples/vectorsearch-biqguery/main.go b/go/samples/vectorsearch-biqguery/main.go index cc188c5214..1e78f67ccc 100644 --- a/go/samples/vectorsearch-biqguery/main.go +++ b/go/samples/vectorsearch-biqguery/main.go @@ -84,7 +84,7 @@ func main() { Location: vectorsearchPlugin.Location, }, vectorsearchPlugin)) - model := googlegenai.VertexAIModel(g, "gemini-2.0-flash") + model := googlegenai.VertexAIModel(g, "gemini-2.5-flash") // Create a BigQuery client. bqClient, err := bigquery.NewClient(ctx, vectorsearchPlugin.ProjectID) diff --git a/go/samples/vectorsearch-firestore/main.go b/go/samples/vectorsearch-firestore/main.go index 8e5ac6a92f..4aa8863fc9 100644 --- a/go/samples/vectorsearch-firestore/main.go +++ b/go/samples/vectorsearch-firestore/main.go @@ -84,7 +84,7 @@ func main() { Location: vectorsearchPlugin.Location, }, vectorsearchPlugin)) - model := googlegenai.VertexAIModel(g, "gemini-2.0-flash") + model := googlegenai.VertexAIModel(g, "gemini-2.5-flash") databaseId := "${FIRESTORE_DATABASE_ID}" // Replace with your Firestore database ID collectionName := "${FIRESTORE_COLLECTION_NAME}" // Replace with your Firestore collection name From 1b705f684557690b817639c8708ba481998ada1b Mon Sep 17 00:00:00 2001 From: Hugo Aguirre Parra Date: Fri, 31 Oct 2025 00:41:49 +0000 Subject: [PATCH 2/2] update live tests --- go/plugins/googlegenai/googleai_live_test.go | 4 ++-- go/plugins/googlegenai/vertexai_live_test.go | 6 +++--- 2 files changed, 5 insertions(+), 5 deletions(-) diff --git a/go/plugins/googlegenai/googleai_live_test.go b/go/plugins/googlegenai/googleai_live_test.go index e8cb09e92d..499400a079 100644 --- a/go/plugins/googlegenai/googleai_live_test.go +++ b/go/plugins/googlegenai/googleai_live_test.go @@ -359,7 +359,7 @@ func TestGoogleAILive(t *testing.T) { ai.WithMessages( ai.NewUserMessage( ai.NewTextPart("do you what animal is in the image?"), - ai.NewMediaPart("image/jpg", "data:image/jpg;base64,"+i), + ai.NewMediaPart("image/jpeg", "data:image/jpeg;base64,"+i), ), ), ) @@ -396,7 +396,7 @@ func TestGoogleAILive(t *testing.T) { ai.WithMessages( ai.NewUserMessage( ai.NewTextPart("do you know who's in the image?"), - ai.NewDataPart("data:image/jpg;base64,"+i), + ai.NewDataPart("data:image/jpeg;base64,"+i), ), ), ) diff --git a/go/plugins/googlegenai/vertexai_live_test.go b/go/plugins/googlegenai/vertexai_live_test.go index 6f4c7c2b21..49c9709186 100644 --- a/go/plugins/googlegenai/vertexai_live_test.go +++ b/go/plugins/googlegenai/vertexai_live_test.go @@ -48,7 +48,7 @@ func TestVertexAILive(t *testing.T) { genkit.WithPlugins(&googlegenai.VertexAI{ProjectID: projectID, Location: location}), ) - embedder := googlegenai.VertexAIEmbedder(g, "textembedding-gecko@003") + embedder := googlegenai.VertexAIEmbedder(g, "gemini-embedding-001") gablorkenTool := genkit.DefineTool(g, "gablorken", "use this tool when the user asks to calculate a gablorken", func(ctx *ai.ToolContext, input struct { @@ -332,7 +332,7 @@ func TestVertexAILive(t *testing.T) { t.Skipf("thinking in Vertex AI is only supported in these regions: [global, us-central1], got: %q", location) } - m := googlegenai.VertexAIModel(g, "gemini-2.5-flash-preview-05-20") + m := googlegenai.VertexAIModel(g, "gemini-2.5-flash") resp, err := genkit.Generate(ctx, g, ai.WithConfig( genai.GenerateContentConfig{ @@ -364,7 +364,7 @@ func TestVertexAILive(t *testing.T) { t.Skipf("thinking in Vertex AI is only supported in these regions: [global, us-central1], got: %q", location) } - m := googlegenai.VertexAIModel(g, "gemini-2.5-flash-preview-05-20") + m := googlegenai.VertexAIModel(g, "gemini-2.5-flash") resp, err := genkit.Generate(ctx, g, ai.WithConfig( genai.GenerateContentConfig{