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

Fix and remove falkor

Browse source
This commit is contained in:
vasilije 2025-04-02 21:09:22 +02:00
parent e7a14b9c60
commit 2355d1bfea
2 changed files with 7 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
cognee/tasks/ingestion/ingest_data.py
View file

@ -15,7 +15,9 @@ import inspect
import json
async def ingest_data(data: Any, dataset_name: str, user: User, node_set: Optional[List[str]] = None):
async def ingest_data(
data: Any, dataset_name: str, user: User, node_set: Optional[List[str]] = None
):
destination = get_dlt_destination()
pipeline = dlt.pipeline(
@ -47,7 +49,9 @@ async def ingest_data(data: Any, dataset_name: str, user: User, node_set: Option
"node_set": json.dumps(node_set) if node_set else None,
}
async def store_data_to_dataset(data: Any, dataset_name: str, user: User, node_set: Optional[List[str]] = None):
async def store_data_to_dataset(
data: Any, dataset_name: str, user: User, node_set: Optional[List[str]] = None
):
if not isinstance(data, list):
# Convert data to a list as we work with lists further down.
data = [data]

2
cognee/tasks/node_set/__init__.py
View file

@ -1 +1 @@
from .apply_node_set import apply_node_set
from .apply_node_set import apply_node_set