Update agent.py

agent.py

@@ -13,6 +13,7 @@ from langchain_core.messages import SystemMessage, HumanMessage
 from langchain_core.tools import tool
 from langchain.tools.retriever import create_retriever_tool
 from supabase.client import create_client
+from langchain_core.messages import AIMessage
 import re
 import traceback
 
@@ -234,9 +235,10 @@ def build_graph(provider: str = "groq"):
                 final_output = f"FINAL ANSWER: {match.group(1).strip()}"
             else:
                 print("⚠️ 'FINAL ANSWER:' not found. Raw content will be used as fallback.")
-                final_output = f"FINAL ANSWER: {raw_output or 'Unable to determine answer'}"
+                
+                final_output = "FINAL ANSWER: Unable to determine answer" if not raw_output else f"FINAL ANSWER: {raw_output}"
 
-            return {"messages": [HumanMessage(content=final_output)]}
+            return {"messages": [AIMessage(content=final_output)]}
 
         except Exception as e:
             print(f"🔥 Exception: {e}")