MariaFilipkowska commited on
Commit
cb7ae6c
·
verified ·
1 Parent(s): e5b34d9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -3
app.py CHANGED
@@ -138,7 +138,7 @@ alfred = builder.compile()
138
  system_prompt = SystemMessage(
139
  content="You are a general AI assistant. \
140
  I will ask you a question. Report your thoughts shortly, and finish your answer with the following template: \
141
- FINAL ANSWER: [YOUR FINAL ANSWER]. \
142
  YOUR FINAL ANSWER should be a number OR as few words as possible OR a comma separated list of numbers and/or strings. \
143
  If you are asked for a number, use only digits in your final answer. Don't use comma nor brackets to write your number neither use units such as $ or percent sign unless specified otherwise. \
144
  If you are asked for a string, don't use articles, neither abbreviations (e.g. for cities), and write the digits in plain text unless specified otherwise. \
@@ -159,9 +159,8 @@ class BasicAgent:
159
  answer = "".join(re.findall(r'(FINAL ANSWER.*)', answer, flags=re.M))
160
  answer = answer.replace('FINAL ANSWER:', '')
161
  answer = answer.replace('FINAL ANSWER', '')
162
- #answer = answer.replace('[', '')
163
  answer = answer.replace('*', '')
164
- #fixed_answer = answer.replace(']', '')
165
  print(f"Agent returning fixed answer: {answer}")
166
  return answer
167
 
 
138
  system_prompt = SystemMessage(
139
  content="You are a general AI assistant. \
140
  I will ask you a question. Report your thoughts shortly, and finish your answer with the following template: \
141
+ FINAL ANSWER: YOUR FINAL ANSWER \
142
  YOUR FINAL ANSWER should be a number OR as few words as possible OR a comma separated list of numbers and/or strings. \
143
  If you are asked for a number, use only digits in your final answer. Don't use comma nor brackets to write your number neither use units such as $ or percent sign unless specified otherwise. \
144
  If you are asked for a string, don't use articles, neither abbreviations (e.g. for cities), and write the digits in plain text unless specified otherwise. \
 
159
  answer = "".join(re.findall(r'(FINAL ANSWER.*)', answer, flags=re.M))
160
  answer = answer.replace('FINAL ANSWER:', '')
161
  answer = answer.replace('FINAL ANSWER', '')
162
+ answer = answer.replace('YOUR ', '')
163
  answer = answer.replace('*', '')
 
164
  print(f"Agent returning fixed answer: {answer}")
165
  return answer
166