Enable different chunking methods

This commit is contained in:
Vasilije 2024-08-08 20:20:25 +02:00
parent 1070a09806
commit 920fc0232c

View file

@ -3,7 +3,7 @@ from cognee.modules.data.processing.document_types.PdfDocument import PdfDocumen
if __name__ == "__main__":
test_file_path = os.path.join(os.path.dirname(__file__), "artificial-inteligence.pdf")
pdf_doc = PdfDocument("Test document.pdf", test_file_path)
pdf_doc = PdfDocument("Test document.pdf", test_file_path, chunking_strategy="paragraph")
reader = pdf_doc.get_reader()
for paragraph_data in reader.read():