def transform_messages()

in model-gallery/deploy/mllm/webui_client.py [0:0]

• 23 lines of code
• 6 McCabe index (conditional complexity)

def transform_messages(original_messages):
    transformed_messages = []
    for message in original_messages:
        new_content = []
        for item in message["content"]:
            if "image_url" in item:
                new_item = {
                    "type": "image_url",
                    "image_url": {"url": item["image_url"]},
                }
            elif "text" in item:
                new_item = {"type": "text", "text": item["text"]}
            elif "video_url" in item:
                new_item = {
                    "type": "video_url",
                    "video_url": {"url": item["video_url"]},
                }
            else:
                continue
            new_content.append(new_item)

        new_message = {"role": message["role"], "content": new_content}
        transformed_messages.append(new_message)

    return transformed_messages