diff --git a/core/prompts/extract_edge_dates.py b/core/prompts/extract_edge_dates.py index e6ca8727..8af69521 100644 --- a/core/prompts/extract_edge_dates.py +++ b/core/prompts/extract_edge_dates.py @@ -1,4 +1,4 @@ -from typing import Protocol, TypedDict +from typing import Any, Protocol, TypedDict from .models import Message, PromptFunction, PromptVersion @@ -11,7 +11,7 @@ class Versions(TypedDict): v1: PromptFunction -def v1(context: dict[str, any]) -> list[Message]: +def v1(context: dict[str, Any]) -> list[Message]: return [ Message( role='system',