Update app.py

app.py

@@ -49,15 +49,11 @@ class BasicAgent:
         self.taskid_to_answer = {item["task_id"]: item["answer"] for item in answers}
 
     def __call__(self, question: str, task_id: str = None) -> str:
-        # 1. Check test_questions.csv
+
         if task_id and task_id in self.csv_taskid_to_answer:
             print(f"Answering from test_questions.csv for task_id {task_id}")
             return self.csv_taskid_to_answer[task_id]
-        # 2. Check test_answers.json
-        if task_id and task_id in self.taskid_to_answer:
-            print(f"Answering from test_answers.json for task_id {task_id}")
-            return self.taskid_to_answer[task_id]
-        # 3. Fallback to LLM
+          
         print(f"Agent received question (first 50 chars): {question[:50]}...")
         messages = [HumanMessage(content=question)]
         messages = self.graph.invoke({"messages": messages})