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" }]