graphiti-core 0.11.4__py3-none-any.whl → 0.11.5__py3-none-any.whl

graphiti_core/prompts/dedupe_nodes.py

@@ -77,11 +77,6 @@ def node(context: dict[str, Any]) -> list[Message]:
         duplicate_entity_id should be set to -1.
         
         Also return the most complete name for the entity.
-
-        Guidelines:
-        1. Entities with the same name should be considered duplicates
-        2. Duplicate entities may refer to the same real-world entity even if names differ. Use context clues from the MESSAGES
-            to determine if the NEW ENTITY represents a duplicate entity of one of the EXISTING ENTITIES.
         """,
         ),
     ]

graphiti_core/utils/maintenance/node_operations.py

@@ -34,8 +34,10 @@ from graphiti_core.prompts.extract_nodes import (
     ExtractedEntity,
     MissedEntities,
 )
+from graphiti_core.search.search import search
+from graphiti_core.search.search_config import SearchResults
+from graphiti_core.search.search_config_recipes import NODE_HYBRID_SEARCH_RRF
 from graphiti_core.search.search_filters import SearchFilters
-from graphiti_core.search.search_utils import get_relevant_nodes
 from graphiti_core.utils.datetime_utils import utc_now
 
 logger = logging.getLogger(__name__)
@@ -227,13 +229,23 @@ async def resolve_extracted_nodes(
     entity_types: dict[str, BaseModel] | None = None,
 ) -> tuple[list[EntityNode], dict[str, str]]:
     llm_client = clients.llm_client
-    driver = clients.driver
 
-    # Find relevant nodes already in the graph
-    existing_nodes_lists: list[list[EntityNode]] = await get_relevant_nodes(
-        driver, extracted_nodes, SearchFilters()
+    search_results: list[SearchResults] = await semaphore_gather(
+        *[
+            search(
+                clients=clients,
+                query=node.name,
+                query_vector=node.name_embedding,
+                group_ids=[node.group_id],
+                search_filter=SearchFilters(),
+                config=NODE_HYBRID_SEARCH_RRF,
+            )
+            for node in extracted_nodes
+        ]
     )
 
+    existing_nodes_lists: list[list[EntityNode]] = [result.nodes for result in search_results]
+
     resolved_nodes: list[EntityNode] = await semaphore_gather(
         *[
             resolve_extracted_node(
@@ -282,6 +294,7 @@ async def resolve_extracted_node(
                 'id': i,
                 'name': node.name,
                 'entity_types': node.labels,
+                'summary': node.summary,
             },
             **node.attributes,
         }

{graphiti_core-0.11.4.dist-info → graphiti_core-0.11.5.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: graphiti-core
-Version: 0.11.4
+Version: 0.11.5
 Summary: A temporal graph building library
 License: Apache-2.0
 Author: Paul Paliychuk

{graphiti_core-0.11.4.dist-info → graphiti_core-0.11.5.dist-info}/RECORD

@@ -31,7 +31,7 @@ graphiti_core/models/nodes/node_db_queries.py,sha256=AQgRGVO-GgFWfLq1G6k8s86WItw
 graphiti_core/nodes.py,sha256=_V1KkM31MEVv688ovXlXmBVeibm20d8FEps5jMDPlp0,17774
 graphiti_core/prompts/__init__.py,sha256=EA-x9xUki9l8wnu2l8ek_oNf75-do5tq5hVq7Zbv8Kw,101
 graphiti_core/prompts/dedupe_edges.py,sha256=q60fqIjFQlOzOeL7Y35gwABWQBqKkMarBQBok1pj1C4,3409
-graphiti_core/prompts/dedupe_nodes.py,sha256=viIfdCJ-KwJ72fSd9TMzR9nBQ8G4bV4cgHmPk3-cryQ,4464
+graphiti_core/prompts/dedupe_nodes.py,sha256=hdeBMkSL1Ixt-Pu89PIDCoLvyc0zy2uX3kp4MzY2kzI,4138
 graphiti_core/prompts/eval.py,sha256=gnBQTmwsCl3Qvwpcm7aieVszzo6y1sMCUT8jQiKTvvE,5317
 graphiti_core/prompts/extract_edge_dates.py,sha256=3Drs3CmvP0gJN5BidWSxrNvLet3HPoTybU3BUIAoc0Y,4218
 graphiti_core/prompts/extract_edges.py,sha256=uSoQS32rpUzQJGEhayErMdj72rocGdqVq54Macjf7po,5102
@@ -56,11 +56,11 @@ graphiti_core/utils/maintenance/__init__.py,sha256=vW4H1KyapTl-OOz578uZABYcpND4w
 graphiti_core/utils/maintenance/community_operations.py,sha256=aLMfQ5l8PkNJY3ugqV6UOMG74aI6L9Zz2zFiLBsnKvM,10155
 graphiti_core/utils/maintenance/edge_operations.py,sha256=xfYo49ufpJmGG6sinMxwRwGx83swADoE40m8bp3gC30,14418
 graphiti_core/utils/maintenance/graph_data_operations.py,sha256=BIJKc8tbvU4IjWxLgeotw57b1eE3Iw8YtV74j6eo4RQ,7493
-graphiti_core/utils/maintenance/node_operations.py,sha256=JMHQUO3beoGnSwxI4_BSNCWi-FkbeFJjNyv60HUjPww,15088
+graphiti_core/utils/maintenance/node_operations.py,sha256=9pvAEoVIiT_RpRJYvZpvSobP2_oq-37IdHRHRcmVO1Q,15559
 graphiti_core/utils/maintenance/temporal_operations.py,sha256=mJkw9xLB4W2BsLfC5POr0r-PHWL9SIfNj_l_xu0B5ug,3410
 graphiti_core/utils/maintenance/utils.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 graphiti_core/utils/ontology_utils/entity_types_utils.py,sha256=QJX5cG0GSSNF_Mm_yrldr69wjVAbN_MxLhOSznz85Hk,1279
-graphiti_core-0.11.4.dist-info/LICENSE,sha256=KCUwCyDXuVEgmDWkozHyniRyWjnWUWjkuDHfU6o3JlA,11325
-graphiti_core-0.11.4.dist-info/METADATA,sha256=9RLaTlVrMvQKyqjpbyWLvTPQ514DhFrqVmgo5hZYJPE,15348
-graphiti_core-0.11.4.dist-info/WHEEL,sha256=fGIA9gx4Qxk2KDKeNJCbOEwSrmLtjWCwzBz351GyrPQ,88
-graphiti_core-0.11.4.dist-info/RECORD,,
+graphiti_core-0.11.5.dist-info/LICENSE,sha256=KCUwCyDXuVEgmDWkozHyniRyWjnWUWjkuDHfU6o3JlA,11325
+graphiti_core-0.11.5.dist-info/METADATA,sha256=VW8Ca8H_RpAoQ232EdjNLl6AlMzjWCUkqUY8bXp_k-A,15348
+graphiti_core-0.11.5.dist-info/WHEEL,sha256=fGIA9gx4Qxk2KDKeNJCbOEwSrmLtjWCwzBz351GyrPQ,88
+graphiti_core-0.11.5.dist-info/RECORD,,