mirror of
https://github.com/labring/FastGPT.git
synced 2025-12-26 04:32:50 +00:00
66 lines
1.7 KiB
TypeScript
66 lines
1.7 KiB
TypeScript
import { ModelTypeEnum } from 'packages/global/core/ai/constants';
|
|
|
|
export default async function setupModels() {
|
|
global.llmModelMap = new Map<string, any>();
|
|
global.embeddingModelMap = new Map<string, any>();
|
|
global.llmModelMap.set('gpt-5', {
|
|
type: ModelTypeEnum.llm,
|
|
model: 'gpt-5',
|
|
name: 'gpt-5',
|
|
avatar: 'gpt-5',
|
|
isActive: true,
|
|
isDefault: true,
|
|
isCustom: false,
|
|
requestUrl: undefined,
|
|
requestAuth: undefined,
|
|
defaultSystemChatPrompt: undefined,
|
|
fieldMap: undefined,
|
|
defaultConfig: undefined,
|
|
provider: 'OpenAI',
|
|
functionCall: false,
|
|
toolChoice: false,
|
|
maxContext: 4096,
|
|
maxResponse: 4096,
|
|
quoteMaxToken: 2048
|
|
});
|
|
global.systemDefaultModel = {
|
|
llm: {
|
|
type: ModelTypeEnum.llm,
|
|
model: 'gpt-5',
|
|
name: 'gpt-5',
|
|
avatar: 'gpt-5',
|
|
isActive: true,
|
|
isDefault: true,
|
|
isCustom: false,
|
|
requestUrl: undefined,
|
|
requestAuth: undefined,
|
|
defaultSystemChatPrompt: undefined,
|
|
fieldMap: undefined,
|
|
defaultConfig: undefined,
|
|
provider: 'OpenAI',
|
|
functionCall: false,
|
|
toolChoice: false,
|
|
maxContext: 4096,
|
|
maxResponse: 4096,
|
|
quoteMaxToken: 2048
|
|
},
|
|
embedding: {
|
|
type: ModelTypeEnum.embedding,
|
|
model: 'text-embedding-ada-002',
|
|
name: 'text-embedding-ada-002',
|
|
avatar: 'text-embedding-ada-002',
|
|
isActive: true,
|
|
isDefault: true,
|
|
isCustom: false,
|
|
requestUrl: undefined,
|
|
requestAuth: undefined,
|
|
defaultConfig: undefined,
|
|
defaultToken: 1,
|
|
maxToken: 100,
|
|
provider: 'OpenAI',
|
|
weight: 1
|
|
}
|
|
};
|
|
global.systemModelList = [global.systemDefaultModel.llm!, global.systemDefaultModel.embedding!];
|
|
}
|