Update agent.py

agent.py

@@ -118,8 +118,13 @@ class LangGraphAgent:
         result = gaia_agent.invoke(input_state, {"recursion_limit": 30})  # prevents infinite looping when the LLM keeps calling tools over and over
         
         content_list = result["messages"][-1].content   # extract content list from the last message
-        texts = [obj["text"] for obj in content_list if obj.get("type") == "text"]  # pull out all text fragments (required for gpt-4o-mini)
-        final_response = "\n".join(texts).strip()  # join and clean
+        if isinstance(content, list):
+            # multimodal list: pull only the text parts
+            texts = [obj["text"] for obj in content if isinstance(obj, dict) and obj.get("type") == "text"]
+            final_response = "\n".join(texts).strip()
+        else:
+            # plain string fallback
+            final_response = str(content).strip()
         
         try:
             return final_response.split("FINAL ANSWER:")[-1].strip()  #  parse out only what's after "FINAL ANSWER:"