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

feat: Add directory resolution as step in cognee add function

Browse source 
Added directory resolution as step in cognee add function

Feature COG-656
This commit is contained in:
Igor Ilic 2024-12-11 17:33:51 +01:00
parent d9d90d91ae
commit ff9fd90cf1
1 changed files with 4 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
cognee/api/v1/add/add_v2.py
View file

@ -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)
]