From a135ddfa487568cc45ed8fc295a85a16d7f8de43 Mon Sep 17 00:00:00 2001
From: Henry Li <henry1943@163.com>
Date: Thu, 29 Jan 2026 15:57:08 +0800
Subject: [PATCH] feat: display mode

---
 frontend/src/components/workspace/input-box.tsx | 16 ++++++++++++++++
 1 file changed, 16 insertions(+)

diff --git a/frontend/src/components/workspace/input-box.tsx b/frontend/src/components/workspace/input-box.tsx
index a65d9a2..e0605d7 100644
--- a/frontend/src/components/workspace/input-box.tsx
+++ b/frontend/src/components/workspace/input-box.tsx
@@ -270,6 +270,22 @@ export function InputBox({
               </DropdownMenuGroup>
             </PromptInputActionMenuContent>
           </PromptInputActionMenu>
+          <div className="border-accent flex items-center gap-1 rounded-md border px-1 py-0.5 text-xs font-normal first-letter:uppercase">
+            <div>
+              {context.mode === "flash" && <ZapIcon className="size-3" />}
+              {context.mode === "thinking" && (
+                <LightbulbIcon className="size-3" />
+              )}
+              {context.mode === "pro" && (
+                <GraduationCapIcon className="size-3" />
+              )}
+            </div>
+            <div>
+              {(context.mode === "flash" && t.inputBox.flashMode) ||
+                (context.mode === "thinking" && t.inputBox.reasoningMode) ||
+                (context.mode === "pro" && t.inputBox.proMode)}
+            </div>
+          </div>
         </PromptInputTools>
         <PromptInputTools>
           <ModelSelector