zack3d/big-agi
1
0
Fork 0
mirror of https://github.com/enricoros/big-AGI.git synced 2026-05-10 21:50:14 -07:00
Code Issues Packages Projects Releases Wiki Activity

Gemini: improve the chat message labels

Browse Source
This commit is contained in:
Enrico Ros
2025-01-07 01:44:46 -08:00
parent f1257556f3
commit da2be7be50
1 changed files with 8 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/common/util/dMessageUtils.tsx
+8
View File
@@ -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';