index.ts

import {Chat, ChatOptions} from "./base";
import {You} from "./you";
import {Forefront} from "./forefront";

export enum Model {
    // define new model here
    You = 'you',
    Forefront = 'forefront',
}

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 Forefront(this.options))
    }

    get(model: Model): Chat | undefined {
        return this.modelMap.get(model);
    }
}