Update app.py
Browse filesupdated the system prompt to GAIA template
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(
|
| 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',
|