Compartir a través de

Consulta con Google Gemini API

Important

La API de Google Gemini solo es compatible con los modelos base de pago por token y los modelos externos. Para una API unificada que funcione en todos los proveedores, use la API de Finalizaciones de Chat.

Google Gemini API proporciona compatibilidad nativa del SDK de Google AI para los modelos de Gemini en Azure Databricks. Use esta API cuando necesite características específicas de Gemini o migren el código existente del SDK de Google AI.

Requisitos

  • Vea Requisitos.
  • Instale el paquete google-genai en su computadora.

Ejemplos de consultas

En los ejemplos siguientes se muestra cómo consultar un punto de conexión de pago por token de Foundation Model API mediante google Gemini API.

Pitón

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

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 compatibles

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

  • Proveedor de modelos de Google

Recursos adicionales

  • Last updated on