diff --git a/tools/ui/src/lib/components/app/models/ModelId.svelte b/tools/ui/src/lib/components/app/models/ModelId.svelte
index 0b9722a73c..de73f9255a 100644
--- a/tools/ui/src/lib/components/app/models/ModelId.svelte
+++ b/tools/ui/src/lib/components/app/models/ModelId.svelte
@@ -39,6 +39,7 @@
 		modelId,
 		showRaw = undefined,
 		showRawTooltip = false,
+		supportsThinking = false,
 		tags,
 		...rest
 	}: Props = $props();
@@ -128,7 +129,7 @@
 			{@render nameAndBadges()}
 		{/if}
 
-		{#if activeCapabilities.length > 0 || activeModalities.length > 0}
+		{#if hasModalityIcons}
 			<span class="inline-flex items-center gap-1.25 text-muted-foreground">
 				{#each activeCapabilities as capability (capability)}
 					{@const CapabilityIcon = CAPABILITY_ICONS[capability]}
diff --git a/tools/ui/src/lib/components/app/models/ModelsSelectorOption.svelte b/tools/ui/src/lib/components/app/models/ModelsSelectorOption.svelte
index 1faeb0c4a5..981eed14d2 100644
--- a/tools/ui/src/lib/components/app/models/ModelsSelectorOption.svelte
+++ b/tools/ui/src/lib/components/app/models/ModelsSelectorOption.svelte
@@ -90,6 +90,7 @@
 		{hideOrgName}
 		{modalities}
 		modelId={option.model}
+		{supportsThinking}
 		showRawTooltip
 		tags={option.tags}
 	/>