EtienneB commited on
Commit
f55b551
·
1 Parent(s): aaf4112

updated prompt and tool name

Browse files
Files changed (3) hide show
  1. agent.py +4 -4
  2. system-prompt-claude-2.txt +1 -1
  3. tools.py +2 -2
agent.py CHANGED
@@ -9,7 +9,7 @@ from langgraph.graph import START, MessagesState, StateGraph
9
  from langgraph.prebuilt import ToolNode, tools_condition
10
 
11
  from tools import (absolute, add, analyze_csv_file, # extract_text_from_image,
12
- analyze_excel_file, arvix_search, audio_transcription,
13
  compound_interest, convert_temperature, divide,
14
  download_file, exponential, factorial, floor_divide,
15
  get_current_time_in_timezone, greatest_common_divisor,
@@ -30,7 +30,7 @@ tools = [
30
  get_current_time_in_timezone, compound_interest,
31
  convert_temperature, factorial, greatest_common_divisor,
32
  is_prime, least_common_multiple, percentage_calculator,
33
- wikipedia_search, analyze_excel_file, arvix_search,
34
  audio_transcription, python_code_parser, analyze_csv_file,
35
  # extract_text_from_image,
36
  reverse_sentence, web_content_extract,
@@ -67,7 +67,7 @@ You are an advanced AI agent equipped with multiple tools to solve complex, mult
67
  - String operations or text processing
68
 
69
  3. **Tool Selection and Evaluation**: Before using any tool, systematically evaluate all available options:
70
- - **Review ALL 30 available tools** in your toolkit before making a selection
71
  - **Match tool capabilities** to the specific requirements of your current step
72
  - **Choose the most appropriate tool** for each task from the complete toolkit
73
  - **Plan multi-tool sequences** - many questions require 2-5 tools in various combinations
@@ -123,7 +123,7 @@ You are an advanced AI agent equipped with multiple tools to solve complex, mult
123
  - **web_search**: Search the web for information
124
  - **web_content_extract**: Extract content from web pages
125
  - **wikipedia_search**: Search Wikipedia for information
126
- - **arvix_search**: Search academic papers on arXiv
127
 
128
  ### Utilities
129
  - **convert_temperature**: Convert between temperature units
 
9
  from langgraph.prebuilt import ToolNode, tools_condition
10
 
11
  from tools import (absolute, add, analyze_csv_file, # extract_text_from_image,
12
+ analyze_excel_file, arxiv_search, audio_transcription,
13
  compound_interest, convert_temperature, divide,
14
  download_file, exponential, factorial, floor_divide,
15
  get_current_time_in_timezone, greatest_common_divisor,
 
30
  get_current_time_in_timezone, compound_interest,
31
  convert_temperature, factorial, greatest_common_divisor,
32
  is_prime, least_common_multiple, percentage_calculator,
33
+ wikipedia_search, analyze_excel_file, arxiv_search,
34
  audio_transcription, python_code_parser, analyze_csv_file,
35
  # extract_text_from_image,
36
  reverse_sentence, web_content_extract,
 
67
  - String operations or text processing
68
 
69
  3. **Tool Selection and Evaluation**: Before using any tool, systematically evaluate all available options:
70
+ - **Review ALL available tools** in your toolkit before making selecting the most appropiate ones.
71
  - **Match tool capabilities** to the specific requirements of your current step
72
  - **Choose the most appropriate tool** for each task from the complete toolkit
73
  - **Plan multi-tool sequences** - many questions require 2-5 tools in various combinations
 
123
  - **web_search**: Search the web for information
124
  - **web_content_extract**: Extract content from web pages
125
  - **wikipedia_search**: Search Wikipedia for information
126
+ - **arxiv_search**: Search academic papers on arXiv
127
 
128
  ### Utilities
129
  - **convert_temperature**: Convert between temperature units
system-prompt-claude-2.txt CHANGED
@@ -26,7 +26,7 @@ You are an advanced AI agent equipped with multiple tools to solve complex, mult
26
  - String operations or text processing
27
 
28
  3. **Tool Selection and Evaluation**: Before using any tool, systematically evaluate all available options:
29
- - **Review ALL 30 available tools** in your toolkit before making a selection
30
  - **Match tool capabilities** to the specific requirements of your current step
31
  - **Choose the most appropriate tool** for each task from the complete toolkit
32
  - **Plan multi-tool sequences** - many questions require 2-5 tools in various combinations
 
26
  - String operations or text processing
27
 
28
  3. **Tool Selection and Evaluation**: Before using any tool, systematically evaluate all available options:
29
+ - **Review ALL available tools** in your toolkit before making a selection
30
  - **Match tool capabilities** to the specific requirements of your current step
31
  - **Choose the most appropriate tool** for each task from the complete toolkit
32
  - **Plan multi-tool sequences** - many questions require 2-5 tools in various combinations
tools.py CHANGED
@@ -627,7 +627,7 @@ def wikipedia_search(query: str) -> str:
627
  return {"wiki_results": formatted_search_docs}
628
 
629
  @tool
630
- def arvix_search(query: str) -> str:
631
  """
632
  Search Arxiv for a query and return maximum 3 result.
633
 
@@ -640,7 +640,7 @@ def arvix_search(query: str) -> str:
640
  f'<Document source="{doc.metadata["source"]}" page="{doc.metadata.get("page", "")}"/>\n{doc.page_content[:1000]}\n</Document>'
641
  for doc in search_docs
642
  ])
643
- return {"arvix_results": formatted_search_docs}
644
 
645
 
646
  @tool
 
627
  return {"wiki_results": formatted_search_docs}
628
 
629
  @tool
630
+ def arxiv_search(query: str) -> str:
631
  """
632
  Search Arxiv for a query and return maximum 3 result.
633
 
 
640
  f'<Document source="{doc.metadata["source"]}" page="{doc.metadata.get("page", "")}"/>\n{doc.page_content[:1000]}\n</Document>'
641
  for doc in search_docs
642
  ])
643
+ return {"arxiv_results": formatted_search_docs}
644
 
645
 
646
  @tool