graphiti-core 0.11.0__tar.gz → 0.11.2__tar.gz

{graphiti_core-0.11.0 → graphiti_core-0.11.2}/PKG-INFO

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

{graphiti_core-0.11.0 → graphiti_core-0.11.2}/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,13 +705,20 @@ 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': node.name, 'name_embedding': node.name_embedding}
-            for node in nodes
-        ],
+        nodes=query_nodes,
         group_id=group_id,
         limit=limit,
         min_score=min_score,

{graphiti_core-0.11.0 → graphiti_core-0.11.2}/graphiti_core/utils/maintenance/node_operations.py

@@ -102,6 +102,7 @@ async def extract_nodes(
         'previous_episodes': [ep.content for ep in previous_episodes],
         'custom_prompt': custom_prompt,
         'entity_types': entity_types_context,
+        'source_description': episode.source_description,
     }
 
     while entities_missed and reflexion_iterations <= MAX_REFLEXION_ITERATIONS:

{graphiti_core-0.11.0 → graphiti_core-0.11.2}/pyproject.toml

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