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
+2 -2
View File
@@ -33,7 +33,7 @@ def _get_mime_type_and_data(base64_string):
pattern = r'data:([^;]+);base64,(.+)' pattern = r'data:([^;]+);base64,(.+)'
match = re.match(pattern, base64_string) match = re.match(pattern, base64_string)
if match: 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) encoded_data = match.group(2)
return mime_type, encoded_data 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) mime_type, encoded_data = _get_mime_type_and_data(image_url)
return { return {
"inline_data": { "inline_data": {
"mime_type": "image/jpeg" if mime_type == "image/jpg" else mime_type, "mime_type": mime_type,
"data": encoded_data "data": encoded_data
} }
} }