xinyin025/LangBot
1
0
Fork 0
mirror of https://github.com/langbot-app/LangBot.git synced 2026-06-10 15:56:03 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: 发送正确的图片格式而不是默认的 image/jpeg

Browse Source
This commit is contained in:
ligen131
2024-08-19 00:00:29 +08:00
parent 2f092f4a87
commit b51ca06c7c
4 changed files with 16 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
pkg/provider/modelmgr/apis/anthropicmsgs.py
View File

@@ -72,12 +72,13 @@ class AnthropicMessages(api.LLMAPIRequester):
for i, ce in enumerate(m.content):
if ce.type == "image_url":
base64_image, image_format = await image.qq_image_url_to_base64(ce.image_url.url)
alter_image_ele = {
"type": "image",
"source": {
"type": "base64",
"media_type": "image/jpeg",
"data": await image.qq_image_url_to_base64(ce.image_url.url)
"media_type": f"image/{image_format}",
"data": base64_image
}
}
msg_dict["content"][i] = alter_image_ele

6
pkg/provider/modelmgr/apis/chatcmpl.py
View File

@@ -136,7 +136,5 @@ class OpenAIChatCompletions(api.LLMAPIRequester):
self,
original_url: str,
) -> str:
base64_image = await image.qq_image_url_to_base64(original_url)
return f"data:image/jpeg;base64,{base64_image}"
base64_image, image_format = await image.qq_image_url_to_base64(original_url)
return f"data:image/{image_format};base64,{base64_image}"

4
pkg/provider/modelmgr/apis/ollamachat.py
View File

@@ -101,5 +101,5 @@ class OllamaChatCompletions(api.LLMAPIRequester):
self,
original_url: str,
) -> str:
base64_image: str = await image.qq_image_url_to_base64(original_url)
return f"data:image/jpeg;base64,{base64_image}"
base64_image, image_format = await image.qq_image_url_to_base64(original_url)
return f"data:image/{image_format};base64,{base64_image}"