llamasrock commited on
Commit
52d913e
·
verified ·
1 Parent(s): bbdf5ac

Update app.py

Browse files

updated the system prompt to GAIA template

Files changed (1) hide show
  1. app.py +8 -1
app.py CHANGED
@@ -58,9 +58,16 @@ class BasicAgent:
58
  fixed_answer = "This is a default answer."
59
  # print(f"Agent returning fixed answer: {fixed_answer}")
60
  # return fixed_answer
 
 
 
 
 
 
 
61
 
62
  await asyncio.sleep(10)
63
- return self.agent.run(question, task_id)
64
  # # Use the Google GenAI client to run the question
65
  # answer = self.client.models.generate_content(
66
  # model='gemini-2.0-flash',
 
58
  fixed_answer = "This is a default answer."
59
  # print(f"Agent returning fixed answer: {fixed_answer}")
60
  # return fixed_answer
61
+
62
+ prompt = f'''You are a general AI assistant. I will ask you a question. Report your thoughts, and finish your answer with the following template:
63
+ FINAL ANSWER: [YOUR FINAL ANSWER]. YOUR FINAL ANSWER should be a number OR as few words as possible OR a comma separated list of numbers and/or strings.
64
+ If you are asked for a number, don't use comma to write your number neither use units such as $ or percent sign unless specified otherwise.
65
+ 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.
66
+ If you are asked for a comma separated list, apply the above rules depending of whether the element to be put in the list is a number or a string.
67
+ {question}'''
68
 
69
  await asyncio.sleep(10)
70
+ return self.agent.run(prompt, task_id)
71
  # # Use the Google GenAI client to run the question
72
  # answer = self.client.models.generate_content(
73
  # model='gemini-2.0-flash',