4329caa17c
Adds a base chunker from which any chunker can inherit. Existing chunkers are refactored to inherit from this base chunker.
17 lines
421 B
Python
17 lines
421 B
Python
from embedchain.chunkers.base_chunker import BaseChunker
|
|
|
|
from langchain.text_splitter import RecursiveCharacterTextSplitter
|
|
|
|
|
|
TEXT_SPLITTER_CHUNK_PARAMS = {
|
|
"chunk_size": 500,
|
|
"chunk_overlap": 0,
|
|
"length_function": len,
|
|
}
|
|
|
|
|
|
class WebsiteChunker(BaseChunker):
|
|
def __init__(self):
|
|
text_splitter = RecursiveCharacterTextSplitter(**TEXT_SPLITTER_CHUNK_PARAMS)
|
|
super().__init__(text_splitter)
|