Update app.py

app.py

@@ -45,11 +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
+        # 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}")
+        
+        ai_msg: str = last_msg.content
+    
         if "FINAL ANSWER:" in ai_msg:
-           return ai_msg.split("FINAL ANSWER:")[-1].strip()
-        else:
-            return ai_msg.strip()
+            return ai_msg.split("FINAL ANSWER:")[-1].strip()
+        
+        return ai_msg.strip()