e009be51e7
* model config * feat: model config ui * perf: rename variable * feat: custom request url * perf: model buffer * perf: init model * feat: json model config * auto login * fix: ts * update packages * package * fix: dockerfile
55 lines
1.9 KiB
TypeScript
55 lines
1.9 KiB
TypeScript
import { SystemModelItemType } from './type';
|
|
|
|
export const getFirstLLMModel = () => {
|
|
return Array.from(global.llmModelMap.values())[0];
|
|
};
|
|
export const getLLMModel = (model?: string) => {
|
|
if (!model) return getFirstLLMModel();
|
|
return global.llmModelMap.get(model) || getFirstLLMModel();
|
|
};
|
|
|
|
export const getDatasetModel = (model?: string) => {
|
|
return (
|
|
Array.from(global.llmModelMap.values())
|
|
?.filter((item) => item.datasetProcess)
|
|
?.find((item) => item.model === model || item.name === model) ?? getFirstLLMModel()
|
|
);
|
|
};
|
|
|
|
export const getFirstEmbeddingModel = () => Array.from(global.embeddingModelMap.values())[0];
|
|
export const getEmbeddingModel = (model?: string) => {
|
|
if (!model) return getFirstEmbeddingModel();
|
|
return global.embeddingModelMap.get(model) || getFirstEmbeddingModel();
|
|
};
|
|
|
|
export const getFirstTTSModel = () => Array.from(global.ttsModelMap.values())[0];
|
|
export function getTTSModel(model?: string) {
|
|
if (!model) return getFirstTTSModel();
|
|
return global.ttsModelMap.get(model) || getFirstTTSModel();
|
|
}
|
|
|
|
export const getFirstSTTModel = () => Array.from(global.sttModelMap.values())[0];
|
|
export function getSTTModel(model?: string) {
|
|
if (!model) return getFirstSTTModel();
|
|
return global.sttModelMap.get(model) || getFirstSTTModel();
|
|
}
|
|
|
|
export const getFirstReRankModel = () => Array.from(global.reRankModelMap.values())[0];
|
|
export function getReRankModel(model?: string) {
|
|
if (!model) return getFirstReRankModel();
|
|
return global.reRankModelMap.get(model) || getFirstReRankModel();
|
|
}
|
|
|
|
export const findAIModel = (model: string): SystemModelItemType | undefined => {
|
|
return (
|
|
global.llmModelMap.get(model) ||
|
|
global.embeddingModelMap.get(model) ||
|
|
global.ttsModelMap.get(model) ||
|
|
global.sttModelMap.get(model) ||
|
|
global.reRankModelMap.get(model)
|
|
);
|
|
};
|
|
export const findModelFromAlldata = (model: string) => {
|
|
return global.systemModelList.find((item) => item.model === model);
|
|
};
|