Browse Source

refac: rename message_template

Michael Poluektov 9 tháng trước cách đây
mục cha
commit
b9b1fdd1a1
1 tập tin đã thay đổi với 3 bổ sung3 xóa
  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"