浏览代码

refac: rename message_template

Michael Poluektov 9 月之前
父节点
当前提交
b9b1fdd1a1
共有 1 个文件被更改,包括 3 次插入3 次删除
  1. 3 3
      backend/utils/misc.py

+ 3 - 3
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"