graph test

app.py

@@ -97,11 +97,11 @@ class BasicAgent:
         return res.get("answer", "N/A")
 
     def log_question(self, state: AnswerState) -> Dict[str, Any]:
-        print(f"Agent received question: {state['question']}...")
+        print(f"[log_question] Agent received question: {state['question']}...")
         return {}
 
     def final_answer(self, state: AnswerState) -> Dict[str, Any]:
-        print(f"Agent returning answer: {state['answer']}")
+        print(f"[final_answer] Agent returning answer: {state['answer']}")
         answer = state["answer"] if state["answer"] else "I don't know"
         return {
             "answer": answer,
@@ -109,6 +109,8 @@ class BasicAgent:
         }
 
     def try_answer(self, state: AnswerState) -> str:
+        print("[final_answer] Agent trying to answer")
+
         messages = [HumanMessage(content=PROMPT.format(question=state["question"]))]
         response = self.model.invoke(messages)
         print(f"Agent response: {response.content}")
@@ -184,7 +186,7 @@ def run_and_submit_all(profile: gr.OAuthProfile | None):
     results_log = []
     answers_payload = []
     print(f"Running agent on {len(questions_data)} questions...")
-    for item in questions_data[:3]:
+    for item in questions_data[:1]:
         task_id = item.get("task_id")
         question_text = item.get("question")
         if not task_id or question_text is None: