Update agent.py

agent.py

@@ -1,5 +1,6 @@
 import os
 from typing import TypedDict, List, Dict, Any, Optional
+from langgraph.prebuilt import create_react_agent
 from langgraph.graph import StateGraph, START, END
 from langchain_google_genai import ChatGoogleGenerativeAI
 from langchain_core.tools import tool
@@ -113,8 +114,8 @@ class BasicAgent:
         """
         self.app = self.__graph_compile__()
         tools = [web_search, visit_webpage, read_file, image_caption]
-        self.model = model.bind_tools(tools) # LLM with tools
-        # self.agent = CodeAgent(tools=[DuckDuckGoSearchTool()], model=model)
+        # self.model = model.bind_tools(tools) # LLM with tools
+        self.agent = create_react_agent(model, tools)
         print("BasicAgent initialized.")
     
     def __call__(self, question: str) -> str:
@@ -124,7 +125,7 @@ class BasicAgent:
             ("human", f"Question: {question}")
         ]
         response = self.model.invoke(prompt_msg)
-        fixed_answer = response.content
+        fixed_answer = response['message'][-1].content
         # fixed_answer = "This is a default answer."
         print(f"Agent returning fixed answer: {fixed_answer}")
         return fixed_answer