Получение ошибки 400 при вызове Google Vertex API с функцией “retrieval” (хранилище данных)

Question

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

Я отправляю POST-запрос на этот конечный пункт:

https://us-central1-aiplatform.googleapis.com/v1/projects/%5C%5C\<project_id\>/locations/us-central1/publishers/google/models/gemini-1.0-pro-001:generateContent

И с полезной нагрузкой:

{
  "contents": [{
    "role": "user",
    "parts": [{
      "text": <prompt>
    }]
  }],
  "model": "projects/<project_id>/locations/us/publishers/google/models/gemini-1.0-pro-001"
}

Работает нормально, но добавление этого в полезную нагрузку:

  "tools": [{
    "retrieval": {
      "vertexAiSearch": {
        "datastore": "projects/<project_id>/locations/us/collections/default_collection/dataStores/<datastore_id>"
      }
    }
  }]

Вызывает ошибку 400:

{
  "error": {
    "code": 400,
    "message": "Запрос содержит недопустимый аргумент.",
    "status": "INVALID_ARGUMENT"
  }
}

Вещи, предложенные в других темах Stack Overflow, которые не помогли:

убедитесь, что приложение и корзина находятся в одном регионе (они в одном)
убедитесь, что есть достаточные права (и подождите, пока изменения прав вступят в силу) (права в порядке)
убедитесь, что приложение и корзина находятся в одном регионе (они в одном)
убедитесь, что достаточные права (и подождите, пока изменения прав вступят в силу) (права в порядке)
убедитесь, что активирована функция Enterprise (я уверен, что она была отмечена при создании хранилища данных)
убедитесь, что создано приложение для поиска, связанное с облачным хранилищем (корзина) – я это сделал, однако приложение для поиска полностью отделено от агентского приложения (я не вижу, как между ними есть связь), так что я уверен, что это не помогает, но все равно.

И просто чтобы подтвердить (так как это может помочь другим), иерархия объектов в Vertex такова:

Конечная точка – местоположение в URL должно быть конкретным местоположением (например, us-central1) или “глобальным”, но не может быть большим регионом (например, “us”) – я выбрал “us-central1”
Проект (просто контейнер для всего остального) – проекты не имеют местоположений.
Приложение (может быть агентом, поиском или чатом) – вы можете выбрать конкретный регион (например, us-central1) или более крупный регион (us). Я выбрал “us”.
Хранилища данных (связывают приложение с данными – данные могут быть веб-сайтом, облачным хранилищем, API и т.д.) – я выбрал “облачное хранилище”.
Облачное хранилище (фактические данные) – вы можете выбрать только более крупную область (например, us) для корзины – поэтому я выбрал “us”.

Логи показывают, что корзина индексируется нормально, я могу спрашивать модель об одном тестовом документе, который я загрузил (используя Workbench), и это работает нормально.

Я пробовал устанавливать все различные типы местоположений. Никаких успехов.

Я знаю, что агент подключается нормально к облачному хранилищу, потому что если есть какие-либо проблемы с правами, вам не дадут сохранить ваши выборы.

Я использую gemini-1.0-pro-001 (не gemini-1.0-pro-002) во всех случаях, потому что gemini-1.0-pro-002 не является опцией в агенте. Я не хотел конфликтов моделей.

Я не могу придумать, что еще проверить.

Получение ошибки 400 при вызове Google Vertex API с функцией “retrieval” (хранилище данных)

Вопрос или проблема

Ответ или решение

Ошибка 400: Понимание проблемы

Анализ запроса

Потенциальные проблемы

Рекомендации по отладке

Заключение