Feat: Hide the suffix of the large model name. #5433 (#5494)

### What problem does this PR solve?

Feat: Hide the suffix of the large model name. #5433

### Type of change


- [x] New Feature (non-breaking change which adds functionality)
This commit is contained in:
balibabu 2025-02-28 18:02:33 +08:00 committed by GitHub
parent 21943ce0e2
commit 7600ebd263
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 9 additions and 3 deletions

View File

@ -14,7 +14,7 @@ import {
} from '@/interfaces/request/llm'; } from '@/interfaces/request/llm';
import userService from '@/services/user-service'; import userService from '@/services/user-service';
import { sortLLmFactoryListBySpecifiedOrder } from '@/utils/common-util'; import { sortLLmFactoryListBySpecifiedOrder } from '@/utils/common-util';
import { getLLMIconName } from '@/utils/llm-util'; import { getLLMIconName, getRealModelName } from '@/utils/llm-util';
import { useMutation, useQuery, useQueryClient } from '@tanstack/react-query'; import { useMutation, useQuery, useQueryClient } from '@tanstack/react-query';
import { Flex, message } from 'antd'; import { Flex, message } from 'antd';
import { DefaultOptionType } from 'antd/es/select'; import { DefaultOptionType } from 'antd/es/select';
@ -45,7 +45,7 @@ export const useSelectLlmOptions = () => {
return { return {
label: key, label: key,
options: value.map((x) => ({ options: value.map((x) => ({
label: x.llm_name, label: getRealModelName(x.llm_name),
value: `${x.llm_name}@${x.fid}`, value: `${x.llm_name}@${x.fid}`,
disabled: !x.available, disabled: !x.available,
})), })),
@ -66,7 +66,7 @@ function buildLlmOptionsWithIcon(x: IThirdOAIModel) {
height={26} height={26}
size={'small'} size={'small'}
/> />
<span>{x.llm_name}</span> <span>{getRealModelName(x.llm_name)}</span>
</Flex> </Flex>
), ),
value: `${x.llm_name}@${x.fid}`, value: `${x.llm_name}@${x.fid}`,
@ -198,6 +198,7 @@ export const useSelectLlmList = () => {
name: key, name: key,
logo: factoryList.find((x) => x.name === key)?.logo ?? '', logo: factoryList.find((x) => x.name === key)?.logo ?? '',
...value, ...value,
llm: value.llm.map((x) => ({ ...x, name: getRealModelName(x.name) })),
})); }));
}, [myLlmList, factoryList]); }, [myLlmList, factoryList]);

View File

@ -11,3 +11,8 @@ export const getLlmNameAndFIdByLlmId = (llmId?: string) => {
return { fId, llmName }; return { fId, llmName };
}; };
// The names of the large models returned by the interface are similar to "deepseek-r1___OpenAI-API"
export function getRealModelName(llmName: string) {
return llmName.split('__').at(0) ?? '';
}