Some type annotation additions

Signed-off-by: Marenz <github@supradigital.org>
This commit is contained in:
Marenz
2024-07-18 17:53:47 +02:00
committed by Mathias L. Baumann
parent 3a32c66923
commit 9776bdda0d
2 changed files with 4 additions and 3 deletions

View File

@@ -62,7 +62,7 @@ def pop_system_message(messages: List[dict]) -> Tuple[dict, List[dict]]:
return get_system_message(messages), remove_system_message(messages)
def add_or_update_system_message(content: str, messages: List[dict]):
def add_or_update_system_message(content: str, messages: List[dict]) -> List[dict]:
"""
Adds a new system message at the beginning of the messages list
or updates the existing system message at the beginning.