Update app.py

app.py

@@ -29,7 +29,37 @@ class BasicAgent:
             "get_files_task_id_tool": lambda task_id: get_files_task_id_tool(task_id),
             "image_processing_tool": lambda file_content: image_processing_tool(file_content)
         }
-        
+    def __call__(self, question: dict) -> str:
+        print(f"Agent received question (first 50 chars): {str(question)[:50]}...")
+        task_id = question["task_id"]
+        question_text = question["question"]
+        conversation = [{"role": "system", "content": self.system_prompt},
+                        {"role": "user", "content": question_text}]
+        while True:
+            prompt = "\n".join([msg["content"] for msg in conversation])
+            response = self._call_llm(prompt)
+            if response.startswith("ANSWER:"):
+                answer = response[len("ANSWER:"):].strip()
+                break
+            elif response.startswith("TOOL:"):
+                tool_call = response[len("TOOL:"):].strip()
+                try:
+                    tool_name, args_str = tool_call.split("(", 1)
+                    args_str = args_str.rstrip(")")
+                    if tool_name == "get_files_task_id_tool":
+                        observation = self.tools[tool_name](task_id)
+                    else:
+                        args = eval(args_str)
+                        observation = self.tools[tool_name](args)
+                    conversation.append({"role": "assistant", "content": response})
+                    conversation.append({"role": "user", "content": f"Observation: {observation}"})
+                except Exception as e:
+                    conversation.append({"role": "user", "content": f"Error executing tool: {e}"})
+            else:
+                conversation.append({"role": "user", "content": "Please provide a tool call or the final answer."})
+        print(f"Agent returning answer: {answer}")
+        return answer
+    
 def web_search_tool(search_terms: str) -> str:
     """
     Retrieves information from the internet using DuckDuckGo Search and returns results in JSON format.