feat: add count_nodes and count_edges methods to GraphDBInterface

2025-10-15 16:03:17 +01:00 · 2025-10-15 16:03:17 +01:00 · 8692cd1338
commit 8692cd1338
parent c73e8964a1
1 changed files with 8 additions and 0 deletions
--- a/cognee/infrastructure/databases/graph/graph_db_interface.py
+++ b/cognee/infrastructure/databases/graph/graph_db_interface.py
@ -159,6 +159,14 @@ class GraphDBInterface(ABC):
    - get_connections
    """

+    @abstractmethod
+    async def count_nodes(self) -> int:
+        raise NotImplementedError
+
+    @abstractmethod
+    async def count_edges(self) -> int:
+        raise NotImplementedError
+
    @abstractmethod
    async def query(self, query: str, params: dict) -> List[Any]:
        """