diff --git a/graphiti_core/graphiti.py b/graphiti_core/graphiti.py index 676f28ea..9f9ec988 100644 --- a/graphiti_core/graphiti.py +++ b/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]) diff --git a/graphiti_core/search/search.py b/graphiti_core/search/search.py index 3b9a47cf..d6661d24 100644 --- a/graphiti_core/search/search.py +++ b/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 [] diff --git a/pyproject.toml b/pyproject.toml index c6161e24..088310d1 100644 --- a/pyproject.toml +++ b/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 ",