From b14f6ffa59dfde7abff125db95cf4c332f36a02d Mon Sep 17 00:00:00 2001 From: Lorenze Jay Date: Sun, 15 Dec 2024 22:32:08 -0800 Subject: [PATCH] run_type docs --- src/crewai/knowledge/source/docling_source.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/src/crewai/knowledge/source/docling_source.py b/src/crewai/knowledge/source/docling_source.py index e6daab714..6362d789a 100644 --- a/src/crewai/knowledge/source/docling_source.py +++ b/src/crewai/knowledge/source/docling_source.py @@ -21,6 +21,7 @@ class DoclingSource(BaseFileKnowledgeSource): document_converter: DocumentConverter = Field(default_factory=DocumentConverter) safe_file_paths: List[Union[Path, str]] = Field(default_factory=list) content: List[DoclingDocument] | None = Field(default=None) + chunks: List[str] = Field(default_factory=List) def model_post_init(self, _) -> None: self.safe_file_paths = self._process_file_paths() @@ -61,7 +62,7 @@ class DoclingSource(BaseFileKnowledgeSource): for chunk in chunker.chunk(doc): yield chunk.text - def _process_file_paths(self) -> List[Path | str]: + def _process_file_paths(self) -> list[Path | str]: processed_paths = [] for path in self.file_paths: if path.startswith("http"):