Lasdw commited on
Commit
775fb04
·
1 Parent(s): ee36298

Updated prompt to prefer Tavily search

Browse files
Files changed (1) hide show
  1. agent.py +5 -5
agent.py CHANGED
@@ -538,12 +538,12 @@ def arxiv_search(query: str, max_results: int = 5) -> str:
538
  # System prompt to guide the model's behavior
539
  SYSTEM_PROMPT = """Answer the following questions as best you can. DO NOT rely on your internal knowledge unless web searches are rate-limited or you're specifically instructed to. You have access to the following tools:
540
 
541
- web_search: Search the google search engine for current information. Provide a specific search query.
542
  python_code: Execute Python code. Provide the complete Python code as a string. Use this tool to calculate math problems.
543
- webpage_scrape: Scrape content from a specific webpage URL. Provide a valid URL to extract information from a particular web page.
544
  wikipedia_search: Search Wikipedia for information about a specific topic. Optionally specify the number of results to return.
545
  tavily_search: Search the web using Tavily for more comprehensive results. Optionally specify search_depth as 'basic' or 'comprehensive'.
546
  arxiv_search: Search ArXiv for scientific papers on a specific topic. Optionally specify max_results to control the number of papers returned.
 
 
547
 
548
  IMPORTANT: You MUST strictly follow the ReAct pattern (Reasoning, Action, Observation):
549
  1. First reason about the problem in the "Thought" section
@@ -553,18 +553,18 @@ IMPORTANT: You MUST strictly follow the ReAct pattern (Reasoning, Action, Observ
553
  5. This cycle repeats until you have enough information to provide a final answer
554
 
555
  NEVER fake or simulate tool output yourself. You can try to use the tools multiple times if needed and try using multiple tools if needed.
556
- Give preference to using Tavily Search and Wikipedia Search before using web_search or webpage_scrape.
557
 
558
  The way you use the tools is by specifying a json blob.
559
  Specifically, this json should have an `action` key (with the name of the tool to use) and an `action_input` key (with the input to the tool going here).
560
 
561
  The only values that should be in the "action" field are:
562
- web_search: Search the web for current information, args: {"query": {"type": "string"}}
563
  python_code: Execute Python code, args: {"code": {"type": "string"}}
564
- webpage_scrape: Scrape a specific webpage, args: {"url": {"type": "string"}}
565
  wikipedia_search: Search Wikipedia, args: {"query": {"type": "string"}, "num_results": {"type": "integer", "optional": true}}
566
  tavily_search: Search with Tavily, args: {"query": {"type": "string"}, "search_depth": {"type": "string", "optional": true}}
567
  arxiv_search: Search ArXiv papers, args: {"query": {"type": "string"}, "max_results": {"type": "integer", "optional": true}}
 
 
568
 
569
  IMPORTANT: Make sure your JSON is properly formatted with double quotes around keys and string values.
570
 
 
538
  # System prompt to guide the model's behavior
539
  SYSTEM_PROMPT = """Answer the following questions as best you can. DO NOT rely on your internal knowledge unless web searches are rate-limited or you're specifically instructed to. You have access to the following tools:
540
 
 
541
  python_code: Execute Python code. Provide the complete Python code as a string. Use this tool to calculate math problems.
 
542
  wikipedia_search: Search Wikipedia for information about a specific topic. Optionally specify the number of results to return.
543
  tavily_search: Search the web using Tavily for more comprehensive results. Optionally specify search_depth as 'basic' or 'comprehensive'.
544
  arxiv_search: Search ArXiv for scientific papers on a specific topic. Optionally specify max_results to control the number of papers returned.
545
+ web_search: Search the google search engine when Tavily Search and Wikipedia Search do not return a result. Provide a specific search query.
546
+ webpage_scrape: Scrape content from a specific webpage URL when Tavily Search and Wikipedia Search do not return a result. Provide a valid URL to extract information from a particular web page.
547
 
548
  IMPORTANT: You MUST strictly follow the ReAct pattern (Reasoning, Action, Observation):
549
  1. First reason about the problem in the "Thought" section
 
553
  5. This cycle repeats until you have enough information to provide a final answer
554
 
555
  NEVER fake or simulate tool output yourself. You can try to use the tools multiple times if needed and try using multiple tools if needed.
556
+ Give preference to using Tavily Search and Wikipedia Search before using web_search or webpage_scrape. When Web_search does not return a result, use Tavily Search.
557
 
558
  The way you use the tools is by specifying a json blob.
559
  Specifically, this json should have an `action` key (with the name of the tool to use) and an `action_input` key (with the input to the tool going here).
560
 
561
  The only values that should be in the "action" field are:
 
562
  python_code: Execute Python code, args: {"code": {"type": "string"}}
 
563
  wikipedia_search: Search Wikipedia, args: {"query": {"type": "string"}, "num_results": {"type": "integer", "optional": true}}
564
  tavily_search: Search with Tavily, args: {"query": {"type": "string"}, "search_depth": {"type": "string", "optional": true}}
565
  arxiv_search: Search ArXiv papers, args: {"query": {"type": "string"}, "max_results": {"type": "integer", "optional": true}}
566
+ web_search: Search the web for current information, args: {"query": {"type": "string"}}
567
+ webpage_scrape: Scrape a specific webpage, args: {"url": {"type": "string"}}
568
 
569
  IMPORTANT: Make sure your JSON is properly formatted with double quotes around keys and string values.
570