graphiti-core 0.7.2__tar.gz → 0.7.4__tar.gz

{graphiti_core-0.7.2 → graphiti_core-0.7.4}/PKG-INFO

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

{graphiti_core-0.7.2 → graphiti_core-0.7.4}/graphiti_core/nodes.py

@@ -514,7 +514,7 @@ def get_episodic_node_from_record(record: Any) -> EpisodicNode:
 
 
 def get_entity_node_from_record(record: Any) -> EntityNode:
-    return EntityNode(
+    entity_node = EntityNode(
         uuid=record['uuid'],
         name=record['name'],
         group_id=record['group_id'],
@@ -525,6 +525,15 @@ def get_entity_node_from_record(record: Any) -> EntityNode:
         attributes=record['attributes'],
     )
 
+    del entity_node.attributes['uuid']
+    del entity_node.attributes['name']
+    del entity_node.attributes['group_id']
+    del entity_node.attributes['name_embedding']
+    del entity_node.attributes['summary']
+    del entity_node.attributes['created_at']
+
+    return entity_node
+
 
 def get_community_node_from_record(record: Any) -> CommunityNode:
     return CommunityNode(

{graphiti_core-0.7.2 → graphiti_core-0.7.4}/graphiti_core/search/search_utils.py

@@ -229,8 +229,8 @@ async def edge_similarity_search(
 
     query: LiteralString = (
         """
-                                                                            MATCH (n:Entity)-[r:RELATES_TO]->(m:Entity)
-                                                                            """
+                                                                                MATCH (n:Entity)-[r:RELATES_TO]->(m:Entity)
+                                                                                """
         + group_filter_query
         + filter_query
         + """\nWITH DISTINCT r, vector.similarity.cosine(r.fact_embedding, $search_vector) AS score
@@ -765,8 +765,9 @@ async def node_distance_reranker(
     # rerank on shortest distance
     filtered_uuids.sort(key=lambda cur_uuid: scores[cur_uuid])
 
-    # add back in filtered center uuid
-    filtered_uuids = [center_node_uuid] + filtered_uuids
+    # add back in filtered center uuid if it was filtered out
+    if center_node_uuid in node_uuids:
+        filtered_uuids = [center_node_uuid] + filtered_uuids
 
     return filtered_uuids
 

{graphiti_core-0.7.2 → graphiti_core-0.7.4}/graphiti_core/utils/maintenance/node_operations.py

@@ -172,7 +172,9 @@ async def extract_nodes(
     new_nodes = []
     for name in extracted_node_names:
         entity_type = node_classifications.get(name)
-        labels = ['Entity'] if entity_type is None else ['Entity', entity_type]
+        labels = (
+            ['Entity'] if entity_type is None or entity_type == 'None' else ['Entity', entity_type]
+        )
 
         new_node = EntityNode(
             name=name,

{graphiti_core-0.7.2 → graphiti_core-0.7.4}/pyproject.toml

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