fix: 重构图片MIME类型转换逻辑

将"image/jpg"到"image/jpeg"的MIME类型转换逻辑从_convert_image函数移至_get_mime_type_and_data函数,避免代码重复并提高一致性。这确保了MIME类型的标准化处理发生在数据提取的同一位置。
This commit is contained in:
snaily
2025-03-18 21:50:27 +08:00
parent 16f16a3ae9
commit 6ba58ce9d1

View File

@@ -33,7 +33,7 @@ def _get_mime_type_and_data(base64_string):
pattern = r'data:([^;]+);base64,(.+)'
match = re.match(pattern, base64_string)
if match:
mime_type = match.group(1)
mime_type = "image/jpeg" if match.group(1) == "image/jpg" else match.group(1)
encoded_data = match.group(2)
return mime_type, encoded_data
@@ -45,7 +45,7 @@ def _convert_image(image_url: str) -> Dict[str, Any]:
mime_type, encoded_data = _get_mime_type_and_data(image_url)
return {
"inline_data": {
"mime_type": "image/jpeg" if mime_type == "image/jpg" else mime_type,
"mime_type": mime_type,
"data": encoded_data
}
}