Update LG_agent.py

LG_agent.py

@@ -157,17 +157,18 @@ def build_graph(max_steps: int = 5):
 #         else:
 #             return "No response."
 
-def __call__(self, question: str) -> str:
+class BasicAgent:
     def __init__(self, max_steps: int = 5):
         self.graph = build_graph(max_steps)
-    
-    response = self.graph({"messages": [HumanMessage(content=question)]})
-    if response.get("messages"):
-        final_message = response["messages"][-1]
-        raw_content = final_message.content if hasattr(final_message, "content") else "No final message."
-        return extract_gaia_answer(raw_content)
-    else:
-        return "No response."
+
+    def __call__(self, question: str) -> str:
+        response = self.graph({"messages": [HumanMessage(content=question)]})
+        if response.get("messages"):
+            final_message = response["messages"][-1]
+            raw_content = final_message.content if hasattr(final_message, "content") else "No final message."
+            return extract_gaia_answer(raw_content)
+        else:
+            return "No response."
 
 
 if __name__ == "__main__":