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}
|