kostasang commited on
Commit
23d26db
·
verified ·
1 Parent(s): 8452087

Upload tools.py

Browse files
Files changed (1) hide show
  1. src/tools.py +5 -2
src/tools.py CHANGED
@@ -74,11 +74,14 @@ def wiki_search(query: str) -> str:
74
 
75
  Args:
76
  query: The search query."""
77
- search_docs = WikipediaLoader(query=query, load_max_docs=2).load()
 
 
 
 
78
  formatted_search_docs = [
79
  {
80
  "source": doc.metadata["source"],
81
- "page": doc.metadata.get("page", ""),
82
  "content": doc.page_content,
83
  }
84
  for doc in search_docs
 
74
 
75
  Args:
76
  query: The search query."""
77
+ search_docs = WikipediaLoader(
78
+ query=query,
79
+ load_max_docs=2,
80
+ doc_content_chars_max=50_000
81
+ ).load()
82
  formatted_search_docs = [
83
  {
84
  "source": doc.metadata["source"],
 
85
  "content": doc.page_content,
86
  }
87
  for doc in search_docs