giulia-fontanella commited on
Commit
a18d9f7
·
verified ·
1 Parent(s): 218c0cf

Update agent.py

Browse files
Files changed (1) hide show
  1. agent.py +4 -1
agent.py CHANGED
@@ -72,7 +72,10 @@ class BasicAgent():
72
  Returns:
73
  A single string containing the search result or summary.
74
  """
75
- sys_msg = SystemMessage(content=f"You are a helpful assistant. You have access to the following tools:\n{textual_description_of_tool} \n You may have access to some optional images. Currently the loaded image is: {image}. \n When providing the final answer, be as concise as possible, provide only the FINAL ANSWER. \n For example: QUESTION: What was the actual enrollment count of the clinical trial on H. pylori in acne vulgaris patients from Jan-May 2018 as listed on the NIH website? \n FINAL ANSWER: 90 \n")
 
 
 
76
 
77
  return {
78
  "messages": [self.chat_with_tools.invoke([sys_msg] + state["messages"])],
 
72
  Returns:
73
  A single string containing the search result or summary.
74
  """
75
+ sys_msg = SystemMessage(content=f"""
76
+ You are a general AI assistant. I will ask you a question. Report your thoughts, and finish your answer with the following template: FINAL ANSWER: [YOUR FINAL ANSWER].
77
+ 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.
78
+ You have access to the following tools:\n{textual_description_of_tool} \n You may have access to some optional images. Currently the loaded image is: {image}.""")
79
 
80
  return {
81
  "messages": [self.chat_with_tools.invoke([sys_msg] + state["messages"])],