Update agent.py

agent.py

@@ -14,6 +14,7 @@ from langchain_core.tools import tool
 from langchain.tools.retriever import create_retriever_tool
 from supabase.client import create_client
 import re
+import traceback
 
 load_dotenv()
 
@@ -211,11 +212,21 @@ def build_graph(provider: str = "groq"):
         ]
         return {"messages": state["messages"] + examples}
 
+    
+
     def assistant(state: MessagesState):
         try:
             messages = [SystemMessage(content=system_prompt.strip())] + state["messages"]
             result = llm_with_tools.invoke(messages)
-            raw_output = result.content.strip()
+
+            # Handle different return types gracefully
+            if hasattr(result, "content"):
+                raw_output = result.content.strip()
+            elif isinstance(result, dict) and "content" in result:
+                raw_output = result["content"].strip()
+            else:
+                raise ValueError(f"Unexpected result format: {repr(result)}")
+
             print("🤖 Raw LLM output:", repr(raw_output))
 
             match = re.search(r"FINAL ANSWER:\s*(.+)", raw_output, re.IGNORECASE)
@@ -226,9 +237,12 @@ def build_graph(provider: str = "groq"):
                 final_output = f"FINAL ANSWER: {raw_output or 'Unable to determine answer'}"
 
             return {"messages": [HumanMessage(content=final_output)]}
+
         except Exception as e:
-            print(f"🔥 Error in assistant node: {e}")
-            return {"messages": [HumanMessage(content=f"FINAL ANSWER: AGENT ERROR: {e}")]}
+            print(f"🔥 Exception: {e}")
+            traceback.print_exc()
+            return {"messages": [HumanMessage(content=f"FINAL ANSWER: AGENT ERROR: {type(e).__name__}: {e}")]}
+
 
     builder = StateGraph(MessagesState)
     builder.add_node("retriever", retriever)