Pular para o conteúdo principal
Última atualização em

Consultar com a API do Google Gemini

importante

A API Google Gemini é compatível apenas com os modelos de pagamento por token da base Gemini. 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
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)

Modelos suportados​

modelos de fundação hospedados no Databricks​

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

Recursos adicionais​