From ad384372a76fba6120f7695841b19f44985718fd Mon Sep 17 00:00:00 2001
From: Daniel Chalef <131175+danielchalef@users.noreply.github.com>
Date: Sat, 27 Sep 2025 14:01:47 -0700
Subject: [PATCH] normalize string formatting in dedupe_nodes.py to use single
 quotes

---
 graphiti_core/prompts/dedupe_nodes.py | 34 +++++++++++++--------------
 1 file changed, 16 insertions(+), 18 deletions(-)

diff --git a/graphiti_core/prompts/dedupe_nodes.py b/graphiti_core/prompts/dedupe_nodes.py
index 549e2f4d..31611473 100644
--- a/graphiti_core/prompts/dedupe_nodes.py
+++ b/graphiti_core/prompts/dedupe_nodes.py
@@ -23,25 +23,23 @@ from .prompt_helpers import to_prompt_json
 
 
 class NodeDuplicate(BaseModel):
-    id: int = Field(..., description="integer id of the entity")
+    id: int = Field(..., description='integer id of the entity')
     duplicate_idx: int = Field(
         ...,
-        description="idx of the duplicate entity. If no duplicate entities are found, default to -1.",
+        description='idx of the duplicate entity. If no duplicate entities are found, default to -1.',
     )
     name: str = Field(
         ...,
-        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 {}.",
+        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 {}.',
     )
     duplicates: list[int] = Field(
         ...,
-        description="idx of all entities that are a duplicate of the entity with the above id.",
+        description='idx of all entities that are a duplicate of the entity with the above id.',
     )
 
 
 class NodeResolutions(BaseModel):
-    entity_resolutions: list[NodeDuplicate] = Field(
-        ..., description="List of resolved nodes"
-    )
+    entity_resolutions: list[NodeDuplicate] = Field(..., description='List of resolved nodes')
 
 
 class Prompt(Protocol):
@@ -59,11 +57,11 @@ class Versions(TypedDict):
 def node(context: dict[str, Any]) -> list[Message]:
     return [
         Message(
-            role="system",
-            content="You are a helpful assistant that determines whether or not a NEW ENTITY is a duplicate of any EXISTING ENTITIES.",
+            role='system',
+            content='You are a helpful assistant that determines whether or not a NEW ENTITY is a duplicate of any EXISTING ENTITIES.',
         ),
         Message(
-            role="user",
+            role='user',
             content=f"""
         <PREVIOUS MESSAGES>
         {to_prompt_json([ep for ep in context['previous_episodes']], ensure_ascii=context.get('ensure_ascii', False), indent=2)}
@@ -119,12 +117,12 @@ def node(context: dict[str, Any]) -> list[Message]:
 def nodes(context: dict[str, Any]) -> list[Message]:
     return [
         Message(
-            role="system",
-            content="You are a helpful assistant that determines whether or not ENTITIES extracted from a conversation are duplicates"
-            " of existing entities.",
+            role='system',
+            content='You are a helpful assistant that determines whether or not ENTITIES extracted from a conversation are duplicates'
+            ' of existing entities.',
         ),
         Message(
-            role="user",
+            role='user',
             content=f"""
         <PREVIOUS MESSAGES>
         {to_prompt_json([ep for ep in context['previous_episodes']], ensure_ascii=context.get('ensure_ascii', True), indent=2)}
@@ -189,11 +187,11 @@ def nodes(context: dict[str, Any]) -> list[Message]:
 def node_list(context: dict[str, Any]) -> list[Message]:
     return [
         Message(
-            role="system",
-            content="You are a helpful assistant that de-duplicates nodes from node lists.",
+            role='system',
+            content='You are a helpful assistant that de-duplicates nodes from node lists.',
         ),
         Message(
-            role="user",
+            role='user',
             content=f"""
         Given the following context, deduplicate a list of nodes:
 
@@ -223,4 +221,4 @@ def node_list(context: dict[str, Any]) -> list[Message]:
     ]
 
 
-versions: Versions = {"node": node, "node_list": node_list, "nodes": nodes}
+versions: Versions = {'node': node, 'node_list': node_list, 'nodes': nodes}