shitton

src/commands/ollama-get-model.ts

@@ -1,15 +1,13 @@
-import {Command} from "../base/command";
-import {Message} from "typescript-telegram-bot-api";
-import {logError, replyToMessage} from "../util/utils";
-import {AiProvider} from "../model/ai-provider";
-import {formatRuntimeModelInfo} from "../ai/provider-model-runtime";
 import {Environment} from "../common/environment";
+import {AiProvider} from "../model/ai-provider";
+import {ProviderGetModelCommand} from "./provider-model-command";
 
-export class OllamaGetModel extends Command {
-    title = Environment.commandTitles.ollamaGetModel;
-    description = Environment.commandDescriptions.ollamaGetModel;
-
-    async execute(msg: Message): Promise<void> {
-        await replyToMessage({message: msg, text: await formatRuntimeModelInfo(AiProvider.OLLAMA)}).catch(logError);
+export class OllamaGetModel extends ProviderGetModelCommand {
+    constructor() {
+        super({
+            provider: AiProvider.OLLAMA,
+            title: Environment.commandTitles.ollamaGetModel,
+            description: Environment.commandDescriptions.ollamaGetModel,
+        });
     }
 }