giulia-fontanella commited on
Commit
d44ef49
·
verified ·
1 Parent(s): 0957bb5

Update agent.py

Browse files
Files changed (1) hide show
  1. agent.py +2 -1
agent.py CHANGED
@@ -40,7 +40,7 @@ class BasicAgent():
40
 
41
  def __call__(self, question: str, file_name : str) -> str:
42
  print(f"Agent received question: {question}. /nProvided file: {file_name}.")
43
- if file_name is not None:
44
  messages=[HumanMessage(content=f"{question}. The filename you have access to is {file_name}.")]
45
  else:
46
  messages=[HumanMessage(content=question)]
@@ -54,6 +54,7 @@ class BasicAgent():
54
 
55
  sys_msg = SystemMessage(content=f"""
56
  You are a general AI assistant. I will ask you a question. Reason step by step and search for the information you need using available tools, one step at a time.
 
57
  Finish your answer with the following template: FINAL ANSWER: [YOUR FINAL ANSWER].
58
  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.
59
  When providing the final answer, ONLY give [YOUR FINAL ANSWER]. Do not add anything else, no additional motivation or explanation, and do not return 'FINAL ANSWER:'.
 
40
 
41
  def __call__(self, question: str, file_name : str) -> str:
42
  print(f"Agent received question: {question}. /nProvided file: {file_name}.")
43
+ if file_name is not None and file_name!='':
44
  messages=[HumanMessage(content=f"{question}. The filename you have access to is {file_name}.")]
45
  else:
46
  messages=[HumanMessage(content=question)]
 
54
 
55
  sys_msg = SystemMessage(content=f"""
56
  You are a general AI assistant. I will ask you a question. Reason step by step and search for the information you need using available tools, one step at a time.
57
+ If you do not have enough information to answer, use the tools available to search for it.
58
  Finish your answer with the following template: FINAL ANSWER: [YOUR FINAL ANSWER].
59
  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.
60
  When providing the final answer, ONLY give [YOUR FINAL ANSWER]. Do not add anything else, no additional motivation or explanation, and do not return 'FINAL ANSWER:'.