fix: honor telegram model overrides in buttons (#8193) (thanks @gildo)

This commit is contained in:
Ayaan Zaidi
2026-02-04 09:16:10 +05:30
committed by Ayaan Zaidi
parent 202c554d09
commit 41a4f1200b
3 changed files with 81 additions and 9 deletions

View File

@@ -8,19 +8,23 @@ import {
} from "../auto-reply/inbound-debounce.js";
import { buildCommandsPaginationKeyboard } from "../auto-reply/reply/commands-info.js";
import { buildModelsProviderData } from "../auto-reply/reply/commands-models.js";
import { resolveStoredModelOverride } from "../auto-reply/reply/model-selection.js";
import { listSkillCommandsForAgents } from "../auto-reply/skill-commands.js";
import { buildCommandsMessagePaginated } from "../auto-reply/status.js";
import { resolveChannelConfigWrites } from "../channels/plugins/config-writes.js";
import { loadConfig } from "../config/config.js";
import { writeConfigFile } from "../config/io.js";
import { loadSessionStore, resolveStorePath } from "../config/sessions.js";
import { danger, logVerbose, warn } from "../globals.js";
import { readChannelAllowFromStore } from "../pairing/pairing-store.js";
import { resolveAgentRoute } from "../routing/resolve-route.js";
import { resolveThreadSessionKeys } from "../routing/session-key.js";
import { withTelegramApiErrorLogging } from "./api-logging.js";
import { firstDefined, isSenderAllowed, normalizeAllowFromWithStore } from "./bot-access.js";
import { RegisterTelegramHandlerParams } from "./bot-native-commands.js";
import { MEDIA_GROUP_TIMEOUT_MS, type MediaGroupEntry } from "./bot-updates.js";
import { resolveMedia } from "./bot/delivery.js";
import { resolveTelegramForumThreadId } from "./bot/helpers.js";
import { buildTelegramGroupPeerId, resolveTelegramForumThreadId } from "./bot/helpers.js";
import { migrateTelegramGroupConfig } from "./group-migration.js";
import { resolveTelegramInlineButtonsScope } from "./inline-buttons.js";
import {
@@ -128,6 +132,60 @@ export const registerTelegramHandlers = ({
},
});
const resolveTelegramSessionModel = (params: {
chatId: number | string;
isGroup: boolean;
isForum: boolean;
messageThreadId?: number;
resolvedThreadId?: number;
}): string | undefined => {
const resolvedThreadId =
params.resolvedThreadId ??
resolveTelegramForumThreadId({
isForum: params.isForum,
messageThreadId: params.messageThreadId,
});
const peerId = params.isGroup
? buildTelegramGroupPeerId(params.chatId, resolvedThreadId)
: String(params.chatId);
const route = resolveAgentRoute({
cfg,
channel: "telegram",
accountId,
peer: {
kind: params.isGroup ? "group" : "dm",
id: peerId,
},
});
const baseSessionKey = route.sessionKey;
const dmThreadId = !params.isGroup ? params.messageThreadId : undefined;
const threadKeys =
dmThreadId != null
? resolveThreadSessionKeys({ baseSessionKey, threadId: String(dmThreadId) })
: null;
const sessionKey = threadKeys?.sessionKey ?? baseSessionKey;
const storePath = resolveStorePath(cfg.session?.store, { agentId: route.agentId });
const store = loadSessionStore(storePath);
const entry = store[sessionKey];
const storedOverride = resolveStoredModelOverride({
sessionEntry: entry,
sessionStore: store,
sessionKey,
});
if (storedOverride) {
return storedOverride.provider
? `${storedOverride.provider}/${storedOverride.model}`
: storedOverride.model;
}
const provider = entry?.modelProvider?.trim();
const model = entry?.model?.trim();
if (provider && model) {
return `${provider}/${model}`;
}
const modelCfg = cfg.agents?.defaults?.model;
return typeof modelCfg === "string" ? modelCfg : modelCfg?.primary;
};
const processMediaGroup = async (entry: MediaGroupEntry) => {
try {
entry.messages.sort((a, b) => a.msg.message_id - b.msg.message_id);
@@ -474,9 +532,14 @@ export const registerTelegramHandlers = ({
const totalPages = calculateTotalPages(models.length, pageSize);
const safePage = Math.max(1, Math.min(page, totalPages));
// Get current model from config for checkmark display
const modelCfg = cfg.agents?.defaults?.model;
const currentModel = typeof modelCfg === "string" ? modelCfg : modelCfg?.primary;
// Resolve current model from session (prefer overrides)
const currentModel = resolveTelegramSessionModel({
chatId,
isGroup,
isForum,
messageThreadId,
resolvedThreadId,
});
const buttons = buildModelsKeyboard({
provider,