pengjunjie/ragflow
1
0
Fork 1
mirror of https://github.com/infiniflow/ragflow.git synced 2025-12-08 20:42:30 +08:00
Code Issues Packages Projects Releases Wiki Activity

feat: add SystemModelSettingModal (#127)

Browse Source 
* feat: add the model

* feat: add SystemModelSettingModal
This commit is contained in:
balibabu
2024-03-15 19:35:59 +08:00
committed by GitHub
parent 2447f95629
commit a0f1e1fa95
10 changed files with 311 additions and 66 deletions
Download Patch File Download Diff File Expand all files Collapse all files

44
web/src/hooks/llmHooks.ts
View File

@ -7,7 +7,7 @@ import {
import { useCallback, useEffect, useMemo } from 'react';
import { useDispatch, useSelector } from 'umi';
export const useFetchLlmList = (modelType: LlmModelType) => {
export const useFetchLlmList = (modelType?: LlmModelType) => {
const dispatch = useDispatch();
const fetchLlmList = useCallback(() => {
@ -85,19 +85,20 @@ export const useFetchLlmFactoryListOnMount = () => {
return list;
};
export type LlmItem = { name: string; logo: string } & IMyLlmValue;
export const useFetchMyLlmListOnMount = () => {
const dispatch = useDispatch();
const llmList = useSelectMyLlmList();
const factoryList = useSelectLlmFactoryList();
const list: Array<{ name: string; logo: string } & IMyLlmValue> =
useMemo(() => {
return Object.entries(llmList).map(([key, value]) => ({
name: key,
logo: factoryList.find((x) => x.name === key)?.logo ?? '',
...value,
}));
}, [llmList, factoryList]);
const list: Array<LlmItem> = useMemo(() => {
return Object.entries(llmList).map(([key, value]) => ({
name: key,
logo: factoryList.find((x) => x.name === key)?.logo ?? '',
...value,
}));
}, [llmList, factoryList]);
const fetchMyLlmList = useCallback(() => {
dispatch({
@ -135,3 +136,28 @@ export const useSaveApiKey = () => {
return saveApiKey;
};
export interface ISystemModelSettingSavingParams {
tenant_id: string;
name?: string;
asr_id: string;
embd_id: string;
img2txt_id: string;
llm_id: string;
}
export const useSaveTenantInfo = () => {
const dispatch = useDispatch();
const saveTenantInfo = useCallback(
(savingParams: ISystemModelSettingSavingParams) => {
return dispatch<any>({
type: 'settingModel/set_tenant_info',
payload: savingParams,
});
},
[dispatch],
);
return saveTenantInfo;
};