graphiti-core 0.7.5__py3-none-any.whl → 0.7.7__py3-none-any.whl

graphiti_core/prompts/dedupe_nodes.py

@@ -64,7 +64,7 @@ def node(context: dict[str, Any]) -> list[Message]:
         {json.dumps(context['existing_nodes'], indent=2)}
         </EXISTING NODES>
         
-        Given the above EXISTING NODES, MESSAGE, and PREVIOUS MESSAGES. Determine if the NEW NODE extracted from the conversation
+        Given the above EXISTING NODES and their attributes, MESSAGE, and PREVIOUS MESSAGES. Determine if the NEW NODE extracted from the conversation
         is a duplicate entity of one of the EXISTING NODES.
 
         <NEW NODE>

graphiti_core/search/search_filters.py

@@ -55,7 +55,7 @@ def node_search_filter_query_constructor(
     filter_params: dict[str, Any] = {}
 
     if filters.node_labels is not None:
-        node_labels = ':'.join(filters.node_labels)
+        node_labels = '|'.join(filters.node_labels)
         node_label_filter = ' AND n:' + node_labels
         filter_query += node_label_filter
 

graphiti_core/utils/maintenance/node_operations.py

@@ -158,13 +158,17 @@ async def extract_nodes(
     node_classifications: dict[str, str | None] = {}
 
     if entity_types is not None:
-        llm_response = await llm_client.generate_response(
-            prompt_library.extract_nodes.classify_nodes(node_classification_context),
-            response_model=EntityClassification,
-        )
-        entities = llm_response.get('entities', [])
-        entity_classifications = llm_response.get('entity_classifications', [])
-        node_classifications.update(dict(zip(entities, entity_classifications)))
+        try:
+            llm_response = await llm_client.generate_response(
+                prompt_library.extract_nodes.classify_nodes(node_classification_context),
+                response_model=EntityClassification,
+            )
+            entities = llm_response.get('entities', [])
+            entity_classifications = llm_response.get('entity_classifications', [])
+            node_classifications.update(dict(zip(entities, entity_classifications)))
+        # catch classification errors and continue if we can't classify
+        except Exception as e:
+            logger.exception(e)
 
     end = time()
     logger.debug(f'Extracted new nodes: {extracted_node_names} in {(end - start) * 1000} ms')
@@ -285,7 +289,10 @@ async def resolve_extracted_node(
     start = time()
 
     # Prepare context for LLM
-    existing_nodes_context = [{'uuid': node.uuid, 'name': node.name} for node in existing_nodes]
+    existing_nodes_context = [
+        {'uuid': node.uuid, 'name': node.name, 'attributes': node.attributes}
+        for node in existing_nodes
+    ]
 
     extracted_node_context = {
         'uuid': extracted_node.uuid,
@@ -362,6 +369,13 @@ async def resolve_extracted_node(
             node = existing_node
             node.name = name
             node.summary = summary_response.get('summary', '')
+
+            new_attributes = existing_node.attributes
+            existing_attributes = existing_node.attributes
+            for attribute_name, attribute_value in existing_attributes.items():
+                if new_attributes.get(attribute_name) is None:
+                    new_attributes[attribute_name] = attribute_value
+
             uuid_map[extracted_node.uuid] = existing_node.uuid
 
     end = time()

{graphiti_core-0.7.5.dist-info → graphiti_core-0.7.7.dist-info}/METADATA

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

{graphiti_core-0.7.5.dist-info → graphiti_core-0.7.7.dist-info}/RECORD

@@ -28,7 +28,7 @@ graphiti_core/models/nodes/node_db_queries.py,sha256=f4_UT6XL8UDt4_CO9YIHeI8pvpw
 graphiti_core/nodes.py,sha256=0lH8SOpnzTtNIvG4ScnJ3SeQudviCsZwsnM867kY1aI,16998
 graphiti_core/prompts/__init__.py,sha256=EA-x9xUki9l8wnu2l8ek_oNf75-do5tq5hVq7Zbv8Kw,101
 graphiti_core/prompts/dedupe_edges.py,sha256=EuX8ngeItBzrlMBOgeHrpExzxIFHD2aoDyaX1ZniF6I,3556
-graphiti_core/prompts/dedupe_nodes.py,sha256=mqvNATL-4Vo33vaxUEZfOq6hXXOiL-ftY0zcx2G-82I,4624
+graphiti_core/prompts/dedupe_nodes.py,sha256=1MHYuqi-eXPnTQd_jEcVKfljSC_F_r5SdVRm_dpdOiI,4645
 graphiti_core/prompts/eval.py,sha256=csW494kKBMvWSm2SYLIRuGgNghhwNR3YwGn3veo3g_Y,3691
 graphiti_core/prompts/extract_edge_dates.py,sha256=td2yx2wnX-nLioMa0mtla3WcRyO71_wSjemT79YZGQ0,4096
 graphiti_core/prompts/extract_edges.py,sha256=vyEdW7JAPOT_eLWUi6nRmxbvucyVoyoYX2SxXfknRUg,3467
@@ -43,7 +43,7 @@ graphiti_core/search/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hS
 graphiti_core/search/search.py,sha256=DX-tcIa0SiKI2HX-b_WdjGE74A8RLWQor4p90dJluUA,12643
 graphiti_core/search/search_config.py,sha256=UZN8jFA4pBlw2O5N1cuhVRBdTwMLR9N3Oyo6sQ4MDVw,3117
 graphiti_core/search/search_config_recipes.py,sha256=yUqiLnn9vFg39M8eVwjVKfBCL_ptGrfDMQ47m_Blb0g,6885
-graphiti_core/search/search_filters.py,sha256=4MJmCXD-blMc71xB4F9K4a72qidDCigADQ_ztdG15kc,5884
+graphiti_core/search/search_filters.py,sha256=JkP7NbM4Dor27dne5vAuxbJic12dIJDtWJxNqmVuRec,5884
 graphiti_core/search/search_utils.py,sha256=Bywp7trqP_gEwqH8I-JHuj3Mljw9P2K6_XooXI75jHI,25739
 graphiti_core/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 graphiti_core/utils/bulk_utils.py,sha256=P4LKO46Yle4tBdNcQ3hDHcSQFaR8UBLfoL-z1M2Wua0,14690
@@ -52,10 +52,10 @@ graphiti_core/utils/maintenance/__init__.py,sha256=TRY3wWWu5kn3Oahk_KKhltrWnh0NA
 graphiti_core/utils/maintenance/community_operations.py,sha256=gIw1M5HGgc2c3TXag5ygPPpAv5WsG-yoC8Lhmfr6FMs,10011
 graphiti_core/utils/maintenance/edge_operations.py,sha256=tNw56vN586JYZMgie6RLRTiHZ680-kWzDIxW8ucL6SU,12780
 graphiti_core/utils/maintenance/graph_data_operations.py,sha256=qds9ALk9PhpQs1CNZTZGpi70mqJ93Y2KhIh9X2r8MUI,6533
-graphiti_core/utils/maintenance/node_operations.py,sha256=ty7_U9SNu4O7mgJgNvj-8THfoYGDDlKOo54EYHOC1wA,14292
+graphiti_core/utils/maintenance/node_operations.py,sha256=aNnk8EZOPnoyRRkH48m4KIlKLsv7jfz4xdyKVBspoaA,14848
 graphiti_core/utils/maintenance/temporal_operations.py,sha256=RdNtubCyYhOVrvcOIq2WppHls1Q-BEjtsN8r38l-Rtc,3691
 graphiti_core/utils/maintenance/utils.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-graphiti_core-0.7.5.dist-info/LICENSE,sha256=KCUwCyDXuVEgmDWkozHyniRyWjnWUWjkuDHfU6o3JlA,11325
-graphiti_core-0.7.5.dist-info/METADATA,sha256=jJE1xBlbtBNv7FfIMpATqP6dnPTinuJwRrn8ppt2JqY,10541
-graphiti_core-0.7.5.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-graphiti_core-0.7.5.dist-info/RECORD,,
+graphiti_core-0.7.7.dist-info/LICENSE,sha256=KCUwCyDXuVEgmDWkozHyniRyWjnWUWjkuDHfU6o3JlA,11325
+graphiti_core-0.7.7.dist-info/METADATA,sha256=m9rSGSFLTsfTkSldAP05q9voPu0_8YBksGtrCABKYOY,10541
+graphiti_core-0.7.7.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+graphiti_core-0.7.7.dist-info/RECORD,,