MariaFilipkowska commited on
Commit
0da0d4e
·
verified ·
1 Parent(s): 2fca3ac

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -199,7 +199,7 @@ class BasicAgent:
199
  message = HumanMessage(content=question)
200
  print(message)
201
  answer = None
202
- while not answer or answer == "" or answer = "insufficient information":
203
  try:
204
  alfred = build_agent(chat1)
205
  answer = alfred.invoke(input={"messages": [system_prompt, message]},config={"recursion_limit": 6})['messages'][-1].content
 
199
  message = HumanMessage(content=question)
200
  print(message)
201
  answer = None
202
+ while not answer or answer == "" or answer == "insufficient information" or answer.lower().startswith("error"):
203
  try:
204
  alfred = build_agent(chat1)
205
  answer = alfred.invoke(input={"messages": [system_prompt, message]},config={"recursion_limit": 6})['messages'][-1].content