Consultar com a API do Google Gemini

importante

A API do Google Gemini é compatível apenas com os modelos básicos de pagamento por token do Gemini e com modelos externos. Para uma API unificada que funcione em todos os provedores, use a API Chat Completions.

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

Requisitos

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

Exemplos de consulta

Os exemplos a seguir mostram como consultar um endpoint de pagamento por token API do Foundation Model 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://example.staging.cloud.databricks.com/serving-endpoints/gemini",
        headers={
            "Authorization": f"Bearer {DATABRICKS_TOKEN}",
        },
    ),
)

response = client.models.generate_content(
    model="databricks-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_host>.databricks.com/serving-endpoints/gemini/v1beta/models/databricks-gemini-2-5-pro:generateContent

Modelos suportados

modelos de fundação hospedados no 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

Modelos externos

• provedor de modelos do Google

Tipos de entrada suportados

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 a seção "Consultar modelos de áudio e vídeo" para exemplos de solicitações de vídeo e áudio e a seção "Consultar modelos de visão" para requisitos de formato e tamanho de imagem. Para obter informações sobre os tipos de entrada por modelo, consulte os modelos de base hospedados no Databricks disponíveis em APIs de modelos de base.

Recursos adicionais

• Consultar um modelo de bate-papo.
• Os modelos de base hospedados no Databricks estão disponíveis nas APIs de Modelos de Base.