@iletai/nzb 1.3.4 → 1.3.5

package/dist/copilot/orchestrator.js

@@ -404,7 +404,8 @@ export async function sendToOrchestrator(prompt, source, callback, onToolEvent,
                 }
                 // Log both sides of the conversation after delivery
                 try {
-                    logConversation(logRole, prompt, sourceLabel);
+                    const telegramMsgId = source.type === "telegram" ? source.messageId : undefined;
+                    logConversation(logRole, prompt, sourceLabel, telegramMsgId);
                 }
                 catch {
                     /* best-effort */

package/dist/store/db.js

@@ -33,9 +33,19 @@ export function getDb() {
         role TEXT NOT NULL CHECK(role IN ('user', 'assistant', 'system')),
         content TEXT NOT NULL,
         source TEXT NOT NULL DEFAULT 'unknown',
+        telegram_msg_id INTEGER,
         ts DATETIME DEFAULT CURRENT_TIMESTAMP
       )
     `);
+        // Migrate: add telegram_msg_id column if missing
+        try {
+            db.prepare(`SELECT telegram_msg_id FROM conversation_log LIMIT 1`).get();
+        }
+        catch {
+            db.exec(`ALTER TABLE conversation_log ADD COLUMN telegram_msg_id INTEGER`);
+        }
+        // Index for fast reply-to lookups
+        db.exec(`CREATE INDEX IF NOT EXISTS idx_conv_telegram_msg ON conversation_log (telegram_msg_id)`);
         db.exec(`
       CREATE TABLE IF NOT EXISTS memories (
         id INTEGER PRIMARY KEY AUTOINCREMENT,
@@ -60,6 +70,7 @@ export function getDb() {
           role TEXT NOT NULL CHECK(role IN ('user', 'assistant', 'system')),
           content TEXT NOT NULL,
           source TEXT NOT NULL DEFAULT 'unknown',
+          telegram_msg_id INTEGER,
           ts DATETIME DEFAULT CURRENT_TIMESTAMP
         )
       `);
@@ -83,11 +94,13 @@ export function getDb() {
             getState: db.prepare(`SELECT value FROM nzb_state WHERE key = ?`),
             setState: db.prepare(`INSERT OR REPLACE INTO nzb_state (key, value) VALUES (?, ?)`),
             deleteState: db.prepare(`DELETE FROM nzb_state WHERE key = ?`),
-            logConversation: db.prepare(`INSERT INTO conversation_log (role, content, source) VALUES (?, ?, ?)`),
+            logConversation: db.prepare(`INSERT INTO conversation_log (role, content, source, telegram_msg_id) VALUES (?, ?, ?, ?)`),
             pruneConversation: db.prepare(`DELETE FROM conversation_log WHERE id NOT IN (SELECT id FROM conversation_log ORDER BY id DESC LIMIT 200)`),
             addMemory: db.prepare(`INSERT INTO memories (category, content, source) VALUES (?, ?, ?)`),
             removeMemory: db.prepare(`DELETE FROM memories WHERE id = ?`),
             memorySummary: db.prepare(`SELECT id, category, content FROM memories ORDER BY category, last_accessed DESC`),
+            getConversationByMsgId: db.prepare(`SELECT id FROM conversation_log WHERE telegram_msg_id = ? LIMIT 1`),
+            getConversationAround: db.prepare(`SELECT role, content, source, ts FROM conversation_log WHERE id BETWEEN ? - 4 AND ? + 4 ORDER BY id ASC`),
         };
     }
     return db;
@@ -106,16 +119,38 @@ export function deleteState(key) {
     getDb(); // ensure init
     stmtCache.deleteState.run(key);
 }
-/** Log a conversation turn (user, assistant, or system). */
-export function logConversation(role, content, source) {
+/** Log a conversation turn (user, assistant, or system) with optional Telegram message ID. */
+export function logConversation(role, content, source, telegramMsgId) {
     getDb(); // ensure init
-    stmtCache.logConversation.run(role, content, source);
+    stmtCache.logConversation.run(role, content, source, telegramMsgId ?? null);
     // Keep last 200 entries to support context recovery after session loss
     logInsertCount++;
     if (logInsertCount % 50 === 0) {
         stmtCache.pruneConversation.run();
     }
 }
+/** Get conversation context around a Telegram message ID (±2 turns). */
+export function getConversationContext(telegramMsgId) {
+    getDb(); // ensure init
+    const row = stmtCache.getConversationByMsgId.get(telegramMsgId);
+    if (!row)
+        return undefined;
+    const rows = stmtCache.getConversationAround.all(row.id, row.id);
+    if (rows.length === 0)
+        return undefined;
+    return rows
+        .map((r) => {
+        const tag = r.role === "user" ? "You" : r.role === "assistant" ? "NZB" : "System";
+        const content = r.content.length > 400 ? r.content.slice(0, 400) + "…" : r.content;
+        return `${tag}: ${content}`;
+    })
+        .join("\n");
+}
+/** Update the most recent assistant log entry with its Telegram message ID (for reply-to lookups). */
+export function updateLastAssistantTelegramMsgId(telegramMsgId) {
+    const db = getDb();
+    db.prepare(`UPDATE conversation_log SET telegram_msg_id = ? WHERE id = (SELECT id FROM conversation_log WHERE role = 'assistant' ORDER BY id DESC LIMIT 1)`).run(telegramMsgId);
+}
 /** Get recent conversation history formatted for injection into system message. */
 export function getRecentConversation(limit = 20) {
     const db = getDb();

package/dist/telegram/bot.js

@@ -439,12 +439,20 @@ export function createBot() {
         const onUsage = (usage) => {
             usageInfo = usage;
         };
-        // If user replies to a message, include that message's text as context
+        // If user replies to a message, include surrounding conversation context
         let userPrompt = ctx.message.text;
         const replyMsg = ctx.message.reply_to_message;
         if (replyMsg && "text" in replyMsg && replyMsg.text) {
-            const quoted = replyMsg.text.length > 500 ? replyMsg.text.slice(0, 500) + "…" : replyMsg.text;
-            userPrompt = `[Replying to: "${quoted}"]\n\n${userPrompt}`;
+            // Try to find full conversation context around the replied message
+            const { getConversationContext } = await import("../store/db.js");
+            const context = getConversationContext(replyMsg.message_id);
+            if (context) {
+                userPrompt = `[Continuing from earlier conversation:]\n---\n${context}\n---\n\n[Your reply]: ${userPrompt}`;
+            }
+            else {
+                const quoted = replyMsg.text.length > 500 ? replyMsg.text.slice(0, 500) + "…" : replyMsg.text;
+                userPrompt = `[Replying to: "${quoted}"]\n\n${userPrompt}`;
+            }
         }
         sendToOrchestrator(userPrompt, { type: "telegram", chatId, messageId: userMessageId }, (text, done) => {
             if (done) {
@@ -505,6 +513,11 @@ export function createBot() {
                                 await bot.api.setMessageReaction(chatId, userMessageId, [{ type: "emoji", emoji: "👍" }]);
                             }
                             catch { }
+                            try {
+                                const { updateLastAssistantTelegramMsgId } = await import("../store/db.js");
+                                updateLastAssistantTelegramMsgId(placeholderMsgId);
+                            }
+                            catch { }
                             return;
                         }
                         catch {
@@ -514,6 +527,11 @@ export function createBot() {
                                     await bot.api.setMessageReaction(chatId, userMessageId, [{ type: "emoji", emoji: "👍" }]);
                                 }
                                 catch { }
+                                try {
+                                    const { updateLastAssistantTelegramMsgId } = await import("../store/db.js");
+                                    updateLastAssistantTelegramMsgId(placeholderMsgId);
+                                }
+                                catch { }
                                 return;
                             }
                             catch {

package/package.json

@@ -1,6 +1,6 @@
 {
 	"name": "@iletai/nzb",
-	"version": "1.3.4",
+	"version": "1.3.5",
 	"description": "NZB — a personal AI assistant for developers, built on the GitHub Copilot SDK",
 	"bin": {
 		"nzb": "dist/cli.js"