From 30f792dcd9955b92be1f0e4c5d7fc87584558147 Mon Sep 17 00:00:00 2001
From: Roo Code <roomote@roocode.com>
Date: Wed, 27 Aug 2025 00:23:42 +0000
Subject: [PATCH] fix: display current LLM model in task header UI

- Added model display in both collapsed and expanded task header views
- Shows model name next to token count in collapsed view
- Added dedicated Model row in expanded task details table
- Added localization support for "Model" label in chat.json

Fixes #7438
---
 webview-ui/src/components/chat/TaskHeader.tsx | 108 +++++++++++-------
 webview-ui/src/i18n/locales/en/chat.json      |   1 +
 2 files changed, 66 insertions(+), 43 deletions(-)

diff --git a/webview-ui/src/components/chat/TaskHeader.tsx b/webview-ui/src/components/chat/TaskHeader.tsx
index 41826bc048..d9e640aafa 100644
--- a/webview-ui/src/components/chat/TaskHeader.tsx
+++ b/webview-ui/src/components/chat/TaskHeader.tsx
@@ -120,51 +120,61 @@ const TaskHeader = ({
 						</div>
 					</div>
 				</div>
-				{!isTaskExpanded && contextWindow > 0 && (
+				{!isTaskExpanded && (
 					<div className="flex items-center gap-2 text-sm" onClick={(e) => e.stopPropagation()}>
-						<StandardTooltip
-							content={
-								<div className="space-y-1">
-									<div>
-										{t("chat:tokenProgress.tokensUsed", {
-											used: formatLargeNumber(contextTokens || 0),
-											total: formatLargeNumber(contextWindow),
-										})}
-									</div>
-									{(() => {
-										const maxTokens = model
-											? getModelMaxOutputTokens({ modelId, model, settings: apiConfiguration })
-											: 0
-										const reservedForOutput = maxTokens || 0
-										const availableSpace = contextWindow - (contextTokens || 0) - reservedForOutput
+						{/* Model display */}
+						{modelId && <span className="text-vscode-descriptionForeground opacity-80">{modelId}</span>}
+						{/* Token progress */}
+						{contextWindow > 0 && (
+							<StandardTooltip
+								content={
+									<div className="space-y-1">
+										<div>
+											{t("chat:tokenProgress.tokensUsed", {
+												used: formatLargeNumber(contextTokens || 0),
+												total: formatLargeNumber(contextWindow),
+											})}
+										</div>
+										{(() => {
+											const maxTokens = model
+												? getModelMaxOutputTokens({
+														modelId,
+														model,
+														settings: apiConfiguration,
+													})
+												: 0
+											const reservedForOutput = maxTokens || 0
+											const availableSpace =
+												contextWindow - (contextTokens || 0) - reservedForOutput
 
-										return (
-											<>
-												{reservedForOutput > 0 && (
-													<div>
-														{t("chat:tokenProgress.reservedForResponse", {
-															amount: formatLargeNumber(reservedForOutput),
-														})}
-													</div>
-												)}
-												{availableSpace > 0 && (
-													<div>
-														{t("chat:tokenProgress.availableSpace", {
-															amount: formatLargeNumber(availableSpace),
-														})}
-													</div>
-												)}
-											</>
-										)
-									})()}
-								</div>
-							}
-							side="top"
-							sideOffset={8}>
-							<span className="mr-1">
-								{formatLargeNumber(contextTokens || 0)} / {formatLargeNumber(contextWindow)}
-							</span>
-						</StandardTooltip>
+											return (
+												<>
+													{reservedForOutput > 0 && (
+														<div>
+															{t("chat:tokenProgress.reservedForResponse", {
+																amount: formatLargeNumber(reservedForOutput),
+															})}
+														</div>
+													)}
+													{availableSpace > 0 && (
+														<div>
+															{t("chat:tokenProgress.availableSpace", {
+																amount: formatLargeNumber(availableSpace),
+															})}
+														</div>
+													)}
+												</>
+											)
+										})()}
+									</div>
+								}
+								side="top"
+								sideOffset={8}>
+								<span className="mr-1">
+									{formatLargeNumber(contextTokens || 0)} / {formatLargeNumber(contextWindow)}
+								</span>
+							</StandardTooltip>
+						)}
 						{!!totalCost && <span>${totalCost.toFixed(2)}</span>}
 					</div>
 				)}
@@ -190,6 +200,18 @@ const TaskHeader = ({
 						<div className="border-t border-b border-vscode-panel-border/50 py-4 mt-2 mb-1">
 							<table className="w-full">
 								<tbody>
+									{/* Model row */}
+									{modelId && (
+										<tr>
+											<th className="font-bold text-left align-top w-1 whitespace-nowrap pl-1 pr-3 h-[24px]">
+												{t("chat:task.model")}
+											</th>
+											<td className="align-top">
+												<span>{modelId}</span>
+											</td>
+										</tr>
+									)}
+
 									{contextWindow > 0 && (
 										<tr>
 											<th
diff --git a/webview-ui/src/i18n/locales/en/chat.json b/webview-ui/src/i18n/locales/en/chat.json
index 74bb3b0c79..07eb7b62fc 100644
--- a/webview-ui/src/i18n/locales/en/chat.json
+++ b/webview-ui/src/i18n/locales/en/chat.json
@@ -6,6 +6,7 @@
 		"collapse": "Collapse task",
 		"seeMore": "See more",
 		"seeLess": "See less",
+		"model": "Model",
 		"tokens": "Tokens",
 		"cache": "Cache",
 		"apiCost": "API Cost",