missing import added (#463)

run-llama · Aug 13, 2023 · a4289c0 · a4289c0
1 parent fd892a6
commit a4289c0
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/llama_hub/tools/wikipedia/base.py b/llama_hub/tools/wikipedia/base.py
@@ -44,6 +44,7 @@ def search_data(self, query: str, lang: str = "en") -> List[Document]:
         Args:
             query (str): the string to search for
         """
+        import wikipedia
         pages = wikipedia.search(query)
         if len(pages) == 0:
             return "Unable to find any details on this search"