prasoonmhwr commited on
Commit
dd17f29
·
1 Parent(s): f7dc6d5

updated agent

Browse files
Files changed (1) hide show
  1. app.py +3 -0
app.py CHANGED
@@ -29,6 +29,9 @@ class BasicAgent:
29
  messages = [HumanMessage(content=question)]
30
  messages = self.graph.invoke({"messages": messages})
31
  print("message",messages)
 
 
 
32
  answer = messages['messages'] #[-1].content
33
  print(answer)
34
  return answer[14:]
 
29
  messages = [HumanMessage(content=question)]
30
  messages = self.graph.invoke({"messages": messages})
31
  print("message",messages)
32
+ if not messages or "messages" not in messages or not messages["messages"]:
33
+ print("⚠️ No messages returned by the graph.")
34
+ return "ERROR: Agent did not return a response."
35
  answer = messages['messages'] #[-1].content
36
  print(answer)
37
  return answer[14:]