🐛 修复飞书适配器转换消息过程中无法正确转化Base64图片

This commit is contained in:
KimigaiiWuyi
2025-04-13 06:06:02 +08:00
parent fc146d3d00
commit c9079b9299

View File

@@ -1,6 +1,8 @@
import json
import uuid
import base64
import lark_oapi as lark
from io import BytesIO
from typing import List
from astrbot.api.event import AstrMessageEvent, MessageChain
from astrbot.api.message_components import Plain, Image as AstrBotImage, At
@@ -27,22 +29,29 @@ class LarkMessageEvent(AstrMessageEvent):
_stage.append({"tag": "at", "user_id": comp.qq, "style": []})
elif isinstance(comp, AstrBotImage):
file_path = ""
image_file = None
if comp.file and comp.file.startswith("file:///"):
file_path = comp.file.replace("file:///", "")
elif comp.file and comp.file.startswith("http"):
image_file_path = await download_image_by_url(comp.file)
file_path = image_file_path
elif comp.file and comp.file.startswith("base64://"):
pass
base64_str = comp.file[9:]
image_data = base64.b64decode(base64_str)
image_file = BytesIO(image_data).getvalue()
else:
file_path = comp.file
if image_file is None:
image_file = open(file_path, "rb")
request = (
CreateImageRequest.builder()
.request_body(
CreateImageRequestBody.builder()
.image_type("message")
.image(open(file_path, "rb"))
.image(image_file)
.build()
)
.build()
@@ -51,7 +60,7 @@ class LarkMessageEvent(AstrMessageEvent):
if not response.success():
logger.error(f"无法上传飞书图片({response.code}): {response.msg}")
image_key = response.data.image_key
print(image_key)
logger.debug(image_key)
ret.append(_stage)
ret.append([{"tag": "img", "image_key": image_key}])
_stage.clear()