From b9b1fdd1a1ff2b0d9984cb78623fc14886d7566e Mon Sep 17 00:00:00 2001 From: Michael Poluektov Date: Wed, 31 Jul 2024 22:01:22 +0100 Subject: [PATCH] refac: rename message_template --- backend/utils/misc.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/backend/utils/misc.py b/backend/utils/misc.py index adbf4f8b2..c4e2eda6f 100644 --- a/backend/utils/misc.py +++ b/backend/utils/misc.py @@ -88,7 +88,7 @@ def add_or_update_system_message(content: str, messages: List[dict]): return messages -def message_template(model: str): +def openai_chat_message_template(model: str): return { "id": f"{model}-{str(uuid.uuid4())}", "created": int(time.time()), @@ -98,14 +98,14 @@ def message_template(model: str): def openai_chat_chunk_message_template(model: str, message: str): - template = message_template(model) + template = openai_chat_message_template(model) template["object"] = "chat.completion.chunk" template["choices"][0]["delta"] = {"content": message} return template def openai_chat_completion_message_template(model: str, message: str): - template = message_template(model) + template = openai_chat_message_template(model) template["object"] = "chat.completion" template["choices"][0]["message"] = {"content": message, "role": "assistant"} template["choices"][0]["finish_reason"] = "stop"