Google Gemini API'siyle sorgulama

Google Gemini API'si, Azure Databricks'te Gemini modelleri için yerel Google AI SDK uyumluluğu sağlar. İkizler'e özgü özelliklere ihtiyacınız olduğunda veya mevcut Google AI SDK kodunu geçirirken bu API'yi kullanın.

Gereksinimler

• Bkz . Gereksinimler.
• google-genai paketini bilgisayarınıza yükleyin.

Sorgu örnekleri

Aşağıdaki örneklerde, Google Gemini API'sini kullanarak bir Foundation Model API'si token başına ödeme uç noktasının nasıl sorgulandığı gösterilmektedir.

Piton

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

Desteklenen modeller

Databricks tarafından barındırılan temel modeller

• 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

Dış modeller

• Google model sağlayıcısı

Desteklenen giriş türleri

Azure Databricks üzerindeki İkizler modelleri metin ve görüntü girişlerini kabul eder. Pro varyantları (databricks-gemini-3-1-pro, databricks-gemini-3-pro, databricks-gemini-2-5-pro) ve Flash varyantları (databricks-gemini-3-1-flash-lite, databricks-gemini-3-flash, databricks-gemini-2-5-flash) video ve ses girişlerini de kabul eder. Video ve ses isteği örnekleri için ses ve video modellerini sorgulama ve görüntü biçimi ve boyut gereksinimleri için görüntü işleme modellerini sorgulama bölümüne bakın. Model başına giriş türleri için bkz. Temel Model API'lerinde kullanılabilen Databricks tarafından barındırılan temel modeller.

Ek kaynaklar

• Sohbet modelini sorgulama.
• Databricks tarafından barındırılan temel modeller, Temel Model API'lerinde kullanılabilir.