Update agent.py

agent.py

@@ -4,8 +4,7 @@ from langgraph.prebuilt import ToolNode
 from langgraph.graph import START, StateGraph
 from langgraph.prebuilt import tools_condition
 from langchain_huggingface import HuggingFaceEndpoint, ChatHuggingFace
-from tools import extract_text, describe_image
-from langchain_community.tools import DuckDuckGoSearchRun
+from tools import extract_text, describe_image, wiki_search, web_search, arxiv_search
 from langchain_openai import ChatOpenAI
 from typing import TypedDict, Annotated, Optional
 
@@ -18,9 +17,8 @@ class BasicAgent():
     def __init__(self, llm):
         chat = ChatHuggingFace(llm=llm, verbose=True)
         
-        search_tool = DuckDuckGoSearchRun()
         vision_llm = ChatOpenAI(model="gpt-4o")
-        self.tools = [extract_text, describe_image, search_tool]
+        self.tools = [extract_text, describe_image, wiki_search, web_search, arxiv_search]
         self.chat_with_tools = vision_llm.bind_tools(self.tools) # need to change
         self._initialize_graph()
         print("BasicAgent initialized.")