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

fix: Resolve issue with llama index type resolution

Browse source 
Resolve issue with llama index type resolution

Fix
This commit is contained in:
Igor Ilic 2024-12-04 11:29:27 +01:00
parent 61aebf79e0
commit ceebcdb251
1 changed files with 3 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

7
cognee/tasks/ingestion/save_data_item_with_metadata_to_storage.py
View file

@ -7,10 +7,9 @@ from cognee.modules.ingestion import save_data_to_file
async def save_data_item_with_metadata_to_storage( async def save_data_item_with_metadata_to_storage(
data_item: Union[BinaryIO, str, Any], dataset_name: str data_item: Union[BinaryIO, str, Any], dataset_name: str
) -> str: ) -> str:
# Dynamic import is used because the llama_index module is optional.
# For the same reason Any is accepted as a data item if "llama_index" in str(type(data_item)):
# Check if data is of type Document or any of it's subclasses # Dynamic import is used because the llama_index module is optional.
if str(type(data_item)).startswith("llama_index"):
from .transform_data import get_data_from_llama_index from .transform_data import get_data_from_llama_index
file_path = get_data_from_llama_index(data_item, dataset_name) file_path = get_data_from_llama_index(data_item, dataset_name)