graphiti-core 0.4.0__tar.gz → 0.4.2__tar.gz

{graphiti_core-0.4.0 → graphiti_core-0.4.2}/PKG-INFO

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

{graphiti_core-0.4.0 → graphiti_core-0.4.2}/graphiti_core/edges.py

@@ -103,6 +103,7 @@ class EpisodicEdge(Edge):
         """,
             uuid=uuid,
             database_=DEFAULT_DATABASE,
+            routing_='r',
         )
 
         edges = [get_episodic_edge_from_record(record) for record in records]
@@ -126,6 +127,7 @@ class EpisodicEdge(Edge):
         """,
             uuids=uuids,
             database_=DEFAULT_DATABASE,
+            routing_='r',
         )
 
         edges = [get_episodic_edge_from_record(record) for record in records]
@@ -149,6 +151,7 @@ class EpisodicEdge(Edge):
         """,
             group_ids=group_ids,
             database_=DEFAULT_DATABASE,
+            routing_='r',
         )
 
         edges = [get_episodic_edge_from_record(record) for record in records]
@@ -230,6 +233,7 @@ class EntityEdge(Edge):
         """,
             uuid=uuid,
             database_=DEFAULT_DATABASE,
+            routing_='r',
         )
 
         edges = [get_entity_edge_from_record(record) for record in records]
@@ -260,6 +264,7 @@ class EntityEdge(Edge):
         """,
             uuids=uuids,
             database_=DEFAULT_DATABASE,
+            routing_='r',
         )
 
         edges = [get_entity_edge_from_record(record) for record in records]
@@ -290,6 +295,7 @@ class EntityEdge(Edge):
         """,
             group_ids=group_ids,
             database_=DEFAULT_DATABASE,
+            routing_='r',
         )
 
         edges = [get_entity_edge_from_record(record) for record in records]
@@ -329,6 +335,7 @@ class CommunityEdge(Edge):
         """,
             uuid=uuid,
             database_=DEFAULT_DATABASE,
+            routing_='r',
         )
 
         edges = [get_community_edge_from_record(record) for record in records]
@@ -350,6 +357,7 @@ class CommunityEdge(Edge):
         """,
             uuids=uuids,
             database_=DEFAULT_DATABASE,
+            routing_='r',
         )
 
         edges = [get_community_edge_from_record(record) for record in records]
@@ -371,6 +379,7 @@ class CommunityEdge(Edge):
         """,
             group_ids=group_ids,
             database_=DEFAULT_DATABASE,
+            routing_='r',
         )
 
         edges = [get_community_edge_from_record(record) for record in records]

{graphiti_core-0.4.0 → graphiti_core-0.4.2}/graphiti_core/nodes.py

@@ -86,7 +86,7 @@ class Node(BaseModel, ABC):
     async def delete(self, driver: AsyncDriver):
         result = await driver.execute_query(
             """
-        MATCH (n {uuid: $uuid})
+        MATCH (n:Entity|Episodic|Community {uuid: $uuid})
         DETACH DELETE n
         """,
             uuid=self.uuid,
@@ -105,6 +105,19 @@ class Node(BaseModel, ABC):
             return self.uuid == other.uuid
         return False
 
+    @classmethod
+    async def delete_by_group_id(cls, driver: AsyncDriver, group_id: str):
+        await driver.execute_query(
+            """
+        MATCH (n:Entity|Episodic|Community {group_id: $group_id})
+        DETACH DELETE n
+        """,
+            group_id=group_id,
+            database_=DEFAULT_DATABASE,
+        )
+
+        return 'SUCCESS'
+
     @classmethod
     async def get_by_uuid(cls, driver: AsyncDriver, uuid: str): ...
 
@@ -159,6 +172,7 @@ class EpisodicNode(Node):
         """,
             uuid=uuid,
             database_=DEFAULT_DATABASE,
+            routing_='r',
         )
 
         episodes = [get_episodic_node_from_record(record) for record in records]
@@ -185,6 +199,7 @@ class EpisodicNode(Node):
         """,
             uuids=uuids,
             database_=DEFAULT_DATABASE,
+            routing_='r',
         )
 
         episodes = [get_episodic_node_from_record(record) for record in records]
@@ -208,6 +223,7 @@ class EpisodicNode(Node):
         """,
             group_ids=group_ids,
             database_=DEFAULT_DATABASE,
+            routing_='r',
         )
 
         episodes = [get_episodic_node_from_record(record) for record in records]
@@ -259,6 +275,7 @@ class EntityNode(Node):
         """,
             uuid=uuid,
             database_=DEFAULT_DATABASE,
+            routing_='r',
         )
 
         nodes = [get_entity_node_from_record(record) for record in records]
@@ -283,6 +300,7 @@ class EntityNode(Node):
         """,
             uuids=uuids,
             database_=DEFAULT_DATABASE,
+            routing_='r',
         )
 
         nodes = [get_entity_node_from_record(record) for record in records]
@@ -304,6 +322,7 @@ class EntityNode(Node):
         """,
             group_ids=group_ids,
             database_=DEFAULT_DATABASE,
+            routing_='r',
         )
 
         nodes = [get_entity_node_from_record(record) for record in records]
@@ -355,6 +374,7 @@ class CommunityNode(Node):
         """,
             uuid=uuid,
             database_=DEFAULT_DATABASE,
+            routing_='r',
         )
 
         nodes = [get_community_node_from_record(record) for record in records]
@@ -379,6 +399,7 @@ class CommunityNode(Node):
         """,
             uuids=uuids,
             database_=DEFAULT_DATABASE,
+            routing_='r',
         )
 
         communities = [get_community_node_from_record(record) for record in records]
@@ -400,6 +421,7 @@ class CommunityNode(Node):
         """,
             group_ids=group_ids,
             database_=DEFAULT_DATABASE,
+            routing_='r',
         )
 
         communities = [get_community_node_from_record(record) for record in records]

{graphiti_core-0.4.0 → graphiti_core-0.4.2}/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-0.4.0 → graphiti_core-0.4.2}/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 → graphiti_core-0.4.2}/pyproject.toml

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