From d8913b94287483fc6603c74278f2e793ba90bb70 Mon Sep 17 00:00:00 2001 From: Erick Friis Date: Thu, 8 Feb 2024 19:09:27 -0800 Subject: [PATCH] templates: simplify tool in gemini-functions-agent (#17282) --- .../gemini_functions_agent/agent.py | 14 +------------- 1 file changed, 1 insertion(+), 13 deletions(-) diff --git a/templates/gemini-functions-agent/gemini_functions_agent/agent.py b/templates/gemini-functions-agent/gemini_functions_agent/agent.py index 79aba8a9445..1f1c756d7c4 100644 --- a/templates/gemini-functions-agent/gemini_functions_agent/agent.py +++ b/templates/gemini-functions-agent/gemini_functions_agent/agent.py @@ -32,19 +32,7 @@ prompt = ChatPromptTemplate.from_messages( ] ) -llm_with_tools = llm.bind( - functions=[ - { - "name": tavily_tool.name, - "description": tavily_tool.description, - "parameters": { - "type": "object", - "properties": {"query": {"type": "string"}}, - "required": ["query"], - }, - } - ] -) +llm_with_tools = llm.bind(functions=[tavily_tool]) def _format_chat_history(chat_history: List[Tuple[str, str]]):