Replace assistant calling with graph invoke

app.py

@@ -59,10 +59,10 @@ class BasicAgent:
     def __call__(self, question: str) -> Tuple[str, TypedDict]:
         print(f"Agent received question (first 50 chars): {question[:50]}...")
         state = State(question=question, messages=[HumanMessage(content=question)])
-        fixed_answer = self.assistant(state)["messages"][0].content
-        print(f"Last message: {self.assistant(state)['messages'][0]}")
-        print(f"Agent returning fixed answer: {fixed_answer}")
-        return fixed_answer, state["messages"]
+        result = self.compiled_graph.invoke(state)
+        final_answer = result["messages"][-1].content
+        print(f"Agent returning fixed answer: {final_answer}")
+        return final_answer, result["messages"]
 
     def assistant(self, state: State):
         # System message