Spaces:
Sleeping
Sleeping
Update tool.py
Browse files
tool.py
CHANGED
|
@@ -57,7 +57,7 @@ def DuckDuckGoSearchTool(query: str) -> str:
|
|
| 57 |
Args:
|
| 58 |
query: The search query.
|
| 59 |
"""
|
| 60 |
-
results = DuckDuckGoSearchAPIWrapper().results(query = query, max_results=
|
| 61 |
formatted_search_docs = "\n\n---\n\n".join(
|
| 62 |
[
|
| 63 |
f'<Document source="{res["link"]}" title="{res["title"]}">\n{res["snippet"]}\n</Document>'
|
|
@@ -71,7 +71,7 @@ def WikipediaSearchTool(query: str) -> str:
|
|
| 71 |
|
| 72 |
Args:
|
| 73 |
query: The search query."""
|
| 74 |
-
search_docs = WikipediaLoader(query=query, load_max_docs=
|
| 75 |
formatted_search_docs = "\n\n---\n\n".join(
|
| 76 |
[
|
| 77 |
f'<Document source="{doc.metadata["source"]}" page="{doc.metadata.get("page", "")}"/>\n{doc.page_content}\n</Document>'
|
|
@@ -85,7 +85,7 @@ def ArxivSearchTool(query: str) -> str:
|
|
| 85 |
|
| 86 |
Args:
|
| 87 |
query: The search query."""
|
| 88 |
-
search_docs = ArxivLoader(query=query, load_max_docs=
|
| 89 |
formatted_search_docs = "\n\n---\n\n".join(
|
| 90 |
[
|
| 91 |
f'<Document source="{doc.metadata["source"]}" page="{doc.metadata.get("page", "")}"/>\n{doc.page_content[:1000]}\n</Document>'
|
|
@@ -99,7 +99,7 @@ def PubmedSearchTool(query: str) -> str:
|
|
| 99 |
|
| 100 |
Args:
|
| 101 |
query: The search query."""
|
| 102 |
-
search_docs = PubMedLoader(query=query, load_max_docs=
|
| 103 |
formatted_search_docs = "\n\n---\n\n".join(
|
| 104 |
[
|
| 105 |
f'<Document source="{doc.metadata["uid"]}" title="{doc.metadata["Title"]}"/>\n{doc.page_content[:1000]}\n</Document>'
|
|
|
|
| 57 |
Args:
|
| 58 |
query: The search query.
|
| 59 |
"""
|
| 60 |
+
results = DuckDuckGoSearchAPIWrapper().results(query = query, max_results=5)
|
| 61 |
formatted_search_docs = "\n\n---\n\n".join(
|
| 62 |
[
|
| 63 |
f'<Document source="{res["link"]}" title="{res["title"]}">\n{res["snippet"]}\n</Document>'
|
|
|
|
| 71 |
|
| 72 |
Args:
|
| 73 |
query: The search query."""
|
| 74 |
+
search_docs = WikipediaLoader(query=query, load_max_docs=5).load()
|
| 75 |
formatted_search_docs = "\n\n---\n\n".join(
|
| 76 |
[
|
| 77 |
f'<Document source="{doc.metadata["source"]}" page="{doc.metadata.get("page", "")}"/>\n{doc.page_content}\n</Document>'
|
|
|
|
| 85 |
|
| 86 |
Args:
|
| 87 |
query: The search query."""
|
| 88 |
+
search_docs = ArxivLoader(query=query, load_max_docs=5).load()
|
| 89 |
formatted_search_docs = "\n\n---\n\n".join(
|
| 90 |
[
|
| 91 |
f'<Document source="{doc.metadata["source"]}" page="{doc.metadata.get("page", "")}"/>\n{doc.page_content[:1000]}\n</Document>'
|
|
|
|
| 99 |
|
| 100 |
Args:
|
| 101 |
query: The search query."""
|
| 102 |
+
search_docs = PubMedLoader(query=query, load_max_docs=5).load()
|
| 103 |
formatted_search_docs = "\n\n---\n\n".join(
|
| 104 |
[
|
| 105 |
f'<Document source="{doc.metadata["uid"]}" title="{doc.metadata["Title"]}"/>\n{doc.page_content[:1000]}\n</Document>'
|