dlaima commited on
Commit
593012b
·
verified ·
1 Parent(s): d2b14c9

Update wikipedia_searcher.py

Browse files
Files changed (1) hide show
  1. wikipedia_searcher.py +15 -3
wikipedia_searcher.py CHANGED
@@ -2,12 +2,24 @@ from langchain_community.tools import WikipediaQueryRun
2
  from langchain_community.utilities import WikipediaAPIWrapper
3
 
4
  class WikipediaSearcher:
5
- def __init__(self):
6
- # WikipediaAPIWrapper handles the direct Wikipedia API calls
 
 
 
 
7
  self.wikipedia = WikipediaQueryRun(api_wrapper=WikipediaAPIWrapper())
8
 
9
  def search(self, query: str) -> str:
10
- # Returns a concise summary or article snippet for the query
 
 
 
 
 
 
 
 
11
  try:
12
  result = self.wikipedia.run(query)
13
  return result
 
2
  from langchain_community.utilities import WikipediaAPIWrapper
3
 
4
  class WikipediaSearcher:
5
+ """
6
+ A simple wrapper class to query Wikipedia and retrieve summaries using LangChain's Wikipedia utilities.
7
+ """
8
+
9
+ def __init__(self) -> None:
10
+ # Initialize the Wikipedia query runner with the API wrapper
11
  self.wikipedia = WikipediaQueryRun(api_wrapper=WikipediaAPIWrapper())
12
 
13
  def search(self, query: str) -> str:
14
+ """
15
+ Search Wikipedia for a given query and return a summary or snippet.
16
+
17
+ Args:
18
+ query (str): The search query string.
19
+
20
+ Returns:
21
+ str: The Wikipedia summary or an error message.
22
+ """
23
  try:
24
  result = self.wikipedia.run(query)
25
  return result