Spaces:
Sleeping
Sleeping
tools fix
Browse files
tools.py
CHANGED
|
@@ -50,7 +50,7 @@ def wiki_search(query: str) -> str:
|
|
| 50 |
documents = WikipediaLoader(query=query, load_max_docs=3).load()
|
| 51 |
processed_documents = "\n\n---\n\n".join(
|
| 52 |
[
|
| 53 |
-
f
|
| 54 |
for document in documents
|
| 55 |
])
|
| 56 |
return {"wiki_results": processed_documents}
|
|
@@ -64,7 +64,7 @@ def arxiv_search(query: str) -> str:
|
|
| 64 |
documents = ArxivLoader(query=query, load_max_docs=3).load()
|
| 65 |
processed_documents = "\n\n---\n\n".join(
|
| 66 |
[
|
| 67 |
-
f
|
| 68 |
for document in documents
|
| 69 |
])
|
| 70 |
return {"arxiv_results": processed_documents}
|
|
@@ -79,7 +79,7 @@ def tavily_web_search(query: str) -> str:
|
|
| 79 |
search_documents = search_engine.invoke(input=query)
|
| 80 |
web_results = "\n\n---\n\n".join(
|
| 81 |
[
|
| 82 |
-
f
|
| 83 |
for document in search_documents
|
| 84 |
])
|
| 85 |
return {"web_results": web_results}
|
|
|
|
| 50 |
documents = WikipediaLoader(query=query, load_max_docs=3).load()
|
| 51 |
processed_documents = "\n\n---\n\n".join(
|
| 52 |
[
|
| 53 |
+
f'Document title: {document.metadata.get("title", "")}. Summary: {document.metadata.get("summary", "")}. Documents details: {document.page_content}'
|
| 54 |
for document in documents
|
| 55 |
])
|
| 56 |
return {"wiki_results": processed_documents}
|
|
|
|
| 64 |
documents = ArxivLoader(query=query, load_max_docs=3).load()
|
| 65 |
processed_documents = "\n\n---\n\n".join(
|
| 66 |
[
|
| 67 |
+
f'Document title: {document.metadata.get("title", "")}. Summary: {document.metadata.get("summary", "")}. Documents details: {document.page_content}'
|
| 68 |
for document in documents
|
| 69 |
])
|
| 70 |
return {"arxiv_results": processed_documents}
|
|
|
|
| 79 |
search_documents = search_engine.invoke(input=query)
|
| 80 |
web_results = "\n\n---\n\n".join(
|
| 81 |
[
|
| 82 |
+
f'Document title: {document["title"]}. Contents: {document["content"]}. Relevance Score: {document["score"]'
|
| 83 |
for document in search_documents
|
| 84 |
])
|
| 85 |
return {"web_results": web_results}
|