Add NotImplementedError fallback for GraphOperationsInterface methods (#1154)

* Add NotImplementedError fallback for GraphOperationsInterface methods

Allow GraphOperationsInterface to be used for selective method overriding
by catching NotImplementedError and falling back to default behavior.

This change wraps all GraphOperationsInterface calls in try/except blocks
so that implementations can raise NotImplementedError for methods they
don't need to customize, allowing the built-in default behavior to execute.

Updated methods in nodes.py:
- Node: delete, delete_by_group_id, delete_by_uuids
- EpisodicNode: save, get_by_uuid, get_by_uuids, get_by_group_ids
- EntityNode: load_name_embedding, save, get_by_uuid, get_by_uuids, get_by_group_ids
- CommunityNode: save, get_by_uuid, get_by_uuids, get_by_group_ids
- SagaNode: save, delete, get_by_uuid, get_by_uuids, get_by_group_ids

Updated methods in edges.py:
- Edge: delete, delete_by_uuids
- EpisodicEdge: save, get_by_uuid, get_by_uuids, get_by_group_ids
- EntityEdge: load_fact_embedding, save, get_by_uuid, get_by_uuids, get_by_group_ids
- CommunityEdge: save, get_by_uuid, get_by_uuids, get_by_group_ids
- HasEpisodeEdge: save, delete, get_by_uuid, get_by_uuids, get_by_group_ids
- NextEpisodeEdge: save, delete, get_by_uuid, get_by_uuids, get_by_group_ids

Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com>

* Add missing GraphOperationsInterface method stubs

Add method stubs for all node and edge CRUD operations that were missing
from the interface:

Node methods:
- node_get_by_uuid, node_get_by_uuids, node_get_by_group_ids
- episodic_node_get_by_uuid, episodic_node_get_by_uuids, episodic_node_get_by_group_ids
- community_node_save, community_node_delete, community_node_save_bulk
- community_node_delete_by_group_id, community_node_delete_by_uuids
- community_node_get_by_uuid, community_node_get_by_uuids, community_node_get_by_group_ids
- saga_node_save, saga_node_delete, saga_node_save_bulk
- saga_node_delete_by_group_id, saga_node_delete_by_uuids
- saga_node_get_by_uuid, saga_node_get_by_uuids, saga_node_get_by_group_ids

Edge methods:
- edge_get_by_uuid, edge_get_by_uuids, edge_get_by_group_ids
- episodic_edge_save, episodic_edge_delete, episodic_edge_delete_by_uuids
- episodic_edge_get_by_uuid, episodic_edge_get_by_uuids, episodic_edge_get_by_group_ids
- community_edge_save, community_edge_delete, community_edge_delete_by_uuids
- community_edge_get_by_uuid, community_edge_get_by_uuids, community_edge_get_by_group_ids
- has_episode_edge_save, has_episode_edge_delete, has_episode_edge_save_bulk
- has_episode_edge_delete_by_uuids, has_episode_edge_get_by_uuid
- has_episode_edge_get_by_uuids, has_episode_edge_get_by_group_ids
- next_episode_edge_save, next_episode_edge_delete, next_episode_edge_save_bulk
- next_episode_edge_delete_by_uuids, next_episode_edge_get_by_uuid
- next_episode_edge_get_by_uuids, next_episode_edge_get_by_group_ids

Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com>

* update

---------

Co-authored-by: Claude Opus 4.5 <noreply@anthropic.com>

Author: prasmussen15

graphiti_core/driver/graph_operations/graph_operations.py

@@ -65,6 +65,29 @@ async def node_delete_by_uuids(
     ) -> None:
         raise NotImplementedError
 
+    # -----------------
+    # Node: Read
+    # -----------------
+
+    async def node_get_by_uuid(self, _cls: Any, driver: Any, uuid: str) -> Any:
+        """Retrieve a single node by UUID."""
+        raise NotImplementedError
+
+    async def node_get_by_uuids(self, _cls: Any, driver: Any, uuids: list[str]) -> list[Any]:
+        """Retrieve multiple nodes by UUIDs."""
+        raise NotImplementedError
+
+    async def node_get_by_group_ids(
+        self,
+        _cls: Any,
+        driver: Any,
+        group_ids: list[str],
+        limit: int | None = None,
+        uuid_cursor: str | None = None,
+    ) -> list[Any]:
+        """Retrieve nodes by group IDs with optional pagination."""
+        raise NotImplementedError
+
     # --------------------------
     # Node: Embeddings (load)
     # --------------------------
@@ -138,6 +161,97 @@ async def episodic_node_delete_by_uuids(
     ) -> None:
         raise NotImplementedError
 
+    # -----------------------
+    # EpisodicNode: Read
+    # -----------------------
+
+    async def episodic_node_get_by_uuid(self, _cls: Any, driver: Any, uuid: str) -> Any:
+        """Retrieve a single episodic node by UUID."""
+        raise NotImplementedError
+
+    async def episodic_node_get_by_uuids(
+        self, _cls: Any, driver: Any, uuids: list[str]
+    ) -> list[Any]:
+        """Retrieve multiple episodic nodes by UUIDs."""
+        raise NotImplementedError
+
+    async def episodic_node_get_by_group_ids(
+        self,
+        _cls: Any,
+        driver: Any,
+        group_ids: list[str],
+        limit: int | None = None,
+        uuid_cursor: str | None = None,
+    ) -> list[Any]:
+        """Retrieve episodic nodes by group IDs with optional pagination."""
+        raise NotImplementedError
+
+    # -----------------------
+    # CommunityNode: Save/Delete
+    # -----------------------
+
+    async def community_node_save(self, node: Any, driver: Any) -> None:
+        """Persist (create or update) a single community node."""
+        raise NotImplementedError
+
+    async def community_node_delete(self, node: Any, driver: Any) -> None:
+        raise NotImplementedError
+
+    async def community_node_save_bulk(
+        self,
+        _cls: Any,
+        driver: Any,
+        transaction: Any,
+        nodes: list[Any],
+        batch_size: int = 100,
+    ) -> None:
+        """Persist (create or update) many community nodes in batches."""
+        raise NotImplementedError
+
+    async def community_node_delete_by_group_id(
+        self,
+        _cls: Any,
+        driver: Any,
+        group_id: str,
+        batch_size: int = 100,
+    ) -> None:
+        raise NotImplementedError
+
+    async def community_node_delete_by_uuids(
+        self,
+        _cls: Any,
+        driver: Any,
+        uuids: list[str],
+        group_id: str | None = None,
+        batch_size: int = 100,
+    ) -> None:
+        raise NotImplementedError
+
+    # -----------------------
+    # CommunityNode: Read
+    # -----------------------
+
+    async def community_node_get_by_uuid(self, _cls: Any, driver: Any, uuid: str) -> Any:
+        """Retrieve a single community node by UUID."""
+        raise NotImplementedError
+
+    async def community_node_get_by_uuids(
+        self, _cls: Any, driver: Any, uuids: list[str]
+    ) -> list[Any]:
+        """Retrieve multiple community nodes by UUIDs."""
+        raise NotImplementedError
+
+    async def community_node_get_by_group_ids(
+        self,
+        _cls: Any,
+        driver: Any,
+        group_ids: list[str],
+        limit: int | None = None,
+        uuid_cursor: str | None = None,
+    ) -> list[Any]:
+        """Retrieve community nodes by group IDs with optional pagination."""
+        raise NotImplementedError
+
     # -----------------------
     # SagaNode: Save/Delete
     # -----------------------
@@ -179,6 +293,29 @@ async def saga_node_delete_by_uuids(
     ) -> None:
         raise NotImplementedError
 
+    # -----------------------
+    # SagaNode: Read
+    # -----------------------
+
+    async def saga_node_get_by_uuid(self, _cls: Any, driver: Any, uuid: str) -> Any:
+        """Retrieve a single saga node by UUID."""
+        raise NotImplementedError
+
+    async def saga_node_get_by_uuids(self, _cls: Any, driver: Any, uuids: list[str]) -> list[Any]:
+        """Retrieve multiple saga nodes by UUIDs."""
+        raise NotImplementedError
+
+    async def saga_node_get_by_group_ids(
+        self,
+        _cls: Any,
+        driver: Any,
+        group_ids: list[str],
+        limit: int | None = None,
+        uuid_cursor: str | None = None,
+    ) -> list[Any]:
+        """Retrieve saga nodes by group IDs with optional pagination."""
+        raise NotImplementedError
+
     # -----------------
     # Edge: Save/Delete
     # -----------------
@@ -210,6 +347,29 @@ async def edge_delete_by_uuids(
     ) -> None:
         raise NotImplementedError
 
+    # -----------------
+    # Edge: Read
+    # -----------------
+
+    async def edge_get_by_uuid(self, _cls: Any, driver: Any, uuid: str) -> Any:
+        """Retrieve a single edge by UUID."""
+        raise NotImplementedError
+
+    async def edge_get_by_uuids(self, _cls: Any, driver: Any, uuids: list[str]) -> list[Any]:
+        """Retrieve multiple edges by UUIDs."""
+        raise NotImplementedError
+
+    async def edge_get_by_group_ids(
+        self,
+        _cls: Any,
+        driver: Any,
+        group_ids: list[str],
+        limit: int | None = None,
+        uuid_cursor: str | None = None,
+    ) -> list[Any]:
+        """Retrieve edges by group IDs with optional pagination."""
+        raise NotImplementedError
+
     # -----------------
     # Edge: Embeddings (load)
     # -----------------
@@ -231,6 +391,96 @@ async def edge_load_embeddings_bulk(
         """
         raise NotImplementedError
 
+    # ---------------------------
+    # EpisodicEdge: Save/Delete
+    # ---------------------------
+
+    async def episodic_edge_save(self, edge: Any, driver: Any) -> None:
+        """Persist (create or update) a single episodic edge (MENTIONS)."""
+        raise NotImplementedError
+
+    async def episodic_edge_delete(self, edge: Any, driver: Any) -> None:
+        raise NotImplementedError
+
+    async def episodic_edge_delete_by_uuids(
+        self,
+        _cls: Any,
+        driver: Any,
+        uuids: list[str],
+        group_id: str | None = None,
+    ) -> None:
+        raise NotImplementedError
+
+    # ---------------------------
+    # EpisodicEdge: Read
+    # ---------------------------
+
+    async def episodic_edge_get_by_uuid(self, _cls: Any, driver: Any, uuid: str) -> Any:
+        """Retrieve a single episodic edge by UUID."""
+        raise NotImplementedError
+
+    async def episodic_edge_get_by_uuids(
+        self, _cls: Any, driver: Any, uuids: list[str]
+    ) -> list[Any]:
+        """Retrieve multiple episodic edges by UUIDs."""
+        raise NotImplementedError
+
+    async def episodic_edge_get_by_group_ids(
+        self,
+        _cls: Any,
+        driver: Any,
+        group_ids: list[str],
+        limit: int | None = None,
+        uuid_cursor: str | None = None,
+    ) -> list[Any]:
+        """Retrieve episodic edges by group IDs with optional pagination."""
+        raise NotImplementedError
+
+    # ---------------------------
+    # CommunityEdge: Save/Delete
+    # ---------------------------
+
+    async def community_edge_save(self, edge: Any, driver: Any) -> None:
+        """Persist (create or update) a single community edge (HAS_MEMBER)."""
+        raise NotImplementedError
+
+    async def community_edge_delete(self, edge: Any, driver: Any) -> None:
+        raise NotImplementedError
+
+    async def community_edge_delete_by_uuids(
+        self,
+        _cls: Any,
+        driver: Any,
+        uuids: list[str],
+        group_id: str | None = None,
+    ) -> None:
+        raise NotImplementedError
+
+    # ---------------------------
+    # CommunityEdge: Read
+    # ---------------------------
+
+    async def community_edge_get_by_uuid(self, _cls: Any, driver: Any, uuid: str) -> Any:
+        """Retrieve a single community edge by UUID."""
+        raise NotImplementedError
+
+    async def community_edge_get_by_uuids(
+        self, _cls: Any, driver: Any, uuids: list[str]
+    ) -> list[Any]:
+        """Retrieve multiple community edges by UUIDs."""
+        raise NotImplementedError
+
+    async def community_edge_get_by_group_ids(
+        self,
+        _cls: Any,
+        driver: Any,
+        group_ids: list[str],
+        limit: int | None = None,
+        uuid_cursor: str | None = None,
+    ) -> list[Any]:
+        """Retrieve community edges by group IDs with optional pagination."""
+        raise NotImplementedError
+
     # ---------------------------
     # HasEpisodeEdge: Save/Delete
     # ---------------------------
@@ -262,6 +512,31 @@ async def has_episode_edge_delete_by_uuids(
     ) -> None:
         raise NotImplementedError
 
+    # ---------------------------
+    # HasEpisodeEdge: Read
+    # ---------------------------
+
+    async def has_episode_edge_get_by_uuid(self, _cls: Any, driver: Any, uuid: str) -> Any:
+        """Retrieve a single has_episode edge by UUID."""
+        raise NotImplementedError
+
+    async def has_episode_edge_get_by_uuids(
+        self, _cls: Any, driver: Any, uuids: list[str]
+    ) -> list[Any]:
+        """Retrieve multiple has_episode edges by UUIDs."""
+        raise NotImplementedError
+
+    async def has_episode_edge_get_by_group_ids(
+        self,
+        _cls: Any,
+        driver: Any,
+        group_ids: list[str],
+        limit: int | None = None,
+        uuid_cursor: str | None = None,
+    ) -> list[Any]:
+        """Retrieve has_episode edges by group IDs with optional pagination."""
+        raise NotImplementedError
+
     # ----------------------------
     # NextEpisodeEdge: Save/Delete
     # ----------------------------
@@ -292,3 +567,28 @@ async def next_episode_edge_delete_by_uuids(
         group_id: str | None = None,
     ) -> None:
         raise NotImplementedError
+
+    # ----------------------------
+    # NextEpisodeEdge: Read
+    # ----------------------------
+
+    async def next_episode_edge_get_by_uuid(self, _cls: Any, driver: Any, uuid: str) -> Any:
+        """Retrieve a single next_episode edge by UUID."""
+        raise NotImplementedError
+
+    async def next_episode_edge_get_by_uuids(
+        self, _cls: Any, driver: Any, uuids: list[str]
+    ) -> list[Any]:
+        """Retrieve multiple next_episode edges by UUIDs."""
+        raise NotImplementedError
+
+    async def next_episode_edge_get_by_group_ids(
+        self,
+        _cls: Any,
+        driver: Any,
+        group_ids: list[str],
+        limit: int | None = None,
+        uuid_cursor: str | None = None,
+    ) -> list[Any]:
+        """Retrieve next_episode edges by group IDs with optional pagination."""
+        raise NotImplementedError