Update agent.py
Browse files
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:'.
|