ORromu commited on
Commit
34081f5
·
verified ·
1 Parent(s): 99407de

Update tool.py

Browse files
Files changed (1) hide show
  1. tool.py +4 -4
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=3)
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=2).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,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=3).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,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=3).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>'
 
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>'