From ce6f51f9c9bb32d41744b1254817fef8f06c2076 Mon Sep 17 00:00:00 2001 From: Asankilp Date: Thu, 21 Nov 2024 01:21:50 +0800 Subject: [PATCH] =?UTF-8?q?=E2=9C=A8=E6=96=B0=E5=A2=9E=E5=85=81=E8=AE=B8?= =?UTF-8?q?=E6=B7=BB=E5=8A=A0=E6=9B=B4=E5=A4=9A=E6=94=AF=E6=8C=81=E5=9B=BE?= =?UTF-8?q?=E7=89=87=E6=A8=A1=E5=9E=8B=E7=9A=84=E9=85=8D=E7=BD=AE=E9=A1=B9?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- README.md | 1 + nonebot_plugin_marshoai/azure.py | 2 +- nonebot_plugin_marshoai/config.py | 1 + pyproject.toml | 2 +- 4 files changed, 4 insertions(+), 2 deletions(-) diff --git a/README.md b/README.md index ad33fefb..604ea10e 100644 --- a/README.md +++ b/README.md @@ -144,6 +144,7 @@ _✨ 使用 OpenAI 标准格式 API 的聊天机器人插件 ✨_ | MARSHOAI_TEMPERATURE | 否 | 无 | 进行推理时的温度参数 | | MARSHOAI_TOP_P | 否 | 无 | 进行推理时的核采样参数 | | MARSHOAI_MAX_TOKENS | 否 | 无 | 返回消息的最大 token 数 | +| MARSHOAI_ADDITIONAL_IMAGE_MODELS | 否 | `[]` | 额外添加的支持图片的模型列表，例如`hunyuan-vision` | ## ❤ 鸣谢&版权说明 diff --git a/nonebot_plugin_marshoai/azure.py b/nonebot_plugin_marshoai/azure.py index 63e5e518..c635c95a 100644 --- a/nonebot_plugin_marshoai/azure.py +++ b/nonebot_plugin_marshoai/azure.py @@ -169,7 +169,7 @@ async def marsho(target: MsgTarget, event: Event, text: Optional[UniMsg] = None) "*你未设置自己的昵称。推荐使用'nickname [昵称]'命令设置昵称来获得个性化(可能）回答。" ).send() - is_support_image_model = model_name.lower() in SUPPORT_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: diff --git a/nonebot_plugin_marshoai/config.py b/nonebot_plugin_marshoai/config.py index cc0e7276..f9b425d3 100644 --- a/nonebot_plugin_marshoai/config.py +++ b/nonebot_plugin_marshoai/config.py @@ -28,6 +28,7 @@ class ConfigModel(BaseModel): marshoai_temperature: float | None = None marshoai_max_tokens: int | None = None marshoai_top_p: float | None = None + marshoai_additional_image_models: list = [] config: ConfigModel = get_plugin_config(ConfigModel) diff --git a/pyproject.toml b/pyproject.toml index cb1d5174..cd99c900 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -1,7 +1,7 @@ [project] name = "nonebot-plugin-marshoai" dynamic = ["version"] -description = "Nonebot2插件，调用Azure OpenAI服务实现猫娘聊天" +description = "Nonebot2插件，调用Azure OpenAI等AI服务实现猫娘聊天" readme = "README.md" requires-python = "<4.0,>=3.9" authors = [{ name = "Asankilp", email = "asankilp@outlook.com" }]