def clean_openai_msg()

in function_app/src/helpers/common.py [0:0]

• 9 lines of code
• 5 McCabe index (conditional complexity)

def clean_openai_msg(message: dict) -> dict:
    """
    Cleans OpenAI messages prior to returning them via the API. This is useful
    for shortening long URLs in image messages.
    :param message:
        The message to clean.

    :return:
        A cleaned message.
    """
    new_msg = message.copy()
    if isinstance(new_msg["content"], list):
        for msg in new_msg["content"]:
            if msg["type"] == "image_url" and msg["image_url"]["url"].startswith(
                "data:image"
            ):
                msg["image_url"]["url"] = msg["image_url"]["url"][:100] + "..."
    return new_msg