Daniel Amendoeira commited on
Commit
0ddf13f
·
verified ·
1 Parent(s): 406ca82

Update agent.py

Browse files
Files changed (1) hide show
  1. agent.py +2 -2
agent.py CHANGED
@@ -10,7 +10,7 @@ from typing import TypedDict, Annotated, Literal
10
  from langchain_community.tools import BraveSearch # web search
11
  from langchain_experimental.tools.python.tool import PythonAstREPLTool # for logic/math problems
12
 
13
- from tools import (calculator_basic, datetime_tools, transcribe_audio, transcribe_youtube, query_image, webpage_content)
14
  from prompt import system_prompt
15
 
16
  # gpt-4.1-nano (cheaper for debugging) with temperature 0 for less randomness
@@ -30,7 +30,7 @@ search_tool = BraveSearch.from_api_key(
30
  )
31
 
32
  community_tools = [search_tool, python_tool]
33
- custom_tools = calculator_basic + datetime_tools + [transcribe_audio, transcribe_youtube, query_image, webpage_content]
34
 
35
  tools = community_tools + custom_tools
36
  llm_with_tools = llm.bind_tools(tools)
 
10
  from langchain_community.tools import BraveSearch # web search
11
  from langchain_experimental.tools.python.tool import PythonAstREPLTool # for logic/math problems
12
 
13
+ from tools import (calculator_basic, datetime_tools, transcribe_audio, transcribe_youtube, query_image, webpage_content, read_excel)
14
  from prompt import system_prompt
15
 
16
  # gpt-4.1-nano (cheaper for debugging) with temperature 0 for less randomness
 
30
  )
31
 
32
  community_tools = [search_tool, python_tool]
33
+ custom_tools = calculator_basic + datetime_tools + [transcribe_audio, transcribe_youtube, query_image, webpage_content, read_excel]
34
 
35
  tools = community_tools + custom_tools
36
  llm_with_tools = llm.bind_tools(tools)