vindruid commited on
Commit
74b2e6e
·
1 Parent(s): 1b5f069

system message

Browse files
Files changed (1) hide show
  1. gaia_agent.py +6 -3
gaia_agent.py CHANGED
@@ -113,11 +113,14 @@ def arvix_search(query: str) -> str:
113
  return {"arvix_results": formatted_search_docs}
114
 
115
  # load the system prompt from the file
116
- with open("system_prompt.txt", "r", encoding="utf-8") as f:
117
- system_prompt = f.read()
118
 
119
  # System message
120
- sys_msg = SystemMessage(content=system_prompt)
 
 
 
 
 
121
 
122
  # build a retriever
123
  embeddings = HuggingFaceEmbeddings(model_name="sentence-transformers/all-mpnet-base-v2") # dim=768
 
113
  return {"arvix_results": formatted_search_docs}
114
 
115
  # load the system prompt from the file
 
 
116
 
117
  # System message
118
+ sys_msg = SystemMessage(content="""You are a helpful assistant tasked with answering questions using a set of tools.
119
+ Now, I will ask you a question. Report your thoughts, and finish your answer with the following template:
120
+ FINAL ANSWER: [YOUR FINAL ANSWER].
121
+ YOUR FINAL ANSWER should be 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.
122
+ Your answer should only start with "FINAL ANSWER: ", then follows with the answer.
123
+ """)
124
 
125
  # build a retriever
126
  embeddings = HuggingFaceEmbeddings(model_name="sentence-transformers/all-mpnet-base-v2") # dim=768