jc7k commited on
Commit
3ddbd8b
·
1 Parent(s): 8fdcd86

Updated prompt

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -38,8 +38,9 @@ class BasicAgent:
38
 
39
  def __call__(self, question: str) -> str:
40
  print(f"Agent received question (first 50 chars): {question[:50]}...")
 
41
 
42
- answer = self.agent.run(question)
43
  if answer:
44
  print(f"Agent returning answer: {answer}")
45
  return answer
 
38
 
39
  def __call__(self, question: str) -> str:
40
  print(f"Agent received question (first 50 chars): {question[:50]}...")
41
+ system_prompt = "You are a general AI assistant. I will ask you a question. Report your thoughts, and finish your answer as a number OR as few words as possible OR a comma separated list of numbers and/or strings. 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. 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. 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. Here is the question: "
42
 
43
+ answer = self.agent.run(system_prompt + question)
44
  if answer:
45
  print(f"Agent returning answer: {answer}")
46
  return answer