MariaFilipkowska commited on
Commit
426f453
·
verified ·
1 Parent(s): 0da0d4e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -180,7 +180,7 @@ class BasicAgent:
180
  print("BasicAgent initialized.")
181
 
182
  def __call__(self, question: str) -> str:
183
-
184
  print(f"Agent received question (first 50 chars): {question[:50]}...")
185
  system_prompt = SystemMessage(
186
  content="You are a general AI assistant. \
@@ -199,7 +199,8 @@ class BasicAgent:
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
 
180
  print("BasicAgent initialized.")
181
 
182
  def __call__(self, question: str) -> str:
183
+
184
  print(f"Agent received question (first 50 chars): {question[:50]}...")
185
  system_prompt = SystemMessage(
186
  content="You are a general AI assistant. \
 
199
  message = HumanMessage(content=question)
200
  print(message)
201
  answer = None
202
+ wrong_answers = ["Requests rate limit exceeded", "", " ", " ", "insufficient information"]
203
+ while not answer or answer in wrong_answers or answer.lower().startswith("error"):
204
  try:
205
  alfred = build_agent(chat1)
206
  answer = alfred.invoke(input={"messages": [system_prompt, message]},config={"recursion_limit": 6})['messages'][-1].content