Merge branch 'dev' into feat/cog-1082-metrics-in-networkx-adapter

Author: alekszievr

cognee/api/v1/cognify/cognify_v2.py

@@ -164,7 +164,7 @@ async def get_default_tasks(
                 summarization_model=cognee_config.summarization_model,
                 task_config={"batch_size": 10},
             ),
-            Task(add_data_points, only_root=True, task_config={"batch_size": 10}),
+            Task(add_data_points, task_config={"batch_size": 10}),
             Task(store_descriptive_metrics, include_optional=True),
         ]
     except Exception as error:

cognee/infrastructure/databases/graph/neo4j_driver/adapter.py

@@ -531,7 +531,7 @@ async def get_filtered_graph_data(self, attribute_filters):
 
         return (nodes, edges)
 
-    async def get_graph_metrics(self):
+    async def get_graph_metrics(self, include_optional=False):
         return {
             "num_nodes": -1,
             "num_edges": -1,

cognee/modules/data/methods/store_descriptive_metrics.py

@@ -1,13 +1,11 @@
 from cognee.infrastructure.engine import DataPoint
-from cognee.modules.data.processing.document_types import Document
 from cognee.infrastructure.databases.relational import get_relational_engine
 from sqlalchemy import select
 from sqlalchemy.sql import func
 from cognee.modules.data.models import Data
 from cognee.modules.data.models import GraphMetrics
 import uuid
 from cognee.infrastructure.databases.graph import get_graph_engine
-from cognee.modules.data.methods import create_dataset
 
 
 async def fetch_token_count(db_engine) -> int: