import {ChatCommand} from "../base/chat-command";
import {Message} from "typescript-telegram-bot-api";
import {boolToEmoji, logError, replyToMessage} from "../util/utils";
import {Environment} from "../common/environment";
import {ollama} from "../index";

export class OllamaGetModel extends ChatCommand {
    title = "/ollamaGetModel";
    description = "Ollama model info";

    async execute(msg: Message): Promise<void> {

        const showResponse = await ollama.show({model: Environment.OLLAMA_MODEL});

        const caps = showResponse.capabilities;

        const text = "```ollama\n" +
            `model: ${Environment.OLLAMA_MODEL}\n\n` +
            `vision: ${boolToEmoji(caps.includes("vision"))}\n` +
            `thinking: ${boolToEmoji(caps.includes("thinking"))}\n` +
            `tools: ${boolToEmoji(caps.includes("tools"))}`
            + "```";

        await replyToMessage({message: msg, text: text, parse_mode: "Markdown"}).catch(logError);
    }
}