def image_message()

in distilvit/gpt4.py [0:0]

• 10 lines of code
• 1 McCabe index (conditional complexity)

    def image_message(self, image):
        buffered = BytesIO()
        image.save(buffered, format="JPEG")
        img_byte = buffered.getvalue()
        img_base64 = base64.b64encode(img_byte)
        encoded_image = img_base64.decode("utf-8")
        return {
            "type": "image_url",
            "image_url": {"url": f"data:image/jpg;base64,{encoded_image}"},
        }