Skip to content

Commit

Permalink
fix formatting
Browse files Browse the repository at this point in the history
  • Loading branch information
danielchalef committed Aug 22, 2024
1 parent 80612c4 commit 03ad8cf
Showing 1 changed file with 13 additions and 13 deletions.
26 changes: 13 additions & 13 deletions core/prompts/dedupe_edges.py
Original file line number Diff line number Diff line change
Expand Up @@ -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]:
Expand Down Expand Up @@ -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:
Expand Down Expand Up @@ -93,8 +93,8 @@ def v2(context: dict[str, any]) -> list[Message]:
]
}}
""",
),
]
),
]


def edge_list(context: dict[str, any]) -> list[Message]:
Expand Down Expand Up @@ -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}

0 comments on commit 03ad8cf

Please sign in to comment.