Thanh Vinh Vo commited on
Commit
e6c31e1
·
1 Parent(s): b2072a9
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -255,10 +255,10 @@ class BasicAgent:
255
  5. If the question asks for a list, please make sure that the elements are separated by a comma(`,`) and a space(` `). For example: `1, 2, 3` instead of `1,2,3`.
256
  6. If the question asks for a number with specific decimal places, please format the number into string with the same decimal places. For example: 3.00 instead of 3.
257
  """
258
- if "food" in question.lower():
259
  prompt = f"""
260
  {prompt}
261
- 7. Be careful about the difference between food and drink items.
262
  """
263
  result = self.manager_agent.run(prompt)
264
  print(f"Agent responded with: {result}")
 
255
  5. If the question asks for a list, please make sure that the elements are separated by a comma(`,`) and a space(` `). For example: `1, 2, 3` instead of `1,2,3`.
256
  6. If the question asks for a number with specific decimal places, please format the number into string with the same decimal places. For example: 3.00 instead of 3.
257
  """
258
+ if "food" in question.lower() or "drink" in question.lower():
259
  prompt = f"""
260
  {prompt}
261
+ 7. Be careful about the difference between food and drink items. For instance: Ice Cream is a food item!
262
  """
263
  result = self.manager_agent.run(prompt)
264
  print(f"Agent responded with: {result}")