Update agent.py

agent.py

@@ -278,29 +278,27 @@ class BasicAgent:
                 If Task ID is included in the question, remember to call the relevant read tools [ie. read_file, excel_read, csv_read, mp3_listen, image_caption]
         """
         self.tools = [duckduck_websearch, serper_websearch, visit_webpage, wiki_search, text_splitter, youtube_transcript, read_file, excel_read, csv_read, mp3_listen, image_caption, python_tool]
-        self.model_with_tools = self.model.bind_tools(self.tools)
-        self.sys_msg = SystemMessage(content=self.sys_prompt)
+        # self.model_with_tools = self.model.bind_tools(self.tools)
+        # self.sys_msg = SystemMessage(content=self.sys_prompt)
         
         self.prompt = ChatPromptTemplate.from_messages([
             ("system", self.sys_prompt),
             ("human", "{input}")
         ])
-        # self.agent = initialize_agent(
-        #     tools=self.tools,
-        #     llm=self.model,
-        #     agent="zero-shot-react-description",  # ReAct agent type
-        #     verbose=True,
-        #     system_prompt=self.prompt,
-        #     handle_parsing_errors="Check your output and make sure it conforms, use the Action/Action Input syntax"
-        # )
-        self.graph = self.__graph_compile__()
+        self.agent = initialize_agent(
+            tools=self.tools,
+            llm=self.model,
+            agent="zero-shot-react-description",  # ReAct agent type
+            verbose=True,
+            system_prompt=self.prompt,
+            handle_parsing_errors="Check your output and make sure it conforms, use the Action/Action Input syntax"
+        )
+        # self.graph = self.__graph_compile__()
         print("BasicAgent initialized.")
     
     def __call__(self, task: dict) -> str:
         task_id, question, file_name = task["task_id"], task["question"], task["file_name"]
         print(f"Agent received question (first 50 chars): {question[:50]}...")
-        # response = self.agent_exe.invoke({"input": f"Question: {question}"})
-        # fixed_answer = response['message'][-1].content
         
         if file_name == "" or file_name is None:
             question = question
@@ -308,10 +306,10 @@ class BasicAgent:
             question = f"{question} with TASK-ID: {task_id}"
             # fixed_answer = self.agent.run(f'{question} with TASK-ID: {task_id}')
         # fixed_answer = "This is a default answer."
-        # fixed_answer = self.agent.run(question)
-        human_message = [HumanMessage(content=question)]
-        messages = self.graph.invoke({"messages": human_message})
-        fixed_answer = messages['messages'][-1].content
+        fixed_answer = self.agent.run(question)
+        # human_message = [HumanMessage(content=question)]
+        # messages = self.graph.invoke({"messages": human_message})
+        # fixed_answer = messages['messages'][-1].content
         print(f"Agent returning fixed answer: {fixed_answer}")
         time.sleep(60)
         return fixed_answer