From f8ba0334e828923389bced66a1b3d4e34cb26747 Mon Sep 17 00:00:00 2001 From: Michael Poluektov Date: Sun, 4 Aug 2024 12:10:02 +0100 Subject: [PATCH] fix: non streaming functions --- backend/utils/misc.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/backend/utils/misc.py b/backend/utils/misc.py index c4e2eda6f..3aadd3fb9 100644 --- a/backend/utils/misc.py +++ b/backend/utils/misc.py @@ -97,18 +97,19 @@ def openai_chat_message_template(model: str): } -def openai_chat_chunk_message_template(model: str, message: str): +def openai_chat_chunk_message_template(model: str, message: str) -> dict: 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): +def openai_chat_completion_message_template(model: str, message: str) -> dict: template = openai_chat_message_template(model) template["object"] = "chat.completion" template["choices"][0]["message"] = {"content": message, "role": "assistant"} template["choices"][0]["finish_reason"] = "stop" + return template def get_gravatar_url(email):