diff --git a/cognee/infrastructure/databases/graph/graph_db_interface.py b/cognee/infrastructure/databases/graph/graph_db_interface.py index 65afdf275..abfdff784 100644 --- 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]: """