chore: improve not found error handling

getzep · Sep 20, 2024 · cc9eb9f · cc9eb9f
1 parent 62b3f5f
commit cc9eb9f
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/server/graph_service/zep_graphiti.py b/server/graph_service/zep_graphiti.py
@@ -46,6 +46,8 @@ async def delete_group(self, group_id: str):
                 await episode.delete(self.driver)
         except EdgeNotFoundError as e:
             raise HTTPException(status_code=404, detail=e.message) from e
+        except NodeNotFoundError as e:
+            raise HTTPException(status_code=404, detail=e.message) from e
 
     async def delete_entity_edge(self, uuid: str):
         try: