minor fixes

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

@@ -571,7 +571,7 @@ async def project_entire_graph(self, graph_name="myGraph"):
         await self.query(query)
 
     async def drop_graph(self, graph_name="myGraph"):
-        if self.graph_exists(graph_name):
+        if await self.graph_exists(graph_name):
             drop_query = f"CALL gds.graph.drop('{graph_name}');"
             await self.query(drop_query)
 
@@ -656,10 +656,13 @@ async def _get_avg_clustering():
             logging.warning("Average clustering calculation is not implemented for neo4j.")
             return -1
 
+        num_nodes = len(nodes[0]["nodes"])
+        num_edges = len(edges[0]["elements"])
+
         mandatory_metrics = {
-            "num_nodes": len(nodes[0]["nodes"]),
-            "num_edges": len(edges[0]["elements"]),
-            "mean_degree": await _get_mean_degree(),
+            "num_nodes": num_nodes,
+            "num_edges": num_edges,
+            "mean_degree": (2 * num_edges) / num_nodes if num_nodes != 0 else None,
             "edge_density": await _get_edge_density(),
             "num_connected_components": await _get_num_connected_components(),
             "sizes_of_connected_components": await _get_size_of_connected_components(),

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

@@ -415,9 +415,9 @@ def _get_avg_shortest_path_length(graph):
 
         def _get_avg_clustering(graph):
             try:
-                return nx.average_clustering(graph.to_undirected())
+                return nx.average_clustering(nx.DiGraph(graph))
             except Exception as e:
-                print(f"Unexpected error in clustering calculation: {e}")
+                logger.warning("Failed to calculate clustering coefficient: %s", e)
                 return None
 
         mandatory_metrics = {