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

Make save_data_item_with_metadata_to_storage async

Browse source
This commit is contained in:
Leon Luithlen 2024-11-27 09:20:44 +01:00
parent 5b5c1ea5c6
commit 9e93ea0794
1 changed files with 2 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

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

@ -4,12 +4,12 @@ from cognee.modules.ingestion import save_data_to_file
from cognee.modules.ingestion.operations.write_metadata import write_metadata from cognee.modules.ingestion.operations.write_metadata import write_metadata
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. # Dynamic import is used because the llama_index module is optional.
# For the same reason Any is accepted as a data item # For the same reason Any is accepted as a data item
metadata_id = write_metadata(data_item) metadata_id = await write_metadata(data_item)
# Check if data is of type Document or any of it's subclasses # Check if data is of type Document or any of it's subclasses
if str(type(data_item)).startswith("llama_index"): if str(type(data_item)).startswith("llama_index"):