diff --git a/cognee/api/v1/add/add_v2.py b/cognee/api/v1/add/add_v2.py index 631d963e5..9d07f31e4 100644 --- a/cognee/api/v1/add/add_v2.py +++ b/cognee/api/v1/add/add_v2.py @@ -2,7 +2,7 @@ from typing import Union, BinaryIO from cognee.modules.users.models import User from cognee.modules.users.methods import get_default_user from cognee.modules.pipelines import run_tasks, Task -from cognee.tasks.ingestion import ingest_data_with_metadata +from cognee.tasks.ingestion import ingest_data_with_metadata, resolve_data_directories from cognee.infrastructure.databases.relational import create_db_and_tables as create_relational_db_and_tables from cognee.infrastructure.databases.vector.pgvector import create_db_and_tables as create_pgvector_db_and_tables @@ -13,6 +13,9 @@ async def add(data: Union[BinaryIO, list[BinaryIO], str, list[str]], dataset_nam if user is None: user = await get_default_user() + # Resolve all directories from data to files + data = resolve_data_directories(data) + tasks = [ Task(ingest_data_with_metadata, dataset_name, user) ]