refactor!: rewrite bot core; add AI (Ollama, Gemini), DB, new commands

2026-01-12 15:32:50 +03:00
parent 9d74ad9861
commit df9471a7e4
137 changed files with 11341 additions and 2025 deletions
@@ -0,0 +1,122 @@
+import {ChatCommand} from "../base/chat-command";
+import {Message} from "typescript-telegram-bot-api";
+import {bot, ollama} from "../index";
+import {
+    collectReplyChainText,
+    editMessageText,
+    escapeMarkdownV2Text,
+    extractText,
+    logError,
+    replyToMessage,
+    startIntervalEditor
+} from "../util/utils";
+import {Environment} from "../common/environment";
+import {MessageStore} from "../common/message-store";
+
+export class OllamaChat extends ChatCommand {
+    regexp = /^\/ollama\s([^]+)/;
+    title = "/ollama";
+    description = "talk to AI (Ollama)";
+
+    async execute(msg: Message, match?: RegExpExecArray | null): Promise<void> {
+        console.log("match", match);
+        return this.executeOllama(msg, match?.[1]);
+    }
+
+    async executeOllama(msg: Message, text: string): Promise<void> {
+        if (!text || text.trim().length === 0) return;
+
+        const chatId = msg.chat.id;
+
+        const messageParts = await collectReplyChainText(msg);
+        console.log("MESSAGE PARTS", messageParts);
+
+        const chatMessages = messageParts.map(part => {
+            return {
+                role: part.bot ? "ASSISTANT" : "USER",
+                content: extractText(part.content, Environment.BOT_PREFIX)
+            };
+        });
+        chatMessages.reverse();
+        chatMessages.unshift({role: "SYSTEM", content: Environment.SYSTEM_PROMPT});
+
+        let waitMessage: Message;
+
+        const startTime = Date.now();
+
+        try {
+            waitMessage = await bot.sendMessage({
+                chat_id: chatId,
+                text: Environment.waitText,
+                reply_parameters: {
+                    chat_id: chatId,
+                    message_id: msg.message_id
+                }
+            });
+
+            const stream = await ollama.chat({
+                model: Environment.OLLAMA_MODEL,
+                stream: true,
+                think: false,
+                keep_alive: 300,
+                messages: chatMessages
+            });
+
+            let currentText = "";
+            let shouldBreak = false;
+
+            const editor = startIntervalEditor({
+                intervalMs: 4500,
+                getText: () => currentText,
+                editFn: async (text) => {
+                    await editMessageText(chatId, waitMessage.message_id, escapeMarkdownV2Text(text), "Markdown");
+                },
+            });
+
+            try {
+                for await (const chunk of stream) {
+                    const content = chunk.message.content;
+                    currentText += content;
+
+                    const length = currentText.length;
+                    if (length > 4096) {
+                        currentText = currentText.slice(0, 4093) + "...";
+                        shouldBreak = true;
+                    }
+
+                    if (shouldBreak || chunk.done) {
+                        console.log("messageText", currentText);
+                        console.log("length", length);
+
+                        if (shouldBreak) {
+                            console.log("break", true);
+                        } else {
+                            console.log("ended", true);
+                        }
+
+                        stream.abort();
+
+                        const diff = Math.abs(Date.now() - startTime) / 1000;
+
+                        await editor.tick();
+                        await editor.stop();
+
+                        waitMessage.reply_to_message = msg;
+                        waitMessage.text = currentText;
+                        MessageStore.put(waitMessage);
+
+                        await replyToMessage(waitMessage, `⏱️ ${diff}s`);
+                        break;
+                    }
+                }
+            } finally {
+                await editor.tick();
+                await editor.stop();
+            }
+        } catch (error) {
+            console.error(error);
+            await replyToMessage(waitMessage, `Произошла ошибка!\n${error.toString()}`).catch(logError);
+        }
+        return Promise.resolve();
+    }
+}