monra's picture
Add the GPT-4 model
8a8fe1d
raw
history blame
999 Bytes
import {Chat, ChatOptions} from "./base";
import {You} from "./you";
import {AiDream} from "./aidream";
import {Forefrontnew} from "./forefront";
import {Mcbbs} from "./mcbbs";
export enum Model {
// define new model here
You = 'you',
Forefront = 'forefront',
AiDream = 'aidream',
Mcbbs = 'mcbbs',
}
export class ChatModelFactory {
private modelMap: Map<Model, Chat>;
private readonly options: ChatOptions | undefined;
constructor(options?: ChatOptions) {
this.modelMap = new Map();
this.options = options;
this.init();
}
init() {
// register new model here
this.modelMap.set(Model.You, new You(this.options))
this.modelMap.set(Model.Forefront, new Forefrontnew(this.options))
this.modelMap.set(Model.AiDream, new AiDream(this.options))
this.modelMap.set(Model.Mcbbs, new Mcbbs(this.options))
}
get(model: Model): Chat | undefined {
return this.modelMap.get(model);
}
}