From e6422272295ced626d98578386a2c04d645e8c19 Mon Sep 17 00:00:00 2001 From: prestonrasmussen Date: Fri, 19 Sep 2025 20:04:28 -0400 Subject: [PATCH] embedding fix --- graphiti_core/utils/bulk_utils.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/graphiti_core/utils/bulk_utils.py b/graphiti_core/utils/bulk_utils.py index 78397e87..ae3ad5f4 100644 --- a/graphiti_core/utils/bulk_utils.py +++ b/graphiti_core/utils/bulk_utils.py @@ -195,18 +195,18 @@ async def add_nodes_and_edges_bulk_tx( else: await tx.run(get_episode_node_save_bulk_query(driver.provider), episodes=episodes) await tx.run( - get_entity_node_save_bulk_query(driver.provider, nodes), + get_entity_node_save_bulk_query( + driver.provider, nodes, has_aoss=bool(driver.aoss_client) + ), nodes=nodes, - has_aoss=bool(driver.aoss_client), ) await tx.run( get_episodic_edge_save_bulk_query(driver.provider), episodic_edges=[edge.model_dump() for edge in episodic_edges], ) await tx.run( - get_entity_edge_save_bulk_query(driver.provider), + get_entity_edge_save_bulk_query(driver.provider, has_aoss=bool(driver.aoss_client)), entity_edges=edges, - has_aoss=bool(driver.aoss_client), ) if driver.aoss_client: