update the tests

SylphAI-Inc · Jun 30, 2024 · 80596b1 · 80596b1
1 parent b655780
commit 80596b1
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/lightrag/components/data_process/__init__.py b/lightrag/components/data_process/__init__.py
@@ -1,11 +1,11 @@
 """Components here are used for data processing/transformation."""
 
-from .document_splitter import DocumentSplitter
+from .text_splitter import TextSplitter
 from .data_components import ToEmbeddings, RetrieverOutputToContextStr
 from lightrag.utils.registry import EntityMapping
 
 
-__all__ = ["DocumentSplitter", "ToEmbeddings", "RetrieverOutputToContextStr"]
+__all__ = ["TextSplitter", "ToEmbeddings", "RetrieverOutputToContextStr"]
 
 for name in __all__:
     EntityMapping.register(name, globals()[name])