Consultar serviços de modelo com a API Gemini do Google

info

Beta

Esse recurso está em Beta. Administradores da conta podem controlar o acesso a este recurso na página Prévias do console da conta. Consulte Gerenciar prévias do Databricks.

importante

A API do Google Gemini é compatível apenas com os modelos fundacionais do Gemini. Para uma API unificada que funciona em todos os provedores, use a API de Conclusões de Chat.

A API do Google Gemini oferece compatibilidade nativa do SDK de AI do Google para modelos Gemini no Databricks. Utilize esta API quando precisar de recursos específicos do Gemini ou estiver migrando um código SDK de AI do Google existente.

Requisitos

• Consulte Requisitos.
• Instale o pacote google-genai no seu compute.

Exemplos de consulta

Os exemplos a seguir mostram como consultar um serviço de modelo usando a API do Google Gemini.

Python

Python

from google import genai
from google.genai import types
import os

DATABRICKS_TOKEN = os.environ.get('DATABRICKS_TOKEN')

client = genai.Client(
    api_key="databricks",
    http_options=types.HttpOptions(
        base_url="https://<workspace-url>/ai-gateway/gemini",
        headers={
            &quot;Authorization&quot;: f&quot;Bearer {DATABRICKS_TOKEN}&quot;,
        },
    ),
)

response = client.models.generate_content(
    model="system.ai.gemini-2-5-pro",
    contents=[
        types.Content(
            role="user",
            parts=[types.Part(text="What is a mixture of experts model?")],
        ),
    ],
    config=types.GenerateContentConfig(
        max_output_tokens=256,
    ),
)

print(response.text)

REST API

Bash

curl \
-u token:$DATABRICKS_TOKEN \
-X POST \
-H "Content-Type: application/json" \
-d '{
  "contents": [
    {
      "role": "user",
      "parts": [{"text": "What is a mixture of experts model?"}]
    }
  ],
  "generationConfig": {
    "maxOutputTokens": 256
  }
}' \
https://<workspace-url>/ai-gateway/gemini/v1beta/models/system.ai.gemini-2-5-pro:generateContent

Modelos compatíveis

Modelos de base hospedados pelo Databricks

• databricks-gemini-3-1-pro
• databricks-gemini-3-1-flash-lite
• databricks-gemini-3-pro
• databricks-gemini-3-flash
• databricks-gemini-2-5-pro
• databricks-gemini-2-5-flash

Tipos de entrada compatíveis

Os modelos Gemini no Databricks aceitam entradas de texto e imagem. As variantes Pro (databricks-gemini-3-1-pro, databricks-gemini-3-pro, databricks-gemini-2-5-pro) e as variantes Flash (databricks-gemini-3-1-flash-lite, databricks-gemini-3-flash, databricks-gemini-2-5-flash) também aceitam entradas de vídeo e áudio. Consulte Consultar modelos básicos por tipo para exemplos de solicitação de vídeo e áudio, e Consultar modelos básicos por tipo para requisitos de formato e tamanho de imagem. Para tipos de entrada por modelo, consulte modelos básicos hospedados pelo Databricks disponíveis nas APIs do Foundation Model.

Recursos adicionais

• Consulte modelos básicos por tipo.
• Modelos de base hospedados pelo Databricks disponíveis nas APIs do Foundation Model.