graphiti-core 0.4.0__py3-none-any.whl → 0.4.1__py3-none-any.whl

graphiti_core/search/search.py

@@ -143,7 +143,7 @@ async def edge_search(
     search_results: list[list[EntityEdge]] = list(
         await asyncio.gather(
             *[
-                edge_fulltext_search(driver, query, None, None, group_ids, 2 * limit),
+                edge_fulltext_search(driver, query, group_ids, 2 * limit),
                 edge_similarity_search(
                     driver, query_vector, None, None, group_ids, 2 * limit, config.sim_min_score
                 ),

graphiti_core/search/search_utils.py

@@ -135,8 +135,6 @@ async def get_communities_by_nodes(
 async def edge_fulltext_search(
     driver: AsyncDriver,
     query: str,
-    source_node_uuid: str | None,
-    target_node_uuid: str | None,
     group_ids: list[str] | None = None,
     limit=RELEVANT_SCHEMA_LIMIT,
 ) -> list[EntityEdge]:
@@ -147,10 +145,8 @@ async def edge_fulltext_search(
 
     cypher_query = Query("""
               CALL db.index.fulltext.queryRelationships("edge_name_and_fact", $query, {limit: $limit}) 
-              YIELD relationship AS rel, score
-              MATCH (n:Entity)-[r {uuid: rel.uuid}]->(m:Entity)
-              WHERE ($source_uuid IS NULL OR n.uuid IN [$source_uuid, $target_uuid])
-              AND ($target_uuid IS NULL OR m.uuid IN [$source_uuid, $target_uuid])
+              YIELD relationship AS r, score
+              WITH r, score, startNode(r) AS n, endNode(r) AS m
               RETURN
                     r.uuid AS uuid,
                     r.group_id AS group_id,
@@ -170,8 +166,6 @@ async def edge_fulltext_search(
     records, _, _ = await driver.execute_query(
         cypher_query,
         query=fuzzy_query,
-        source_uuid=source_node_uuid,
-        target_uuid=target_node_uuid,
         group_ids=group_ids,
         limit=limit,
         database_=DEFAULT_DATABASE,
@@ -600,13 +594,7 @@ async def get_relevant_edges(
             )
             for edge in edges
             if edge.fact_embedding is not None
-        ],
-        *[
-            edge_fulltext_search(
-                driver, edge.fact, source_node_uuid, target_node_uuid, [edge.group_id], limit
-            )
-            for edge in edges
-        ],
+        ]
     )
 
     for result in results:

{graphiti_core-0.4.0.dist-info → graphiti_core-0.4.1.dist-info}/METADATA

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

{graphiti_core-0.4.0.dist-info → graphiti_core-0.4.1.dist-info}/RECORD

@@ -38,10 +38,10 @@ graphiti_core/prompts/models.py,sha256=cvx_Bv5RMFUD_5IUawYrbpOKLPHogai7_bm7YXrSz
 graphiti_core/prompts/summarize_nodes.py,sha256=lNpagCq6Pz71vipBonQZUxBZG6OY4bQbyGZ5UEgdL4s,3488
 graphiti_core/py.typed,sha256=vlmmzQOt7bmeQl9L3XJP4W6Ry0iiELepnOrinKz5KQg,79
 graphiti_core/search/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-graphiti_core/search/search.py,sha256=qkiAZCYhKhTkzyGuSH9j_W2Kz60TGyhLM1hZNCaCeUs,11996
+graphiti_core/search/search.py,sha256=5r7HpIR4cQvwH24YN_DJB6LVNN9hizviE0QXGM6swz8,11984
 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_utils.py,sha256=8Ro90L7fqtw5tsdusIhS_pdYcLX2JVIDiH5dD04G7ug,22968
+graphiti_core/search/search_utils.py,sha256=C9qzdRMP2WFW7O0CivoGHYyEviaCZEGDu4HAAXonxzo,22459
 graphiti_core/utils/__init__.py,sha256=cJAcMnBZdHBQmWrZdU1PQ1YmaL75bhVUkyVpIPuOyns,260
 graphiti_core/utils/bulk_utils.py,sha256=AMi42deCXjIIgdpLEiD5SYCMZfX8XKoOsVYEwXVTX5A,14016
 graphiti_core/utils/maintenance/__init__.py,sha256=TRY3wWWu5kn3Oahk_KKhltrWnh0NACw0FskjqF6OtlA,314
@@ -51,7 +51,7 @@ graphiti_core/utils/maintenance/graph_data_operations.py,sha256=w66_SLlvPapuG91Y
 graphiti_core/utils/maintenance/node_operations.py,sha256=m0q-PPCI5LM0dZMkrcyUENw6WfDBMO61DQ7S6MuNQ8E,11624
 graphiti_core/utils/maintenance/temporal_operations.py,sha256=MvaRLWrBlDeYw8CQrKish1xbYcY5ovpfdqA2hSX7v5k,3367
 graphiti_core/utils/maintenance/utils.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-graphiti_core-0.4.0.dist-info/LICENSE,sha256=KCUwCyDXuVEgmDWkozHyniRyWjnWUWjkuDHfU6o3JlA,11325
-graphiti_core-0.4.0.dist-info/METADATA,sha256=URZ0gW5i_Q9aOe5M60tsxFsYbAwqoW7HJc8oVyrzuFk,10058
-graphiti_core-0.4.0.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-graphiti_core-0.4.0.dist-info/RECORD,,
+graphiti_core-0.4.1.dist-info/LICENSE,sha256=KCUwCyDXuVEgmDWkozHyniRyWjnWUWjkuDHfU6o3JlA,11325
+graphiti_core-0.4.1.dist-info/METADATA,sha256=ujFYwG2A62hiztEWo4aXP92Bc-QAgCFwmmn0BoAGquc,10058
+graphiti_core-0.4.1.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+graphiti_core-0.4.1.dist-info/RECORD,,