From f96ded3abb4de29bda69af38304bae8bb8c4e598 Mon Sep 17 00:00:00 2001 From: Evian Schlenz Date: Wed, 31 Dec 2025 01:48:47 +0100 Subject: [PATCH] Annotate generate_system_prompt --- data/prompting.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/data/prompting.py b/data/prompting.py index f2041e9..ab724b5 100644 --- a/data/prompting.py +++ b/data/prompting.py @@ -1,6 +1,6 @@ import babel.dates -from utils import generate_random_datetime +from utils import Example, generate_random_datetime CURRENT_DATE_PROMPT = { "english": "The current time and date is", @@ -51,7 +51,7 @@ USER_INSTRUCTION_PROMPT = { } -def generate_system_prompt(example: dict, persona: str, language: str, pile_of_system_prompts: dict) -> str: +def generate_system_prompt(example: Example, persona: str, language: str, pile_of_system_prompts: dict[str, str]) -> str: sys_prompt = pile_of_system_prompts[persona] random_datetime = generate_random_datetime() translate_datetime = babel.dates.format_datetime(random_datetime, BABEL_FORMAT[language], locale=BABEL_LOCALE[language])