graphiti-core 0.15.1__py3-none-any.whl → 0.17.0__py3-none-any.whl

graphiti_core/nodes.py

@@ -28,7 +28,7 @@ from typing_extensions import LiteralString
 from graphiti_core.driver.driver import GraphDriver
 from graphiti_core.embedder import EmbedderClient
 from graphiti_core.errors import NodeNotFoundError
-from graphiti_core.helpers import DEFAULT_DATABASE, parse_db_date
+from graphiti_core.helpers import parse_db_date
 from graphiti_core.models.nodes.node_db_queries import (
     COMMUNITY_NODE_SAVE,
     ENTITY_NODE_SAVE,
@@ -103,7 +103,6 @@ class Node(BaseModel, ABC):
         DETACH DELETE n
         """,
             uuid=self.uuid,
-            database_=DEFAULT_DATABASE,
         )
 
         logger.debug(f'Deleted Node: {self.uuid}')
@@ -126,7 +125,6 @@ class Node(BaseModel, ABC):
         DETACH DELETE n
         """,
             group_id=group_id,
-            database_=DEFAULT_DATABASE,
         )
 
         return 'SUCCESS'
@@ -162,7 +160,6 @@ class EpisodicNode(Node):
             created_at=self.created_at,
             valid_at=self.valid_at,
             source=self.source.value,
-            database_=DEFAULT_DATABASE,
         )
 
         logger.debug(f'Saved Node to Graph: {self.uuid}')
@@ -185,7 +182,6 @@ class EpisodicNode(Node):
             e.entity_edges AS entity_edges
         """,
             uuid=uuid,
-            database_=DEFAULT_DATABASE,
             routing_='r',
         )
 
@@ -213,7 +209,6 @@ class EpisodicNode(Node):
             e.entity_edges AS entity_edges
         """,
             uuids=uuids,
-            database_=DEFAULT_DATABASE,
             routing_='r',
         )
 
@@ -254,7 +249,6 @@ class EpisodicNode(Node):
             group_ids=group_ids,
             uuid=uuid_cursor,
             limit=limit,
-            database_=DEFAULT_DATABASE,
             routing_='r',
         )
 
@@ -279,7 +273,6 @@ class EpisodicNode(Node):
             e.entity_edges AS entity_edges
         """,
             entity_node_uuid=entity_node_uuid,
-            database_=DEFAULT_DATABASE,
             routing_='r',
         )
 
@@ -309,9 +302,7 @@ class EntityNode(Node):
             MATCH (n:Entity {uuid: $uuid})
             RETURN n.name_embedding AS name_embedding
         """
-        records, _, _ = await driver.execute_query(
-            query, uuid=self.uuid, database_=DEFAULT_DATABASE, routing_='r'
-        )
+        records, _, _ = await driver.execute_query(query, uuid=self.uuid, routing_='r')
 
         if len(records) == 0:
             raise NodeNotFoundError(self.uuid)
@@ -334,7 +325,6 @@ class EntityNode(Node):
             ENTITY_NODE_SAVE,
             labels=self.labels + ['Entity'],
             entity_data=entity_data,
-            database_=DEFAULT_DATABASE,
         )
 
         logger.debug(f'Saved Node to Graph: {self.uuid}')
@@ -352,7 +342,6 @@ class EntityNode(Node):
         records, _, _ = await driver.execute_query(
             query,
             uuid=uuid,
-            database_=DEFAULT_DATABASE,
             routing_='r',
         )
 
@@ -371,7 +360,6 @@ class EntityNode(Node):
         """
             + ENTITY_NODE_RETURN,
             uuids=uuids,
-            database_=DEFAULT_DATABASE,
             routing_='r',
         )
 
@@ -403,7 +391,6 @@ class EntityNode(Node):
             group_ids=group_ids,
             uuid=uuid_cursor,
             limit=limit,
-            database_=DEFAULT_DATABASE,
             routing_='r',
         )
 
@@ -425,7 +412,6 @@ class CommunityNode(Node):
             summary=self.summary,
             name_embedding=self.name_embedding,
             created_at=self.created_at,
-            database_=DEFAULT_DATABASE,
         )
 
         logger.debug(f'Saved Node to Graph: {self.uuid}')
@@ -446,9 +432,7 @@ class CommunityNode(Node):
             MATCH (c:Community {uuid: $uuid})
             RETURN c.name_embedding AS name_embedding
         """
-        records, _, _ = await driver.execute_query(
-            query, uuid=self.uuid, database_=DEFAULT_DATABASE, routing_='r'
-        )
+        records, _, _ = await driver.execute_query(query, uuid=self.uuid, routing_='r')
 
         if len(records) == 0:
             raise NodeNotFoundError(self.uuid)
@@ -468,7 +452,6 @@ class CommunityNode(Node):
             n.summary AS summary
         """,
             uuid=uuid,
-            database_=DEFAULT_DATABASE,
             routing_='r',
         )
 
@@ -492,7 +475,6 @@ class CommunityNode(Node):
             n.summary AS summary
         """,
             uuids=uuids,
-            database_=DEFAULT_DATABASE,
             routing_='r',
         )
 
@@ -529,7 +511,6 @@ class CommunityNode(Node):
             group_ids=group_ids,
             uuid=uuid_cursor,
             limit=limit,
-            database_=DEFAULT_DATABASE,
             routing_='r',
         )
 

graphiti_core/prompts/dedupe_edges.py

@@ -23,9 +23,9 @@ from .models import Message, PromptFunction, PromptVersion
 
 
 class EdgeDuplicate(BaseModel):
-    duplicate_fact_id: int = Field(
+    duplicate_facts: list[int] = Field(
         ...,
-        description='id of the duplicate fact. If no duplicate facts are found, default to -1.',
+        description='List of ids of any duplicate facts. If no duplicate facts are found, default to empty list.',
     )
     contradicted_facts: list[int] = Field(
         ...,
@@ -75,8 +75,9 @@ def edge(context: dict[str, Any]) -> list[Message]:
         </NEW EDGE>
         
         Task:
-        If the New Edges represents the same factual information as any edge in Existing Edges, return the id of the duplicate fact.
-        If the NEW EDGE is not a duplicate of any of the EXISTING EDGES, return -1.
+        If the New Edges represents the same factual information as any edge in Existing Edges, return the id of the duplicate fact
+            as part of the list of duplicate_facts.
+        If the NEW EDGE is not a duplicate of any of the EXISTING EDGES, return an empty list.
 
         Guidelines:
         1. The facts do not need to be completely identical to be duplicates, they just need to express the same information.

graphiti_core/prompts/dedupe_nodes.py

@@ -32,9 +32,9 @@ class NodeDuplicate(BaseModel):
         ...,
         description='Name of the entity. Should be the most complete and descriptive name of the entity. Do not include any JSON formatting in the Entity name such as {}.',
     )
-    additional_duplicates: list[int] = Field(
+    duplicates: list[int] = Field(
         ...,
-        description='idx of additional duplicate entities. Use this list if the entity has multiple duplicates among existing entities.',
+        description='idx of all duplicate entities.',
     )
 
 
@@ -94,7 +94,7 @@ def node(context: dict[str, Any]) -> list[Message]:
          1. Compare `new_entity` against each item in `existing_entities`.
          2. If it refers to the same real‐world object or concept, collect its index.
          3. Let `duplicate_idx` = the *first* collected index, or –1 if none.
-         4. Let `additional_duplicates` = the list of *any other* collected indices (empty list if none).
+         4. Let `duplicates` = the list of *all* collected indices (empty list if none).
         
         Also return the full name of the NEW ENTITY (whether it is the name of the NEW ENTITY, a node it
         is a duplicate of, or a combination of the two).

graphiti_core/search/search_utils.py

@@ -31,7 +31,6 @@ from graphiti_core.graph_queries import (
     get_vector_cosine_func_query,
 )
 from graphiti_core.helpers import (
-    DEFAULT_DATABASE,
     RUNTIME_QUERY,
     lucene_sanitize,
     normalize_l2,
@@ -116,7 +115,6 @@ async def get_mentioned_nodes(
     records, _, _ = await driver.execute_query(
         query,
         uuids=episode_uuids,
-        database_=DEFAULT_DATABASE,
         routing_='r',
     )
 
@@ -143,7 +141,6 @@ async def get_communities_by_nodes(
     records, _, _ = await driver.execute_query(
         query,
         uuids=node_uuids,
-        database_=DEFAULT_DATABASE,
         routing_='r',
     )
 
@@ -198,7 +195,6 @@ async def edge_fulltext_search(
         query=fuzzy_query,
         group_ids=group_ids,
         limit=limit,
-        database_=DEFAULT_DATABASE,
         routing_='r',
     )
 
@@ -274,7 +270,6 @@ async def edge_similarity_search(
         group_ids=group_ids,
         limit=limit,
         min_score=min_score,
-        database_=DEFAULT_DATABASE,
         routing_='r',
     )
 
@@ -329,7 +324,6 @@ async def edge_bfs_search(
         bfs_origin_node_uuids=bfs_origin_node_uuids,
         depth=bfs_max_depth,
         limit=limit,
-        database_=DEFAULT_DATABASE,
         routing_='r',
     )
 
@@ -371,7 +365,6 @@ async def node_fulltext_search(
         query=fuzzy_query,
         group_ids=group_ids,
         limit=limit,
-        database_=DEFAULT_DATABASE,
         routing_='r',
     )
 
@@ -425,7 +418,6 @@ async def node_similarity_search(
         group_ids=group_ids,
         limit=limit,
         min_score=min_score,
-        database_=DEFAULT_DATABASE,
         routing_='r',
     )
 
@@ -465,7 +457,6 @@ async def node_bfs_search(
         bfs_origin_node_uuids=bfs_origin_node_uuids,
         depth=bfs_max_depth,
         limit=limit,
-        database_=DEFAULT_DATABASE,
         routing_='r',
     )
     nodes = [get_entity_node_from_record(record) for record in records]
@@ -511,7 +502,6 @@ async def episode_fulltext_search(
         query=fuzzy_query,
         group_ids=group_ids,
         limit=limit,
-        database_=DEFAULT_DATABASE,
         routing_='r',
     )
     episodes = [get_episodic_node_from_record(record) for record in records]
@@ -551,7 +541,6 @@ async def community_fulltext_search(
         query=fuzzy_query,
         group_ids=group_ids,
         limit=limit,
-        database_=DEFAULT_DATABASE,
         routing_='r',
     )
     communities = [get_community_node_from_record(record) for record in records]
@@ -603,7 +592,6 @@ async def community_similarity_search(
         group_ids=group_ids,
         limit=limit,
         min_score=min_score,
-        database_=DEFAULT_DATABASE,
         routing_='r',
     )
     communities = [get_community_node_from_record(record) for record in records]
@@ -764,7 +752,6 @@ async def get_relevant_nodes(
         group_id=group_id,
         limit=limit,
         min_score=min_score,
-        database_=DEFAULT_DATABASE,
         routing_='r',
     )
 
@@ -834,7 +821,6 @@ async def get_relevant_edges(
         edges=[edge.model_dump() for edge in edges],
         limit=limit,
         min_score=min_score,
-        database_=DEFAULT_DATABASE,
         routing_='r',
     )
 
@@ -905,7 +891,6 @@ async def get_edge_invalidation_candidates(
         edges=[edge.model_dump() for edge in edges],
         limit=limit,
         min_score=min_score,
-        database_=DEFAULT_DATABASE,
         routing_='r',
     )
     invalidation_edges_dict: dict[str, list[EntityEdge]] = {
@@ -955,7 +940,6 @@ async def node_distance_reranker(
         query,
         node_uuids=filtered_uuids,
         center_uuid=center_node_uuid,
-        database_=DEFAULT_DATABASE,
         routing_='r',
     )
     if driver.provider == 'falkordb':
@@ -997,7 +981,6 @@ async def episode_mentions_reranker(
     results, _, _ = await driver.execute_query(
         query,
         node_uuids=sorted_uuids,
-        database_=DEFAULT_DATABASE,
         routing_='r',
     )
 
@@ -1060,7 +1043,7 @@ async def get_embeddings_for_nodes(
                     """
 
     results, _, _ = await driver.execute_query(
-        query, node_uuids=[node.uuid for node in nodes], database_=DEFAULT_DATABASE, routing_='r'
+        query, node_uuids=[node.uuid for node in nodes], routing_='r'
     )
 
     embeddings_dict: dict[str, list[float]] = {}
@@ -1086,7 +1069,6 @@ async def get_embeddings_for_communities(
     results, _, _ = await driver.execute_query(
         query,
         community_uuids=[community.uuid for community in communities],
-        database_=DEFAULT_DATABASE,
         routing_='r',
     )
 
@@ -1113,7 +1095,6 @@ async def get_embeddings_for_edges(
     results, _, _ = await driver.execute_query(
         query,
         edge_uuids=[edge.uuid for edge in edges],
-        database_=DEFAULT_DATABASE,
         routing_='r',
     )