gmakstutis/graphiti
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

update unit tests

Browse source
This commit is contained in:
prestonrasmussen 2025-09-12 16:48:24 -04:00
parent bdc6fbb540
commit 000c8ef866
1 changed files with 3 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
graphiti_core/nodes.py
View file

@ -111,9 +111,7 @@ class Node(BaseModel, ABC):
uuid=self.uuid, uuid=self.uuid,
) )
edge_uuids: list[str] = [] edge_uuids: list[str] = records[0].get('edge_uuids', []) if records else []
if records and records.get('edge_uuids'):
edge_uuids = records['edge_uuids']
if driver.aoss_client: if driver.aoss_client:
# Delete the node from OpenSearch indices # Delete the node from OpenSearch indices
@ -303,14 +301,14 @@ class Node(BaseModel, ABC):
MATCH (n:Entity|Episodic|Community) MATCH (n:Entity|Episodic|Community)
WHERE n.uuid IN $uuids WHERE n.uuid IN $uuids
MATCH (n)-[r]-() MATCH (n)-[r]-()
RETURN collect(r.uuid) AS edgeUuids RETURN collect(r.uuid) AS edge_uuids
""", """,
uuids=uuids, uuids=uuids,
) )
record = await result.single() record = await result.single()
edge_uuids: list[str] = ( edge_uuids: list[str] = (
record['edgeUuids'] if record and record['edgeUuids'] else [] record['edge_uuids'] if record and record['edge_uuids'] else []
) )
# Now delete the nodes in batches # Now delete the nodes in batches