graphiti-core 0.11.1__tar.gz → 0.11.3__tar.gz

{graphiti_core-0.11.1 → graphiti_core-0.11.3}/PKG-INFO

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

{graphiti_core-0.11.1 → graphiti_core-0.11.3}/graphiti_core/search/search_utils.py

@@ -341,10 +341,10 @@ async def node_fulltext_search(
 
     query = (
         """
-                                    CALL db.index.fulltext.queryNodes("node_name_and_summary", $query, {limit: $limit}) 
-                                    YIELD node AS n, score
-                                    WHERE n:Entity
-                                    """
+                                        CALL db.index.fulltext.queryNodes("node_name_and_summary", $query, {limit: $limit}) 
+                                        YIELD node AS n, score
+                                        WHERE n:Entity
+                                        """
         + filter_query
         + ENTITY_NODE_RETURN
         + """
@@ -676,7 +676,7 @@ async def get_relevant_nodes(
         WHERE score > $min_score
         WITH node, collect(n)[..$limit] AS top_vector_nodes, collect(n.uuid) AS vector_node_uuids
         
-        CALL db.index.fulltext.queryNodes("node_name_and_summary", 'group_id:"' + $group_id + '" AND ' + node.name, {limit: $limit}) 
+        CALL db.index.fulltext.queryNodes("node_name_and_summary", node.fulltext_query, {limit: $limit}) 
         YIELD node AS m
         WHERE m.group_id = $group_id
         WITH node, top_vector_nodes, vector_node_uuids, collect(m) AS fulltext_nodes
@@ -705,18 +705,21 @@ async def get_relevant_nodes(
         """
     )
 
+    query_nodes = [
+        {
+            'uuid': node.uuid,
+            'name': node.name,
+            'name_embedding': node.name_embedding,
+            'fulltext_query': fulltext_query(node.name, [node.group_id]),
+        }
+        for node in nodes
+    ]
+
     results, _, _ = await driver.execute_query(
         query,
         query_params,
-        nodes=[
-            {
-                'uuid': node.uuid,
-                'name': lucene_sanitize(node.name),
-                'name_embedding': node.name_embedding,
-            }
-            for node in nodes
-        ],
-        group_id=lucene_sanitize(group_id),
+        nodes=query_nodes,
+        group_id=group_id,
         limit=limit,
         min_score=min_score,
         database_=DEFAULT_DATABASE,

{graphiti_core-0.11.1 → graphiti_core-0.11.3}/graphiti_core/utils/maintenance/node_operations.py

@@ -140,11 +140,12 @@ async def extract_nodes(
             for entity in missing_entities:
                 custom_prompt += f'\n{entity},'
 
+    filtered_extracted_entities = [entity for entity in extracted_entities if entity.name.strip()]
     end = time()
-    logger.debug(f'Extracted new nodes: {extracted_entities} in {(end - start) * 1000} ms')
+    logger.debug(f'Extracted new nodes: {filtered_extracted_entities} in {(end - start) * 1000} ms')
     # Convert the extracted data into EntityNode objects
     extracted_nodes = []
-    for extracted_entity in extracted_entities:
+    for extracted_entity in filtered_extracted_entities:
         entity_type_name = entity_types_context[extracted_entity.entity_type_id].get(
             'entity_type_name'
         )

{graphiti_core-0.11.1 → graphiti_core-0.11.3}/pyproject.toml

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