github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-07-21 12:01:47 +00:00
Code Issues Packages Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
Erick Friis 2024-02-08 19:09:27 -08:00 committed by GitHub
parent 1032faba5f
commit d8913b9428
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
templates/gemini-functions-agent/gemini_functions_agent/agent.py
View File

@ -32,19 +32,7 @@ prompt = ChatPromptTemplate.from_messages(
] ]
) )
llm_with_tools = llm.bind( llm_with_tools = llm.bind(functions=[tavily_tool])
functions=[
{
"name": tavily_tool.name,
"description": tavily_tool.description,
"parameters": {
"type": "object",
"properties": {"query": {"type": "string"}},
"required": ["query"],
},
}
]
)
def _format_chat_history(chat_history: List[Tuple[str, str]]): def _format_chat_history(chat_history: List[Tuple[str, str]]):