GithubBackup/MoviePilot
1
0
Fork 0
mirror of https://github.com/jxxghp/MoviePilot.git synced 2026-07-07 07:41:28 +08:00
Code Issues Packages Projects Releases Wiki Activity

Expand image and edit support across messaging channels

Browse Source
This commit is contained in:
jxxghp
2026-04-11 22:10:54 +08:00
parent bf2d2cbd03
commit 2f53fd3108
12 changed files with 952 additions and 45 deletions
Download Patch File Download Diff File Expand all files Collapse all files

42
app/chain/message.py
View File

@@ -1371,7 +1371,15 @@ class MessageChain(ChainBase):
base64_images = []
for img in images:
try:
if img.startswith("tg://file_id/"):
if img.startswith("data:"):
base64_images.append(img)
logger.info(
"图片无需下载: channel=%s, source=%s, input=%s",
channel.value if channel else None,
source,
self._summarize_images([img])[0],
)
elif img.startswith("tg://file_id/"):
file_id = img.replace("tg://file_id/", "")
base64_data = self.run_module(
"download_file_to_base64", file_id=file_id, source=source
@@ -1384,6 +1392,23 @@ class MessageChain(ChainBase):
source,
img,
)
elif img.startswith("wxwork://media_id/") or img.startswith(
"wxbot://image/"
):
data_url = self.run_module(
"download_wechat_image_to_data_url",
image_ref=img,
source=source,
)
if data_url:
base64_images.append(data_url)
logger.info(
"图片下载成功: channel=%s, source=%s, input=%s, output=%s",
channel.value if channel else None,
source,
img,
self._summarize_images([data_url])[0],
)
elif channel == MessageChannel.Slack:
data_url = self.run_module(
"download_file_to_data_url", file_url=img, source=source
@@ -1397,6 +1422,21 @@ class MessageChain(ChainBase):
img,
self._summarize_images([data_url])[0],
)
elif img.startswith("vocechat://file/"):
data_url = self.run_module(
"download_vocechat_image_to_data_url",
image_ref=img,
source=source,
)
if data_url:
base64_images.append(data_url)
logger.info(
"图片下载成功: channel=%s, source=%s, input=%s, output=%s",
channel.value if channel else None,
source,
img,
self._summarize_images([data_url])[0],
)
elif img.startswith("http"):
resp = RequestUtils(timeout=30).get_res(img)
if resp and resp.content: