Consulta com a API do Google Gemini

Importante

A API Google Gemini é compatível apenas com modelos de fundação Gemini pay per token e modelos externos. Para uma API unificada que funcione entre fornecedores, use a API Open Responses ou a API Chat Completions.

A API Google Gemini fornece compatibilidade nativa do Google AI SDK para modelos Gemini em Azure Databricks. Use esta API quando precisar de funcionalidades específicas do Gemini ou estiver a migrar o código existente do Google AI SDK.

Requerimentos

  • Consulte Requisitos.
  • Instala o google-genai pacote no teu computador.

Exemplos de consulta

Os exemplos seguintes mostram como consultar um endpoint pay-per-token da API do Foundation Model usando a Google Gemini API.

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-3-1-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)

API REST

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-3-1-pro:generateContent

Modelos suportados

Modelos de fundação alojados em Databricks

  • databricks-gemini-3-5-flash-lite
  • databricks-gemini-3-8-flash
  • databricks-gemini-3-7-flash
  • databricks-gemini-3-6-flash
  • databricks-gemini-3-1-pro
  • databricks-gemini-3-1-flash-lite
  • databricks-gemini-3-flash

Modelos externos

  • Fornecedor de modelos Google

Tipos de entrada suportados

Os modelos Gemini no Azure Databricks aceitam entradas de texto e imagem. A variante Pro (databricks-gemini-3-1-pro) e as variantes Flash (databricks-gemini-3-5-flash-lite, databricks-gemini-3-1-flash-lite, databricks-gemini-3-flash) também aceitam entradas de vídeo e áudio. Consulte como consultar modelos de áudio e vídeo para ver exemplos de pedidos de vídeo e áudio, e como consultar modelos de visão para conhecer os requisitos de formato e tamanho das imagens. Para tipos de entrada por modelo, consulte Lista detalhada de modelos suportados pelas APIs de Modelos da Fundação Databricks.

Recursos adicionais

  • Última atualização a