notsakeeb commited on
Commit
4b115b9
·
verified ·
1 Parent(s): 52b2b63

Update ToolSet.py

Browse files
Files changed (1) hide show
  1. ToolSet.py +2 -2
ToolSet.py CHANGED
@@ -108,7 +108,7 @@ def get_web_search_result(query: str) -> str:
108
 
109
  @tool
110
  def wiki_search(query: str) -> str:
111
- """Search Wikipedia for a query and return maximum 5 results. Use this tool only if the query specifies Wiki or Wikipedia.
112
  Args:
113
  query: The search query.
114
 
@@ -132,7 +132,7 @@ def arxiv_search(query: str) -> str:
132
  Args:
133
  query: The search query.
134
  Returns:
135
- An array of documents
136
  """
137
  print("arxiv_search")
138
  search_docs = ArxivLoader(query=query, load_max_docs=3).load()
 
108
 
109
  @tool
110
  def wiki_search(query: str) -> str:
111
+ """Search Wikipedia for a query and return results based on that given query.
112
  Args:
113
  query: The search query.
114
 
 
132
  Args:
133
  query: The search query.
134
  Returns:
135
+ An array of documents.
136
  """
137
  print("arxiv_search")
138
  search_docs = ArxivLoader(query=query, load_max_docs=3).load()