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

embedding fix

Browse source
This commit is contained in:
prestonrasmussen 2025-09-19 20:04:28 -04:00
parent 3ea6f9f9a8
commit e642227229
1 changed files with 4 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
graphiti_core/utils/bulk_utils.py
View file

@ -195,18 +195,18 @@ async def add_nodes_and_edges_bulk_tx(
else: else:
await tx.run(get_episode_node_save_bulk_query(driver.provider), episodes=episodes) await tx.run(get_episode_node_save_bulk_query(driver.provider), episodes=episodes)
await tx.run( 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, nodes=nodes,
has_aoss=bool(driver.aoss_client),
) )
await tx.run( await tx.run(
get_episodic_edge_save_bulk_query(driver.provider), get_episodic_edge_save_bulk_query(driver.provider),
episodic_edges=[edge.model_dump() for edge in episodic_edges], episodic_edges=[edge.model_dump() for edge in episodic_edges],
) )
await tx.run( 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, entity_edges=edges,
has_aoss=bool(driver.aoss_client),
) )
if driver.aoss_client: if driver.aoss_client: