129 lines
2.9 KiB
TypeScript
129 lines
2.9 KiB
TypeScript
import type { ServiceName, ModelDataType, ModelSchema } from '@/types/mongoSchema';
|
||
import type { RedisModelDataItemType } from '@/types/redis';
|
||
|
||
export enum ChatModelNameEnum {
|
||
GPT35 = 'gpt-3.5-turbo',
|
||
VECTOR_GPT = 'VECTOR_GPT',
|
||
GPT3 = 'text-davinci-003',
|
||
VECTOR = 'text-embedding-ada-002'
|
||
}
|
||
|
||
export const ChatModelNameMap = {
|
||
[ChatModelNameEnum.GPT35]: 'gpt-3.5-turbo',
|
||
[ChatModelNameEnum.VECTOR_GPT]: 'gpt-3.5-turbo',
|
||
[ChatModelNameEnum.GPT3]: 'text-davinci-003',
|
||
[ChatModelNameEnum.VECTOR]: 'text-embedding-ada-002'
|
||
};
|
||
|
||
export type ModelConstantsData = {
|
||
serviceCompany: `${ServiceName}`;
|
||
name: string;
|
||
model: `${ChatModelNameEnum}`;
|
||
trainName: string; // 空字符串代表不能训练
|
||
maxToken: number;
|
||
contextMaxToken: number;
|
||
maxTemperature: number;
|
||
trainedMaxToken: number; // 训练后最大多少tokens
|
||
price: number; // 多少钱 / 1token，单位: 0.00001元
|
||
};
|
||
|
||
export const modelList: ModelConstantsData[] = [
|
||
{
|
||
serviceCompany: 'openai',
|
||
name: 'chatGPT',
|
||
model: ChatModelNameEnum.GPT35,
|
||
trainName: '',
|
||
maxToken: 4000,
|
||
contextMaxToken: 7500,
|
||
trainedMaxToken: 2000,
|
||
maxTemperature: 2,
|
||
price: 3
|
||
},
|
||
{
|
||
serviceCompany: 'openai',
|
||
name: '知识库',
|
||
model: ChatModelNameEnum.VECTOR_GPT,
|
||
trainName: 'vector',
|
||
maxToken: 4000,
|
||
contextMaxToken: 7500,
|
||
trainedMaxToken: 2000,
|
||
maxTemperature: 1,
|
||
price: 3
|
||
}
|
||
// {
|
||
// serviceCompany: 'openai',
|
||
// name: 'GPT3',
|
||
// model: ChatModelNameEnum.GPT3,
|
||
// trainName: 'davinci',
|
||
// maxToken: 4000,
|
||
// contextMaxToken: 7500,
|
||
// trainedMaxToken: 2000,
|
||
// maxTemperature: 2,
|
||
// price: 30
|
||
// }
|
||
];
|
||
|
||
export enum TrainingStatusEnum {
|
||
pending = 'pending',
|
||
succeed = 'succeed',
|
||
errored = 'errored',
|
||
canceled = 'canceled'
|
||
}
|
||
|
||
export enum ModelStatusEnum {
|
||
running = 'running',
|
||
training = 'training',
|
||
pending = 'pending',
|
||
closed = 'closed'
|
||
}
|
||
|
||
export const formatModelStatus = {
|
||
[ModelStatusEnum.running]: {
|
||
colorTheme: 'green',
|
||
text: '运行中'
|
||
},
|
||
[ModelStatusEnum.training]: {
|
||
colorTheme: 'blue',
|
||
text: '训练中'
|
||
},
|
||
[ModelStatusEnum.pending]: {
|
||
colorTheme: 'gray',
|
||
text: '加载中'
|
||
},
|
||
[ModelStatusEnum.closed]: {
|
||
colorTheme: 'red',
|
||
text: '已关闭'
|
||
}
|
||
};
|
||
|
||
export const ModelDataStatusMap: Record<RedisModelDataItemType['status'], string> = {
|
||
ready: '训练完成',
|
||
waiting: '训练中'
|
||
};
|
||
|
||
export const defaultModel: ModelSchema = {
|
||
_id: '',
|
||
userId: '',
|
||
name: '',
|
||
avatar: '',
|
||
status: ModelStatusEnum.pending,
|
||
updateTime: Date.now(),
|
||
trainingTimes: 0,
|
||
systemPrompt: '',
|
||
intro: '',
|
||
temperature: 5,
|
||
service: {
|
||
company: 'openai',
|
||
trainId: '',
|
||
chatModel: ChatModelNameEnum.GPT35,
|
||
modelName: ChatModelNameEnum.GPT35
|
||
},
|
||
security: {
|
||
domain: ['*'],
|
||
contextMaxLen: 1,
|
||
contentMaxLen: 1,
|
||
expiredTime: 9999,
|
||
maxLoadAmount: 1
|
||
}
|
||
};
|