diff --git a/core/prompts/dedupe_edges.py b/core/prompts/dedupe_edges.py index 0c750dce..b3b1069a 100644 --- a/core/prompts/dedupe_edges.py +++ b/core/prompts/dedupe_edges.py @@ -6,13 +6,13 @@ class Prompt(Protocol): v1: PromptVersion - v2: PromptVersion + v2: PromptVersion class Versions(TypedDict): v1: PromptFunction v2: PromptFunction - edge_list: PromptFunction + edge_list: PromptFunction def v1(context: dict[str, any]) -> list[Message]: @@ -56,14 +56,14 @@ def v1(context: dict[str, any]) -> list[Message]: def v2(context: dict[str, any]) -> list[Message]: - return [ - Message( - role="system", - content="You are a helpful assistant that de-duplicates relationship from edge lists.", - ), - Message( - role="user", - content=f""" + return [ + Message( + role='system', + content='You are a helpful assistant that de-duplicates relationship from edge lists.', + ), + Message( + role='user', + content=f""" Given the following context, deduplicate edges from a list of new edges given a list of existing edges: Existing Edges: @@ -93,8 +93,8 @@ def v2(context: dict[str, any]) -> list[Message]: ] }} """, - ), - ] + ), + ] def edge_list(context: dict[str, any]) -> list[Message]: @@ -133,4 +133,4 @@ def edge_list(context: dict[str, any]) -> list[Message]: ] -versions: Versions = {"v1": v1, "v2": v2, "edge_list": edge_list} +versions: Versions = {'v1': v1, 'v2': v2, 'edge_list': edge_list}