Update app.py

app.py

@@ -47,7 +47,12 @@ sys_msg = SystemMessage(content="You are a helpful assistant tasked with perform
 
 # Node
 def assistant(state: MessagesState):
-   return {"messages": [llm_with_tools.invoke([sys_msg] + state["messages"])]}
+    try:
+        result = llm_with_tools.invoke([sys_msg] + state["messages"])
+        return {"messages": [result]}
+    except Exception as e:
+        return {"messages": [SystemMessage(content=f"AGENT ERROR: {str(e)}")]}
+
 
 # Build graph
 builder = StateGraph(MessagesState)