mirror of
https://git.mirrors.martin98.com/https://github.com/infiniflow/ragflow.git
synced 2025-06-30 16:55:10 +08:00
### 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:
parent
21943ce0e2
commit
7600ebd263
@ -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]);
|
||||||
|
|
||||||
|
@ -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) ?? '';
|
||||||
|
}
|
||||||
|
Loading…
x
Reference in New Issue
Block a user