605219f8c7
* feat: Add real world dates extraction * fix: Linter * fix: 💄 mypy errors * chore: handle invalid dates returned by the llm * chore: Polish prompt * reformat * style: 💄 reformat
15 lines
298 B
Python
15 lines
298 B
Python
from typing import Any, Callable, Protocol
|
|
|
|
from pydantic import BaseModel
|
|
|
|
|
|
class Message(BaseModel):
|
|
role: str
|
|
content: str
|
|
|
|
|
|
class PromptVersion(Protocol):
|
|
def __call__(self, context: dict[str, Any]) -> list[Message]: ...
|
|
|
|
|
|
PromptFunction = Callable[[dict[str, Any]], list[Message]]
|