graph test

app.py

@@ -68,13 +68,15 @@ class BasicAgent:
         self.graph.add_node("final_answer", self.final_answer)
 
         self.graph.add_edge(START, "log_question")
-        self.graph.add_edge("log_question", "try_answer")
 
         # Add conditional edges
         self.graph.add_conditional_edges(
-            "try_answer",
+            "log_question",
             self.try_answer,
-            {"FINAL_ANSWER": "final_answer", "SOME_TOOL": "final_answer"},
+            {
+                "FINAL_ANSWER": "final_answer",
+                "SOME_TOOL": "final_answer",
+            },
         )
 
         self.graph.add_edge("final_answer", END)
@@ -91,17 +93,19 @@ class BasicAgent:
         )
         return str(res)
 
-    def log_question(self, state: AnswerState) -> None:
-        print(f"Agent received question (first 50 chars): {state['question']}...")
+    def log_question(self, state: AnswerState) -> Dict[str, Any]:
+        print(f"Agent received question: {state['question']}...")
+        return {}
 
     def final_answer(self, state: AnswerState) -> str:
         print(f"Agent returning answer: {state['answer']}")
         return state["answer"] if state["answer"] else "I don't know"
 
     def try_answer(self, state: AnswerState) -> str:
-
         messages = [HumanMessage(content=PROMPT.format(question=state["question"]))]
         response = self.model.invoke(messages)
+        print(f"Agent response: {response.content}")
+
         txt = response.text().split("FINAL ANSWER:")
         print(f"Agent returning answer: {txt}")
         answer = txt[-1].strip() if len(txt) > 1 else "I don't know."