32 lines
1.1 KiB
TypeScript
32 lines
1.1 KiB
TypeScript
import {Command} from "../base/command";
|
|
import {Message} from "typescript-telegram-bot-api";
|
|
import {logError, replyToMessage} from "../util/utils";
|
|
import {Environment} from "../common/environment";
|
|
import {googleAi} from "../index";
|
|
import {AiModelCapabilities} from "../model/ai-model-capabilities";
|
|
|
|
export class GeminiGetModel extends Command {
|
|
title = "/geminiGetModel";
|
|
description = "Get current Gemini model";
|
|
|
|
async execute(msg: Message): Promise<void> {
|
|
await replyToMessage({message: msg, text: `Текущая модель: "${Environment.GEMINI_MODEL}"`}).catch(logError);
|
|
}
|
|
|
|
async getModelCapabilities(): Promise<AiModelCapabilities | null> {
|
|
try {
|
|
const info = await googleAi.models.get({model: Environment.GEMINI_MODEL});
|
|
console.log(info);
|
|
|
|
return {
|
|
vision: {supported: true},
|
|
ocr: undefined,
|
|
thinking: {supported: info.thinking},
|
|
tools: undefined
|
|
};
|
|
} catch (e) {
|
|
logError(e);
|
|
return null;
|
|
}
|
|
}
|
|
} |