RuaZhou commited on
Commit
1e62b39
·
verified ·
1 Parent(s): 65338b5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -33,12 +33,12 @@ DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
33
  ############### tool
34
  @tool
35
  def arvix_search(query: str) -> str:
36
- """Search Arxiv papers for a query and return maximum 3 result.
37
 
38
  Args:
39
  query: The search query."""
40
 
41
- search_docs = ArxivLoader(query=query, load_max_docs=3).load()
42
 
43
  print(f"Search query arvix: {query}")
44
  print(f"Search results count arvix: {len(search_docs)}, type: {type(search_docs)}")
@@ -239,11 +239,11 @@ def wiki_search(query: str) -> str:
239
 
240
  @tool
241
  def web_search(query: str) -> str:
242
- """Search Tavily on the web for a query and return maximum 3 results.
243
 
244
  Args:
245
  query: The search query."""
246
- search_docs = TavilySearch(max_results=3).invoke({'query': query})
247
 
248
  print(f"Search query: {query}")
249
  print(f"Search results count: {len(search_docs)}")
 
33
  ############### tool
34
  @tool
35
  def arvix_search(query: str) -> str:
36
+ """Search Arxiv papers for a query and return maximum 10 result.
37
 
38
  Args:
39
  query: The search query."""
40
 
41
+ search_docs = ArxivLoader(query=query, load_max_docs=10).load()
42
 
43
  print(f"Search query arvix: {query}")
44
  print(f"Search results count arvix: {len(search_docs)}, type: {type(search_docs)}")
 
239
 
240
  @tool
241
  def web_search(query: str) -> str:
242
+ """Search Tavily on the web for a query and return maximum 10 results.
243
 
244
  Args:
245
  query: The search query."""
246
+ search_docs = TavilySearch(max_results=10).invoke({'query': query})
247
 
248
  print(f"Search query: {query}")
249
  print(f"Search results count: {len(search_docs)}")