fix: fix utils

This commit is contained in:
Krrish Dholakia 2025-02-25 22:54:46 -08:00
parent ac6e503461
commit b8c5b31a4e

View file

@ -5939,7 +5939,7 @@ def validate_and_fix_openai_messages(messages: List):
Handles missing role for assistant messages. Handles missing role for assistant messages.
""" """
for message in messages: for message in messages:
if "role" not in message: if not message.get("role"):
message["role"] = "assistant" message["role"] = "assistant"
return validate_chat_completion_messages(messages=messages) return validate_chat_completion_messages(messages=messages)