qwen-alpha 1.0.14 → 1.0.16

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "qwen-alpha",
-  "version": "1.0.14",
+  "version": "1.0.16",
   "description": "Telegram bot for Qwen Code integration — AI-powered code review, bug detection, and code generation",
   "main": "src/index.js",
   "bin": {

package/src/bot/handlers/message.js

@@ -55,26 +55,22 @@ async function messageHandler(ctx) {
     let contextMessages = [];
     let session = ctx.state.session;
 
-    // Если сессии нет в группе - создаём новую
-    if (!session && !isPrivate) {
-      session = sessionService.create({
-        userId,
-        chatId,
-        rootMessageId: ctx.message.message_id,
-        chatType: ctx.chat.type,
-        chatTitle: ctx.chat.title,
-      });
-      ctx.state.session = session;
-      statsService.incrementSessionCreated();
-    }
-
     if (session) {
-      // Если это reply на сообщение в сессии
+      // Если это reply на сообщение в сессии — получаем полную цепочку
       const replyToMessageId = ctx.message?.reply_to_message?.message_id;
 
       if (replyToMessageId && session.message_tree[replyToMessageId]) {
-        // Получаем цепочку сообщений
         contextMessages = sessionService.getMessageChain(session, replyToMessageId);
+      } else {
+        // Если не reply, берём последние N сообщений из сессии
+        const allMessages = Object.values(session.message_tree)
+          .sort((a, b) => new Date(a.created_at) - new Date(b.created_at))
+          .slice(-10); // Последние 10 сообщений
+        
+        contextMessages = allMessages.map(msg => ({
+          role: msg.user_id === 'bot' ? 'assistant' : 'user',
+          content: msg.text || '',
+        }));
       }
     }
 
@@ -137,9 +133,18 @@ async function messageHandler(ctx) {
         text: prompt,
         type: 'user_question',
       });
-      
-      // Добавляем ответ бота (будет добавлен после отправки)
-      // sessionService.addMessage({...})
+
+      // Добавляем ответ бота
+      const botMessageId = ctx.message.message_id + 1; // Приблизительно
+      sessionService.addMessage({
+        sessionId: session.session_id,
+        chatId,
+        messageId: botMessageId,
+        userId: 'bot',
+        text: responseText,
+        type: 'bot_response',
+        parent_id: ctx.message.message_id,
+      });
     }
     
     // Обновление статистики

package/src/bot/middleware/session.js

@@ -38,27 +38,32 @@ async function sessionMiddleware(ctx, next) {
       ctx.state.sessionKey = sessionKey;
     }
   } else {
-    // Групповой чат - поиск сессии по reply или создание новой
-    const replyToMessageId = ctx.message?.reply_to_message?.message_id;
-
-    if (replyToMessageId) {
-      // Поиск сессии по сообщению, на которое ответили
-      const session = sessionService.findByMessage(chatId, replyToMessageId);
-
-      if (session) {
-        ctx.state.session = session;
-        ctx.state.sessionKey = `chat:${chatId}`;
-        ctx.state.replyToSession = true;
-      }
+    // Групповой чат - всегда используем последнюю активную сессию
+    const chatSessions = sessionService.getChatSessions(chatId);
+    let activeSession = chatSessions.find(s => s.status === 'active');
+    
+    // Если сессии нет - создаём новую
+    if (!activeSession) {
+      activeSession = sessionService.create({
+        userId,
+        chatId,
+        rootMessageId: ctx.message?.message_id || 1,
+        chatType: ctx.chat.type,
+        chatTitle: ctx.chat.title,
+      });
+    }
+    
+    if (activeSession) {
+      ctx.state.session = activeSession;
+      ctx.state.sessionKey = `chat:${chatId}`;
     }
     
-    // Если сессии нет, ищем последнюю активную сессию чата
-    if (!ctx.state.session) {
-      const chatSessions = sessionService.getChatSessions(chatId);
-      const activeSession = chatSessions.find(s => s.status === 'active');
-      if (activeSession) {
-        ctx.state.session = activeSession;
-        ctx.state.sessionKey = `chat:${chatId}`;
+    // Если есть reply, проверяем принадлежит ли оно этой сессии
+    const replyToMessageId = ctx.message?.reply_to_message?.message_id;
+    if (replyToMessageId && activeSession) {
+      // Проверяем есть ли сообщение в дереве сессии
+      if (activeSession.message_tree[replyToMessageId]) {
+        ctx.state.replyToSession = true;
       }
     }
   }

package/src/services/qwenService.js

@@ -84,15 +84,18 @@ class QwenService {
         .map(msg => `${msg.role === 'assistant' ? 'Assistant' : 'User'}: ${msg.content}`)
         .join('\n');
 
-      fullPrompt = `${contextText}\n\nUser: ${code}`;
+      // Явно указываем что это продолжение диалога
+      fullPrompt = `Продолжи диалог. Контекст:\n${contextText}\n\nUser: ${code}`;
     }
 
     // Создаём временный файл с промптом
     const tempFile = path.join(os.tmpdir(), `qwen-alpha-${Date.now()}-${Math.random().toString(36).substr(2, 9)}.txt`);
     fs.writeFileSync(tempFile, fullPrompt, 'utf-8');
 
-    // Команда для Qwen с чтением из файла
-    const command = `cat '${tempFile}' | qwen -p "Проанализируй код и дай рекомендации" -o json`;
+    // Команда для Qwen — без системного промпта если есть контекст
+    const command = contextMessages.length > 0
+      ? `cat '${tempFile}' | qwen -o json`
+      : `cat '${tempFile}' | qwen -p "Проанализируй код и дай рекомендации" -o json`;
 
     logger.debug({ codeLength: code.length, contextLength: contextMessages.length, tempFile }, 'Running Qwen analysis');