UniqAI/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui synced 2025-02-11 15:33:38 +00:00
Code Issues Actions 4 Packages Projects Releases Wiki Activity

refac: rename message_template

Browse Source
This commit is contained in:
Michael Poluektov 2024-07-31 22:01:22 +01:00
parent 2e0fa1c6a0
commit b9b1fdd1a1
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
backend/utils/misc.py
View File

@ -88,7 +88,7 @@ def add_or_update_system_message(content: str, messages: List[dict]):
return messages return messages
def message_template(model: str): def openai_chat_message_template(model: str):
return { return {
"id": f"{model}-{str(uuid.uuid4())}", "id": f"{model}-{str(uuid.uuid4())}",
"created": int(time.time()), "created": int(time.time()),
@ -98,14 +98,14 @@ def message_template(model: str):
def openai_chat_chunk_message_template(model: str, message: 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["object"] = "chat.completion.chunk"
template["choices"][0]["delta"] = {"content": message} template["choices"][0]["delta"] = {"content": message}
return template return template
def openai_chat_completion_message_template(model: str, message: str): 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["object"] = "chat.completion"
template["choices"][0]["message"] = {"content": message, "role": "assistant"} template["choices"][0]["message"] = {"content": message, "role": "assistant"}
template["choices"][0]["finish_reason"] = "stop" template["choices"][0]["finish_reason"] = "stop"