From 5648bfa0246f033a178e40b9cd89527b825fe0a9 Mon Sep 17 00:00:00 2001 From: Bulat Kurbanov Date: Wed, 19 Feb 2025 15:29:18 +0100 Subject: [PATCH] Fix --- .../stream_notifications/messages_proc.py | 16 +++++++++++----- 1 file changed, 11 insertions(+), 5 deletions(-) diff --git a/src/modules/stream_notifications/messages_proc.py b/src/modules/stream_notifications/messages_proc.py index 9284995..f550881 100644 --- a/src/modules/stream_notifications/messages_proc.py +++ b/src/modules/stream_notifications/messages_proc.py @@ -96,11 +96,17 @@ async def get_completion(messages: list[dict]) -> str: logger.info(f"Getting completion for message: {messages}") data_messages = [ - { - "role": "assistant" if message["user"] == "kurbezz" else "user", - "content": message["text"] - } - for message in messages + # { + # "role": "system", + # "content": "Don't use markdown! Don't use blocked words on Twitch! Make answers short and clear!" + # }, + *( + { + "role": "assistant" if message["user"] == "kurbezz" else "user", + "content": message["text"] + } + for message in messages + ), ] async with AsyncClient() as client: