update chinese_text_splitter.py

2023-05-21 21:24:25 +08:00 · 2023-05-21 21:24:25 +08:00 · f986b756ff
parent 14295392d0
commit f986b756ff
1 changed files with 1 additions and 1 deletions
--- a/textsplitter/chinese_text_splitter.py
+++ b/textsplitter/chinese_text_splitter.py
@ -5,7 +5,7 @@ from configs.model_config import SENTENCE_SIZE


 class ChineseTextSplitter(CharacterTextSplitter):
-    def __init__(self, pdf: bool = False, sentence_size: int = None, **kwargs):
+    def __init__(self, pdf: bool = False, sentence_size: int = SENTENCE_SIZE, **kwargs):
        super().__init__(**kwargs)
        self.pdf = pdf
        self.sentence_size = sentence_size