From da2be7be50b151bc29d240e818c5b7d392ccc377 Mon Sep 17 00:00:00 2001
From: Enrico Ros <enrico.ros@gmail.com>
Date: Tue, 7 Jan 2025 01:44:46 -0800
Subject: [PATCH] Gemini: improve the chat message labels

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

diff --git a/src/common/util/dMessageUtils.tsx b/src/common/util/dMessageUtils.tsx
index 1ce413e3d..687d71206 100644
--- a/src/common/util/dMessageUtils.tsx
+++ b/src/common/util/dMessageUtils.tsx
@@ -344,6 +344,14 @@ export function prettyShortChatModelName(model: string | undefined): string {
   // [Anthropic]
   const prettyAnthropic = _prettyAnthropicModelName(model);
   if (prettyAnthropic) return prettyAnthropic;
+  // [Gemini]
+  if (model.includes('gemini-')) {
+    return model.replaceAll('-', ' ')
+      .replace('gemini', 'Gemini')
+      .replace('pro', 'Pro')
+      .replace('flash', 'Flash')
+      .replace('thinking', 'Thinking');
+  }
   // [Deepseek]
   if (model.includes('deepseek-chat')) return 'Deepseek Chat';
   if (model.includes('deepseek-coder')) return 'Deepseek Coder';