def convert_gpt4v_to_gpt4_chunk()

in aidial_adapter_openai/gpt4_multi_modal/gpt4_vision.py [0:0]

• 15 lines of code
• 7 McCabe index (conditional complexity)

def convert_gpt4v_to_gpt4_chunk(obj: dict) -> Optional[dict]:
    ret = obj.copy()
    if (
        ret.get("choices", []) == []
        or ret.get("id", "") == ""
        or ret.get("model", "") == ""
        or ret.get("created", 0) == 0
    ):
        return None

    if ret.get("prompt_filter_results"):
        del ret["prompt_filter_results"]

    ret["choices"] = [
        convert_gpt4v_to_gpt4_choice(choice) for choice in obj["choices"]
    ]

    return ret