Graph metrics getter template in graph db interface and adapters

cognee/infrastructure/databases/graph/graph_db_interface.py

@@ -54,3 +54,7 @@ async def delete_graph(
     @abstractmethod
     async def get_graph_data(self):
         raise NotImplementedError
+
+    @abstractmethod
+    async def get_graph_metrics(self):
+        raise NotImplementedError

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

@@ -522,3 +522,17 @@ async def get_filtered_graph_data(self, attribute_filters):
         ]
 
         return (nodes, edges)
+
+    async def get_graph_metrics(self):
+        return {
+            "num_nodes": -1,
+            "num_edges": -1,
+            "mean_degree": -1,
+            "edge_density": -1,
+            "num_connected_components": -1,
+            "sizes_of_connected_components": -1,
+            "num_selfloops": -1,
+            "diameter": -1,
+            "avg_shortest_path_length": -1,
+            "avg_clustering": -1,
+        }

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

@@ -385,3 +385,17 @@ async def get_filtered_graph_data(
         ]
 
         return filtered_nodes, filtered_edges
+
+    async def get_graph_metrics(self):
+        return {
+            "num_nodes": -1,
+            "num_edges": -1,
+            "mean_degree": -1,
+            "edge_density": -1,
+            "num_connected_components": -1,
+            "sizes_of_connected_components": -1,
+            "num_selfloops": -1,
+            "diameter": -1,
+            "avg_shortest_path_length": -1,
+            "avg_clustering": -1,
+        }

cognee/tasks/storage/descriptive_metrics.py

@@ -24,22 +24,12 @@ async def fetch_token_count(db_engine) -> int:
     return token_count_sum
 
 
-async def calculate_graph_metrics(graph_data):
-    nodes, edges = graph_data
-    graph_metrics = {
-        "num_nodes": len(nodes),
-        "num_edges": len(edges),
-    }
-    return graph_metrics
-
-
 async def store_descriptive_metrics(data_points: list[DataPoint]):
     db_engine = get_relational_engine()
     graph_engine = await get_graph_engine()
-    graph_data = await graph_engine.get_graph_data()
 
     token_count_sum = await fetch_token_count(db_engine)
-    graph_metrics = await calculate_graph_metrics(graph_data)
+    graph_metrics = await graph_engine.get_graph_metrics()
 
     table_name = "graph_metrics_table"
     metrics_dict = {"id": uuid.uuid4(), "num_tokens": token_count_sum} | graph_metrics