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

fix: Fix docling import so other executions don't fail

Browse source
This commit is contained in:
Andrej Milicevic 2025-09-29 17:57:58 +02:00
parent 8ef3bf6393
commit d352807a9d
1 changed files with 10 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

14
cognee/tasks/ingestion/save_data_item_to_storage.py
View file

@ -27,11 +27,17 @@ async def save_data_item_to_storage(data_item: Union[BinaryIO, str, Any]) -> str
return await get_data_from_llama_index(data_item)
from docling_core.types import DoclingDocument
try:
from docling_core.types import DoclingDocument
if isinstance(data_item, DoclingDocument):
# Convert DoclingDocument to plain text and continue processing file
data_item = data_item.export_to_text()
if isinstance(data_item, DoclingDocument):
# Convert DoclingDocument to plain text and continue processing file
data_item = data_item.export_to_text()
except ImportError:
logger.debug(
"Docling import failed. Docling optional dependency not installed, so we don't expect Docling documents. "
"Continuing normal operation."
)
# data is a file object coming from upload.
if hasattr(data_item, "file"):