Update app.py

app.py

@@ -45,22 +45,22 @@ class BasicAgent:
         print(f"Agent received question (first 50 chars): {question[:50]}...")
         messages = [self.systemMsg, HumanMessage(content=question)]
         response = self.graph.invoke({"messages": messages})
-        # ai_msg = response["messages"][-1].content
-        # if "FINAL ANSWER:" in ai_msg:
-        #    return ai_msg.split("FINAL ANSWER:")[-1].strip()
-        # else:
-        #     return ai_msg.strip()
-        last_msg = response["messages"][-1]
+        ai_msg = response["messages"][-1].content
+        if "FINAL ANSWER:" in ai_msg:
+           return ai_msg.split("FINAL ANSWER:")[-1].strip()
+        else:
+            return ai_msg.strip()
+        # last_msg = response["messages"][-1]
     
-        if not hasattr(last_msg, "content"):
-            raise ValueError(f"Graph returned invalid message type: {last_msg}")
+        # if not hasattr(last_msg, "content"):
+        #     raise ValueError(f"Graph returned invalid message type: {last_msg}")
         
-        ai_msg: str = last_msg.content
+        # ai_msg: str = last_msg.content
     
-        if "FINAL ANSWER:" in ai_msg:
-            return ai_msg.split("FINAL ANSWER:")[-1].strip()
+        # if "FINAL ANSWER:" in ai_msg:
+        #     return ai_msg.split("FINAL ANSWER:")[-1].strip()
         
-        return ai_msg.strip()
+        # return ai_msg.strip()