KPatelis commited on
Commit
92357df
·
1 Parent(s): d44f842
Files changed (1) hide show
  1. tools.py +3 -3
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"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,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"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,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"Document title: {document["title"]}. Contents: {document["content"]}. Relevance Score: {document["score"]}"
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}