diff --git a/endpoints/OAI/utils/chat_completion.py b/endpoints/OAI/utils/chat_completion.py index 34de3e8..32e0696 100644 --- a/endpoints/OAI/utils/chat_completion.py +++ b/endpoints/OAI/utils/chat_completion.py @@ -217,7 +217,7 @@ async def format_prompt_with_template( # Deal with list in messages.content # Just replace the content list with the very first text message for message in data.messages: - if message["role"] == "user" and isinstance(message["content"], list): + if isinstance(message["content"], list): message["content"] = next( ( content["text"]