Поделиться через

Запрос с помощью API Google Gemini

Это важно

API Google Gemini совместим только с базовыми моделями Gemini по оплате за токен и внешними моделями. Для единого API, который работает во всех поставщиках, используйте API завершения чата.

API Google Gemini обеспечивает нативную совместимость с пакетом SDK Google AI для моделей Gemini в Azure Databricks. Используйте этот API, если вам нужны специальные функции Gemini или переносите существующий код Google AI SDK.

Требования

  • См. раздел Требования.
  • Установите пакет google-genai на компьютер.

Примеры запросов

Следующие примеры показывают, как выполнять запросы к конечной точке API базовой модели с оплатой за каждый токен, используя API Google Gemini.

Питон

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

Поддерживаемые модели

Модели фундамента, размещенные в 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

Внешние модели

  • Поставщик моделей Google

Дополнительные ресурсы

  • Last updated on