import {Message} from "typescript-telegram-bot-api";
import {ChatCommand} from "../base/chat-command";
import {Requirements} from "../base/requirements";
import {Requirement} from "../base/requirement";
import {AiProvider} from "../model/ai-provider";
import {runUnifiedAi} from "../ai/unified-ai-runner";
import {Environment} from "../common/environment";

export class MistralChat extends ChatCommand {
    command = ["mistral", "mistral-chat"];
    argsMode = "required" as const;

    requirements = Requirements.Build(Requirement.BOT_CREATOR);

    title = Environment.commandTitles.mistralChat;
    description = Environment.commandDescriptions.mistralChat;

    async execute(msg: Message, match?: RegExpExecArray): Promise<void> {
        return this.executeMistral(msg, match?.[3] || "");
    }

    async executeMistral(msg: Message, text: string, stream: boolean = true): Promise<void> {
        await runUnifiedAi({provider: AiProvider.MISTRAL, msg, text, stream});
    }
}