Antoine101 commited on
Commit
dc7594a
·
verified ·
1 Parent(s): f2b8d4b

Update tools.py

Browse files
Files changed (1) hide show
  1. tools.py +6 -2
tools.py CHANGED
@@ -23,8 +23,12 @@ def wiki_loader(query: str, lang: str='en', load_max_docs: int=2):
23
  loader = WikipediaLoader(query=query, lang=lang, load_max_docs=load_max_docs)
24
  # Load the documents
25
  documents = loader.load()
26
- print(documents)
27
- return documents
 
 
 
 
28
  except Exception as e:
29
  print(f"An error occurred: {e}")
30
  return []
 
23
  loader = WikipediaLoader(query=query, lang=lang, load_max_docs=load_max_docs)
24
  # Load the documents
25
  documents = loader.load()
26
+ result = []
27
+ for doc in documents:
28
+ content = doc.page_content
29
+ result.append(content)
30
+ print(result)
31
+ return result
32
  except Exception as e:
33
  print(f"An error occurred: {e}")
34
  return []