Consultar com a API do Google Gemini

importante

A API Google Gemini é compatível apenas com modelos de fundação Gemini de pagamento por token e modelos externos. Para uma API unificada que funciona entre provedores, use a API de Respostas Abertas ou a API de Conclusão de Chat.

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

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 de áudio e vídeo para exemplos de solicitação de vídeo e áudio, e Consultar modelos de visão para requisitos de formato e tamanho de imagem. Para tipos de entrada por modelo, consulte modelos de fundação hospedados no Databricks disponíveis nas APIs do Foundation Model.

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.