NeoAnd/adk-python
1
0
Fork 0
mirror of https://github.com/EvolutionAPI/adk-python.git synced 2025-07-13 15:14:50 -06:00
Code Issues Actions Packages Projects Releases Wiki Activity

feat: expose more config of VertexAiSearchTool from latest Google GenAI SDK

Browse Source 
for fixing: https://github.com/google/adk-python/issues/955

PiperOrigin-RevId: 764568728
This commit is contained in:
Xiang (Sean) Zhou 2025-05-28 22:32:13 -07:00 committed by Copybara-Service
parent b92e068155
commit 2b5c89b3a9
1 changed files with 8 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
src/google/adk/tools/vertex_ai_search_tool.py
View File

@ -40,6 +40,8 @@ class VertexAiSearchTool(BaseTool):
*,
data_store_id: Optional[str] = None,
search_engine_id: Optional[str] = None,
filter: Optional[str] = None,
max_results: Optional[int] = None,
):
"""Initializes the Vertex AI Search tool.
@ -64,6 +66,8 @@ class VertexAiSearchTool(BaseTool):
)
self.data_store_id = data_store_id
self.search_engine_id = search_engine_id
self.filter = filter
self.max_results = max_results
@override
async def process_llm_request(
@ -84,7 +88,10 @@ class VertexAiSearchTool(BaseTool):
types.Tool(
retrieval=types.Retrieval(
vertex_ai_search=types.VertexAISearch(
datastore=self.data_store_id, engine=self.search_engine_id
datastore=self.data_store_id,
engine=self.search_engine_id,
filter=self.filter,
max_results=self.max_results,
)
)
)