From 914c8ffcd7a3bcc542fe4e83216ae9c2061a26f2 Mon Sep 17 00:00:00 2001 From: Yannick Stephan Date: Sun, 9 Feb 2025 11:48:27 +0100 Subject: [PATCH] cleaned code --- lightrag/lightrag.py | 13 +++++-------- 1 file changed, 5 insertions(+), 8 deletions(-) diff --git a/lightrag/lightrag.py b/lightrag/lightrag.py index c2f45fe7..2fbba5ac 100644 --- a/lightrag/lightrag.py +++ b/lightrag/lightrag.py @@ -194,8 +194,6 @@ class LightRAG: ], list[dict[str, Any]], ] = chunking_by_token_size - - chunking_func_kwargs: dict = field(default_factory=dict) def __post_init__(self): os.makedirs(self.log_dir, exist_ok=True) @@ -581,12 +579,11 @@ class LightRAG: } for dp in self.chunking_func( doc["content"], - split_by_character=split_by_character, - split_by_character_only=split_by_character_only, - overlap_token_size=self.chunk_overlap_token_size, - max_token_size=self.chunk_token_size, - tiktoken_model=self.tiktoken_model_name, - **self.chunking_func_kwargs, + split_by_character, + split_by_character_only, + self.chunk_overlap_token_size, + self.chunk_token_size, + self.tiktoken_model_name, ) }