LiteyukiStudio/nonebot-plugin-marshoai
5
4
Fork 0
mirror of https://github.com/LiteyukiStudio/nonebot-plugin-marshoai.git synced 2025-09-24 15:36:23 +00:00
Code Issues Packages Projects Releases Wiki Activity

🐛 处理可能的意外 application/octet-stream ，兼容matcha

Browse Source
This commit is contained in:
Asankilp
2025-09-05 15:29:07 +08:00
parent caff43ff7b
commit 93bfb966ea
1 changed files with 3 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
nonebot_plugin_marshoai/util.py
View File

@ -83,6 +83,8 @@ async def get_image_raw_and_type(
content_type = response.headers.get("Content-Type") content_type = response.headers.get("Content-Type")
if not content_type: if not content_type:
content_type = mimetypes.guess_type(url)[0] content_type = mimetypes.guess_type(url)[0]
if content_type == "application/octet-stream": # matcha 兼容
content_type = "image/jpeg"
# image_format = content_type.split("/")[1] if content_type else "jpeg" # image_format = content_type.split("/")[1] if content_type else "jpeg"
return response.content, str(content_type) return response.content, str(content_type)
else: else:
@ -125,7 +127,7 @@ async def make_chat_openai(
model_name: 指定AI模型名 model_name: 指定AI模型名
tools: 工具列表 tools: 工具列表
""" """
print(msg) # print(msg)
return await client.chat.completions.create( # type: ignore return await client.chat.completions.create( # type: ignore
messages=msg, messages=msg,
model=model_name, model=model_name,