Spaces:
Running
Running
Daniel Amendoeira
commited on
Update agent.py
Browse files
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)
|