ORromu commited on
Commit
7bfa566
·
verified ·
1 Parent(s): 7dc9ee0

Update agent.py

Browse files
Files changed (1) hide show
  1. agent.py +2 -1
agent.py CHANGED
@@ -1,5 +1,5 @@
1
  from typing import TypedDict, Annotated
2
- from tool import add, substract, multiply, divide, DuckDuckGoSearchTool, WikipediaSearchTool, ArxivSearchTool, PubmedSearchTool
3
  from os import getenv
4
  from langgraph.graph.message import add_messages
5
  from langchain_core.messages import AnyMessage, SystemMessage, HumanMessage, AIMessage
@@ -28,6 +28,7 @@ tools = [add,
28
  multiply,
29
  divide,
30
  DuckDuckGoSearchTool,
 
31
  WikipediaSearchTool,
32
  ArxivSearchTool,
33
  PubmedSearchTool]
 
1
  from typing import TypedDict, Annotated
2
+ from tool import add, substract, multiply, divide, DuckDuckGoSearchTool, TavilySearchTool, WikipediaSearchTool, ArxivSearchTool, PubmedSearchTool
3
  from os import getenv
4
  from langgraph.graph.message import add_messages
5
  from langchain_core.messages import AnyMessage, SystemMessage, HumanMessage, AIMessage
 
28
  multiply,
29
  divide,
30
  DuckDuckGoSearchTool,
31
+ TavilySearchTool,
32
  WikipediaSearchTool,
33
  ArxivSearchTool,
34
  PubmedSearchTool]