graphiti-core 0.11.0__py3-none-any.whl → 0.11.2__py3-none-any.whl

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/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.dist-info → graphiti_core-0.11.2.dist-info}/METADATA

@@ -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.dist-info → graphiti_core-0.11.2.dist-info}/RECORD

@@ -48,7 +48,7 @@ graphiti_core/search/search_config.py,sha256=VvKg6AB_RPhoe56DBBXHRBXHThAVJ_OLFCy
 graphiti_core/search/search_config_recipes.py,sha256=4GquRphHhJlpXQhAZOySYnCzBWYoTwxlJj44eTOavZQ,7443
 graphiti_core/search/search_filters.py,sha256=JkP7NbM4Dor27dne5vAuxbJic12dIJDtWJxNqmVuRec,5884
 graphiti_core/search/search_helpers.py,sha256=G5Ceaq5Pfgx0Weelqgeylp_pUHwiBnINaUYsDbURJbE,2636
-graphiti_core/search/search_utils.py,sha256=94WN65cFn9IeMcYqx_BERs3YPVmFfHB2tAxQyQjgeAI,30955
+graphiti_core/search/search_utils.py,sha256=h9pgLwOpFe6rugFNXITxvDQ0qhx6M6xgQ4KW0NLBqes,31099
 graphiti_core/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 graphiti_core/utils/bulk_utils.py,sha256=Z1QzpcopVHxttRmVeYdVDHRjjY7z0cW1wwHB-0yQ6Ng,14820
 graphiti_core/utils/datetime_utils.py,sha256=Ti-2tnrDFRzBsbfblzsHybsM3jaDLP4-VT2t0VhpIzU,1357
@@ -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=3rstKAv85Hshv3jgbUSU8OBTFmYn4BQYA9bd5ZfDURU,14319
 graphiti_core/utils/maintenance/graph_data_operations.py,sha256=BIJKc8tbvU4IjWxLgeotw57b1eE3Iw8YtV74j6eo4RQ,7493
-graphiti_core/utils/maintenance/node_operations.py,sha256=g2HytAgTcI6-EN9i-GnTtpdm6eYr83p33WUVdmfclss,14909
+graphiti_core/utils/maintenance/node_operations.py,sha256=lhNEobEbEX0RrcqNUt9OpakigXNzHpEbf6EG4vQRB0A,14967
 graphiti_core/utils/maintenance/temporal_operations.py,sha256=DBCmKZscR6_GxSv0wBK5PNHoyJfuF4RrrFd6yr02x5s,3311
 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.0.dist-info/LICENSE,sha256=KCUwCyDXuVEgmDWkozHyniRyWjnWUWjkuDHfU6o3JlA,11325
-graphiti_core-0.11.0.dist-info/METADATA,sha256=K5UkaIT3XKjZqfjP5p9i0a4iQZThf--4Y8FjWtUCv4o,15348
-graphiti_core-0.11.0.dist-info/WHEEL,sha256=fGIA9gx4Qxk2KDKeNJCbOEwSrmLtjWCwzBz351GyrPQ,88
-graphiti_core-0.11.0.dist-info/RECORD,,
+graphiti_core-0.11.2.dist-info/LICENSE,sha256=KCUwCyDXuVEgmDWkozHyniRyWjnWUWjkuDHfU6o3JlA,11325
+graphiti_core-0.11.2.dist-info/METADATA,sha256=93_6qBvl0gAPqmBa9jBsHZwX2emFq4CYW8d--WS1pLY,15348
+graphiti_core-0.11.2.dist-info/WHEEL,sha256=fGIA9gx4Qxk2KDKeNJCbOEwSrmLtjWCwzBz351GyrPQ,88
+graphiti_core-0.11.2.dist-info/RECORD,,