22 lines
633 B
Python
22 lines
633 B
Python
from typing import Optional
|
|
from embedchain.chunkers.base_chunker import BaseChunker
|
|
from embedchain.config.AddConfig import ChunkerConfig
|
|
|
|
from langchain.text_splitter import RecursiveCharacterTextSplitter
|
|
|
|
|
|
TEXT_SPLITTER_CHUNK_PARAMS = {
|
|
"chunk_size": 1000,
|
|
"chunk_overlap": 0,
|
|
"length_function": len,
|
|
}
|
|
|
|
|
|
class PdfFileChunker(BaseChunker):
|
|
''' Chunker for PDF file. '''
|
|
def __init__(self, config: Optional[ChunkerConfig] = None):
|
|
if config is None:
|
|
config = TEXT_SPLITTER_CHUNK_PARAMS
|
|
text_splitter = RecursiveCharacterTextSplitter(**config)
|
|
super().__init__(text_splitter)
|