feat: Add Telegram bot progress updates with step callbacks

src/bot/telegram-bot.ts

@@ -5,7 +5,16 @@ import { PipelineExecutor } from '../pipelines/pipeline-executor';
 export class TelegramBot {
   private bot: Telegraf;
   private pipelineExecutor: PipelineExecutor;
-  private activePipelines: Map<number, Pipeline> = new Map();
+  private activePipelines: Map<number, { pipeline: Pipeline; chatId: number }> =
+    new Map();
+  private stepCallbacks: Map<
+    string,
+    (
+      stepName: string,
+      status: 'started' | 'completed',
+      result?: string,
+    ) => Promise<void>
+  > = new Map();
 
   constructor(token: string, mcpServerUrl: string) {
     this.bot = new Telegraf(token);
@@ -13,6 +22,39 @@ export class TelegramBot {
     this.setupHandlers();
   }
 
+  private async sendMessage(chatId: number, message: string): Promise<void> {
+    try {
+      await this.bot.telegram.sendMessage(chatId, message, {
+        parse_mode: 'HTML',
+      });
+    } catch (error) {
+      console.error('Error sending Telegram message:', error);
+    }
+  }
+
+  private setupStepCallback(
+    _pipelineId: string,
+    chatId: number,
+  ): (
+    stepName: string,
+    status: 'started' | 'completed',
+    result?: string,
+  ) => Promise<void> {
+    return async (
+      stepName: string,
+      status: 'started' | 'completed',
+      result?: string,
+    ) => {
+      const emoji = status === 'started' ? '🔄' : '✅';
+      const message =
+        status === 'started'
+          ? `<b>${stepName}</b> ${emoji}`
+          : `<b>${stepName}</b> ${emoji}\n\n<pre>${result?.substring(0, 3000) || 'Готово'}</pre>`;
+
+      await this.sendMessage(chatId, message);
+    };
+  }
+
   private setupHandlers(): void {
     this.bot.start((ctx) => {
       ctx.reply(
@@ -67,17 +109,33 @@ export class TelegramBot {
             },
           ],
         };
-        this.activePipelines.set(chatId, userPipeline);
+        const pipelineId = `pipeline_${Date.now()}_${chatId}`;
+        const stepCallback = this.setupStepCallback(pipelineId, chatId);
+        this.stepCallbacks.set(pipelineId, stepCallback);
+        this.activePipelines.set(chatId, { pipeline: userPipeline, chatId });
 
         await ctx.sendChatAction('typing');
+        await ctx.reply('🚀 Запускаю анализ...');
 
         const results = await this.pipelineExecutor.executeChain({
           pipeline: userPipeline,
           variables: {},
           systemPrompts: [],
           openAIApiKey: process.env.OPENAI_API_KEY || '',
+          onStepStart: async (stepName) => {
+            await stepCallback(stepName, 'started');
+            await ctx.sendChatAction('typing');
+          },
+          onStepComplete: async (stepName, result, isLastStep) => {
+            // Skip the completion notification for the last step as it will be sent as the final response
+            if (!isLastStep) {
+              await stepCallback(stepName, 'completed', result);
+            }
+          },
         });
 
+        this.stepCallbacks.delete(pipelineId);
+
         const lastResult = results[results.length - 1];
         if (lastResult) {
           // remove ``` in start and ``` in end