aynot commited on
Commit
f77a3e2
·
1 Parent(s): 51e99a5

updated prompt

Browse files
Files changed (1) hide show
  1. agent.py +17 -1
agent.py CHANGED
@@ -39,7 +39,23 @@ class MyAgent:
39
  return answer.strip()
40
 
41
  def make_prompt(self, question):
42
- return self.gaia_prompt + '\n' + question
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
43
 
44
  def make_additional_args(self, additional_data: dict | None = None):
45
  if additional_data is None:
 
39
  return answer.strip()
40
 
41
  def make_prompt(self, question):
42
+
43
+ custom_prompt = """
44
+ I will ask you a question.
45
+ Your answer should be a number OR as few words as possible OR a comma separated list of numbers and/or strings.
46
+ 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.
47
+ If you are asked for a string, don't use articles, neither abbreviations (e.g. for cities).
48
+ 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.
49
+
50
+ Examples:
51
+ Q: What is the capital of France?
52
+ A: Paris
53
+
54
+ Q: What is 2+2 equal to?
55
+ A: 4
56
+ """
57
+
58
+ return custom_prompt + '\n' + question
59
 
60
  def make_additional_args(self, additional_data: dict | None = None):
61
  if additional_data is None: