zlei6/ragflow
1
0
Fork 0
mirror of https://github.com/infiniflow/ragflow.git synced 2026-07-23 08:56:42 +08:00
Code Issues Packages Projects Releases Wiki Activity

Feat: Add model verify (#13005)

Browse Source 
### What problem does this PR solve?

Feat: Add model verify

### Type of change

- [x] New Feature (non-breaking change which adds functionality)

---------

Co-authored-by: Liu An <asiro@qq.com>
This commit is contained in:
chanx
2026-02-05 15:53:20 +08:00
committed by GitHub
parent 90b726c988
commit 89fdb1d498
22 changed files with 1083 additions and 220 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
web/src/hooks/use-llm-request.tsx
View File

@@ -271,11 +271,12 @@ export interface IApiKeySavingParams {
llm_name?: string;
model_type?: string;
base_url?: string;
verify?: boolean;
}
export const useSaveApiKey = () => {
const queryClient = useQueryClient();
const { t } = useTranslation();
// const { t } = useTranslation();
const {
data,
isPending: loading,
@@ -285,14 +286,14 @@ export const useSaveApiKey = () => {
mutationFn: async (params: IApiKeySavingParams) => {
const { data } = await userService.set_api_key(params);
if (data.code === 0) {
message.success(t('message.modified'));
// message.success(t('message.modified'));
queryClient.invalidateQueries({ queryKey: [LLMApiAction.MyLlmList] });
queryClient.invalidateQueries({
queryKey: [LLMApiAction.MyLlmListDetailed],
});
queryClient.invalidateQueries({ queryKey: [LLMApiAction.FactoryList] });
}
return data.code;
return data;
},
});
@@ -330,25 +331,25 @@ export const useSaveTenantInfo = () => {
export const useAddLlm = () => {
const queryClient = useQueryClient();
const { t } = useTranslation();
// const { t } = useTranslation();
const {
data,
isPending: loading,
mutateAsync,
} = useMutation({
mutationKey: [LLMApiAction.AddLlm],
mutationFn: async (params: IAddLlmRequestBody) => {
mutationFn: async (params: IAddLlmRequestBody & { verify?: boolean }) => {
const { data } = await userService.add_llm(params);
if (data.code === 0) {
if (data.code === 0 && !params.verify) {
queryClient.invalidateQueries({ queryKey: [LLMApiAction.MyLlmList] });
queryClient.invalidateQueries({
queryKey: [LLMApiAction.MyLlmListDetailed],
});
queryClient.invalidateQueries({ queryKey: [LLMApiAction.FactoryList] });
queryClient.invalidateQueries({ queryKey: [LLMApiAction.LlmList] });
message.success(t('message.modified'));
// message.success(t('message.modified'));
}
return data.code;
return data;
},
});