From 2355d1bfea5bcfd923ef31d4557108a802bdf33e Mon Sep 17 00:00:00 2001 From: vasilije Date: Wed, 2 Apr 2025 21:09:22 +0200 Subject: [PATCH] Fix and remove falkor --- cognee/tasks/ingestion/ingest_data.py | 8 ++++++-- cognee/tasks/node_set/__init__.py | 2 +- 2 files changed, 7 insertions(+), 3 deletions(-) diff --git a/cognee/tasks/ingestion/ingest_data.py b/cognee/tasks/ingestion/ingest_data.py index 15af1ce2e..8708217d8 100644 --- a/cognee/tasks/ingestion/ingest_data.py +++ b/cognee/tasks/ingestion/ingest_data.py @@ -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] diff --git a/cognee/tasks/node_set/__init__.py b/cognee/tasks/node_set/__init__.py index e0582a8d5..4ad30c370 100644 --- a/cognee/tasks/node_set/__init__.py +++ b/cognee/tasks/node_set/__init__.py @@ -1 +1 @@ -from .apply_node_set import apply_node_set \ No newline at end of file +from .apply_node_set import apply_node_set