fix: update legacy llama-index Document!

TogetherCrew · May 23, 2024 · 6525673 · 6525673
1 parent ba715a0
commit 6525673
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/dags/hivemind_etl_helpers/src/db/mediawiki/mediawiki_reader.py b/dags/hivemind_etl_helpers/src/db/mediawiki/mediawiki_reader.py
@@ -4,7 +4,7 @@
 
 import wikipedia
 from llama_index.legacy.readers.base import BasePydanticReader
-from llama_index.legacy.schema import Document
+from llama_index.core import Document
 
 
 class MediaWikiReader(BasePydanticReader):