graphiti-core 0.18.5__tar.gz → 0.18.7__tar.gz

{graphiti_core-0.18.5 → graphiti_core-0.18.7}/CONTRIBUTING.md

@@ -36,6 +36,9 @@ Join our [Discord server](https://discord.com/invite/W8Kw6bsgXQ) community and p
 
 ## What happens next?
 
+### Notes for Large Changes
+> Please keep the changes as concise as possible. For major architectural changes (>500 LOC), we would expect a GitHub issue (RFC) discussing the technical design and justification. Otherwise, we will tag it with rfc-required and might not go through the PR.
+
 Once you've found an issue tagged with "good first issue" or "help wanted," or prepared an example to share, here's how to turn that into a contribution:
 
 1. Share your approach in the issue discussion or [Discord](https://discord.com/invite/W8Kw6bsgXQ) before diving deep into code. This helps ensure your solution adheres to the architecture of Graphiti from the start and saves you from potential rework.

{graphiti_core-0.18.5 → graphiti_core-0.18.7}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: graphiti-core
-Version: 0.18.5
+Version: 0.18.7
 Summary: A temporal graph building library
 Project-URL: Homepage, https://help.getzep.com/graphiti/graphiti/overview
 Project-URL: Repository, https://github.com/getzep/graphiti

{graphiti_core-0.18.5 → graphiti_core-0.18.7}/graphiti_core/edges.py

@@ -64,6 +64,21 @@ class Edge(BaseModel, ABC):
 
         return result
 
+    @classmethod
+    async def delete_by_uuids(cls, driver: GraphDriver, uuids: list[str]):
+        result = await driver.execute_query(
+            """
+            MATCH (n)-[e:MENTIONS|RELATES_TO|HAS_MEMBER]->(m)
+            WHERE e.uuid IN $uuids
+            DELETE e
+            """,
+            uuids=uuids,
+        )
+
+        logger.debug(f'Deleted Edges: {uuids}')
+
+        return result
+
     def __hash__(self):
         return hash(self.uuid)
 

{graphiti_core-0.18.5 → graphiti_core-0.18.7}/graphiti_core/graphiti.py

@@ -28,6 +28,7 @@ from graphiti_core.driver.driver import GraphDriver
 from graphiti_core.driver.neo4j_driver import Neo4jDriver
 from graphiti_core.edges import (
     CommunityEdge,
+    Edge,
     EntityEdge,
     EpisodicEdge,
     create_entity_edge_embeddings,
@@ -46,6 +47,7 @@ from graphiti_core.nodes import (
     EntityNode,
     EpisodeType,
     EpisodicNode,
+    Node,
     create_entity_node_embeddings,
 )
 from graphiti_core.search.search import SearchConfig, search
@@ -158,7 +160,7 @@ class Graphiti:
             If not set, the Graphiti default is used.
         ensure_ascii : bool, optional
             Whether to escape non-ASCII characters in JSON serialization for prompts. Defaults to False.
-            Set to False to preserve non-ASCII characters (e.g., Korean, Japanese, Chinese) in their
+            Set as False to preserve non-ASCII characters (e.g., Korean, Japanese, Chinese) in their
             original form, making them readable in LLM logs and improving model understanding.
 
         Returns
@@ -1066,12 +1068,7 @@ class Graphiti:
                 if record['episode_count'] == 1:
                     nodes_to_delete.append(node)
 
-        await semaphore_gather(
-            *[node.delete(self.driver) for node in nodes_to_delete],
-            max_coroutines=self.max_coroutines,
-        )
-        await semaphore_gather(
-            *[edge.delete(self.driver) for edge in edges_to_delete],
-            max_coroutines=self.max_coroutines,
-        )
+        await Node.delete_by_uuids(self.driver, [node.uuid for node in nodes_to_delete])
+
+        await Edge.delete_by_uuids(self.driver, [edge.uuid for edge in edges_to_delete])
         await episode.delete(self.driver)

{graphiti_core-0.18.5 → graphiti_core-0.18.7}/graphiti_core/nodes.py

@@ -142,6 +142,33 @@ class Node(BaseModel, ABC):
                     batch_size=batch_size,
                 )
 
+    @classmethod
+    async def delete_by_uuids(cls, driver: GraphDriver, uuids: list[str], batch_size: int = 100):
+        if driver.provider == GraphProvider.FALKORDB:
+            for label in ['Entity', 'Episodic', 'Community']:
+                await driver.execute_query(
+                    f"""
+                       MATCH (n:{label})
+                       WHERE n.uuid IN $uuids
+                       DETACH DELETE n
+                       """,
+                    uuids=uuids,
+                )
+        else:
+            async with driver.session() as session:
+                await session.run(
+                    """
+                    MATCH (n:Entity|Episodic|Community)
+                    WHERE n.uuid IN $uuids
+                    CALL {
+                        WITH n
+                        DETACH DELETE n
+                    } IN TRANSACTIONS OF $batch_size ROWS
+                    """,
+                    uuids=uuids,
+                    batch_size=batch_size,
+                )
+
     @classmethod
     async def get_by_uuid(cls, driver: GraphDriver, uuid: str): ...
 

{graphiti_core-0.18.5 → graphiti_core-0.18.7}/graphiti_core/search/search.py

@@ -177,31 +177,42 @@ async def edge_search(
 ) -> tuple[list[EntityEdge], list[float]]:
     if config is None:
         return [], []
-    search_results: list[list[EntityEdge]] = list(
-        await semaphore_gather(
-            *[
-                edge_fulltext_search(driver, query, search_filter, group_ids, 2 * limit),
-                edge_similarity_search(
-                    driver,
-                    query_vector,
-                    None,
-                    None,
-                    search_filter,
-                    group_ids,
-                    2 * limit,
-                    config.sim_min_score,
-                ),
-                edge_bfs_search(
-                    driver,
-                    bfs_origin_node_uuids,
-                    config.bfs_max_depth,
-                    search_filter,
-                    group_ids,
-                    2 * limit,
-                ),
-            ]
+
+    # Build search tasks based on configured search methods
+    search_tasks = []
+    if EdgeSearchMethod.bm25 in config.search_methods:
+        search_tasks.append(
+            edge_fulltext_search(driver, query, search_filter, group_ids, 2 * limit)
         )
-    )
+    if EdgeSearchMethod.cosine_similarity in config.search_methods:
+        search_tasks.append(
+            edge_similarity_search(
+                driver,
+                query_vector,
+                None,
+                None,
+                search_filter,
+                group_ids,
+                2 * limit,
+                config.sim_min_score,
+            )
+        )
+    if EdgeSearchMethod.bfs in config.search_methods:
+        search_tasks.append(
+            edge_bfs_search(
+                driver,
+                bfs_origin_node_uuids,
+                config.bfs_max_depth,
+                search_filter,
+                group_ids,
+                2 * limit,
+            )
+        )
+
+    # Execute only the configured search methods
+    search_results: list[list[EntityEdge]] = []
+    if search_tasks:
+        search_results = list(await semaphore_gather(*search_tasks))
 
     if EdgeSearchMethod.bfs in config.search_methods and bfs_origin_node_uuids is None:
         source_node_uuids = [edge.source_node_uuid for result in search_results for edge in result]
@@ -289,24 +300,35 @@ async def node_search(
 ) -> tuple[list[EntityNode], list[float]]:
     if config is None:
         return [], []
-    search_results: list[list[EntityNode]] = list(
-        await semaphore_gather(
-            *[
-                node_fulltext_search(driver, query, search_filter, group_ids, 2 * limit),
-                node_similarity_search(
-                    driver, query_vector, search_filter, group_ids, 2 * limit, config.sim_min_score
-                ),
-                node_bfs_search(
-                    driver,
-                    bfs_origin_node_uuids,
-                    search_filter,
-                    config.bfs_max_depth,
-                    group_ids,
-                    2 * limit,
-                ),
-            ]
+
+    # Build search tasks based on configured search methods
+    search_tasks = []
+    if NodeSearchMethod.bm25 in config.search_methods:
+        search_tasks.append(
+            node_fulltext_search(driver, query, search_filter, group_ids, 2 * limit)
         )
-    )
+    if NodeSearchMethod.cosine_similarity in config.search_methods:
+        search_tasks.append(
+            node_similarity_search(
+                driver, query_vector, search_filter, group_ids, 2 * limit, config.sim_min_score
+            )
+        )
+    if NodeSearchMethod.bfs in config.search_methods:
+        search_tasks.append(
+            node_bfs_search(
+                driver,
+                bfs_origin_node_uuids,
+                search_filter,
+                config.bfs_max_depth,
+                group_ids,
+                2 * limit,
+            )
+        )
+
+    # Execute only the configured search methods
+    search_results: list[list[EntityNode]] = []
+    if search_tasks:
+        search_results = list(await semaphore_gather(*search_tasks))
 
     if NodeSearchMethod.bfs in config.search_methods and bfs_origin_node_uuids is None:
         origin_node_uuids = [node.uuid for result in search_results for node in result]

{graphiti_core-0.18.5 → graphiti_core-0.18.7}/graphiti_core/search/search_filters.py

@@ -28,6 +28,8 @@ class ComparisonOperator(Enum):
     less_than = '<'
     greater_than_equal = '>='
     less_than_equal = '<='
+    is_null = 'IS NULL'
+    is_not_null = 'IS NOT NULL'
 
 
 class DateFilter(BaseModel):
@@ -64,6 +66,19 @@ def node_search_filter_query_constructor(
     return filter_query, filter_params
 
 
+def date_filter_query_constructor(
+    value_name: str, param_name: str, operator: ComparisonOperator
+) -> str:
+    query = '(' + value_name + ' '
+
+    if operator == ComparisonOperator.is_null or operator == ComparisonOperator.is_not_null:
+        query += operator.value + ')'
+    else:
+        query += operator.value + ' ' + param_name + ')'
+
+    return query
+
+
 def edge_search_filter_query_constructor(
     filters: SearchFilters,
 ) -> tuple[str, dict[str, Any]]:
@@ -85,10 +100,16 @@ def edge_search_filter_query_constructor(
         valid_at_filter = '\nAND ('
         for i, or_list in enumerate(filters.valid_at):
             for j, date_filter in enumerate(or_list):
-                filter_params['valid_at_' + str(j)] = date_filter.date
+                if date_filter.comparison_operator not in [
+                    ComparisonOperator.is_null,
+                    ComparisonOperator.is_not_null,
+                ]:
+                    filter_params['valid_at_' + str(j)] = date_filter.date
 
             and_filters = [
-                '(e.valid_at ' + date_filter.comparison_operator.value + f' $valid_at_{j})'
+                date_filter_query_constructor(
+                    'e.valid_at', f'$valid_at_{j}', date_filter.comparison_operator
+                )
                 for j, date_filter in enumerate(or_list)
             ]
             and_filter_query = ''
@@ -110,10 +131,16 @@ def edge_search_filter_query_constructor(
         invalid_at_filter = ' AND ('
         for i, or_list in enumerate(filters.invalid_at):
             for j, date_filter in enumerate(or_list):
-                filter_params['invalid_at_' + str(j)] = date_filter.date
+                if date_filter.comparison_operator not in [
+                    ComparisonOperator.is_null,
+                    ComparisonOperator.is_not_null,
+                ]:
+                    filter_params['invalid_at_' + str(j)] = date_filter.date
 
             and_filters = [
-                '(e.invalid_at ' + date_filter.comparison_operator.value + f' $invalid_at_{j})'
+                date_filter_query_constructor(
+                    'e.invalid_at', f'$invalid_at_{j}', date_filter.comparison_operator
+                )
                 for j, date_filter in enumerate(or_list)
             ]
             and_filter_query = ''
@@ -135,10 +162,16 @@ def edge_search_filter_query_constructor(
         created_at_filter = ' AND ('
         for i, or_list in enumerate(filters.created_at):
             for j, date_filter in enumerate(or_list):
-                filter_params['created_at_' + str(j)] = date_filter.date
+                if date_filter.comparison_operator not in [
+                    ComparisonOperator.is_null,
+                    ComparisonOperator.is_not_null,
+                ]:
+                    filter_params['created_at_' + str(j)] = date_filter.date
 
             and_filters = [
-                '(e.created_at ' + date_filter.comparison_operator.value + f' $created_at_{j})'
+                date_filter_query_constructor(
+                    'e.created_at', f'$created_at_{j}', date_filter.comparison_operator
+                )
                 for j, date_filter in enumerate(or_list)
             ]
             and_filter_query = ''
@@ -160,10 +193,16 @@ def edge_search_filter_query_constructor(
         expired_at_filter = ' AND ('
         for i, or_list in enumerate(filters.expired_at):
             for j, date_filter in enumerate(or_list):
-                filter_params['expired_at_' + str(j)] = date_filter.date
+                if date_filter.comparison_operator not in [
+                    ComparisonOperator.is_null,
+                    ComparisonOperator.is_not_null,
+                ]:
+                    filter_params['expired_at_' + str(j)] = date_filter.date
 
             and_filters = [
-                '(e.expired_at ' + date_filter.comparison_operator.value + f' $expired_at_{j})'
+                date_filter_query_constructor(
+                    'e.expired_at', f'$expired_at_{j}', date_filter.comparison_operator
+                )
                 for j, date_filter in enumerate(or_list)
             ]
             and_filter_query = ''

{graphiti_core-0.18.5 → graphiti_core-0.18.7}/pyproject.toml

@@ -1,7 +1,7 @@
 [project]
 name = "graphiti-core"
 description = "A temporal graph building library"
-version = "0.18.5"
+version = "0.18.7"
 authors = [
     { name = "Paul Paliychuk", email = "paul@getzep.com" },
     { name = "Preston Rasmussen", email = "preston@getzep.com" },

{graphiti_core-0.18.5 → graphiti_core-0.18.7}/signatures/version1/cla.json

@@ -303,6 +303,14 @@
       "created_at": "2025-08-08T07:52:27Z",
       "repoId": 840056306,
       "pullRequestNo": 816
+    },
+    {
+      "name": "CaroLe-prw",
+      "id": 42695882,
+      "comment_id": 3187949734,
+      "created_at": "2025-08-14T10:29:25Z",
+      "repoId": 840056306,
+      "pullRequestNo": 833
     }
   ]
 }

{graphiti_core-0.18.5 → graphiti_core-0.18.7}/tests/test_graphiti_int.py

@@ -65,7 +65,11 @@ async def test_graphiti_init(driver):
 
     search_filter = SearchFilters(
         node_labels=['Person', 'City'],
-        created_at=[[DateFilter(date=utc_now(), comparison_operator=ComparisonOperator.less_than)]],
+        created_at=[
+            [DateFilter(date=None, comparison_operator=ComparisonOperator.is_null)],
+            [DateFilter(date=utc_now(), comparison_operator=ComparisonOperator.less_than)],
+            [DateFilter(date=None, comparison_operator=ComparisonOperator.is_not_null)],
+        ],
     )
 
     results = await graphiti.search_(

{graphiti_core-0.18.5 → graphiti_core-0.18.7}/uv.lock

@@ -1,5 +1,5 @@
 version = 1
-revision = 3
+revision = 2
 requires-python = ">=3.10, <4"
 resolution-markers = [
     "python_full_version >= '3.14'",
@@ -746,7 +746,7 @@ wheels = [
 
 [[package]]
 name = "graphiti-core"
-version = "0.18.5"
+version = "0.18.7"
 source = { editable = "." }
 dependencies = [
     { name = "diskcache" },