diff --git a/nonebot_plugin_marshoai/constants.py b/nonebot_plugin_marshoai/constants.py index 463756a4..9c5c98b0 100755 --- a/nonebot_plugin_marshoai/constants.py +++ b/nonebot_plugin_marshoai/constants.py @@ -23,8 +23,9 @@ SUPPORT_IMAGE_MODELS: list = [ "phi-3.5-vision-instruct", "llama-3.2-90b-vision-instruct", "llama-3.2-11b-vision-instruct", + "gemini-2.0-flash-exp", ] -REASONING_MODELS: list = ["o1-preview", "o1-mini"] +NO_SYSPROMPT_MODELS: list = ["o1", "o1-preview", "o1-mini"] INTRODUCTION: str = f"""MarshoAI-NoneBot by LiteyukiStudio 你好喵~我是一只可爱的猫娘AI,名叫小棉~🐾! 我的主页在这里哦~↓↓↓ diff --git a/nonebot_plugin_marshoai/marsho.py b/nonebot_plugin_marshoai/marsho.py index 7a67096e..0cdeafba 100644 --- a/nonebot_plugin_marshoai/marsho.py +++ b/nonebot_plugin_marshoai/marsho.py @@ -248,7 +248,7 @@ async def marsho( model_name.lower() in SUPPORT_IMAGE_MODELS + config.marshoai_additional_image_models ) - is_reasoning_model = model_name.lower() in REASONING_MODELS + is_reasoning_model = model_name.lower() in NO_SYSPROMPT_MODELS usermsg = [] if is_support_image_model else "" for i in text: # type: ignore if i.type == "text":