From e1dd45e41840ddac272716f8c06dc56e255fb9a1 Mon Sep 17 00:00:00 2001
From: t8y2 <1156263951@qq.com>
Date: Fri, 22 May 2026 21:18:32 +0800
Subject: [PATCH] fix(editor): update AI mode icons

---
 .../src/components/editor/AiAssistant.vue     |  8 +++++---
 .../app-tests/aiAssistantModeMenu.test.ts     | 20 +++++++++++++++++++
 2 files changed, 25 insertions(+), 3 deletions(-)
 create mode 100644 packages/app-tests/aiAssistantModeMenu.test.ts

diff --git a/apps/desktop/src/components/editor/AiAssistant.vue b/apps/desktop/src/components/editor/AiAssistant.vue
index dc2768fcf..3c73eb8e8 100644
--- a/apps/desktop/src/components/editor/AiAssistant.vue
+++ b/apps/desktop/src/components/editor/AiAssistant.vue
@@ -954,7 +954,7 @@ const messageRenderer = computed(() => {
                 class="flex items-center gap-1 rounded-full border px-2 py-0.5 text-[11px] text-muted-foreground hover:bg-muted hover:text-foreground"
                 :title="activeModeHint"
               >
-                <Bot class="h-3 w-3" />
+                <component :is="assistantMode === 'agent' ? Bot : MessageSquarePlus" class="h-3 w-3" />
                 <span>{{ t(`ai.modes.${assistantMode}`) }}</span>
                 <svg
                   class="h-3 w-3 opacity-50"
@@ -972,7 +972,8 @@ const messageRenderer = computed(() => {
             </DropdownMenuTrigger>
             <DropdownMenuContent align="start" class="w-40">
               <DropdownMenuItem class="text-xs gap-1.5" :title="t('ai.modeHints.ask')" @click="assistantMode = 'ask'">
-                <Check class="h-3 w-3" :class="{ 'opacity-0': assistantMode !== 'ask' }" />
+                <Check class="h-3 w-3 shrink-0" :class="{ 'opacity-0': assistantMode !== 'ask' }" />
+                <MessageSquarePlus class="h-3 w-3 shrink-0 text-muted-foreground" />
                 <span>{{ t("ai.modes.ask") }}</span>
               </DropdownMenuItem>
               <DropdownMenuItem
@@ -980,7 +981,8 @@ const messageRenderer = computed(() => {
                 :title="t('ai.modeHints.agent')"
                 @click="assistantMode = 'agent'"
               >
-                <Check class="h-3 w-3" :class="{ 'opacity-0': assistantMode !== 'agent' }" />
+                <Check class="h-3 w-3 shrink-0" :class="{ 'opacity-0': assistantMode !== 'agent' }" />
+                <Bot class="h-3 w-3 shrink-0 text-muted-foreground" />
                 <span>{{ t("ai.modes.agent") }}</span>
               </DropdownMenuItem>
             </DropdownMenuContent>
diff --git a/packages/app-tests/aiAssistantModeMenu.test.ts b/packages/app-tests/aiAssistantModeMenu.test.ts
new file mode 100644
index 000000000..379b0e7b2
--- /dev/null
+++ b/packages/app-tests/aiAssistantModeMenu.test.ts
@@ -0,0 +1,20 @@
+import { readFileSync } from "node:fs";
+import assert from "node:assert/strict";
+import test from "node:test";
+
+test("AI assistant mode menu shows an icon for ask and agent options", () => {
+  const source = readFileSync("apps/desktop/src/components/editor/AiAssistant.vue", "utf8");
+
+  assert.match(
+    source,
+    /<DropdownMenuItem[\s\S]*?ai\.modeHints\.ask[\s\S]*?<Check[\s\S]*?assistantMode !== 'ask'[\s\S]*?<MessageSquarePlus class="h-3 w-3 shrink-0 text-muted-foreground"[\s\S]*?ai\.modes\.ask[\s\S]*?<\/DropdownMenuItem>/,
+  );
+  assert.match(
+    source,
+    /<DropdownMenuItem[\s\S]*?ai\.modeHints\.agent[\s\S]*?<Check[\s\S]*?assistantMode !== 'agent'[\s\S]*?<Bot class="h-3 w-3 shrink-0 text-muted-foreground"[\s\S]*?ai\.modes\.agent[\s\S]*?<\/DropdownMenuItem>/,
+  );
+  assert.match(
+    source,
+    /<component :is="assistantMode === 'agent' \? Bot : MessageSquarePlus" class="h-3 w-3" \/>/,
+  );
+});