Update agent.py

agent.py

@@ -4,7 +4,8 @@ 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 *
+from tools import extract_text
+from langchain_community.tools import DuckDuckGoSearchRun
 
 
 class AgentState(TypedDict):
@@ -12,9 +13,13 @@ class AgentState(TypedDict):
 
 
 class Agent():
-    def __init__(self, llm, tools):
+    def __init__(self, llm):
         chat = ChatHuggingFace(llm=llm, verbose=True)
-        chat_with_tools = chat.bind_tools(tools)
+        
+        search_tool = DuckDuckGoSearchRun()
+        vision_llm = ChatOpenAI(model="gpt-4o")
+
+        chat_with_tools = chat.bind_tools([extract_text,search_tool])
         self._initialize_graph()
 
     def _initialize_graph(self):