UNPKG

openclaw

Version:

Multi-channel AI gateway with extensible messaging integrations

181 lines (180 loc) 6.91 kB
import { l as normalizeOptionalString } from "./string-coerce-CIXf7egm.js"; //#region src/sessions/model-overrides.ts const MODEL_SELECTION_LOCKED_MESSAGE = "Model selection is locked for this session."; const MODEL_SELECTION_LOCKED_RESET_MESSAGE = "This session cannot be reset while model selection is locked."; const MODEL_SELECTION_LOCKED_PARENT_FORK_MESSAGE = "Model-selection-locked sessions cannot create child sessions from parent context."; /** Raised when a caller attempts to mutate a locked session model selection. */ var ModelSelectionLockedError = class extends Error { constructor(message = MODEL_SELECTION_LOCKED_MESSAGE) { super(message); this.name = "ModelSelectionLockedError"; } }; function isModelSelectionLocked(entry) { return entry?.modelSelectionLocked === true; } /** A locked harness owns both model selection and transcript lineage. */ function assertModelSelectionUnlocked(entry, message = MODEL_SELECTION_LOCKED_MESSAGE) { if (isModelSelectionLocked(entry)) throw new ModelSelectionLockedError(message); } function clearFallbackOrigin(entry) { let updated = false; if (entry.modelOverrideFallbackOriginProvider !== void 0) { delete entry.modelOverrideFallbackOriginProvider; updated = true; } if (entry.modelOverrideFallbackOriginModel !== void 0) { delete entry.modelOverrideFallbackOriginModel; updated = true; } return updated; } /** Applies a model/auth-profile override to a session entry and clears stale runtime fields. */ function applyModelOverrideToSessionEntry(params) { const { entry, selection, profileOverride } = params; assertModelSelectionUnlocked(entry); const profileOverrideSource = params.profileOverrideSource ?? "user"; const selectionSource = params.selectionSource ?? "user"; let updated = false; let selectionUpdated = false; let profileUpdated = false; if (selection.isDefault) { if (entry.providerOverride) { delete entry.providerOverride; updated = true; selectionUpdated = true; } if (entry.modelOverride) { delete entry.modelOverride; updated = true; selectionUpdated = true; } if (entry.modelOverrideSource) { delete entry.modelOverrideSource; updated = true; } if (entry.modelOverrideRouteResolution) { delete entry.modelOverrideRouteResolution; updated = true; } updated = clearFallbackOrigin(entry) || updated; } else { if (entry.providerOverride !== selection.provider) { entry.providerOverride = selection.provider; updated = true; selectionUpdated = true; } if (entry.modelOverride !== selection.model) { entry.modelOverride = selection.model; updated = true; selectionUpdated = true; } if (entry.modelOverrideSource !== selectionSource) { entry.modelOverrideSource = selectionSource; updated = true; } if (entry.modelOverrideRouteResolution !== "resolved") { entry.modelOverrideRouteResolution = "resolved"; updated = true; } updated = clearFallbackOrigin(entry) || updated; } const runtimeModel = normalizeOptionalString(entry.model) ?? ""; const runtimeProvider = normalizeOptionalString(entry.modelProvider) ?? ""; const runtimePresent = runtimeModel.length > 0 || runtimeProvider.length > 0; const runtimeAligned = runtimeModel === selection.model && (runtimeProvider.length === 0 || runtimeProvider === selection.provider); if (runtimePresent && (selectionUpdated || !runtimeAligned)) { if (entry.model !== void 0) { delete entry.model; updated = true; } if (entry.modelProvider !== void 0) { delete entry.modelProvider; updated = true; } } if (selection.isDefault && runtimePresent && !runtimeAligned) selectionUpdated = true; if (selectionUpdated || runtimePresent && !runtimeAligned) { if (entry.contextTokens !== void 0) { delete entry.contextTokens; updated = true; } if (entry.contextTokensSource !== void 0) { delete entry.contextTokensSource; updated = true; } if (entry.contextBudgetStatus !== void 0) { delete entry.contextBudgetStatus; updated = true; } } if (profileOverride) { if (entry.authProfileOverride !== profileOverride) { entry.authProfileOverride = profileOverride; updated = true; profileUpdated = true; } if (entry.authProfileOverrideSource !== profileOverrideSource) { entry.authProfileOverrideSource = profileOverrideSource; updated = true; profileUpdated = true; } if (entry.authProfileOverrideCompactionCount !== void 0) { delete entry.authProfileOverrideCompactionCount; updated = true; } } else if (!params.preserveAuthProfileOverride) { if (entry.authProfileOverride) { delete entry.authProfileOverride; updated = true; profileUpdated = true; } if (entry.authProfileOverrideSource) { delete entry.authProfileOverrideSource; updated = true; profileUpdated = true; } if (entry.authProfileOverrideCompactionCount !== void 0) { delete entry.authProfileOverrideCompactionCount; updated = true; } } if (updated) { if ((selectionUpdated || profileUpdated) && params.markLiveSwitchPending) entry.liveModelSwitchPending = true; delete entry.fallbackNotice; entry.updatedAt = Date.now(); } return { updated }; } function wrappedOverrideModel(provider, model) { return `${provider}/${model}`; } /** Repairs overrides where legacy provider/model fields were stored as provider/model strings. */ function repairProviderWrappedModelOverride(params) { const overrideProvider = normalizeOptionalString(params.entry.providerOverride); const overrideModel = normalizeOptionalString(params.entry.modelOverride); if (!overrideProvider || !overrideModel) return { updated: false }; const wrappedModel = wrappedOverrideModel(overrideProvider, overrideModel); const runtimeProvider = normalizeOptionalString(params.entry.modelProvider); const runtimeModel = normalizeOptionalString(params.entry.model); if (runtimeProvider && runtimeModel === wrappedModel && runtimeProvider !== overrideProvider) return applyModelOverrideToSessionEntry({ entry: params.entry, selection: { provider: runtimeProvider, model: runtimeModel, isDefault: runtimeProvider === params.defaultProvider && runtimeModel === params.defaultModel }, selectionSource: params.entry.modelOverrideSource === "auto" ? "auto" : "user" }); if (params.defaultProvider !== overrideProvider && params.defaultModel === wrappedModel) return applyModelOverrideToSessionEntry({ entry: params.entry, selection: { provider: params.defaultProvider, model: params.defaultModel, isDefault: true } }); return { updated: false }; } //#endregion export { applyModelOverrideToSessionEntry as a, repairProviderWrappedModelOverride as c, ModelSelectionLockedError as i, MODEL_SELECTION_LOCKED_PARENT_FORK_MESSAGE as n, assertModelSelectionUnlocked as o, MODEL_SELECTION_LOCKED_RESET_MESSAGE as r, isModelSelectionLocked as s, MODEL_SELECTION_LOCKED_MESSAGE as t };