vertex_ai_search

Retriever wrapper for Google Vertex AI Search.

Set the following environment variables before the tests: export PROJECT_ID=… - set to your Google Cloud project ID export DATA_STORE_ID=… - the ID of the search engine to use for the test

Classes

vertex_ai_search.VertexAIMultiTurnSearchRetriever	Google Vertex AI Search retriever for multi-turn conversations.
vertex_ai_search.VertexAISearchRetriever	Google Vertex AI Search retriever.
vertex_ai_search.VertexAISearchSummaryTool	Class that exposes a tool to interface with an App in Vertex Search and Conversation and get the summary of the documents retrieved.