From 6da05b23c1e72ac68e0ac103f3ad5dbcaf91f983 Mon Sep 17 00:00:00 2001
From: Asankilp <asankilp@outlook.com>
Date: Wed, 1 Jan 2025 13:34:37 +0800
Subject: [PATCH] =?UTF-8?q?=E2=9C=A8=20=E6=9B=B4=E6=96=B0=E6=A8=A1?=
 =?UTF-8?q?=E5=9E=8B=E5=B8=B8=E9=87=8F?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 nonebot_plugin_marshoai/constants.py | 3 ++-
 nonebot_plugin_marshoai/marsho.py    | 2 +-
 2 files changed, 3 insertions(+), 2 deletions(-)

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":