gemini1.5からgemini2.0にアップデートした際に「Unexpected error: retrieval parameter is not supported in Gemini API 」に苦労した話

はじめに

Python SDKを用いてgemini1.5からgemini2.0へアップデートする際、「Unexpected error: retrieval parameter is not supported in Gemini API」エラーが発生し、解決に時間を要しました。その際の知見を共有します。

エラーの詳細 – 何が問題だったのか？

前提

!pip install --upgrade google-genai

修正前のコード

from google import genaifrom google import genai
from google.genai import types

def generate():
    client = genai.Client(api_key="APIキー")

    model = "gemini-2.0-flash-001"
    contents = [
        types.Content(
            role="user",
            parts=[
                types.Part.from_text(text="""プロンプト""")
            ]
        )
    ]
    tools = [
        types.Tool(retrieval=types.Retrieval(vertex_ai_search=types.VertexAISearch(datastore="データストアID"))),
    ]
    generate_content_config = types.GenerateContentConfig(
        temperature=0.2,
        top_p=0.8,
        max_output_tokens=1024,
        response_modalities=["TEXT"],
        safety_settings=[
            types.SafetySetting(
                category="HARM_CATEGORY_HATE_SPEECH",
                threshold="OFF"
            ),
            types.SafetySetting(
                category="HARM_CATEGORY_DANGEROUS_CONTENT",
                threshold="OFF"
            ),
            types.SafetySetting(
                category="HARM_CATEGORY_SEXUALLY_EXPLICIT",
                threshold="OFF"
            ),
            types.SafetySetting(
                category="HARM_CATEGORY_HARASSMENT",
                threshold="OFF"
            )
        ],
        tools=tools,
    )

    for chunk in client.models.generate_content_stream(
        model=model,
        contents=contents,
        config=generate_content_config,
    ):
        if not chunk.candidates or not chunk.candidates[0].content or not chunk.candidates[0].content.parts:
            continue
        print(chunk.candidates[0].content.parts[0].text, end="")

generate()

ドキュメントの確認、言語やライブラリのバージョンチェックを行いましたが、「Unexpected error: retrieval parameter is not supported in Gemini API」エラーが解消されませんでした。

解決策 – Gemini 2.0は当時（2025年4月時点）日本ではまだ提供されていませんでした。

そこで、genai.Clientの初期化を下記のように変更しました。

client = genai.Client(api_key="APIキー")

↓

client = genai.Client(
  vertexai=True,
  project="プロジェクト名",
location="us-central1",
)

これにより、正常に回答が得られるようになりました。

まとめ

今回は、サービスがどこのリージョンで提供されているかの意識が薄かったことでエラー解消に時間がかかってしまいました。実装していると、サービスがどこのリージョンで提供されているかは抜けがちな観点かと思いましたので気をつけることが重要だと感じました。