Skip to main content
Add your Google API key to begin using Gemini/Gemma models in the platform.
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.

Add the Integration Using the Arize Skills

Add a Gemini integration from your coding agent using the Arize Skills:

Create a Gemini integration named 'my-gemini' with my API key AI...

Add the Integration from Arize AX

Select Gemini from the Providers List

The AI Provider integrations tab

Fill Out Your Integration Details

The AI integration details tab
Give your integration a name, set your Gemini API key, and check the box to agree to the terms and conditions. You can create and manage your API key in Google AI Studio under API Keys.

Advanced Settings

The AI integration details tab
Expand the Advanced Settings section to configure the following:
  • Supports Function Calling - Turn this on to allow the integration to use function calling features. This is on by default.
  • Authorized Orgs - Configure which organizations have access to this integration.
  • Authorized Spaces - Configure which spaces have access to this integration.

Add the Integration

Finally check the box to agree to the terms and conditions, then select the Add Integration button to create the integration.