Merge pull request #14 from EillesWan/main

把回复消息中的 图片 全部插入消息体

.gitignore

@@ -1,3 +1,7 @@
+
+# Other Things
+test.md
+
 # Byte-compiled / optimized / DLL files
 __pycache__/
 *.py[cod]

.vscode/settings.json

@@ -0,0 +1,3 @@
+{
+    "python.analysis.typeCheckingMode": "standard"
+}

nonebot_plugin_marshoai/__init__.py

@@ -1,15 +1,19 @@
 from nonebot.plugin import require
+
 require("nonebot_plugin_alconna")
 require("nonebot_plugin_localstore")
-from .azure import *
+
-#from .hunyuan import *
 from nonebot import get_driver, logger
+import nonebot_plugin_localstore as store
+
+# from .hunyuan import *
+from .azure import *
 from .config import config
 from .metadata import metadata
-import nonebot_plugin_localstore as store
 
 __author__ = "Asankilp"
 __plugin_meta__ = metadata
+
 driver = get_driver()
 
 

nonebot_plugin_marshoai/azure.py

@@ -1,5 +1,6 @@
-import contextlib
+import uuid
 import traceback
+import contextlib
 from typing import Optional
 from pathlib import Path
 
@@ -15,15 +16,21 @@ from azure.ai.inference.models import (
     ChatCompletionsToolCall,
 )
 from azure.core.credentials import AzureKeyCredential
-from nonebot import on_command, on_message, logger
+from nonebot import on_command, on_message, logger, get_driver
 from nonebot.adapters import Message, Event
 from nonebot.params import CommandArg
 from nonebot.permission import SUPERUSER
 from nonebot.rule import Rule, to_me
-from nonebot_plugin_alconna import on_alconna, MsgTarget
+from nonebot_plugin_alconna import (
-from nonebot_plugin_alconna.uniseg import UniMessage, UniMsg
+    on_alconna,
+    MsgTarget,
+    UniMessage,
+    UniMsg,
+    Text as TextMsg,
+    Image as ImageMsg,
+)
 import nonebot_plugin_localstore as store
-from nonebot import get_driver
+
 
 from .constants import *
 from .metadata import metadata
@@ -37,15 +44,23 @@ async def at_enable():
 
 driver = get_driver()
 
-changemodel_cmd = on_command("changemodel", permission=SUPERUSER, priority=10, block=True)
+changemodel_cmd = on_command(
+    "changemodel", permission=SUPERUSER, priority=10, block=True
+)
 resetmem_cmd = on_command("reset", priority=10, block=True)
 # setprompt_cmd = on_command("prompt",permission=SUPERUSER)
 praises_cmd = on_command("praises", permission=SUPERUSER, priority=10, block=True)
 add_usermsg_cmd = on_command("usermsg", permission=SUPERUSER, priority=10, block=True)
-add_assistantmsg_cmd = on_command("assistantmsg", permission=SUPERUSER, priority=10, block=True)
+add_assistantmsg_cmd = on_command(
+    "assistantmsg", permission=SUPERUSER, priority=10, block=True
+)
 contexts_cmd = on_command("contexts", permission=SUPERUSER, priority=10, block=True)
-save_context_cmd = on_command("savecontext", permission=SUPERUSER, priority=10, block=True)
+save_context_cmd = on_command(
-load_context_cmd = on_command("loadcontext", permission=SUPERUSER, priority=10, block=True)
+    "savecontext", permission=SUPERUSER, priority=10, block=True
+)
+load_context_cmd = on_command(
+    "loadcontext", permission=SUPERUSER, priority=10, block=True
+)
 marsho_cmd = on_alconna(
     Alconna(
         config.marshoai_default_name,
@@ -53,7 +68,7 @@ marsho_cmd = on_alconna(
     ),
     aliases=config.marshoai_aliases,
     priority=10,
-    block=True
+    block=True,
 )
 marsho_at = on_message(rule=to_me() & at_enable, priority=11)
 nickname_cmd = on_alconna(
@@ -62,9 +77,11 @@ nickname_cmd = on_alconna(
         Args["name?", str],
     ),
     priority=10,
-    block = True
+    block=True,
+)
+refresh_data_cmd = on_command(
+    "refresh_data", permission=SUPERUSER, priority=10, block=True
 )
-refresh_data_cmd = on_command("refresh_data", permission=SUPERUSER, priority=10, block=True)
 
 command_start = driver.config.command_start
 model_name = config.marshoai_default_model
@@ -86,7 +103,9 @@ async def _preload_tools():
         tools.load_tools(store.get_plugin_data_dir() / "tools")
         for tool_dir in config.marshoai_toolset_dir:
             tools.load_tools(tool_dir)
-        logger.info("如果启用小棉工具后使用的模型出现报错，请尝试将 MARSHOAI_ENABLE_TOOLS 设为 false。")
+        logger.info(
+            "如果启用小棉工具后使用的模型出现报错，请尝试将 MARSHOAI_ENABLE_TOOLS 设为 false。"
+        )
 
 
 @add_usermsg_cmd.handle()
@@ -132,7 +151,9 @@ async def save_context(target: MsgTarget, arg: Message = CommandArg()):
 @load_context_cmd.handle()
 async def load_context(target: MsgTarget, arg: Message = CommandArg()):
     if msg := arg.extract_plain_text():
-        await get_backup_context(target.id, target.private)  # 为了将当前会话添加到"已恢复过备份"的列表而添加，防止上下文被覆盖（好奇怪QwQ
+        await get_backup_context(
+            target.id, target.private
+        )  # 为了将当前会话添加到"已恢复过备份"的列表而添加，防止上下文被覆盖（好奇怪QwQ
         context.set_context(
             await load_context_from_json(msg, "contexts"), target.id, target.private
         )
@@ -178,11 +199,113 @@ async def refresh_data():
     await refresh_data_cmd.finish("已刷新数据")
 
 
+"""
+以下函数依照 Apache 2.0 协议授权
+
+函数： get_back_uuidcodeblock、send_markdown
+
+版权所有 © 2024 金羿ELS
+Copyright (R) 2024 Eilles(EillesWan@outlook.com)
+
+Licensed under the Apache License, Version 2.0 (the "License");
+you may not use this file except in compliance with the License.
+You may obtain a copy of the License at
+
+   http://www.apache.org/licenses/LICENSE-2.0
+
+Unless required by applicable law or agreed to in writing, software
+distributed under the License is distributed on an "AS IS" BASIS,
+WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+See the License for the specific language governing permissions and
+limitations under the License.
+"""
+
+
+async def get_back_uuidcodeblock(msg: str, code_blank_uuid_map: list[tuple[str, str]]):
+
+    for torep, rep in code_blank_uuid_map:
+        msg = msg.replace(torep, rep)
+
+    return msg
+
+
+async def send_markdown(msg: str):
+    """
+    人工智能给出的回答一般不会包含 HTML 嵌入其中，但是包含图片或者 LaTeX 公式、代码块，都很正常。
+    这个函数会把这些都以图片形式嵌入消息体。
+    """
+
+    if not IMG_TAG_PATTERN.search(msg):  # 没有图片标签
+        await UniMessage(msg).send(reply_to=True)
+        return
+
+    result_msg = UniMessage()
+    code_blank_uuid_map = [
+        (uuid.uuid4().hex, cbp.group()) for cbp in CODE_BLOCK_PATTERN.finditer(msg)
+    ]
+
+    last_tag_index = 0
+
+    # 代码块渲染麻烦，先不处理
+    for rep, torep in code_blank_uuid_map:
+        msg = msg.replace(torep, rep)
+
+    # for to_rep in CODE_SINGLE_PATTERN.finditer(msg):
+    #     code_blank_uuid_map.append((rep := uuid.uuid4().hex, to_rep.group()))
+    #     msg = msg.replace(to_rep.group(), rep)
+
+    print("#####################\n", msg, "\n\n")
+
+    # 插入图片
+    for each_img_tag in IMG_TAG_PATTERN.finditer(msg):
+        img_tag = await get_back_uuidcodeblock(
+            each_img_tag.group(), code_blank_uuid_map
+        )
+        image_description = img_tag[2 : img_tag.find("]")]
+        image_url = img_tag[img_tag.find("(") + 1 : -1]
+
+        result_msg.append(
+            TextMsg(
+                await get_back_uuidcodeblock(
+                    msg[last_tag_index : msg.find(img_tag)], code_blank_uuid_map
+                )
+            )
+        )
+
+        last_tag_index = msg.find(img_tag) + len(img_tag)
+
+        if image_ := await get_image_raw_and_type(image_url):
+
+            result_msg.append(
+                ImageMsg(
+                    raw=image_[0], mimetype=image_[1], name=image_description + ".png"
+                )
+            )
+            result_msg.append(TextMsg("（{}）".format(image_description)))
+
+        else:
+            result_msg.append(TextMsg(img_tag))
+
+    result_msg.append(
+        TextMsg(await get_back_uuidcodeblock(msg[last_tag_index:], code_blank_uuid_map))
+    )
+
+    await result_msg.send(reply_to=True)
+
+
+"""
+Apache 2.0 协议授权部分结束
+"""
+
+
 @marsho_at.handle()
 @marsho_cmd.handle()
 async def marsho(target: MsgTarget, event: Event, text: Optional[UniMsg] = None):
     global target_list
-    if event.get_message().extract_plain_text() and (not text and event.get_message().extract_plain_text() != config.marshoai_default_name):
+    if event.get_message().extract_plain_text() and (
+        not text
+        and event.get_message().extract_plain_text() != config.marshoai_default_name
+    ):
         text = event.get_message()
     if not text:
         # 发送说明
@@ -204,7 +327,10 @@ async def marsho(target: MsgTarget, event: Event, text: Optional[UniMsg] = None)
                     "*你未设置自己的昵称。推荐使用'nickname [昵称]'命令设置昵称来获得个性化(可能）回答。"
                 ).send()
 
-        is_support_image_model = model_name.lower() in SUPPORT_IMAGE_MODELS + config.marshoai_additional_image_models
+        is_support_image_model = (
+            model_name.lower()
+            in SUPPORT_IMAGE_MODELS + config.marshoai_additional_image_models
+        )
         is_reasoning_model = model_name.lower() in REASONING_MODELS
         usermsg = [] if is_support_image_model else ""
         for i in text:
@@ -217,14 +343,18 @@ async def marsho(target: MsgTarget, event: Event, text: Optional[UniMsg] = None)
                 if is_support_image_model:
                     usermsg.append(
                         ImageContentItem(
-                            image_url=ImageUrl(url=str(await get_image_b64(i.data["url"])))
+                            image_url=ImageUrl(
+                                url=str(await get_image_b64(i.data["url"]))
+                            )
                         )
                     )
                 elif config.marshoai_enable_support_image_tip:
                     await UniMessage("*此模型不支持图片处理。").send()
         backup_context = await get_backup_context(target.id, target.private)
         if backup_context:
-            context.set_context(backup_context, target.id, target.private)  # 加载历史记录
+            context.set_context(
+                backup_context, target.id, target.private
+            )  # 加载历史记录
             logger.info(f"已恢复会话 {target.id} 的上下文备份~")
         context_msg = context.build(target.id, target.private)
         if not is_reasoning_model:
@@ -234,46 +364,74 @@ async def marsho(target: MsgTarget, event: Event, text: Optional[UniMsg] = None)
             client=client,
             model_name=model_name,
             msg=context_msg + [UserMessage(content=usermsg)],
-            tools=tools.get_tools_list()
+            tools=tools.get_tools_list(),
         )
         # await UniMessage(str(response)).send()
         choice = response.choices[0]
-        if (choice["finish_reason"] == CompletionsFinishReason.STOPPED):  # 当对话成功时，将dict的上下文添加到上下文类中
+        if choice["finish_reason"] == CompletionsFinishReason.STOPPED:
+            # 当对话成功时，将dict的上下文添加到上下文类中
             context.append(
                 UserMessage(content=usermsg).as_dict(), target.id, target.private
             )
             context.append(choice.message.as_dict(), target.id, target.private)
             if [target.id, target.private] not in target_list:
                 target_list.append([target.id, target.private])
-            await UniMessage(str(choice.message.content)).send(reply_to=True)
+
+            # 对话成功发送消息
+            await send_markdown(str(choice.message.content))
         elif choice["finish_reason"] == CompletionsFinishReason.CONTENT_FILTERED:
-            await UniMessage("*已被内容过滤器过滤。请调整聊天内容后重试。").send(reply_to=True)
+
+            # 对话失败，消息过滤
+
+            await UniMessage("*已被内容过滤器过滤。请调整聊天内容后重试。").send(
+                reply_to=True
+            )
             return
         elif choice["finish_reason"] == CompletionsFinishReason.TOOL_CALLS:
+
+            # 需要获取额外信息，调用函数工具
             tool_msg = []
             while choice.message.tool_calls != None:
-                tool_msg.append(AssistantMessage(tool_calls=response.choices[0].message.tool_calls))
+                tool_msg.append(
+                    AssistantMessage(tool_calls=response.choices[0].message.tool_calls)
+                )
                 for tool_call in choice.message.tool_calls:
-                    if isinstance(tool_call, ChatCompletionsToolCall): # 循环调用工具直到不需要调用
+                    if isinstance(
-                        function_args = json.loads(tool_call.function.arguments.replace("'", '"'))
+                        tool_call, ChatCompletionsToolCall
-                        logger.info(f"调用函数 {tool_call.function.name} ,参数为 {function_args}")
+                    ):  # 循环调用工具直到不需要调用
-                        await UniMessage(f"调用函数 {tool_call.function.name} ,参数为 {function_args}").send()
+                        function_args = json.loads(
-                        func_return = await tools.call(tool_call.function.name, function_args) # 获取返回值
+                            tool_call.function.arguments.replace("'", '"')
-                        tool_msg.append(ToolMessage(tool_call_id=tool_call.id, content=func_return))
+                        )
+                        logger.info(
+                            f"调用函数 {tool_call.function.name} ,参数为 {function_args}"
+                        )
+                        await UniMessage(
+                            f"调用函数 {tool_call.function.name} ,参数为 {function_args}"
+                        ).send()
+                        func_return = await tools.call(
+                            tool_call.function.name, function_args
+                        )  # 获取返回值
+                        tool_msg.append(
+                            ToolMessage(tool_call_id=tool_call.id, content=func_return)
+                        )
                 response = await make_chat(
                     client=client,
                     model_name=model_name,
                     msg=context_msg + [UserMessage(content=usermsg)] + tool_msg,
-                    tools=tools.get_tools_list()
+                    tools=tools.get_tools_list(),
                 )
                 choice = response.choices[0]
             if choice["finish_reason"] == CompletionsFinishReason.STOPPED:
+
+                # 对话成功 添加上下文
                 context.append(
                     UserMessage(content=usermsg).as_dict(), target.id, target.private
                 )
                 # context.append(tool_msg, target.id, target.private)
                 context.append(choice.message.as_dict(), target.id, target.private)
-                await UniMessage(str(choice.message.content)).send(reply_to=True)
+
+                # 发送消息
+                await send_markdown(str(choice.message.content))
             else:
                 await marsho_cmd.finish(f"意外的完成原因:{choice['finish_reason']}")
         else:
@@ -288,7 +446,6 @@ with contextlib.suppress(ImportError):  # 优化先不做（）
     import nonebot.adapters.onebot.v11  # type: ignore
     from .azure_onebot import poke_notify
 
-
     @poke_notify.handle()
     async def poke(event: Event):
 
@@ -327,5 +484,7 @@ async def auto_backup_context():
             target_uid = "private_" + target_id
         else:
             target_uid = "group_" + target_id
-        await save_context_to_json(f"back_up_context_{target_uid}", contexts_data, "contexts/backup")
+        await save_context_to_json(
+            f"back_up_context_{target_uid}", contexts_data, "contexts/backup"
+        )
         logger.info(f"已保存会话 {target_id} 的上下文备份，将在下次对话时恢复~")

nonebot_plugin_marshoai/config.py

@@ -55,19 +55,19 @@ destination_file = destination_folder / "config.yaml"
 
 
 def copy_config(source_template, destination_file):
-    '''
+    """
     复制模板配置文件到config
-    '''
+    """
     shutil.copy(source_template, destination_file)
 
 
 def check_yaml_is_changed(source_template):
-    '''
+    """
     检查配置文件是否需要更新
-    '''
+    """
-    with open(config_file_path, 'r', encoding="utf-8") as f:
+    with open(config_file_path, "r", encoding="utf-8") as f:
         old = yaml.load(f)
-    with open(source_template, 'r', encoding="utf-8") as f:
+    with open(source_template, "r", encoding="utf-8") as f:
         example_ = yaml.load(f)
     keys1 = set(example_.keys())
     keys2 = set(old.keys())
@@ -78,9 +78,9 @@ def check_yaml_is_changed(source_template):
 
 
 def merge_configs(old_config, new_config):
-    '''
+    """
     合并配置文件
-    '''
+    """
     for key, value in new_config.items():
         if key in old_config:
             continue
@@ -89,6 +89,7 @@ def merge_configs(old_config, new_config):
             old_config[key] = value
     return old_config
 
+
 config: ConfigModel = get_plugin_config(ConfigModel)
 if config.marshoai_use_yaml_config:
     if not config_file_path.exists():
@@ -102,15 +103,15 @@ if config.marshoai_use_yaml_config:
             yaml_2 = YAML()
             logger.info("插件新的配置已更新, 正在更新")
 
-            with open(config_file_path, 'r', encoding="utf-8") as f:
+            with open(config_file_path, "r", encoding="utf-8") as f:
                 old_config = yaml_2.load(f)
 
-            with open(source_template, 'r', encoding="utf-8") as f:
+            with open(source_template, "r", encoding="utf-8") as f:
                 new_config = yaml_2.load(f)
 
             merged_config = merge_configs(old_config, new_config)
 
-            with open(destination_file, 'w', encoding="utf-8") as f:
+            with open(destination_file, "w", encoding="utf-8") as f:
                 yaml_2.dump(merged_config, f)
 
     with open(config_file_path, "r", encoding="utf-8") as f:
@@ -118,4 +119,6 @@ if config.marshoai_use_yaml_config:
 
         config = ConfigModel(**yaml_config)
 else:
-    logger.info("MarshoAI 支持新的 YAML 配置系统，若要使用，请将 MARSHOAI_USE_YAML_CONFIG 配置项设置为 true。")
+    logger.info(
+        "MarshoAI 支持新的 YAML 配置系统，若要使用，请将 MARSHOAI_USE_YAML_CONFIG 配置项设置为 true。"
+    )

nonebot_plugin_marshoai/constants.py

@@ -1,4 +1,6 @@
+import re
 from .config import config
+
 USAGE: str = f"""MarshoAI-NoneBot Beta by Asankilp
 用法：
   {config.marshoai_default_name} <聊天内容> : 与 Marsho 进行对话。当模型为 GPT-4o(-mini) 等时，可以带上图片进行对话。
@@ -15,7 +17,13 @@ USAGE: str = f"""MarshoAI-NoneBot Beta by Asankilp
   refresh_data : 从文件刷新已加载的昵称与夸赞名单。
 ※本AI的回答"按原样"提供，不提供任何担保。AI也会犯错，请仔细甄别回答的准确性。"""
 
-SUPPORT_IMAGE_MODELS: list = ["gpt-4o","gpt-4o-mini","phi-3.5-vision-instruct","llama-3.2-90b-vision-instruct","llama-3.2-11b-vision-instruct"]
+SUPPORT_IMAGE_MODELS: list = [
+    "gpt-4o",
+    "gpt-4o-mini",
+    "phi-3.5-vision-instruct",
+    "llama-3.2-90b-vision-instruct",
+    "llama-3.2-11b-vision-instruct",
+]
 REASONING_MODELS: list = ["o1-preview", "o1-mini"]
 INTRODUCTION: str = """你好喵~我是一只可爱的猫娘AI，名叫小棉~🐾！
 我的代码在这里哦~↓↓↓
@@ -25,3 +33,19 @@ https://github.com/LiteyukiStudio/nonebot-plugin-marshoai
 https://github.com/Meloland/melobot
 我与 Melobot 酱贴贴的代码在这里喵~↓↓↓
 https://github.com/LiteyukiStudio/marshoai-melo"""
+
+
+# 正则匹配代码块
+CODE_BLOCK_PATTERN = re.compile(
+    r"```(.*?)```|`(.*?)`", re.DOTALL
+)
+# 正则匹配完整图片标签字段
+IMG_TAG_PATTERN = re.compile(r"!\[[^\]]*\]\([^()]*\)")
+# # 正则匹配图片标签中的图片url字段
+# INTAG_URL_PATTERN = re.compile(r'\(([^)]*)')
+# # 正则匹配图片标签中的文本描述字段
+# INTAG_TEXT_PATTERN = re.compile(r'!\[([^\]]*)\]')
+# 正则匹配 LaTeX 公式内容
+LATEX_PATTERN = re.compile(
+    r"\\begin\{equation\}(.*?)\\end\{equation\}|(?<!\$)(\$(.*?)\$|\$\$(.*?)\$\$|\\\[(.*?)\\\]|\\\[.*?\\\]|\\\((.*?)\\\))",
+)

nonebot_plugin_marshoai/models.py

@@ -4,20 +4,19 @@ import os
 import re
 import json
 import importlib
+
 # import importlib.util
 import traceback
 from nonebot import logger
 
+
 class MarshoContext:
     """
     Marsho 的上下文类
     """
 
     def __init__(self):
-        self.contents = {
+        self.contents = {"private": {}, "non-private": {}}
-            "private": {},
-            "non-private": {}
-        }
 
     def _get_target_dict(self, is_private):
         return self.contents["private"] if is_private else self.contents["non-private"]
@@ -60,6 +59,7 @@ class MarshoTools:
     """
     Marsho 的工具类
     """
+
     def __init__(self):
         self.tools_list = []
         self.imported_packages = {}
@@ -74,16 +74,20 @@ class MarshoTools:
 
         for package_name in os.listdir(tools_dir):
             package_path = os.path.join(tools_dir, package_name)
-            if os.path.isdir(package_path) and os.path.exists(os.path.join(package_path, '__init__.py')):
+            if os.path.isdir(package_path) and os.path.exists(
-                json_path = os.path.join(package_path, 'tools.json')
+                os.path.join(package_path, "__init__.py")
+            ):
+                json_path = os.path.join(package_path, "tools.json")
                 if os.path.exists(json_path):
                     try:
-                        with open(json_path, 'r', encoding="utf-8") as json_file:
+                        with open(json_path, "r", encoding="utf-8") as json_file:
                             data = json.load(json_file)
                             for i in data:
                                 self.tools_list.append(i)
                             # 导入包
-                            spec = importlib.util.spec_from_file_location(package_name, os.path.join(package_path, "__init__.py"))
+                            spec = importlib.util.spec_from_file_location(
+                                package_name, os.path.join(package_path, "__init__.py")
+                            )
                             package = importlib.util.module_from_spec(spec)
                             spec.loader.exec_module(package)
                             self.imported_packages[package_name] = package
@@ -94,7 +98,9 @@ class MarshoTools:
                         logger.error(f"加载工具包时发生错误: {e}")
                         traceback.print_exc()
                 else:
-                    logger.warning(f"在工具包 {package_path} 下找不到tools.json，跳过加载。")
+                    logger.warning(
+                        f"在工具包 {package_path} 下找不到tools.json，跳过加载。"
+                    )
             else:
                 logger.warning(f"{package_path} 不是有效的工具包路径，跳过加载。")
 
@@ -125,5 +131,3 @@ class MarshoTools:
         if not self.tools_list or not config.marshoai_enable_tools:
             return None
         return self.tools_list
-
-

nonebot_plugin_marshoai/tools/marshoai-basic/__init__.py

@@ -1,20 +1,23 @@
 import os
-from datetime import datetime
 from zhDateTime import DateTime
+
+
 async def get_weather(location: str):
     return f"{location}的温度是114514℃。"
 
+
 async def get_current_env():
     ver = os.popen("uname -a").read()
     return str(ver)
 
+
 async def get_current_time():
-    current_time = datetime.now().strftime("%Y.%m.%d %H:%M:%S")
+    current_time = DateTime.now().strftime("%Y.%m.%d %H:%M:%S")
-    current_weekday = datetime.now().weekday()
+    current_weekday = DateTime.now().weekday()
 
     weekdays = ["星期一", "星期二", "星期三", "星期四", "星期五", "星期六", "星期日"]
     current_weekday_name = weekdays[current_weekday]
 
-    current_lunar_date = (DateTime.now().to_lunar().date_hanzify()[5:])
+    current_lunar_date = DateTime.now().to_lunar().date_hanzify()[5:]
     time_prompt = f"现在的时间是{current_time}，{current_weekday_name}，农历{current_lunar_date}。"
     return time_prompt

nonebot_plugin_marshoai/util.py

@@ -2,13 +2,12 @@ import base64
 import mimetypes
 import os
 import json
-from typing import Any
+from typing import Any, Optional
 import httpx
 import nonebot_plugin_localstore as store
-from datetime import datetime
 
 from nonebot.log import logger
-from zhDateTime import DateTime  # type: ignore
+from zhDateTime import DateTime
 from azure.ai.inference.aio import ChatCompletionsClient
 from azure.ai.inference.models import SystemMessage
 from .config import config
@@ -17,30 +16,66 @@ nickname_json = None  # 记录昵称
 praises_json = None  # 记录夸赞名单
 loaded_target_list = []  # 记录已恢复备份的上下文的列表
 
-
-async def get_image_b64(url):
 # noinspection LongLine
-    headers = {
+chromium_headers = {
     "User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/91.0.4472.124 Safari/537.36"
 }
 
+
+async def get_image_raw_and_type(
+    url: str, timeout: int = 10
+) -> Optional[tuple[bytes, str]]:
+    """
+    获取图片的二进制数据
+
+    参数:
+        url: str 图片链接
+        timeout: int 超时时间 秒
+
+    return:
+        tuple[bytes, str]: 图片二进制数据, 图片MIME格式
+
+    """
+
     async with httpx.AsyncClient() as client:
-        response = await client.get(url, headers=headers)
+        response = await client.get(url, headers=chromium_headers, timeout=timeout)
         if response.status_code == 200:
             # 获取图片数据
-            image_data = response.content
             content_type = response.headers.get("Content-Type")
             if not content_type:
                 content_type = mimetypes.guess_type(url)[0]
             # image_format = content_type.split("/")[1] if content_type else "jpeg"
-            base64_image = base64.b64encode(image_data).decode("utf-8")
+            return response.content, str(content_type)
-            data_url = f"data:{content_type};base64,{base64_image}"
+        else:
+            return None
+
+
+async def get_image_b64(url: str, timeout: int = 10) -> Optional[str]:
+    """
+    获取图片的base64编码
+
+    参数:
+        url: 图片链接
+        timeout: 超时时间 秒
+
+    return: 图片base64编码
+    """
+
+    if data_type := await get_image_raw_and_type(url, timeout):
+        # image_format = content_type.split("/")[1] if content_type else "jpeg"
+        base64_image = base64.b64encode(data_type[0]).decode("utf-8")
+        data_url = "data:{};base64,{}".format(data_type[1], base64_image)
         return data_url
     else:
         return None
 
 
-async def make_chat(client: ChatCompletionsClient, msg: list, model_name: str, tools: list = None):
+async def make_chat(
+    client: ChatCompletionsClient,
+    msg: list,
+    model_name: str,
+    tools: Optional[list] = None,
+):
     """调用ai获取回复
 
     参数:
@@ -60,7 +95,9 @@ async def make_chat(client: ChatCompletionsClient, msg: list, model_name: str, t
 def get_praises():
     global praises_json
     if praises_json is None:
-        praises_file = store.get_plugin_data_file("praises.json")  # 夸赞名单文件使用localstore存储
+        praises_file = store.get_plugin_data_file(
+            "praises.json"
+        )  # 夸赞名单文件使用localstore存储
         if not os.path.exists(praises_file):
             init_data = {
                 "like": [
@@ -207,5 +244,7 @@ async def get_backup_context(target_id: str, target_private: bool) -> list:
         target_uid = f"group_{target_id}"
     if target_uid not in loaded_target_list:
         loaded_target_list.append(target_uid)
-        return await load_context_from_json(f"back_up_context_{target_uid}", "contexts/backup")
+        return await load_context_from_json(
+            f"back_up_context_{target_uid}", "contexts/backup"
+        )
     return []

nonebot_plugin_marshoai/util_hunyuan.py

@@ -3,11 +3,17 @@ import types
 from tencentcloud.common import credential
 from tencentcloud.common.profile.client_profile import ClientProfile
 from tencentcloud.common.profile.http_profile import HttpProfile
-from tencentcloud.common.exception.tencent_cloud_sdk_exception import TencentCloudSDKException
+from tencentcloud.common.exception.tencent_cloud_sdk_exception import (
+    TencentCloudSDKException,
+)
 from tencentcloud.hunyuan.v20230901 import hunyuan_client, models
 from .config import config
+
+
 def generate_image(prompt: str):
-    cred = credential.Credential(config.marshoai_tencent_secretid, config.marshoai_tencent_secretkey)
+    cred = credential.Credential(
+        config.marshoai_tencent_secretid, config.marshoai_tencent_secretkey
+    )
     # 实例化一个http选项，可选的，没有特殊需求可以跳过
     httpProfile = HttpProfile()
     httpProfile.endpoint = "hunyuan.tencentcloudapi.com"
@@ -18,11 +24,7 @@ def generate_image(prompt: str):
     client = hunyuan_client.HunyuanClient(cred, "ap-guangzhou", clientProfile)
 
     req = models.TextToImageLiteRequest()
-    params = {
+    params = {"Prompt": prompt, "RspImgType": "url", "Resolution": "1080:1920"}
-        "Prompt": prompt,
-        "RspImgType": "url",
-        "Resolution": "1080:1920"
-    }
     req.from_json_string(json.dumps(params))
 
     # 返回的resp是一个TextToImageLiteResponse的实例，与请求对象对应