HCho commited on
Commit
5d24331
·
verified ·
1 Parent(s): 7cc1748

Update agent.py

Browse files
Files changed (1) hide show
  1. agent.py +2 -2
agent.py CHANGED
@@ -38,7 +38,7 @@ def divide(a: int,b: int) -> float:
38
 
39
  @tool
40
  def web_search(query: str) -> str:
41
- """ Search for a query on web and return best 2 result."""
42
 
43
  search_results = TavilySearchResults(max_results = 3).invoke(input=query)
44
 
@@ -86,7 +86,7 @@ def arxiv_search(query: str) -> str:
86
 
87
 
88
 
89
- system_prompt = """You are a general AI assistant. I will ask you a question. Use your tools and think carefully to report your thoughts, and finish your answer with the following template:
90
  FINAL ANSWER: [YOUR FINAL ANSWER]. YOUR FINAL ANSWER should be a number OR as few words as possible OR a comma separated list of numbers and/or strings.
91
  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.
92
  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.
 
38
 
39
  @tool
40
  def web_search(query: str) -> str:
41
+ """ Search for a query on web and return best 3 result."""
42
 
43
  search_results = TavilySearchResults(max_results = 3).invoke(input=query)
44
 
 
86
 
87
 
88
 
89
+ system_prompt = """You are a general AI assistant. I will ask you a question. Think carefully, step by step, and use your given tools to report your thoughts, and finish your answer with the following template:
90
  FINAL ANSWER: [YOUR FINAL ANSWER]. YOUR FINAL ANSWER should be a number OR as few words as possible OR a comma separated list of numbers and/or strings.
91
  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.
92
  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.