def _get_image_size()

in aidial_adapter_bedrock/llm/model/claude/v3/tokenizer.py [0:0]

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

def _get_image_size(image_data: Union[str, Base64FileInput]) -> Tuple[int, int]:
    try:
        if not isinstance(image_data, str):
            raise ValueError("Images as files aren't yet supported.")

        image_bytes = base64.b64decode(image_data)
        with Image.open(io.BytesIO(image_bytes)) as img:
            return img.size
    except Exception:
        log.exception("Cannot compute image size, assuming 1000x1000")
        return 1000, 1000