update returns

graphiti_core/driver/graph_operations/graph_operations.py

@@ -80,7 +80,7 @@ class GraphOperationsInterface(BaseModel):
         driver: Any,
         nodes: list[Any],
         batch_size: int = 100,
-    ) -> None:
+    ) -> dict[str, list[float]]:
         """
         Load embedding vectors for many nodes in batches. Mutates the provided node instances.
         """
@@ -184,7 +184,7 @@ class GraphOperationsInterface(BaseModel):
         driver: Any,
         edges: list[Any],
         batch_size: int = 100,
-    ) -> None:
+    ) -> dict[str, list[float]]:
         """
         Load embedding vectors for many edges in batches. Mutates the provided edge instances.
         """

graphiti_core/search/search_utils.py

@@ -1880,7 +1880,7 @@ async def get_embeddings_for_nodes(
     driver: GraphDriver, nodes: list[EntityNode]
 ) -> dict[str, list[float]]:
     if driver.graph_operations_interface:
-        await driver.graph_operations_interface.node_load_embeddings_bulk(driver, nodes)
+        return await driver.graph_operations_interface.node_load_embeddings_bulk(driver, nodes)
     elif driver.provider == GraphProvider.NEPTUNE:
         query = """
         MATCH (n:Entity)
@@ -1952,7 +1952,7 @@ async def get_embeddings_for_edges(
     driver: GraphDriver, edges: list[EntityEdge]
 ) -> dict[str, list[float]]:
     if driver.graph_operations_interface:
-        await driver.graph_operations_interface.edge_load_embeddings_bulk(driver, edges)
+        return await driver.graph_operations_interface.edge_load_embeddings_bulk(driver, edges)
     elif driver.provider == GraphProvider.NEPTUNE:
         query = """
         MATCH (n:Entity)-[e:RELATES_TO]-(m:Entity)

uv.lock

@@ -783,7 +783,7 @@ wheels = [
 
 [[package]]
 name = "graphiti-core"
-version = "0.22.1rc1"
+version = "0.22.1rc2"
 source = { editable = "." }
 dependencies = [
     { name = "diskcache" },