Skip to content

Commit

Permalink
feat: Add delete group endpoint
Browse files Browse the repository at this point in the history
  • Loading branch information
paul-paliychuk committed Sep 20, 2024
1 parent 69964d3 commit 94dc013
Show file tree
Hide file tree
Showing 2 changed files with 20 additions and 0 deletions.
6 changes: 6 additions & 0 deletions server/graph_service/routers/ingest.py
Original file line number Diff line number Diff line change
Expand Up @@ -90,6 +90,12 @@ async def delete_entity_edge(uuid: str, graphiti: ZepGraphitiDep):
return Result(message='Entity Edge deleted', success=True)


@router.delete('/group/{group_id}', status_code=status.HTTP_200_OK)
async def delete_group(group_id: str, graphiti: ZepGraphitiDep):
await graphiti.delete_group(group_id)
return Result(message='Group deleted', success=True)


@router.delete('/episode/{uuid}', status_code=status.HTTP_200_OK)
async def delete_episode(uuid: str, graphiti: ZepGraphitiDep):
await graphiti.delete_episodic_node(uuid)
Expand Down
14 changes: 14 additions & 0 deletions server/graph_service/zep_graphiti.py
Original file line number Diff line number Diff line change
Expand Up @@ -33,6 +33,20 @@ async def get_entity_edge(self, uuid: str):
except EdgeNotFoundError as e:
raise HTTPException(status_code=404, detail=e.message) from e

async def delete_group(self, group_id: str):
try:
edges = await EntityEdge.get_by_group_ids(self.driver, [group_id])
nodes = await EntityNode.get_by_group_ids(self.driver, [group_id])
episodes = await EpisodicNode.get_by_group_ids(self.driver, [group_id])
for edge in edges:
await edge.delete(self.driver)
for node in nodes:
await node.delete(self.driver)
for episode in episodes:
await episode.delete(self.driver)
except EdgeNotFoundError as e:
raise HTTPException(status_code=404, detail=e.message) from e

async def delete_entity_edge(self, uuid: str):
try:
edge = await EntityEdge.get_by_uuid(self.driver, uuid)
Expand Down

0 comments on commit 94dc013

Please sign in to comment.