From ff9fd90cf1bf4083925be991b22512e3bcf06474 Mon Sep 17 00:00:00 2001 From: Igor Ilic Date: Wed, 11 Dec 2024 17:33:51 +0100 Subject: [PATCH] feat: Add directory resolution as step in cognee add function Added directory resolution as step in cognee add function Feature COG-656 --- cognee/api/v1/add/add_v2.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) 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) ]