gmakstutis/cognee
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Enable different chunking methods

Browse source
This commit is contained in:
Vasilije 2024-08-08 20:12:34 +02:00
parent e80d39167b
commit 7c7b8a319b
2 changed files with 2 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
cognee/modules/data/processing/document_types/AudioDocument.py
View file

@ -11,7 +11,7 @@ from cognee.tasks.chunking.chunking_registry import get_chunking_function
class AudioReader:
id: UUID
file_path: str
chunking_function:callable
chunking_strategy:str
def __init__(self, id: UUID, file_path: str, chunking_strategy:str = "paragraph"):
self.id = id

1
cognee/modules/data/processing/document_types/PdfDocument.py
View file

@ -12,6 +12,7 @@ from .Document import Document
class PdfReader():
id: UUID
file_path: str
chunking_strategy: str
def __init__(self, id: UUID, file_path: str, chunking_strategy:str = "paragraph"):
self.id = id