Improve short model naming

enricoros · Jul 9, 2023 · c01b1da · c01b1da · vercel · Jul 9, 2023
1 parent 1c7f700
commit c01b1da
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/src/apps/chat/components/message/ChatMessage.tsx b/src/apps/chat/components/message/ChatMessage.tsx
@@ -280,7 +280,7 @@ export function ChatMessage(props: { message: DMessage, isBottom: boolean, onMes
 
       {/* Avatar */}
       {showAvatars && <Stack
-        sx={{ alignItems: 'center', minWidth: { xs: 50, md: 64 }, textAlign: 'center' }}
+        sx={{ alignItems: 'center', minWidth: { xs: 50, md: 64 }, maxWidth: 80, textAlign: 'center' }}
         onMouseEnter={() => setIsHovering(true)} onMouseLeave={() => setIsHovering(false)}
         onClick={event => setMenuAnchor(event.currentTarget)}>
 
@@ -296,7 +296,7 @@ export function ChatMessage(props: { message: DMessage, isBottom: boolean, onMes
           <Tooltip title={messageOriginLLM || 'unk-model'} variant='solid'>
             <Typography level='body2' sx={{
               fontSize: { xs: 'xs', sm: 'sm' }, fontWeight: 500,
-              lineBreak: 'anywhere',
+              overflowWrap: 'anywhere',
               ...(messageTyping ? { animation: `${cssRainbowColorKeyframes} 5s linear infinite` } : {}),
             }}>
               {prettyBaseModel(messageOriginLLM)}

diff --git a/src/common/util/conversationToMarkdown.ts b/src/common/util/conversationToMarkdown.ts
@@ -6,7 +6,9 @@ export function prettyBaseModel(model: string | undefined): string {
   if (!model) return '';
   if (model.includes('gpt-4-32k')) return 'gpt-4-32k';
   if (model.includes('gpt-4')) return 'gpt-4';
+  if (model.includes('gpt-3.5-turbo-16k')) return '3.5 Turbo 16k';
   if (model.includes('gpt-3.5-turbo')) return '3.5 Turbo';
+  if (model.endsWith('.bin')) return model.slice(0, -4);
   return model;
 }