Integration examples for Google AI (Gemini) using the go-ai SDK.
export GOOGLE_GENERATIVE_AI_API_KEY=your-api-key
go run main.go| Constant | Model ID |
|---|---|
google.ModelGemini25Pro |
gemini-2.5-pro |
google.ModelGemini25Flash |
gemini-2.5-flash |
google.ModelGemini25FlashLite |
gemini-2.5-flash-lite |
| Constant | Model ID |
|---|---|
google.ModelGemini3ProPreview |
gemini-3-pro-preview |
google.ModelGemini3FlashPreview |
gemini-3-flash-preview |
google.ModelGemini31ProPreview |
gemini-3.1-pro-preview (New) |
google.ModelGemini31ProPreviewCustom |
gemini-3.1-pro-preview-customtools (New) |
Full list of constants is in pkg/providers/google/model_ids.go.
| Constant | Model ID |
|---|---|
google.ModelImagen40Generate001 |
imagen-4.0-generate-001 |
google.ModelImagen40UltraGenerate001 |
imagen-4.0-ultra-generate-001 |
google.ModelImagen40FastGenerate001 |
imagen-4.0-fast-generate-001 |
| Constant | Model ID |
|---|---|
google.ModelGemini25FlashImage |
gemini-2.5-flash-image |
google.ModelGemini3ProImagePreview |
gemini-3-pro-image-preview |
google.ModelGemini31FlashImagePreview |
gemini-3.1-flash-image-preview (New) |
Image generation supports the following aspect ratios via provider.ImageGenerateOptions.AspectRatio:
Standard (Imagen + Gemini):
google.ImageAspectRatio1x1—1:1google.ImageAspectRatio3x4—3:4google.ImageAspectRatio4x3—4:3google.ImageAspectRatio9x16—9:16google.ImageAspectRatio16x9—16:9
Extended (Gemini image models only, added in #12897):
google.ImageAspectRatio2x3—2:3google.ImageAspectRatio3x2—3:2google.ImageAspectRatio4x5—4:5google.ImageAspectRatio5x4—5:4google.ImageAspectRatio21x9—21:9google.ImageAspectRatio1x8—1:8google.ImageAspectRatio8x1—8:1google.ImageAspectRatio1x4—1:4google.ImageAspectRatio4x1—4:1
The output resolution can be set via ProviderOptions:
opts := &provider.ImageGenerateOptions{
Prompt: "A photo",
ProviderOptions: map[string]interface{}{
"google": map[string]interface{}{
"imageSize": google.ImageSize2K, // "512", "1K", "2K", or "4K"
},
},
}See main.go for working examples including:
- Text generation with Gemini Pro
- Image generation with
gemini-3.1-flash-image-preview - Aspect ratio usage