36 lines
1.3 KiB
TypeScript
36 lines
1.3 KiB
TypeScript
import {ChatCommand} from "../base/chat-command";
|
|
import {Requirements} from "../base/requirements";
|
|
import {Requirement} from "../base/requirement";
|
|
import {Message} from "typescript-telegram-bot-api";
|
|
import {googleAi} from "../index";
|
|
import {logError, replyToMessage} from "../util/utils";
|
|
|
|
export class GeminiListModels extends ChatCommand {
|
|
title = "/geminiListModels";
|
|
description = "List all Gemini models";
|
|
|
|
requirements = Requirements.Build(Requirement.BOT_CREATOR);
|
|
|
|
async execute(msg: Message): Promise<void> {
|
|
try {
|
|
const listResponse = await googleAi.models.list();
|
|
console.log(listResponse);
|
|
|
|
const modelsString = listResponse.page
|
|
.sort((a, b) => a.name.localeCompare(b.name))
|
|
.map(e => `\`${e.name}\``)
|
|
.join("\n");
|
|
|
|
const text = "Доступные модели:\n\n" + modelsString;
|
|
|
|
await replyToMessage({
|
|
message: msg,
|
|
text: text,
|
|
parse_mode: "Markdown"
|
|
});
|
|
} catch (e) {
|
|
console.error(e);
|
|
await replyToMessage({message: msg, text: "Не получилось загрузить список моделей"}).catch(logError);
|
|
}
|
|
}
|
|
} |