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

4 months ago · 12d3159dd6
parent febf9540b9
commit 12d3159dd6
1 changed files with 1 additions and 1 deletions
--- a/templates/gemini-functions-agent/gemini_functions_agent/agent.py
+++ b/templates/gemini-functions-agent/gemini_functions_agent/agent.py
@ -32,7 +32,7 @@ prompt = ChatPromptTemplate.from_messages(
    ]
 )

-llm_with_tools = llm.bind(functions=[tavily_tool])
+llm_with_tools = llm.bind(functions=tools)


 def _format_chat_history(chat_history: List[Tuple[str, str]]):