add gemini-2.0-flash-exp model

Author: hugoaguirre

go/plugins/googlegenai/models.go

@@ -21,6 +21,7 @@ const (
 	gemini20FlashLitePrev        = "gemini-2.0-flash-lite-preview"
 	gemini20ProExp0205           = "gemini-2.0-pro-exp-02-05"
 	gemini20FlashThinkingExp0121 = "gemini-2.0-flash-thinking-exp-01-21"
+	gemini20FlashExp             = "gemini-2.0-flash-exp"
 
 	gemini25ProExp0305 = "gemini-2.5-pro-exp-03-25"
 )
@@ -47,6 +48,7 @@ var (
 		gemini20ProExp0205,
 		gemini20FlashThinkingExp0121,
 		gemini25ProExp0305,
+		gemini20FlashExp,
 	}
 
 	supportedGeminiModels = map[string]ai.ModelInfo{
@@ -83,6 +85,12 @@ var (
 			},
 			Supports: &gemini.Multimodal,
 		},
+		gemini20FlashExp: {
+			Label:    "Gemini 2.0 Flash Exp",
+			Versions: []string{},
+			Supports: &gemini.Multimodal,
+		},
+
 		gemini20FlashLite: {
 			Label: "Gemini 2.0 Flash Lite",
 			Versions: []string{

go/plugins/internal/gemini/gemini.go

@@ -272,6 +272,7 @@ func Generate(
 func convertRequest(client *genai.Client, model string, input *ai.ModelRequest, cache *genai.CachedContent) (*genai.GenerateContentConfig, error) {
 	gc := genai.GenerateContentConfig{}
 	gc.CandidateCount = genai.Ptr[int32](1)
+	gc.ResponseModalities = []string{"TEXT", "IMAGE"}
 	if c, ok := input.Config.(*ai.GenerationCommonConfig); ok && c != nil {
 		if c.MaxOutputTokens != 0 {
 			gc.MaxOutputTokens = genai.Ptr[int32](int32(c.MaxOutputTokens))
@@ -512,7 +513,7 @@ func translateResponse(resp *genai.GenerateContentResponse) *ai.ModelResponse {
 	r.Usage = &ai.GenerationUsage{}
 	if u := resp.UsageMetadata; u != nil {
 		r.Usage.InputTokens = int(*u.PromptTokenCount)
-		r.Usage.OutputTokens = int(*u.CandidatesTokenCount)
+		// r.Usage.OutputTokens = int(*u.CandidatesTokenCount)
 		r.Usage.TotalTokens = int(u.TotalTokenCount)
 	}
 	return r

go/samples/basic-gemini/main.go

@@ -17,7 +17,6 @@ package main
 import (
 	"context"
 	"errors"
-	"fmt"
 	"log"
 
 	"github.com/firebase/genkit/go/ai"
@@ -39,18 +38,17 @@ func main() {
 
 	// Define a simple flow that generates jokes about a given topic
 	genkit.DefineFlow(g, "jokesFlow", func(ctx context.Context, input string) (string, error) {
-		m := googlegenai.GoogleAIModel(g, "gemini-2.0-flash")
+		m := googlegenai.GoogleAIModel(g, "gemini-2.0-flash-exp")
 		if m == nil {
 			return "", errors.New("jokesFlow: failed to find model")
 		}
 
 		resp, err := genkit.Generate(ctx, g,
 			ai.WithModel(m),
 			ai.WithConfig(&ai.GenerationCommonConfig{
-				Temperature: 1,
-				Version:     "gemini-2.0-flash-001",
+				Temperature: 0.5,
 			}),
-			ai.WithPromptText(fmt.Sprintf(`Tell silly short jokes about %s`, input)))
+			ai.WithPromptText(input))
 		if err != nil {
 			return "", err
 		}