andre/crewAI
1
0
Fork 0
mirror of https://github.com/crewAIInc/crewAI.git synced 2026-01-10 00:28:31 +00:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

fixed logic

Browse Source
This commit is contained in:
Lorenze Jay
2024-12-15 22:48:45 -08:00
parent 610ea40c2d
commit f1c9caa8ec
1 changed files with 4 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
src/crewai/knowledge/source/docling_source.py
View File

@@ -39,11 +39,12 @@ class DoclingSource(BaseFileKnowledgeSource):
)
self.content = self.load_content()
def load_content(self):
def load_content(self) -> List[DoclingDocument] | None:
try:
self.content = self.convert_source_to_docling_documents()
return self.convert_source_to_docling_documents()
except Exception as e:
self._logger.log("error", f"Error loading content: {e}")
return None
def add(self) -> None:
if self.content is None:
@@ -76,7 +77,7 @@ class DoclingSource(BaseFileKnowledgeSource):
except Exception as e:
raise ValueError(f"Invalid URL: {path}. Error: {str(e)}")
else:
local_path = Path(KNOWLEDGE_DIRECTORY).joinpath(path)
local_path = Path(KNOWLEDGE_DIRECTORY + "/" + path)
if local_path.exists():
processed_paths.append(local_path)
else: