Consultar com a API do Google Gemini

Importante

A API do Google Gemini é compatível apenas com modelos de base Gemini com cobrança por token e modelos externos. Para uma API unificada que funcione entre provedores, use a API de Respostas Abertas ou a API de Conclusões de Chat.

A API do Google Gemini fornece compatibilidade nativa do SDK de IA do Google para modelos Gemini no Azure Databricks. Use essa API quando precisar de recursos específicos do Gemini ou estiver migrando o código existente do SDK de IA do Google.

Requirements

  • Confira os Requisitos
  • Instale o google-genai pacote em sua computação.

Exemplos de consulta

Os exemplos a seguir mostram como consultar um endpoint de pagamento por token da API de modelo fundamental usando a API do Google Gemini.

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 com suporte

Modelos de base hospedados pelo 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

  • Provedor de modelos do Google

Tipos de entrada com suporte

Os modelos gêmeos em Azure Databricks aceitam entradas de texto e imagem. As variantes 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 modelos de áudio e vídeo de consulta para exemplos de solicitação de áudio e vídeo e modelos de visão de consulta para requisitos de tamanho e formato de imagem. Para tipos de entrada por modelo, veja Lista detalhada de modelos suportados pelas APIs de Modelos da Fundação Databricks.

Recursos adicionais

  • Última atualização em