This commit is contained in:
Timothy Jaeryang Baek
2026-07-27 02:49:08 -04:00
parent 7d77efe0f1
commit ba556bd8f0
3 changed files with 34 additions and 21 deletions
+4 -3
View File
@@ -663,9 +663,9 @@ def strip_empty_content_blocks(messages: list[dict]) -> list[dict]:
return messages
def openai_chat_message_template(model: str):
def openai_chat_message_template(model: str, message_id: str | None = None):
return {
'id': f'{model}-{str(uuid.uuid4())}',
'id': message_id if message_id else f'{model}-{str(uuid.uuid4())}',
'created': int(time.time()),
'model': model,
'choices': [{'index': 0, 'logprobs': None, 'finish_reason': None}],
@@ -678,8 +678,9 @@ def openai_chat_chunk_message_template(
reasoning_content: str | None = None,
tool_calls: list[dict | None] = None,
usage: dict | None = None,
message_id: str | None = None,
) -> dict:
template = openai_chat_message_template(model)
template = openai_chat_message_template(model, message_id)
template['object'] = 'chat.completion.chunk'
template['choices'][0]['index'] = 0