templates: simplify tool in gemini-functions-agent (#17282)

4 months ago · d8913b9428
parent 1032faba5f
commit d8913b9428
1 changed files with 1 additions and 13 deletions
--- 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]]):