From c5ebc9e2e1810d94e03fe5a97f4be7b804cc64f9 Mon Sep 17 00:00:00 2001
From: prestonrasmussen <prasmuss15@gmail.com>
Date: Mon, 8 Sep 2025 14:21:01 -0400
Subject: [PATCH] add episode bulk search results

---
 graphiti_core/graphiti.py | 12 ++++++++++--
 pyproject.toml            |  2 +-
 uv.lock                   |  2 +-
 3 files changed, 12 insertions(+), 4 deletions(-)

diff --git a/graphiti_core/graphiti.py b/graphiti_core/graphiti.py
index ec6b5d89..68dc1d01 100644
--- a/graphiti_core/graphiti.py
+++ b/graphiti_core/graphiti.py
@@ -572,7 +572,6 @@ class Graphiti:
         except Exception as e:
             raise e
 
-    ##### EXPERIMENTAL #####
     async def add_episode_bulk(
         self,
         bulk_episodes: list[RawEpisode],
@@ -581,7 +580,7 @@ class Graphiti:
         excluded_entity_types: list[str] | None = None,
         edge_types: dict[str, type[BaseModel]] | None = None,
         edge_type_map: dict[tuple[str, str], list[str]] | None = None,
-    ):
+    ) -> AddEpisodeResults:
         """
         Process multiple episodes in bulk and update the graph.
 
@@ -849,6 +848,15 @@ class Graphiti:
             end = time()
             logger.info(f'Completed add_episode_bulk in {(end - start) * 1000} ms')
 
+            return AddEpisodeResults(
+                episode=episodes,
+                episodic_edges=resolved_episodic_edges,
+                nodes=final_hydrated_nodes,
+                edges=resolved_edges + invalidated_edges,
+                communities=[],
+                community_edges=[],
+            )
+
         except Exception as e:
             raise e
 
diff --git a/pyproject.toml b/pyproject.toml
index e1e1d9cf..d8672d62 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -1,7 +1,7 @@
 [project]
 name = "graphiti-core"
 description = "A temporal graph building library"
-version = "0.20.2"
+version = "0.20.3"
 authors = [
     { name = "Paul Paliychuk", email = "paul@getzep.com" },
     { name = "Preston Rasmussen", email = "preston@getzep.com" },
diff --git a/uv.lock b/uv.lock
index ad38b5c4..a001c9ea 100644
--- a/uv.lock
+++ b/uv.lock
@@ -783,7 +783,7 @@ wheels = [
 
 [[package]]
 name = "graphiti-core"
-version = "0.20.2"
+version = "0.20.3"
 source = { editable = "." }
 dependencies = [
     { name = "diskcache" },