gpt4free-ts/model/index.ts

import {Chat, ChatOptions} from "./base";
import {You} from "./you";
import {Mcbbs} from "./mcbbs";
import {ChatDemo} from "./chatdemo";
import {Phind} from "./phind";
import {Vita} from "./vita";
import {FakeOpen} from "./fakeopen";
import {Better} from "./better";

export enum Site {
    // define new model here
    You = 'you',
    Phind = 'phind',
    Mcbbs = 'mcbbs',
    ChatDemo = 'chatdemo',
    Vita = 'vita',
    FakeOpen = 'fakeopen',
    Better = 'better',
}

export class ChatModelFactory {
    private modelMap: Map<Site, 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(Site.You, new You(this.options))
        this.modelMap.set(Site.Phind, new Phind(this.options))
        this.modelMap.set(Site.Mcbbs, new Mcbbs(this.options))
        this.modelMap.set(Site.ChatDemo, new ChatDemo(this.options))
        this.modelMap.set(Site.Vita, new Vita(this.options))
        this.modelMap.set(Site.FakeOpen, new FakeOpen(this.options))
        this.modelMap.set(Site.Better, new Better(this.options))
    }

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