Merge pull request #101 from chonkie-ai/development

[fix] Add LateChunker support to chunker and module exports
chonkie-ai · Dec 23, 2024 · faa1778 · faa1778
2 parents c2aa9e3 + 4be8308
commit faa1778
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/src/chonkie/__init__.py b/src/chonkie/__init__.py
@@ -7,6 +7,7 @@
     SentenceChunker,
     TokenChunker,
     WordChunker,
+    LateChunker,
 )
 from .embeddings import (
     AutoEmbeddings,
@@ -26,6 +27,7 @@
     SemanticSentence,
     Sentence,
     SentenceChunk,
+    LateChunk,
 )
 
 __version__ = "0.3.0"
@@ -47,6 +49,7 @@
     "SemanticChunk",
     "Sentence",
     "SemanticSentence",
+    "LateChunk",
 ]
 
 # Add all chunker classes to __all__
@@ -57,6 +60,7 @@
     "SentenceChunker",
     "SemanticChunker",
     "SDPMChunker",
+    "LateChunker",
 ]
 
 # Add all embeddings classes to __all__

diff --git a/src/chonkie/chunker/__init__.py b/src/chonkie/chunker/__init__.py
@@ -6,6 +6,7 @@
 from .sentence import SentenceChunker
 from .token import TokenChunker
 from .word import WordChunker
+from .late import LateChunker
 
 __all__ = [
     "BaseChunker",
@@ -14,4 +15,5 @@
     "SentenceChunker",
     "SemanticChunker",
     "SDPMChunker",
+    "LateChunker",
 ]