@grinev/opencode-telegram-bot 0.15.0 → 0.16.1

package/dist/bot/streaming/response-streamer.js

@@ -2,16 +2,22 @@ import { logger } from "../../utils/logger.js";
 function buildStateKey(sessionId, messageId) {
     return `${sessionId}:${messageId}`;
 }
+function clonePart(part) {
+    return {
+        text: part.text,
+        entities: part.entities ? [...part.entities] : undefined,
+        fallbackText: part.fallbackText,
+        source: part.source,
+    };
+}
 function normalizePayload(payload) {
-    const normalizedParts = payload.parts
-        .map((part) => part.trim())
-        .filter((part) => part.length > 0);
+    const normalizedParts = payload.parts.map(clonePart).filter((part) => part.text.length > 0);
     if (normalizedParts.length === 0) {
+        logger.debug("[ResponseStreamer] Dropped empty streaming payload after normalization");
         return null;
     }
     return {
         parts: normalizedParts,
-        format: payload.format,
         sendOptions: payload.sendOptions,
         editOptions: payload.editOptions,
     };
@@ -37,8 +43,8 @@ function getRetryAfterMs(error) {
     }
     return seconds * 1000;
 }
-function createSignature(text, format) {
-    return `${format}\n${text}`;
+function createSignature(part) {
+    return `${part.text}\n${JSON.stringify(part.entities ?? null)}`;
 }
 function delay(ms) {
     return new Promise((resolve) => {
@@ -47,14 +53,14 @@ function delay(ms) {
 }
 export class ResponseStreamer {
     throttleMs;
-    sendText;
-    editText;
+    sendPart;
+    editPart;
     deleteText;
     states = new Map();
     constructor(options) {
         this.throttleMs = Math.max(0, Math.floor(options.throttleMs));
-        this.sendText = options.sendText;
-        this.editText = options.editText;
+        this.sendPart = options.sendPart;
+        this.editPart = options.editPart;
         this.deleteText = options.deleteText;
     }
     enqueue(sessionId, messageId, payload) {
@@ -73,6 +79,7 @@ export class ResponseStreamer {
         const notStreamed = { streamed: false, telegramMessageIds: [] };
         const state = this.states.get(buildStateKey(sessionId, messageId));
         if (!state) {
+            logger.debug(`[ResponseStreamer] Complete skipped, no active stream state: session=${sessionId}, message=${messageId}`);
             return notStreamed;
         }
         if (payload) {
@@ -90,6 +97,7 @@ export class ResponseStreamer {
             return notStreamed;
         }
         if (state.telegramMessageIds.length === 0) {
+            logger.debug(`[ResponseStreamer] Complete returned not streamed: session=${sessionId}, message=${messageId}, reason=no_visible_partials`);
             this.cancelState(state);
             this.states.delete(state.key);
             return notStreamed;
@@ -204,10 +212,11 @@ export class ResponseStreamer {
             if (!payload) {
                 return state.telegramMessageIds.length > 0;
             }
-            const targetSignatures = payload.parts.map((part) => createSignature(part, payload.format));
+            const targetSignatures = payload.parts.map((part) => createSignature(part));
             const unchanged = targetSignatures.length === state.lastSentSignatures.length &&
                 targetSignatures.every((signature, index) => signature === state.lastSentSignatures[index]);
             if (unchanged) {
+                logger.debug(`[ResponseStreamer] Skipped unchanged payload: session=${state.sessionId}, message=${state.messageId}, parts=${payload.parts.length}`);
                 return state.telegramMessageIds.length > 0;
             }
             try {
@@ -259,20 +268,20 @@ export class ResponseStreamer {
     }
     async syncMessages(state, payload, targetSignatures) {
         for (let index = 0; index < payload.parts.length; index++) {
-            const text = payload.parts[index];
+            const part = payload.parts[index];
             const nextSignature = targetSignatures[index];
             const currentMessageId = state.telegramMessageIds[index];
             if (currentMessageId) {
                 if (state.lastSentSignatures[index] === nextSignature) {
                     continue;
                 }
-                await this.editText(currentMessageId, text, payload.format, payload.editOptions);
-                state.lastSentSignatures[index] = nextSignature;
+                const result = await this.editPart(currentMessageId, part, payload.editOptions);
+                state.lastSentSignatures[index] = result.deliveredSignature;
                 continue;
             }
-            const messageId = await this.sendText(text, payload.format, payload.sendOptions);
-            state.telegramMessageIds[index] = messageId;
-            state.lastSentSignatures[index] = nextSignature;
+            const result = await this.sendPart(part, payload.sendOptions);
+            state.telegramMessageIds[index] = result.messageId;
+            state.lastSentSignatures[index] = result.deliveredSignature;
         }
         for (let index = state.telegramMessageIds.length - 1; index >= payload.parts.length; index--) {
             const messageId = state.telegramMessageIds[index];

package/dist/bot/utils/assistant-rendering.js

@@ -0,0 +1,117 @@
+import { config } from "../../config.js";
+import { logger } from "../../utils/logger.js";
+import { chunkTelegramRenderedBlocks } from "../../telegram/render/chunker.js";
+import { renderTelegramBlocks, renderTelegramParts } from "../../telegram/render/pipeline.js";
+export function createPlainRenderedBlock(text) {
+    return {
+        blockType: "plain",
+        mode: "plain",
+        text,
+        fallbackText: text,
+        source: "plain",
+    };
+}
+export function createPlainRenderedParts(text, maxPartLength) {
+    return chunkTelegramRenderedBlocks([createPlainRenderedBlock(text)], { maxPartLength });
+}
+function useAssistantEntitiesFormat() {
+    return config.bot.messageFormatMode === "markdown";
+}
+function renderAssistantBlocksSafe(text) {
+    if (!text) {
+        return [];
+    }
+    try {
+        return renderTelegramBlocks(text);
+    }
+    catch (error) {
+        logger.warn("[AssistantRender] Block rendering failed, falling back to plain streaming block", error);
+        return [createPlainRenderedBlock(text)];
+    }
+}
+export function renderAssistantFinalPartsSafe(text, maxPartLength = 4096) {
+    if (!text) {
+        return [];
+    }
+    const formatMode = useAssistantEntitiesFormat() ? "entities" : "raw";
+    if (!useAssistantEntitiesFormat()) {
+        const parts = createPlainRenderedParts(text, maxPartLength);
+        logger.debug("[AssistantRender] Built final assistant parts in raw mode", {
+            formatMode,
+            textLength: text.length,
+            partCount: parts.length,
+        });
+        return parts;
+    }
+    try {
+        const parts = renderTelegramParts(text, { maxPartLength });
+        logger.debug("[AssistantRender] Built final assistant parts in entities mode", {
+            formatMode,
+            textLength: text.length,
+            partCount: parts.length,
+            richParts: parts.filter((part) => part.source === "entities").length,
+            plainParts: parts.filter((part) => part.source === "plain").length,
+        });
+        return parts;
+    }
+    catch (error) {
+        logger.warn("[AssistantRender] Part rendering failed, falling back to plain text parts", error);
+        const parts = createPlainRenderedParts(text, maxPartLength);
+        logger.debug("[AssistantRender] Built final assistant parts in raw fallback mode", {
+            formatMode,
+            textLength: text.length,
+            partCount: parts.length,
+        });
+        return parts;
+    }
+}
+function getStableStreamingBoundary(messageText) {
+    if (!messageText) {
+        return 0;
+    }
+    if (messageText.endsWith("\n\n")) {
+        return messageText.length;
+    }
+    const lastBlockSeparatorIndex = messageText.lastIndexOf("\n\n");
+    return lastBlockSeparatorIndex >= 0 ? lastBlockSeparatorIndex + 2 : 0;
+}
+export function prepareAssistantStreamingPayload(messageText, maxPartLength) {
+    if (!messageText) {
+        return null;
+    }
+    const formatMode = useAssistantEntitiesFormat() ? "entities" : "raw";
+    if (!useAssistantEntitiesFormat()) {
+        const parts = createPlainRenderedParts(messageText, maxPartLength);
+        logger.debug("[AssistantRender] Built streaming assistant payload in raw mode", {
+            formatMode,
+            textLength: messageText.length,
+            partCount: parts.length,
+        });
+        return parts.length > 0 ? { parts } : null;
+    }
+    const stableBoundary = getStableStreamingBoundary(messageText);
+    const blocks = [];
+    if (stableBoundary > 0) {
+        blocks.push(...renderAssistantBlocksSafe(messageText.slice(0, stableBoundary)));
+    }
+    const unstableTail = stableBoundary > 0 ? messageText.slice(stableBoundary) : messageText;
+    if (unstableTail) {
+        blocks.push(createPlainRenderedBlock(unstableTail));
+    }
+    const parts = chunkTelegramRenderedBlocks(blocks, { maxPartLength });
+    logger.debug("[AssistantRender] Built streaming assistant payload in entities mode", {
+        formatMode,
+        textLength: messageText.length,
+        stableBoundary,
+        tailLength: unstableTail.length,
+        blockCount: blocks.length,
+        partCount: parts.length,
+        richParts: parts.filter((part) => part.source === "entities").length,
+        plainParts: parts.filter((part) => part.source === "plain").length,
+    });
+    return parts.length > 0 ? { parts } : null;
+}
+export function prepareAssistantFinalStreamingPayload(messageText, maxPartLength) {
+    const parts = renderAssistantFinalPartsSafe(messageText, maxPartLength);
+    return parts.length > 0 ? { parts } : null;
+}

package/dist/bot/utils/assistant-run-footer.js

@@ -0,0 +1,9 @@
+import { getAgentDisplayName } from "../../agent/types.js";
+function formatElapsedSeconds(elapsedMs) {
+    const safeElapsedMs = Math.max(0, elapsedMs);
+    return `${(safeElapsedMs / 1000).toFixed(1)}s`;
+}
+export function formatAssistantRunFooter({ agent, providerID, modelID, elapsedMs, }) {
+    const agentDisplay = getAgentDisplayName(agent);
+    return `${agentDisplay} · 🤖 ${providerID}/${modelID} · 🕒 ${formatElapsedSeconds(elapsedMs)}`;
+}

package/dist/bot/utils/browser-roots.js

@@ -0,0 +1,140 @@
+import path from "node:path";
+import { realpath } from "node:fs/promises";
+import os from "node:os";
+import { logger } from "../../utils/logger.js";
+/**
+ * Allowed root directories for the `/open` command.
+ *
+ * Configured via `OPEN_BROWSER_ROOTS` env var (comma-separated absolute paths).
+ * Defaults to `[os.homedir()]` when the variable is not set.
+ *
+ * All navigation in the directory browser is restricted to stay within one
+ * of the configured roots. This prevents the bot from browsing arbitrary
+ * locations on the filesystem.
+ */
+let resolvedRoots = null;
+function isWindows() {
+    return process.platform === "win32";
+}
+function resolveConfiguredPath(p) {
+    return path.resolve(expandTilde(p));
+}
+/**
+ * Expand a leading `~` or `~/` to the user's home directory.
+ * Does not handle `~user/` syntax — only the current user's home.
+ */
+function expandTilde(p) {
+    if (p === "~") {
+        return os.homedir();
+    }
+    if (p.startsWith("~/") || p.startsWith("~\\")) {
+        return path.join(os.homedir(), p.slice(2));
+    }
+    return p;
+}
+/**
+ * Normalize a path for comparison: expand tilde, resolve, then lowercase on Windows.
+ */
+function normalizePath(p) {
+    const resolved = resolveConfiguredPath(p);
+    return isWindows() ? resolved.toLowerCase() : resolved;
+}
+/**
+ * Initialize (or re-initialize) the allowed browser roots from the raw
+ * env string. Each entry is resolved to an absolute path via `path.resolve`.
+ *
+ * Entries that cannot be resolved are logged and skipped.
+ */
+export function initBrowserRoots(raw) {
+    if (!raw || raw.trim() === "") {
+        resolvedRoots = [resolveConfiguredPath(os.homedir())];
+        logger.debug(`[BrowserRoots] No OPEN_BROWSER_ROOTS configured, defaulting to home: ${resolvedRoots[0]}`);
+        return;
+    }
+    const entries = raw
+        .split(",")
+        .map((s) => s.trim())
+        .filter((s) => s.length > 0);
+    const roots = [];
+    for (const entry of entries) {
+        roots.push(resolveConfiguredPath(entry));
+    }
+    if (roots.length === 0) {
+        resolvedRoots = [resolveConfiguredPath(os.homedir())];
+        logger.warn("[BrowserRoots] All configured roots were invalid, falling back to home directory");
+    }
+    else {
+        resolvedRoots = roots;
+        logger.info(`[BrowserRoots] Configured roots: ${roots.join(", ")}`);
+    }
+}
+/**
+ * Return the configured browser roots. Lazily initializes from env if
+ * `initBrowserRoots` was never called.
+ */
+export function getBrowserRoots() {
+    if (resolvedRoots === null) {
+        initBrowserRoots(process.env.OPEN_BROWSER_ROOTS);
+    }
+    return resolvedRoots;
+}
+/**
+ * Return the original (non-normalized) root paths for display/navigation.
+ * On Windows the display roots are lowercased because `normalizePath` is
+ * used during init — this is acceptable for button labels.
+ */
+export function getBrowserRootPaths() {
+    return getBrowserRoots();
+}
+/**
+ * Check whether a target path is inside one of the allowed roots.
+ *
+ * Uses `path.resolve` on the target for consistent comparison, then
+ * checks that the target equals a root or is a descendant (starts with
+ * root + separator).
+ *
+ * For full symlink-proof validation, callers can optionally resolve the
+ * target with `fs.realpath` before calling this function.
+ */
+export function isWithinAllowedRoot(targetPath) {
+    const normalizedTarget = normalizePath(targetPath);
+    const roots = getBrowserRoots();
+    for (const root of roots) {
+        const normalizedRoot = normalizePath(root);
+        if (normalizedTarget === normalizedRoot) {
+            return true;
+        }
+        // Check both separators — `path.sep` is always `/` on Unix but a
+        // lowercased Windows path may contain either `\` or `/`.
+        if (normalizedTarget.startsWith(normalizedRoot + "/") ||
+            normalizedTarget.startsWith(normalizedRoot + "\\")) {
+            return true;
+        }
+    }
+    return false;
+}
+/**
+ * Like `isWithinAllowedRoot` but resolves symlinks first using `realpath`.
+ * Falls back to the plain path if `realpath` fails (e.g. ENOENT).
+ */
+export async function isWithinAllowedRootSafe(targetPath) {
+    let resolved = targetPath;
+    try {
+        resolved = await realpath(targetPath);
+    }
+    catch {
+        // Path doesn't exist yet or can't be resolved — use as-is
+    }
+    return isWithinAllowedRoot(resolved);
+}
+/**
+ * Check whether a path is exactly one of the allowed roots (not a descendant).
+ */
+export function isAllowedRoot(targetPath) {
+    const normalizedTarget = normalizePath(targetPath);
+    return getBrowserRoots().some((root) => normalizePath(root) === normalizedTarget);
+}
+/** Reset state — for testing only. */
+export function __resetBrowserRootsForTests() {
+    resolvedRoots = null;
+}

package/dist/bot/utils/file-tree.js

@@ -0,0 +1,92 @@
+import { promises as fs } from "node:fs";
+import os from "node:os";
+import path from "node:path";
+import { t } from "../../i18n/index.js";
+export const MAX_ENTRIES_PER_PAGE = 8;
+export function getHomeDirectory() {
+    return os.homedir();
+}
+export function pathToDisplayPath(absolutePath) {
+    const home = getHomeDirectory();
+    if (absolutePath === home) {
+        return "~";
+    }
+    if (absolutePath.startsWith(home + path.sep)) {
+        return "~" + absolutePath.slice(home.length);
+    }
+    return absolutePath;
+}
+export async function scanDirectory(dirPath, page = 0) {
+    try {
+        const entries = await fs.readdir(dirPath, { withFileTypes: true });
+        const subdirs = [];
+        for (const entry of entries) {
+            if (entry.isDirectory() && !entry.name.startsWith(".")) {
+                subdirs.push({
+                    name: entry.name,
+                    fullPath: path.join(dirPath, entry.name),
+                });
+            }
+        }
+        subdirs.sort((a, b) => a.name.localeCompare(b.name, undefined, { sensitivity: "base" }));
+        const parentPath = path.dirname(dirPath);
+        const hasParent = dirPath !== path.parse(dirPath).root;
+        // Clamp page to valid range so stale or crafted callbacks never produce
+        // an out-of-bounds page indicator like "(4/2)".
+        const totalPages = Math.max(1, Math.ceil(subdirs.length / MAX_ENTRIES_PER_PAGE));
+        const safePage = Math.max(0, Math.min(page, totalPages - 1));
+        const start = safePage * MAX_ENTRIES_PER_PAGE;
+        const pageEntries = subdirs.slice(start, start + MAX_ENTRIES_PER_PAGE);
+        return {
+            entries: pageEntries,
+            totalCount: subdirs.length,
+            page: safePage,
+            currentPath: dirPath,
+            displayPath: pathToDisplayPath(dirPath),
+            hasParent,
+            parentPath: hasParent ? parentPath : null,
+        };
+    }
+    catch (error) {
+        if (error instanceof Error) {
+            if ("code" in error) {
+                const code = error.code;
+                if (code === "ENOENT" || code === "ELOOP") {
+                    return { error: `Directory not found: ${dirPath}`, code: "ENOENT" };
+                }
+                if (code === "EACCES" || code === "EPERM") {
+                    return { error: `Permission denied: ${dirPath}`, code: "EACCES" };
+                }
+                if (code === "ENOTDIR") {
+                    return { error: `Not a directory: ${dirPath}`, code: "ENOTDIR" };
+                }
+            }
+        }
+        return {
+            error: error instanceof Error ? error.message : "Unknown error",
+            code: "UNKNOWN",
+        };
+    }
+}
+export function buildEntryLabel(entry) {
+    return `📁 ${entry.name}`;
+}
+export function buildTreeHeader(displayPath, totalCount, page, totalPages) {
+    let header = `📂 ${displayPath}`;
+    if (totalPages > 1) {
+        header += `  (${page + 1}/${totalPages})`;
+    }
+    if (totalCount === 0) {
+        header += `\n${t("open.no_subfolders")}`;
+    }
+    else if (totalCount === 1) {
+        header += `\n${t("open.subfolder_count", { count: String(totalCount) })}`;
+    }
+    else {
+        header += `\n${t("open.subfolders_count", { count: String(totalCount) })}`;
+    }
+    return header;
+}
+export function isScanError(result) {
+    return "error" in result;
+}

package/dist/bot/utils/finalize-assistant-response.js

@@ -1,35 +1,29 @@
 import { logger } from "../../utils/logger.js";
-export async function finalizeAssistantResponse({ sessionId, messageId, messageText, responseStreamer, flushPendingServiceMessages, prepareStreamingPayload, formatSummary, formatRawSummary, resolveFormat, getReplyKeyboard, sendText, deleteMessages, }) {
-    let streamedMessageIds = [];
+export async function finalizeAssistantResponse({ sessionId, messageId, messageText, responseStreamer, flushPendingServiceMessages, prepareStreamingPayload, renderFinalParts, getReplyKeyboard, sendRenderedPart, }) {
+    logger.debug(`[FinalizeResponse] Final assistant raw text received: session=${sessionId}, message=${messageId}`, messageText);
+    const keyboard = getReplyKeyboard();
+    const replyOptions = keyboard ? { reply_markup: keyboard } : undefined;
+    const silentReplyOptions = {
+        disable_notification: true,
+        ...(replyOptions ?? {}),
+    };
+    const streamSendOptions = {
+        ...silentReplyOptions,
+    };
     const preparedStreamPayload = prepareStreamingPayload(messageText);
     if (preparedStreamPayload) {
-        preparedStreamPayload.sendOptions = { disable_notification: true };
+        preparedStreamPayload.sendOptions = streamSendOptions;
         preparedStreamPayload.editOptions = undefined;
     }
     const result = await responseStreamer.complete(sessionId, messageId, preparedStreamPayload ?? undefined);
-    if (result.streamed) {
-        streamedMessageIds = result.telegramMessageIds;
-    }
     await flushPendingServiceMessages();
-    // When the response was streamed, delete the streamed messages and re-send
-    // via the non-streamed path so the reply keyboard carries the latest context.
-    if (streamedMessageIds.length > 0) {
-        try {
-            await deleteMessages(streamedMessageIds);
-        }
-        catch (err) {
-            logger.warn("[FinalizeResponse] Failed to delete streamed messages, sending with keyboard anyway:", err);
-        }
+    if (result.streamed) {
+        logger.debug(`[FinalizeResponse] Finalized streamed assistant message in place: session=${sessionId}, message=${messageId}, telegramMessages=${result.telegramMessageIds.length}`);
+        return true;
     }
-    const parts = formatSummary(messageText);
-    const rawParts = formatRawSummary(messageText);
-    const format = resolveFormat();
-    for (let partIndex = 0; partIndex < parts.length; partIndex++) {
-        const part = parts[partIndex];
-        const rawFallbackText = rawParts[partIndex];
-        const keyboard = getReplyKeyboard();
-        const options = keyboard ? { reply_markup: keyboard } : undefined;
-        await sendText(part, rawFallbackText, options, format);
+    const parts = renderFinalParts(messageText);
+    for (const part of parts) {
+        await sendRenderedPart(part, silentReplyOptions);
     }
     return false;
 }

package/dist/bot/utils/send-with-markdown-fallback.js

@@ -110,7 +110,7 @@ export async function sendMessageWithMarkdownFallback({ api, chatId, text, rawFa
         if (parseMode === "MarkdownV2") {
             const escapedText = escapeTelegramMarkdownV2(text);
             if (escapedText !== text) {
-                logger.warn("[Bot] Markdown parse failed, retrying assistant message with escaped MarkdownV2", error);
+                logger.warn("[Bot] Markdown parse failed, retrying message with escaped MarkdownV2", error);
                 try {
                     return await api.sendMessage(chatId, escapedText, markdownOptions);
                 }
@@ -118,12 +118,12 @@ export async function sendMessageWithMarkdownFallback({ api, chatId, text, rawFa
                     if (!isTelegramMarkdownParseError(escapedError)) {
                         throw escapedError;
                     }
-                    logger.warn("[Bot] Escaped Markdown parse failed, retrying assistant message in raw mode", escapedError);
+                    logger.warn("[Bot] Escaped Markdown parse failed, retrying message in raw mode", escapedError);
                     return api.sendMessage(chatId, fallbackText, stripMarkdownFormattingOptions(options));
                 }
             }
         }
-        logger.warn("[Bot] Markdown parse failed, retrying assistant message in raw mode", error);
+        logger.warn("[Bot] Markdown parse failed, retrying message in raw mode", error);
         return api.sendMessage(chatId, fallbackText, stripMarkdownFormattingOptions(options));
     }
 }

package/dist/bot/utils/switch-project.js

@@ -0,0 +1,48 @@
+import { setCurrentProject } from "../../settings/manager.js";
+import { clearSession } from "../../session/manager.js";
+import { summaryAggregator } from "../../summary/aggregator.js";
+import { pinnedMessageManager } from "../../pinned/manager.js";
+import { keyboardManager } from "../../keyboard/manager.js";
+import { getStoredAgent, resolveProjectAgent } from "../../agent/manager.js";
+import { getStoredModel } from "../../model/manager.js";
+import { formatVariantForButton } from "../../variant/manager.js";
+import { clearAllInteractionState } from "../../interaction/cleanup.js";
+import { createMainKeyboard } from "./keyboard.js";
+import { logger } from "../../utils/logger.js";
+/**
+ * Shared logic for switching the active project.
+ *
+ * Called by both `/projects` (selecting an existing project) and `/open`
+ * (browsing and adding a new directory). Performs the full state transition:
+ * persists the project, clears the session, resets the pinned message and
+ * keyboard, and returns a fresh reply keyboard for the caller to attach to
+ * its confirmation message.
+ *
+ * @param ctx       grammY callback context (used for `ctx.chat` / `ctx.api`)
+ * @param project   the project to switch to
+ * @param reason    short tag for `clearAllInteractionState` (e.g. "project_switched")
+ */
+export async function switchToProject(ctx, project, reason) {
+    setCurrentProject(project);
+    clearSession();
+    summaryAggregator.clear();
+    clearAllInteractionState(reason);
+    try {
+        await pinnedMessageManager.clear();
+    }
+    catch (err) {
+        logger.error("[Bot] Error clearing pinned message:", err);
+    }
+    if (ctx.chat) {
+        keyboardManager.initialize(ctx.api, ctx.chat.id);
+    }
+    await pinnedMessageManager.refreshContextLimit();
+    const contextLimit = pinnedMessageManager.getContextLimit();
+    keyboardManager.updateContext(0, contextLimit);
+    const currentAgent = await resolveProjectAgent(getStoredAgent());
+    const currentModel = getStoredModel();
+    const contextInfo = { tokensUsed: 0, tokensLimit: contextLimit };
+    const variantName = formatVariantForButton(currentModel.variant || "default");
+    keyboardManager.updateAgent(currentAgent);
+    return createMainKeyboard(currentAgent, currentModel, contextInfo, variantName);
+}