Commit b9b1fdd1 authored by Michael Poluektov's avatar Michael Poluektov
Browse files

refac: rename message_template

parent 2e0fa1c6
......@@ -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"
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment