diff --git a/src/renderer/src/components/Popups/SelectModelPopup.tsx b/src/renderer/src/components/Popups/SelectModelPopup.tsx index 47c2506f..c8c581cc 100644 --- a/src/renderer/src/components/Popups/SelectModelPopup.tsx +++ b/src/renderer/src/components/Popups/SelectModelPopup.tsx @@ -6,7 +6,7 @@ import { useProviders } from '@renderer/hooks/useProvider' import { getModelUniqId } from '@renderer/services/model' import { Model } from '@renderer/types' import { Avatar, Divider, Empty, Input, InputRef, Menu, MenuProps, Modal } from 'antd' -import { first, reverse, sortBy, upperFirst } from 'lodash' +import { first, reverse, sortBy } from 'lodash' import { useEffect, useRef, useState } from 'react' import { useTranslation } from 'react-i18next' import styled from 'styled-components' @@ -42,7 +42,7 @@ const PopupContainer: React.FC = ({ model, resolve }) => { key: getModelUniqId(m), label: ( - {upperFirst(m?.name)} {isVisionModel(m) && } + {m?.name} {isVisionModel(m) && } ), icon: ( diff --git a/src/renderer/src/pages/home/Messages/MessageHeader.tsx b/src/renderer/src/pages/home/Messages/MessageHeader.tsx index b6be7f8c..9c908f33 100644 --- a/src/renderer/src/pages/home/Messages/MessageHeader.tsx +++ b/src/renderer/src/pages/home/Messages/MessageHeader.tsx @@ -9,7 +9,6 @@ import { Assistant, Message, Model } from '@renderer/types' import { firstLetter, removeLeadingEmoji } from '@renderer/utils' import { Avatar } from 'antd' import dayjs from 'dayjs' -import { upperFirst } from 'lodash' import { FC, useCallback, useMemo } from 'react' import { useTranslation } from 'react-i18next' import styled from 'styled-components' @@ -36,7 +35,7 @@ const MessageHeader: FC = ({ assistant, model, message }) => { const getUserName = useCallback(() => { if (isLocalAi && message.role !== 'user') return APP_NAME - if (message.role === 'assistant') return upperFirst(model?.name || model?.id) + if (message.role === 'assistant') return model?.name || model?.id || '' return userName || t('common.you') }, [message.role, model?.id, model?.name, t, userName]) diff --git a/src/renderer/src/pages/home/components/SelectModelButton.tsx b/src/renderer/src/pages/home/components/SelectModelButton.tsx index 12e3ddc8..3a18b671 100644 --- a/src/renderer/src/pages/home/components/SelectModelButton.tsx +++ b/src/renderer/src/pages/home/components/SelectModelButton.tsx @@ -6,7 +6,6 @@ import { isVisionModel } from '@renderer/config/models' import { useAssistant } from '@renderer/hooks/useAssistant' import { Assistant } from '@renderer/types' import { Button } from 'antd' -import { upperFirst } from 'lodash' import { FC } from 'react' import { useTranslation } from 'react-i18next' import styled from 'styled-components' @@ -34,7 +33,7 @@ const SelectModelButton: FC = ({ assistant }) => { return ( - {model ? upperFirst(model.name) : t('button.select_model')} + {model ? model.name : t('button.select_model')} {isVisionModel(model) && } ) diff --git a/src/renderer/src/pages/settings/ModelSettings.tsx b/src/renderer/src/pages/settings/ModelSettings.tsx index 6a3dd26a..bc26839c 100644 --- a/src/renderer/src/pages/settings/ModelSettings.tsx +++ b/src/renderer/src/pages/settings/ModelSettings.tsx @@ -4,7 +4,7 @@ import { useProviders } from '@renderer/hooks/useProvider' import { getModelUniqId, hasModel } from '@renderer/services/model' import { Model } from '@renderer/types' import { Select } from 'antd' -import { find, sortBy, upperFirst } from 'lodash' +import { find, sortBy } from 'lodash' import { FC, useMemo } from 'react' import { useTranslation } from 'react-i18next' @@ -23,7 +23,7 @@ const ModelSettings: FC = () => { label: p.isSystem ? t(`provider.${p.id}`) : p.name, title: p.name, options: sortBy(p.models, 'name').map((m) => ({ - label: upperFirst(m.name), + label: m.name, value: getModelUniqId(m) })) }))