@pixelbyte-software/pixcode 1.53.7 → 1.53.9

package/dist-server/server/services/telegram/control-center.js

@@ -9,11 +9,14 @@ const TERMINAL_RUN_STATES = new Set(['completed', 'failed', 'canceled']);
 const CALLBACK_TTL_MS = 10 * 60 * 1000;
 const MAX_CALLBACK_ACTIONS = 1000;
 const MAX_TELEGRAM_TEXT = 3600;
+const MAX_ACTIVITY_OUTPUT_CHARS = 48_000;
+const MAX_SSE_BUFFER_CHARS = 256_000;
 const ACTIVITY_EDIT_THROTTLE_MS = 1200;
 const ACTIVITY_HEARTBEAT_MS = 8000;
 const INTENT_ROUTER_TIMEOUT_MS = 45_000;
 const callbackActions = new Map();
 const runMonitors = new Map();
+const activeLongTasks = new Map();
 const MODEL_FALLBACKS = Object.fromEntries(PROVIDERS.map((provider) => [provider, getStaticProviderModels(provider)]));
 const AUTH_HELP = {
     claude: '`claude login`',
@@ -59,6 +62,15 @@ function truncate(text, max = MAX_TELEGRAM_TEXT) {
     const value = String(text || '').trim();
     return value.length > max ? `${value.slice(0, max - 20)}\n\n…truncated…` : value;
 }
+function appendBoundedText(current, chunk, maxChars = MAX_ACTIVITY_OUTPUT_CHARS) {
+    const nextChunk = String(chunk || '');
+    if (!nextChunk)
+        return { text: current || '', truncated: false };
+    const combined = `${current || ''}${nextChunk}`;
+    if (combined.length <= maxChars)
+        return { text: combined, truncated: false };
+    return { text: combined.slice(-maxChars), truncated: true };
+}
 function languageFor(link) {
     return SUPPORTED_LANGUAGES.includes(link?.language) ? link.language : 'en';
 }
@@ -278,6 +290,9 @@ async function localAgentStream(userId, body, onEvent) {
         if (done)
             break;
         buffer += decoder.decode(value, { stream: true });
+        if (buffer.length > MAX_SSE_BUFFER_CHARS) {
+            buffer = buffer.slice(-MAX_SSE_BUFFER_CHARS);
+        }
         let boundary = buffer.indexOf('\n\n');
         while (boundary !== -1) {
             const block = buffer.slice(0, boundary);
@@ -322,6 +337,7 @@ function createActivityState({ lang, type = 'agent', provider, project, prompt,
         workflowId: null,
         events: [],
         output: '',
+        outputTruncated: false,
         error: null,
     };
 }
@@ -350,11 +366,20 @@ function trimTelegramOutput(text, max, suffix = '') {
     const room = Math.max(300, max - ending.length - 4);
     return `${value.slice(0, room).trim()}\n\n${ending}`;
 }
+function appendActivityOutput(activity, chunk) {
+    if (!activity)
+        return;
+    const next = appendBoundedText(activity.output, chunk);
+    activity.output = next.text;
+    activity.outputTruncated = Boolean(activity.outputTruncated || next.truncated);
+}
 function renderActivity(activity, { finalText = null } = {}) {
     const output = finalText || activity.output;
     if (activity.type === 'agent' && output && !activity.error) {
         if (activity.status === 'done') {
-            return trimTelegramOutput(output, 3400, t(activity.lang, 'control.activity.outputTooLong'));
+            return trimTelegramOutput(output, 3400, activity.outputTruncated
+                ? t(activity.lang, 'control.activity.outputHistoryTrimmed')
+                : t(activity.lang, 'control.activity.outputTooLong'));
         }
         const footer = `⏳ ${t(activity.lang, 'control.activity.liveFooter', { elapsed: formatElapsed(activity.startedAt) })}`;
         const body = trimTelegramOutput(output, 3200, t(activity.lang, 'control.activity.outputShortened'));
@@ -760,13 +785,13 @@ function applyAgentStreamEvent(activity, event) {
     if (event.kind === 'stream_delta' || event.kind === 'text') {
         activity.status = 'running';
         activity.phase = t(activity.lang, 'control.activity.responding');
-        activity.output = `${activity.output}${extractTextFromEvent(event)}`;
+        appendActivityOutput(activity, extractTextFromEvent(event));
         return;
     }
     if (event.type === 'claude-response' && event.data?.type === 'assistant') {
         activity.status = 'running';
         activity.phase = t(activity.lang, 'control.activity.responding');
-        activity.output = `${activity.output}${extractTextFromEvent(event)}`;
+        appendActivityOutput(activity, extractTextFromEvent(event));
         return;
     }
     if (event.kind === 'tool_use') {
@@ -891,6 +916,43 @@ async function runAgent({ bot, chatId, link, prompt, activity = null }) {
         active.output = t(lang, 'control.noAssistantText');
     await editTelegramActivity({ bot, chatId, activity: active, force: true });
 }
+function longTaskKey(chatId) {
+    return String(chatId);
+}
+async function launchLongTelegramTask({ bot, chatId, link, kind, activity = null, task }) {
+    const key = longTaskKey(chatId);
+    const existing = activeLongTasks.get(key);
+    const lang = languageFor(link);
+    if (existing) {
+        await send(bot, chatId, t(lang, 'control.longTaskRunning', {
+            kind: existing.kind,
+            elapsed: formatElapsed(existing.startedAt),
+        }), {
+            editMessageId: activity?.messageId,
+            parse_mode: undefined,
+        });
+        return true;
+    }
+    const record = {
+        kind,
+        startedAt: Date.now(),
+    };
+    activeLongTasks.set(key, record);
+    Promise.resolve()
+        .then(task)
+        .catch(async (error) => {
+        console.error('[telegram-control] long task failed:', error);
+        await send(bot, chatId, t(lang, 'control.longTaskFailed'), {
+            editMessageId: activity?.messageId,
+            parse_mode: undefined,
+        }).catch(() => { });
+    })
+        .finally(() => {
+        if (activeLongTasks.get(key) === record)
+            activeLongTasks.delete(key);
+    });
+    return true;
+}
 export async function runWorkflow({ bot, chatId, link, input, activity = null }) {
     const lang = languageFor(link);
     const state = getState(link.user_id);
@@ -1097,8 +1159,14 @@ async function handleRoutedIntent({ bot, chatId, link, text, activity }) {
     const intent = await resolveTelegramAiIntent({ bot, chatId, link, text, activity });
     const editMessageId = activity?.messageId;
     if (intent.action === 'agent_prompt') {
-        await runAgent({ bot, chatId, link, prompt: intent.prompt || text, activity });
-        return true;
+        return launchLongTelegramTask({
+            bot,
+            chatId,
+            link,
+            kind: 'agent',
+            activity,
+            task: () => runAgent({ bot, chatId, link, prompt: intent.prompt || text, activity }),
+        });
     }
     if (intent.action === 'show_menu') {
         await showMainMenu({ bot, chatId, link, editMessageId });
@@ -1175,8 +1243,14 @@ async function handleRoutedIntent({ bot, chatId, link, text, activity }) {
         return true;
     }
     if (intent.action === 'run_workflow') {
-        await runWorkflow({ bot, chatId, link, input: intent.workflowInput || text, activity });
-        return true;
+        return launchLongTelegramTask({
+            bot,
+            chatId,
+            link,
+            kind: 'workflow',
+            activity,
+            task: () => runWorkflow({ bot, chatId, link, input: intent.workflowInput || text, activity }),
+        });
     }
     if (intent.action === 'show_sessions') {
         await showSessions({ bot, chatId, link, editMessageId });
@@ -1186,8 +1260,14 @@ async function handleRoutedIntent({ bot, chatId, link, text, activity }) {
         await startNewChat({ bot, chatId, link, editMessageId });
         return true;
     }
-    await runAgent({ bot, chatId, link, prompt: text, activity });
-    return true;
+    return launchLongTelegramTask({
+        bot,
+        chatId,
+        link,
+        kind: 'agent',
+        activity,
+        task: () => runAgent({ bot, chatId, link, prompt: text, activity }),
+    });
 }
 async function fetchRun(userId, runId) {
     return localApi(userId, `/api/orchestration/workflows/runs/${runId}`);
@@ -1393,12 +1473,22 @@ async function handleAwaitingInput({ bot, chatId, link, text }) {
         return false;
     updateTelegramControlState(link.user_id, { awaiting: null });
     if (awaiting.type === 'agent_prompt') {
-        await runAgent({ bot, chatId, link, prompt: text });
-        return true;
+        return launchLongTelegramTask({
+            bot,
+            chatId,
+            link,
+            kind: 'agent',
+            task: () => runAgent({ bot, chatId, link, prompt: text }),
+        });
     }
     if (awaiting.type === 'workflow_prompt') {
-        await runWorkflow({ bot, chatId, link, input: text });
-        return true;
+        return launchLongTelegramTask({
+            bot,
+            chatId,
+            link,
+            kind: 'workflow',
+            task: () => runWorkflow({ bot, chatId, link, input: text }),
+        });
     }
     return false;
 }
@@ -1494,8 +1584,13 @@ async function handleCommand({ bot, chatId, link, text }) {
             await send(bot, chatId, t(lang, 'control.sendAgentPrompt'));
             return true;
         }
-        await runAgent({ bot, chatId, link, prompt: argText });
-        return true;
+        return launchLongTelegramTask({
+            bot,
+            chatId,
+            link,
+            kind: 'agent',
+            task: () => runAgent({ bot, chatId, link, prompt: argText }),
+        });
     }
     if (command === '/workflow' || command === '/orchestrate') {
         if (!argText) {
@@ -1503,8 +1598,13 @@ async function handleCommand({ bot, chatId, link, text }) {
             await send(bot, chatId, t(lang, 'control.sendWorkflowPrompt'));
             return true;
         }
-        await runWorkflow({ bot, chatId, link, input: argText });
-        return true;
+        return launchLongTelegramTask({
+            bot,
+            chatId,
+            link,
+            kind: 'workflow',
+            task: () => runWorkflow({ bot, chatId, link, input: argText }),
+        });
     }
     if (command === '/cancel') {
         if (!argText) {
@@ -1563,6 +1663,10 @@ export async function handleTelegramControlMessage(args) {
             await send(args.bot, chatId, t(languageFor(args.link), 'error.generic')).catch(() => { });
             return true;
         }
+    }).catch(async (error) => {
+        console.error('[telegram-control] message job failed:', error);
+        await send(args.bot, chatId, t(languageFor(args.link), 'error.generic')).catch(() => { });
+        return true;
     });
 }
 async function showRunDetail({ bot, chatId, link, runId, editMessageId }) {
@@ -1769,6 +1873,11 @@ export async function handleTelegramControlCallback(args) {
             await args.bot?.answerCallbackQuery(args.query?.id, { text: t(lang, 'error.generic') }).catch(() => { });
             await send(args.bot, chatId, t(lang, 'error.generic')).catch(() => { });
         }
+    }).catch(async (error) => {
+        console.error('[telegram-control] callback job failed:', error);
+        const lang = languageFor(args.link);
+        await args.bot?.answerCallbackQuery(args.query?.id, { text: t(lang, 'error.generic') }).catch(() => { });
+        await send(args.bot, chatId, t(lang, 'error.generic')).catch(() => { });
     });
 }
 //# sourceMappingURL=control-center.js.map