Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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
|
| 37 |
|
| 38 |
Args:
|
| 39 |
query: The search query."""
|
| 40 |
|
| 41 |
-
search_docs = ArxivLoader(query=query, load_max_docs=
|
| 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
|
| 243 |
|
| 244 |
Args:
|
| 245 |
query: The search query."""
|
| 246 |
-
search_docs = TavilySearch(max_results=
|
| 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)}")
|