set content to empty after episode extractions have happened (#191)

* set content to empty after episode extractions have happened

* format

* bump version

graphiti_core/graphiti.py

@@ -319,8 +319,6 @@ class Graphiti:
                 valid_at=reference_time,
             )
             episode.uuid = uuid if uuid is not None else episode.uuid
-            if not self.store_raw_episode_content:
-                episode.content = ''
 
             # Extract entities as nodes
 
@@ -442,6 +440,9 @@ class Graphiti:
             episode.entity_edges = [edge.uuid for edge in entity_edges]
 
             # Future optimization would be using batch operations to save nodes and edges
+            if not self.store_raw_episode_content:
+                episode.content = ''
+
             await episode.save(self.driver)
             await asyncio.gather(*[node.save(self.driver) for node in nodes])
             await asyncio.gather(*[edge.save(self.driver) for edge in episodic_edges])

graphiti_core/search/search.py

@@ -53,12 +53,12 @@ logger = logging.getLogger(__name__)
 
 
 async def search(
-        driver: AsyncDriver,
-        embedder: EmbedderClient,
-        query: str,
-        group_ids: list[str] | None,
-        config: SearchConfig,
-        center_node_uuid: str | None = None,
+    driver: AsyncDriver,
+    embedder: EmbedderClient,
+    query: str,
+    group_ids: list[str] | None,
+    config: SearchConfig,
+    center_node_uuid: str | None = None,
 ) -> SearchResults:
     start = time()
     query = query.replace('\n', ' ')
@@ -107,13 +107,13 @@ async def search(
 
 
 async def edge_search(
-        driver: AsyncDriver,
-        embedder: EmbedderClient,
-        query: str,
-        group_ids: list[str] | None,
-        config: EdgeSearchConfig | None,
-        center_node_uuid: str | None = None,
-        limit=DEFAULT_SEARCH_LIMIT,
+    driver: AsyncDriver,
+    embedder: EmbedderClient,
+    query: str,
+    group_ids: list[str] | None,
+    config: EdgeSearchConfig | None,
+    center_node_uuid: str | None = None,
+    limit=DEFAULT_SEARCH_LIMIT,
 ) -> list[EntityEdge]:
     if config is None:
         return []
@@ -176,13 +176,13 @@ async def edge_search(
 
 
 async def node_search(
-        driver: AsyncDriver,
-        embedder: EmbedderClient,
-        query: str,
-        group_ids: list[str] | None,
-        config: NodeSearchConfig | None,
-        center_node_uuid: str | None = None,
-        limit=DEFAULT_SEARCH_LIMIT,
+    driver: AsyncDriver,
+    embedder: EmbedderClient,
+    query: str,
+    group_ids: list[str] | None,
+    config: NodeSearchConfig | None,
+    center_node_uuid: str | None = None,
+    limit=DEFAULT_SEARCH_LIMIT,
 ) -> list[EntityNode]:
     if config is None:
         return []
@@ -230,12 +230,12 @@ async def node_search(
 
 
 async def community_search(
-        driver: AsyncDriver,
-        embedder: EmbedderClient,
-        query: str,
-        group_ids: list[str] | None,
-        config: CommunitySearchConfig | None,
-        limit=DEFAULT_SEARCH_LIMIT,
+    driver: AsyncDriver,
+    embedder: EmbedderClient,
+    query: str,
+    group_ids: list[str] | None,
+    config: CommunitySearchConfig | None,
+    limit=DEFAULT_SEARCH_LIMIT,
 ) -> list[CommunityNode]:
     if config is None:
         return []

pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "graphiti-core"
-version = "0.3.10"
+version = "0.3.11"
 description = "A temporal graph building library"
 authors = [
     "Paul Paliychuk <paul@getzep.com>",