From bc86214c5ef53365deddf04d54fe12635d23fa53 Mon Sep 17 00:00:00 2001
From: Enrico Ros <enrico.ros@gmail.com>
Date: Mon, 13 Oct 2025 18:51:30 -0700
Subject: [PATCH] LLMs: xAI: grok names

---
 src/common/util/dMessageUtils.tsx | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/src/common/util/dMessageUtils.tsx b/src/common/util/dMessageUtils.tsx
index d1612da23..399084dbf 100644
--- a/src/common/util/dMessageUtils.tsx
+++ b/src/common/util/dMessageUtils.tsx
@@ -466,10 +466,11 @@ export function prettyShortChatModelName(model: string | undefined): string {
   }
   // [xAI]
   if (model.includes('grok-')) {
-    if (model.includes('grok-3') || model.includes('grok-2')) {
+    if (['grok-code', 'grok-4', 'grok-3', 'grok-2'].some(m => model.includes(m))) {
       return model
         .replace('xai-', '')
         .replace('-beta', '')
+        .replace('-non-reasoning', '')
         .split('-').map(s => s.charAt(0).toUpperCase() + s.slice(1)).join(' ');
     }
     if (model.includes('grok-beta')) return 'Grok Beta';