Skip to main content
Add your Google API key to begin using Gemini/Gemma models in the platform.
Note: By adding this integration, your data may be sent to Google for certain actions within Arize (e.g., prompt playground) and your account may be billed for usage.

Select Gemini integration

Select the Gemini integration card

Fill out your integration

Set up a Gemini integration with API key
You can create and manage your API key in Google AI Studio under API Keys.

Supported models

  • gemini-3.1-pro-preview
  • gemini-3.1-flash-lite-preview
  • gemini-3-pro-preview
  • gemini-3-flash-preview
  • gemini-2.5-pro
  • gemini-2.5-flash
  • gemini-2.5-flash-lite
  • gemini-2.5-flash-lite-preview-06-17
  • gemini-2.0-flash
  • gemma-3-27b-it
  • gemma-3-12b-it