@bastani/atomic 0.9.11-alpha.4 → 0.9.11-alpha.6

package/dist/core/agent-session-models.js

@@ -1,8 +1,7 @@
 import { clampThinkingLevel, getSupportedThinkingLevels, modelsAreEqual } from "@earendil-works/pi-ai/compat";
-import { getModelDefaultContextWindow, getSupportedContextWindows, selectContextWindow } from "./context-window.js";
 import { formatNoApiKeyFoundMessage } from "./auth-guidance.js";
 import { DEFAULT_THINKING_LEVEL } from "./defaults.js";
-import { COPILOT_CONTEXT_WINDOW_SELECTION_OPTIONS, THINKING_LEVELS } from "./agent-session-types.js";
+import { THINKING_LEVELS } from "./agent-session-types.js";
 export async function _getRequiredRequestAuth(model) {
     const result = await this._modelRegistry.getApiKeyAndHeaders(model);
     if (!result.ok) {
@@ -11,7 +10,7 @@ export async function _getRequiredRequestAuth(model) {
         }
         throw new Error(result.error);
     }
-    if (result.apiKey) {
+    if (result.apiKey || result.headers) {
         return { apiKey: result.apiKey, headers: result.headers, baseUrl: result.baseUrl };
     }
     const isOAuth = this._modelRegistry.isUsingOAuth(model);
@@ -61,10 +60,9 @@ export async function setModel(model) {
     }
     const previousModel = this.model;
     const thinkingLevel = this._getThinkingLevelForModelSwitch();
-    const nextModel = this._withContextWindowForModelSwitch(model);
+    const nextModel = model;
     this.agent.state.model = nextModel;
     this.sessionManager.appendModelChange(nextModel.provider, nextModel.id);
-    this._appendContextWindowChangeIfChanged(previousModel, nextModel);
     this.settingsManager.setDefaultModelAndProvider(nextModel.provider, nextModel.id);
     // Re-clamp thinking level for new model's capabilities
     this.setThinkingLevel(thinkingLevel);
@@ -96,11 +94,10 @@ export async function _cycleScopedModel(direction) {
     const nextIndex = direction === "forward" ? (currentIndex + 1) % len : (currentIndex - 1 + len) % len;
     const next = scopedModels[nextIndex];
     const thinkingLevel = this._getThinkingLevelForModelSwitch(next.thinkingLevel);
-    const nextModel = this._withContextWindowForModelSwitch(next.model);
+    const nextModel = next.model;
     // Apply model
     this.agent.state.model = nextModel;
     this.sessionManager.appendModelChange(nextModel.provider, nextModel.id);
-    this._appendContextWindowChangeIfChanged(currentModel, nextModel);
     this.settingsManager.setDefaultModelAndProvider(nextModel.provider, nextModel.id);
     // Apply thinking level.
     // - Explicit scoped model thinking level overrides current session level
@@ -122,11 +119,10 @@ export async function _cycleAvailableModel(direction) {
         currentIndex = 0;
     const len = availableModels.length;
     const nextIndex = direction === "forward" ? (currentIndex + 1) % len : (currentIndex - 1 + len) % len;
-    const selectedModel = this._withContextWindowForModelSwitch(availableModels[nextIndex]);
+    const selectedModel = availableModels[nextIndex];
     const thinkingLevel = this._getThinkingLevelForModelSwitch();
     this.agent.state.model = selectedModel;
     this.sessionManager.appendModelChange(selectedModel.provider, selectedModel.id);
-    this._appendContextWindowChangeIfChanged(currentModel, selectedModel);
     this.settingsManager.setDefaultModelAndProvider(selectedModel.provider, selectedModel.id);
     // Re-clamp thinking level for new model's capabilities
     this.setThinkingLevel(thinkingLevel);
@@ -206,133 +202,6 @@ export function _clampThinkingLevel(level, _availableLevels) {
     return this.model ? clampThinkingLevel(this.model, level) : "off";
 }
 // =========================================================================
-// Context Window Management
-// =========================================================================
-export function getAvailableContextWindows() {
-    return this.model ? getSupportedContextWindows(this.model) : [];
-}
-export function supportsContextWindowSelection() {
-    return this.getAvailableContextWindows().length > 1;
-}
-export function setContextWindow(contextWindow, options = {}) {
-    if (!this.model) {
-        throw new Error("No model selected");
-    }
-    const selected = selectContextWindow(this.model, contextWindow, COPILOT_CONTEXT_WINDOW_SELECTION_OPTIONS);
-    if ("error" in selected) {
-        throw new Error(selected.error);
-    }
-    const previousContextWindow = this.model.contextWindow;
-    const isChanging = previousContextWindow !== selected.contextWindow;
-    this.agent.state.model = selected.model;
-    if (isChanging) {
-        this.sessionManager.appendContextWindowChange(selected.contextWindow);
-        this._emit({ type: "context_window_changed", contextWindow: selected.contextWindow });
-    }
-    if (options.persistDefault === true) {
-        this.settingsManager.setDefaultContextWindowForModel(selected.model.provider, selected.model.id, selected.contextWindow);
-    }
-}
-export function _withContextWindowForModelSwitch(model) {
-    // A source model's scalar contextWindow can be its natural default (for example a 1m-default
-    // model). Do not treat that alone as an opt-in to larger windows on a 400k-default target.
-    const settingsDefaultContextWindow = this._getSettingsContextWindowRequestForModel(model)?.contextWindow;
-    const candidates = [];
-    const targetDefaultContextWindow = getModelDefaultContextWindow(model);
-    if (model.contextWindow !== targetDefaultContextWindow) {
-        // Preserve an explicit context-window selection already applied to the target model
-        // (for example a caller passing selectContextWindow(target, 1m).model).
-        candidates.push(model.contextWindow);
-    }
-    const currentModel = this.model;
-    if (currentModel && this._shouldCarryCurrentContextWindowForModelSwitch(currentModel, settingsDefaultContextWindow)) {
-        candidates.push(currentModel.contextWindow);
-    }
-    if (settingsDefaultContextWindow !== undefined) {
-        candidates.push(settingsDefaultContextWindow);
-    }
-    candidates.push(targetDefaultContextWindow);
-    for (const candidate of candidates) {
-        const selected = selectContextWindow(model, candidate, COPILOT_CONTEXT_WINDOW_SELECTION_OPTIONS);
-        if (!("error" in selected))
-            return selected.model;
-    }
-    return model;
-}
-export function _shouldCarryCurrentContextWindowForModelSwitch(currentModel, settingsDefaultContextWindow) {
-    if (currentModel.contextWindow !== getModelDefaultContextWindow(currentModel)) {
-        return true;
-    }
-    if (this.sessionManager.getBranch().some((entry) => entry.type === "context_window_change")) {
-        return true;
-    }
-    return (settingsDefaultContextWindow !== undefined &&
-        currentModel.contextWindow === settingsDefaultContextWindow &&
-        getSupportedContextWindows(currentModel).includes(settingsDefaultContextWindow));
-}
-export function _getSettingsContextWindowRequestForModel(model) {
-    const modelContextWindow = this.settingsManager.getDefaultContextWindowForModel(model.provider, model.id);
-    if (modelContextWindow !== undefined) {
-        return { contextWindow: modelContextWindow, source: "model-settings" };
-    }
-    const globalContextWindow = this.settingsManager.getDefaultContextWindow();
-    return globalContextWindow === undefined
-        ? undefined
-        : { contextWindow: globalContextWindow, source: "global-settings" };
-}
-export function _getContextWindowReplayForModel(model, requestedContextWindow, source) {
-    if (requestedContextWindow !== undefined) {
-        const selected = selectContextWindow(model, requestedContextWindow, COPILOT_CONTEXT_WINDOW_SELECTION_OPTIONS);
-        if (!("error" in selected)) {
-            return { model: selected.model, contextWindow: selected.contextWindow, wouldWarn: false };
-        }
-        return this._getDefaultContextWindowReplayForModel(model, source !== "global-settings");
-    }
-    return this._getDefaultContextWindowReplayForModel(model, false);
-}
-export function _getDefaultContextWindowReplayForModel(model, wouldWarn) {
-    const defaultContextWindow = getModelDefaultContextWindow(model);
-    const selected = selectContextWindow(model, defaultContextWindow, COPILOT_CONTEXT_WINDOW_SELECTION_OPTIONS);
-    if (!("error" in selected)) {
-        return { model: selected.model, contextWindow: selected.contextWindow, wouldWarn };
-    }
-    return {
-        model: { ...model, contextWindow: defaultContextWindow, defaultContextWindow },
-        contextWindow: defaultContextWindow,
-        wouldWarn,
-    };
-}
-export function _getResumeContextWindowReplayForModel(model) {
-    const sessionContext = this.sessionManager.buildSessionContext();
-    if (sessionContext.contextWindow !== undefined) {
-        return this._getContextWindowReplayForModel(model, sessionContext.contextWindow, "session");
-    }
-    const settingsContextWindow = this._getSettingsContextWindowRequestForModel(model);
-    return this._getContextWindowReplayForModel(model, settingsContextWindow?.contextWindow, settingsContextWindow?.source);
-}
-export function _applyContextWindowReplay(contextWindow) {
-    if (!this.model)
-        return;
-    const previousContextWindow = this.model.contextWindow;
-    const settingsContextWindow = this._getSettingsContextWindowRequestForModel(this.model);
-    const requestedContextWindow = contextWindow ?? settingsContextWindow?.contextWindow;
-    const source = contextWindow !== undefined ? "session" : settingsContextWindow?.source;
-    const replay = this._getContextWindowReplayForModel(this.model, requestedContextWindow, source);
-    this.agent.state.model = replay.model;
-    if (previousContextWindow !== replay.contextWindow) {
-        this._emit({ type: "context_window_changed", contextWindow: replay.contextWindow });
-    }
-}
-export function _appendContextWindowChangeIfChanged(previousModel, nextModel) {
-    const replay = this._getResumeContextWindowReplayForModel(nextModel);
-    if (!replay.wouldWarn && nextModel.contextWindow === replay.contextWindow)
-        return;
-    this.sessionManager.appendContextWindowChange(nextModel.contextWindow);
-    if (previousModel?.contextWindow !== nextModel.contextWindow) {
-        this._emit({ type: "context_window_changed", contextWindow: nextModel.contextWindow });
-    }
-}
-// =========================================================================
 // Queue Mode Management
 // =========================================================================
 /**
@@ -353,16 +222,5 @@ export const agentSessionModelsMethods = {
     supportsThinking,
     _getThinkingLevelForModelSwitch,
     _clampThinkingLevel,
-    getAvailableContextWindows,
-    supportsContextWindowSelection,
-    setContextWindow,
-    _withContextWindowForModelSwitch,
-    _shouldCarryCurrentContextWindowForModelSwitch,
-    _getSettingsContextWindowRequestForModel,
-    _getContextWindowReplayForModel,
-    _getDefaultContextWindowReplayForModel,
-    _getResumeContextWindowReplayForModel,
-    _applyContextWindowReplay,
-    _appendContextWindowChangeIfChanged,
 };
 //# sourceMappingURL=agent-session-models.js.map

package/dist/core/agent-session-models.js.map

@@ -1 +1 @@
-{"version":3,"file":"agent-session-models.js","sourceRoot":"","sources":["../../src/core/agent-session-models.ts"],"names":[],"mappings":"AAGA,OAAO,EAAE,kBAAkB,EAAE,0BAA0B,EAAE,cAAc,EAAE,MAAM,8BAA8B,CAAC;AAC9G,OAAO,EAAE,4BAA4B,EAAE,0BAA0B,EAAE,mBAAmB,EAAE,MAAM,qBAAqB,CAAC;AACpH,OAAO,EAAE,0BAA0B,EAAE,MAAM,oBAAoB,CAAC;AAChE,OAAO,EAAE,sBAAsB,EAAE,MAAM,eAAe,CAAC;AAEvD,OAAO,EAAE,wCAAwC,EAAE,eAAe,EAA0F,MAAM,0BAA0B,CAAC;AAE7L,MAAM,CAAC,KAAK,UAAU,uBAAuB,CAAqB,KAAiB;IAKlF,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,cAAc,CAAC,mBAAmB,CAAC,KAAK,CAAC,CAAC;IACpE,IAAI,CAAC,MAAM,CAAC,EAAE,EAAE,CAAC;QAChB,IAAI,MAAM,CAAC,KAAK,CAAC,UAAU,CAAC,kBAAkB,CAAC,EAAE,CAAC;YACjD,MAAM,IAAI,KAAK,CAAC,0BAA0B,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC,CAAC;QAC7D,CAAC;QACD,MAAM,IAAI,KAAK,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;IAC/B,CAAC;IACD,IAAI,MAAM,CAAC,MAAM,EAAE,CAAC;QACnB,OAAO,EAAE,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,EAAE,MAAM,CAAC,OAAO,EAAE,OAAO,EAAE,MAAM,CAAC,OAAO,EAAE,CAAC;IACpF,CAAC;IAED,MAAM,OAAO,GAAG,IAAI,CAAC,cAAc,CAAC,YAAY,CAAC,KAAK,CAAC,CAAC;IACxD,IAAI,OAAO,EAAE,CAAC;QACb,MAAM,IAAI,KAAK,CACd,8BAA8B,KAAK,CAAC,QAAQ,KAAK;YAChD,0DAA0D;YAC1D,eAAe,KAAK,CAAC,QAAQ,uBAAuB,CACrD,CAAC;IACH,CAAC;IACD,MAAM,IAAI,KAAK,CAAC,0BAA0B,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC,CAAC;AAC7D,CAAC;AAED;;;;;;;GAOG;AAEH,MAAM,UAAU,iBAAiB,CAChC,SAAqB,EACrB,aAAqC,EACrC,MAAgD;IAEhD,IAAI,cAAc,CAAC,aAAa,EAAE,SAAS,CAAC;QAAE,OAAO;IACrD,IAAI,CAAC,KAAK,CAAC;QACV,IAAI,EAAE,eAAe;QACrB,KAAK,EAAE,SAAS;QAChB,aAAa;QACb,MAAM;KACN,CAAC,CAAC;AACJ,CAAC;AAGD,MAAM,CAAC,KAAK,UAAU,gBAAgB,CACrC,SAAqB,EACrB,aAAqC,EACrC,MAAgD;IAEhD,IAAI,cAAc,CAAC,aAAa,EAAE,SAAS,CAAC;QAAE,OAAO;IACrD,MAAM,IAAI,CAAC,gBAAgB,CAAC,IAAI,CAAC;QAChC,IAAI,EAAE,cAAc;QACpB,KAAK,EAAE,SAAS;QAChB,aAAa;QACb,MAAM;KACN,CAAC,CAAC;AACJ,CAAC;AAED;;;;GAIG;AAEH,MAAM,CAAC,KAAK,UAAU,QAAQ,CAAqB,KAAiB;IACnE,IAAI,CAAC,IAAI,CAAC,cAAc,CAAC,iBAAiB,CAAC,KAAK,CAAC,EAAE,CAAC;QACnD,MAAM,IAAI,KAAK,CAAC,kBAAkB,KAAK,CAAC,QAAQ,IAAI,KAAK,CAAC,EAAE,EAAE,CAAC,CAAC;IACjE,CAAC;IAED,MAAM,aAAa,GAAG,IAAI,CAAC,KAAK,CAAC;IACjC,MAAM,aAAa,GAAG,IAAI,CAAC,+BAA+B,EAAE,CAAC;IAC7D,MAAM,SAAS,GAAG,IAAI,CAAC,gCAAgC,CAAC,KAAK,CAAC,CAAC;IAC/D,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,KAAK,GAAG,SAAS,CAAC;IACnC,IAAI,CAAC,cAAc,CAAC,iBAAiB,CAAC,SAAS,CAAC,QAAQ,EAAE,SAAS,CAAC,EAAE,CAAC,CAAC;IACxE,IAAI,CAAC,mCAAmC,CAAC,aAAa,EAAE,SAAS,CAAC,CAAC;IACnE,IAAI,CAAC,eAAe,CAAC,0BAA0B,CAAC,SAAS,CAAC,QAAQ,EAAE,SAAS,CAAC,EAAE,CAAC,CAAC;IAElF,uDAAuD;IACvD,IAAI,CAAC,gBAAgB,CAAC,aAAa,CAAC,CAAC;IACrC,IAAI,CAAC,uCAAuC,EAAE,CAAC;IAE/C,IAAI,CAAC,iBAAiB,CAAC,SAAS,EAAE,aAAa,EAAE,KAAK,CAAC,CAAC;IACxD,MAAM,IAAI,CAAC,gBAAgB,CAAC,SAAS,EAAE,aAAa,EAAE,KAAK,CAAC,CAAC;AAC9D,CAAC;AAED;;;;;GAKG;AAEH,MAAM,CAAC,KAAK,UAAU,UAAU,CAAqB,SAAS,GAA2B,SAAS;IACjG,IAAI,IAAI,CAAC,aAAa,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QACnC,OAAO,IAAI,CAAC,iBAAiB,CAAC,SAAS,CAAC,CAAC;IAC1C,CAAC;IACD,OAAO,IAAI,CAAC,oBAAoB,CAAC,SAAS,CAAC,CAAC;AAC7C,CAAC;AAGD,MAAM,CAAC,KAAK,UAAU,iBAAiB,CAAqB,SAAiC;IAC5F,MAAM,YAAY,GAAG,IAAI,CAAC,aAAa,CAAC,MAAM,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,IAAI,CAAC,cAAc,CAAC,iBAAiB,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC;IAChH,IAAI,YAAY,CAAC,MAAM,IAAI,CAAC;QAAE,OAAO,SAAS,CAAC;IAE/C,MAAM,YAAY,GAAG,IAAI,CAAC,KAAK,CAAC;IAChC,IAAI,YAAY,GAAG,YAAY,CAAC,SAAS,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,cAAc,CAAC,EAAE,CAAC,KAAK,EAAE,YAAY,CAAC,CAAC,CAAC;IAE1F,IAAI,YAAY,KAAK,CAAC,CAAC;QAAE,YAAY,GAAG,CAAC,CAAC;IAC1C,MAAM,GAAG,GAAG,YAAY,CAAC,MAAM,CAAC;IAChC,MAAM,SAAS,GAAG,SAAS,KAAK,SAAS,CAAC,CAAC,CAAC,CAAC,YAAY,GAAG,CAAC,CAAC,GAAG,GAAG,CAAC,CAAC,CAAC,CAAC,YAAY,GAAG,CAAC,GAAG,GAAG,CAAC,GAAG,GAAG,CAAC;IACtG,MAAM,IAAI,GAAG,YAAY,CAAC,SAAS,CAAC,CAAC;IACrC,MAAM,aAAa,GAAG,IAAI,CAAC,+BAA+B,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;IAC/E,MAAM,SAAS,GAAG,IAAI,CAAC,gCAAgC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IAEpE,cAAc;IACd,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,KAAK,GAAG,SAAS,CAAC;IACnC,IAAI,CAAC,cAAc,CAAC,iBAAiB,CAAC,SAAS,CAAC,QAAQ,EAAE,SAAS,CAAC,EAAE,CAAC,CAAC;IACxE,IAAI,CAAC,mCAAmC,CAAC,YAAY,EAAE,SAAS,CAAC,CAAC;IAClE,IAAI,CAAC,eAAe,CAAC,0BAA0B,CAAC,SAAS,CAAC,QAAQ,EAAE,SAAS,CAAC,EAAE,CAAC,CAAC;IAElF,wBAAwB;IACxB,yEAAyE;IACzE,kFAAkF;IAClF,iDAAiD;IACjD,IAAI,CAAC,gBAAgB,CAAC,aAAa,CAAC,CAAC;IACrC,IAAI,CAAC,uCAAuC,EAAE,CAAC;IAE/C,IAAI,CAAC,iBAAiB,CAAC,SAAS,EAAE,YAAY,EAAE,OAAO,CAAC,CAAC;IACzD,MAAM,IAAI,CAAC,gBAAgB,CAAC,SAAS,EAAE,YAAY,EAAE,OAAO,CAAC,CAAC;IAE9D,OAAO,EAAE,KAAK,EAAE,SAAS,EAAE,aAAa,EAAE,IAAI,CAAC,aAAa,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC;AAChF,CAAC;AAGD,MAAM,CAAC,KAAK,UAAU,oBAAoB,CAAqB,SAAiC;IAC/F,MAAM,eAAe,GAAG,MAAM,IAAI,CAAC,cAAc,CAAC,YAAY,EAAE,CAAC;IACjE,IAAI,eAAe,CAAC,MAAM,IAAI,CAAC;QAAE,OAAO,SAAS,CAAC;IAElD,MAAM,YAAY,GAAG,IAAI,CAAC,KAAK,CAAC;IAChC,IAAI,YAAY,GAAG,eAAe,CAAC,SAAS,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,cAAc,CAAC,CAAC,EAAE,YAAY,CAAC,CAAC,CAAC;IAErF,IAAI,YAAY,KAAK,CAAC,CAAC;QAAE,YAAY,GAAG,CAAC,CAAC;IAC1C,MAAM,GAAG,GAAG,eAAe,CAAC,MAAM,CAAC;IACnC,MAAM,SAAS,GAAG,SAAS,KAAK,SAAS,CAAC,CAAC,CAAC,CAAC,YAAY,GAAG,CAAC,CAAC,GAAG,GAAG,CAAC,CAAC,CAAC,CAAC,YAAY,GAAG,CAAC,GAAG,GAAG,CAAC,GAAG,GAAG,CAAC;IACtG,MAAM,aAAa,GAAG,IAAI,CAAC,gCAAgC,CAAC,eAAe,CAAC,SAAS,CAAC,CAAC,CAAC;IAExF,MAAM,aAAa,GAAG,IAAI,CAAC,+BAA+B,EAAE,CAAC;IAC7D,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,KAAK,GAAG,aAAa,CAAC;IACvC,IAAI,CAAC,cAAc,CAAC,iBAAiB,CAAC,aAAa,CAAC,QAAQ,EAAE,aAAa,CAAC,EAAE,CAAC,CAAC;IAChF,IAAI,CAAC,mCAAmC,CAAC,YAAY,EAAE,aAAa,CAAC,CAAC;IACtE,IAAI,CAAC,eAAe,CAAC,0BAA0B,CAAC,aAAa,CAAC,QAAQ,EAAE,aAAa,CAAC,EAAE,CAAC,CAAC;IAE1F,uDAAuD;IACvD,IAAI,CAAC,gBAAgB,CAAC,aAAa,CAAC,CAAC;IACrC,IAAI,CAAC,uCAAuC,EAAE,CAAC;IAE/C,IAAI,CAAC,iBAAiB,CAAC,aAAa,EAAE,YAAY,EAAE,OAAO,CAAC,CAAC;IAC7D,MAAM,IAAI,CAAC,gBAAgB,CAAC,aAAa,EAAE,YAAY,EAAE,OAAO,CAAC,CAAC;IAElE,OAAO,EAAE,KAAK,EAAE,aAAa,EAAE,aAAa,EAAE,IAAI,CAAC,aAAa,EAAE,QAAQ,EAAE,KAAK,EAAE,CAAC;AACrF,CAAC;AAED,4EAA4E;AAC5E,4BAA4B;AAC5B,4EAA4E;AAE5E;;;;GAIG;AAEH,MAAM,UAAU,gBAAgB,CAAqB,KAAoB;IACxE,MAAM,eAAe,GAAG,IAAI,CAAC,0BAA0B,EAAE,CAAC;IAC1D,MAAM,cAAc,GAAG,eAAe,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,CAAC,mBAAmB,CAAC,KAAK,EAAE,eAAe,CAAC,CAAC;IAElH,oCAAoC;IACpC,MAAM,aAAa,GAAG,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,aAAa,CAAC;IACrD,MAAM,UAAU,GAAG,cAAc,KAAK,aAAa,CAAC;IAEpD,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,aAAa,GAAG,cAAc,CAAC;IAEhD,IAAI,UAAU,EAAE,CAAC;QAChB,IAAI,CAAC,cAAc,CAAC,yBAAyB,CAAC,cAAc,CAAC,CAAC;QAC9D,IAAI,CAAC,uCAAuC,EAAE,CAAC;QAC/C,IAAI,IAAI,CAAC,gBAAgB,EAAE,IAAI,cAAc,KAAK,KAAK,EAAE,CAAC;YACzD,IAAI,CAAC,eAAe,CAAC,uBAAuB,CAAC,cAAc,CAAC,CAAC;QAC9D,CAAC;QACD,IAAI,CAAC,KAAK,CAAC,EAAE,IAAI,EAAE,wBAAwB,EAAE,KAAK,EAAE,cAAc,EAAE,CAAC,CAAC;QACtE,KAAK,IAAI,CAAC,gBAAgB,CAAC,IAAI,CAAC;YAC/B,IAAI,EAAE,uBAAuB;YAC7B,KAAK,EAAE,cAAc;YACrB,aAAa;SACb,CAAC,CAAC;IACJ,CAAC;AACF,CAAC;AAED;;;GAGG;AAEH,MAAM,UAAU,kBAAkB;IACjC,IAAI,CAAC,IAAI,CAAC,gBAAgB,EAAE;QAAE,OAAO,SAAS,CAAC;IAE/C,MAAM,MAAM,GAAG,IAAI,CAAC,0BAA0B,EAAE,CAAC;IACjD,MAAM,YAAY,GAAG,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;IACxD,MAAM,SAAS,GAAG,CAAC,YAAY,GAAG,CAAC,CAAC,GAAG,MAAM,CAAC,MAAM,CAAC;IACrD,MAAM,SAAS,GAAG,MAAM,CAAC,SAAS,CAAC,CAAC;IAEpC,IAAI,CAAC,gBAAgB,CAAC,SAAS,CAAC,CAAC;IACjC,OAAO,SAAS,CAAC;AAClB,CAAC;AAED;;;GAGG;AAEH,MAAM,UAAU,0BAA0B;IACzC,IAAI,CAAC,IAAI,CAAC,KAAK;QAAE,OAAO,eAAe,CAAC;IACxC,OAAO,0BAA0B,CAAC,IAAI,CAAC,KAAK,CAAoB,CAAC;AAClE,CAAC;AAED;;GAEG;AAEH,MAAM,UAAU,gBAAgB;IAC/B,OAAO,CAAC,CAAC,IAAI,CAAC,KAAK,EAAE,SAAS,CAAC;AAChC,CAAC;AAGD,MAAM,UAAU,+BAA+B,CAAqB,aAA6B;IAChG,IAAI,aAAa,KAAK,SAAS,EAAE,CAAC;QACjC,OAAO,aAAa,CAAC;IACtB,CAAC;IACD,IAAI,CAAC,IAAI,CAAC,gBAAgB,EAAE,EAAE,CAAC;QAC9B,OAAO,IAAI,CAAC,eAAe,CAAC,uBAAuB,EAAE,IAAI,sBAAsB,CAAC;IACjF,CAAC;IACD,OAAO,IAAI,CAAC,aAAa,CAAC;AAC3B,CAAC;AAGD,MAAM,UAAU,mBAAmB,CAAqB,KAAoB,EAAE,gBAAiC;IAC9G,OAAO,IAAI,CAAC,KAAK,CAAC,CAAC,CAAE,kBAAkB,CAAC,IAAI,CAAC,KAAK,EAAE,KAAK,CAAmB,CAAC,CAAC,CAAC,KAAK,CAAC;AACtF,CAAC;AAED,4EAA4E;AAC5E,4BAA4B;AAC5B,4EAA4E;AAG5E,MAAM,UAAU,0BAA0B;IACzC,OAAO,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,0BAA0B,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;AACjE,CAAC;AAGD,MAAM,UAAU,8BAA8B;IAC7C,OAAO,IAAI,CAAC,0BAA0B,EAAE,CAAC,MAAM,GAAG,CAAC,CAAC;AACrD,CAAC;AAGD,MAAM,UAAU,gBAAgB,CAAqB,aAAqB,EAAE,OAAO,GAAiC,EAAE;IACrH,IAAI,CAAC,IAAI,CAAC,KAAK,EAAE,CAAC;QACjB,MAAM,IAAI,KAAK,CAAC,mBAAmB,CAAC,CAAC;IACtC,CAAC;IACD,MAAM,QAAQ,GAAG,mBAAmB,CAAC,IAAI,CAAC,KAAK,EAAE,aAAa,EAAE,wCAAwC,CAAC,CAAC;IAC1G,IAAI,OAAO,IAAI,QAAQ,EAAE,CAAC;QACzB,MAAM,IAAI,KAAK,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC;IACjC,CAAC;IAED,MAAM,qBAAqB,GAAG,IAAI,CAAC,KAAK,CAAC,aAAa,CAAC;IACvD,MAAM,UAAU,GAAG,qBAAqB,KAAK,QAAQ,CAAC,aAAa,CAAC;IACpE,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,KAAK,GAAG,QAAQ,CAAC,KAAK,CAAC;IAExC,IAAI,UAAU,EAAE,CAAC;QAChB,IAAI,CAAC,cAAc,CAAC,yBAAyB,CAAC,QAAQ,CAAC,aAAa,CAAC,CAAC;QACtE,IAAI,CAAC,KAAK,CAAC,EAAE,IAAI,EAAE,wBAAwB,EAAE,aAAa,EAAE,QAAQ,CAAC,aAAa,EAAE,CAAC,CAAC;IACvF,CAAC;IACD,IAAI,OAAO,CAAC,cAAc,KAAK,IAAI,EAAE,CAAC;QACrC,IAAI,CAAC,eAAe,CAAC,+BAA+B,CAAC,QAAQ,CAAC,KAAK,CAAC,QAAQ,EAAE,QAAQ,CAAC,KAAK,CAAC,EAAE,EAAE,QAAQ,CAAC,aAAa,CAAC,CAAC;IAC1H,CAAC;AACF,CAAC;AAGD,MAAM,UAAU,gCAAgC,CAAqB,KAAiB;IACrF,6FAA6F;IAC7F,2FAA2F;IAC3F,MAAM,4BAA4B,GAAG,IAAI,CAAC,wCAAwC,CAAC,KAAK,CAAC,EAAE,aAAa,CAAC;IACzG,MAAM,UAAU,GAAa,EAAE,CAAC;IAChC,MAAM,0BAA0B,GAAG,4BAA4B,CAAC,KAAK,CAAC,CAAC;IACvE,IAAI,KAAK,CAAC,aAAa,KAAK,0BAA0B,EAAE,CAAC;QACxD,oFAAoF;QACpF,wEAAwE;QACxE,UAAU,CAAC,IAAI,CAAC,KAAK,CAAC,aAAa,CAAC,CAAC;IACtC,CAAC;IACD,MAAM,YAAY,GAAG,IAAI,CAAC,KAAK,CAAC;IAChC,IAAI,YAAY,IAAI,IAAI,CAAC,8CAA8C,CAAC,YAAY,EAAE,4BAA4B,CAAC,EAAE,CAAC;QACrH,UAAU,CAAC,IAAI,CAAC,YAAY,CAAC,aAAa,CAAC,CAAC;IAC7C,CAAC;IACD,IAAI,4BAA4B,KAAK,SAAS,EAAE,CAAC;QAChD,UAAU,CAAC,IAAI,CAAC,4BAA4B,CAAC,CAAC;IAC/C,CAAC;IACD,UAAU,CAAC,IAAI,CAAC,0BAA0B,CAAC,CAAC;IAE5C,KAAK,MAAM,SAAS,IAAI,UAAU,EAAE,CAAC;QACpC,MAAM,QAAQ,GAAG,mBAAmB,CAAC,KAAK,EAAE,SAAS,EAAE,wCAAwC,CAAC,CAAC;QACjG,IAAI,CAAC,CAAC,OAAO,IAAI,QAAQ,CAAC;YAAE,OAAO,QAAQ,CAAC,KAAK,CAAC;IACnD,CAAC;IACD,OAAO,KAAK,CAAC;AACd,CAAC;AAGD,MAAM,UAAU,8CAA8C,CAC7D,YAAwB,EACxB,4BAAgD;IAEhD,IAAI,YAAY,CAAC,aAAa,KAAK,4BAA4B,CAAC,YAAY,CAAC,EAAE,CAAC;QAC/E,OAAO,IAAI,CAAC;IACb,CAAC;IACD,IAAI,IAAI,CAAC,cAAc,CAAC,SAAS,EAAE,CAAC,IAAI,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,IAAI,KAAK,uBAAuB,CAAC,EAAE,CAAC;QAC7F,OAAO,IAAI,CAAC;IACb,CAAC;IACD,OAAO,CACN,4BAA4B,KAAK,SAAS;QAC1C,YAAY,CAAC,aAAa,KAAK,4BAA4B;QAC3D,0BAA0B,CAAC,YAAY,CAAC,CAAC,QAAQ,CAAC,4BAA4B,CAAC,CAC/E,CAAC;AACH,CAAC;AAGD,MAAM,UAAU,wCAAwC,CAAqB,KAAiB;IAC7F,MAAM,kBAAkB,GAAG,IAAI,CAAC,eAAe,CAAC,+BAA+B,CAAC,KAAK,CAAC,QAAQ,EAAE,KAAK,CAAC,EAAE,CAAC,CAAC;IAC1G,IAAI,kBAAkB,KAAK,SAAS,EAAE,CAAC;QACtC,OAAO,EAAE,aAAa,EAAE,kBAAkB,EAAE,MAAM,EAAE,gBAAgB,EAAE,CAAC;IACxE,CAAC;IACD,MAAM,mBAAmB,GAAG,IAAI,CAAC,eAAe,CAAC,uBAAuB,EAAE,CAAC;IAC3E,OAAO,mBAAmB,KAAK,SAAS;QACvC,CAAC,CAAC,SAAS;QACX,CAAC,CAAC,EAAE,aAAa,EAAE,mBAAmB,EAAE,MAAM,EAAE,iBAAiB,EAAE,CAAC;AACtE,CAAC;AAGD,MAAM,UAAU,+BAA+B,CAC9C,KAAiB,EACjB,sBAA0C,EAC1C,MAA6C;IAE7C,IAAI,sBAAsB,KAAK,SAAS,EAAE,CAAC;QAC1C,MAAM,QAAQ,GAAG,mBAAmB,CAAC,KAAK,EAAE,sBAAsB,EAAE,wCAAwC,CAAC,CAAC;QAC9G,IAAI,CAAC,CAAC,OAAO,IAAI,QAAQ,CAAC,EAAE,CAAC;YAC5B,OAAO,EAAE,KAAK,EAAE,QAAQ,CAAC,KAAK,EAAE,aAAa,EAAE,QAAQ,CAAC,aAAa,EAAE,SAAS,EAAE,KAAK,EAAE,CAAC;QAC3F,CAAC;QACD,OAAO,IAAI,CAAC,sCAAsC,CAAC,KAAK,EAAE,MAAM,KAAK,iBAAiB,CAAC,CAAC;IACzF,CAAC;IAED,OAAO,IAAI,CAAC,sCAAsC,CAAC,KAAK,EAAE,KAAK,CAAC,CAAC;AAClE,CAAC;AAGD,MAAM,UAAU,sCAAsC,CACrD,KAAiB,EACjB,SAAkB;IAElB,MAAM,oBAAoB,GAAG,4BAA4B,CAAC,KAAK,CAAC,CAAC;IACjE,MAAM,QAAQ,GAAG,mBAAmB,CAAC,KAAK,EAAE,oBAAoB,EAAE,wCAAwC,CAAC,CAAC;IAC5G,IAAI,CAAC,CAAC,OAAO,IAAI,QAAQ,CAAC,EAAE,CAAC;QAC5B,OAAO,EAAE,KAAK,EAAE,QAAQ,CAAC,KAAK,EAAE,aAAa,EAAE,QAAQ,CAAC,aAAa,EAAE,SAAS,EAAE,CAAC;IACpF,CAAC;IACD,OAAO;QACN,KAAK,EAAE,EAAE,GAAG,KAAK,EAAE,aAAa,EAAE,oBAAoB,EAAE,oBAAoB,EAAE;QAC9E,aAAa,EAAE,oBAAoB;QACnC,SAAS;KACT,CAAC;AACH,CAAC;AAGD,MAAM,UAAU,qCAAqC,CACpD,KAAiB;IAEjB,MAAM,cAAc,GAAG,IAAI,CAAC,cAAc,CAAC,mBAAmB,EAAE,CAAC;IACjE,IAAI,cAAc,CAAC,aAAa,KAAK,SAAS,EAAE,CAAC;QAChD,OAAO,IAAI,CAAC,+BAA+B,CAAC,KAAK,EAAE,cAAc,CAAC,aAAa,EAAE,SAAS,CAAC,CAAC;IAC7F,CAAC;IACD,MAAM,qBAAqB,GAAG,IAAI,CAAC,wCAAwC,CAAC,KAAK,CAAC,CAAC;IACnF,OAAO,IAAI,CAAC,+BAA+B,CAAC,KAAK,EAAE,qBAAqB,EAAE,aAAa,EAAE,qBAAqB,EAAE,MAAM,CAAC,CAAC;AACzH,CAAC;AAGD,MAAM,UAAU,yBAAyB,CAAqB,aAAiC;IAC9F,IAAI,CAAC,IAAI,CAAC,KAAK;QAAE,OAAO;IACxB,MAAM,qBAAqB,GAAG,IAAI,CAAC,KAAK,CAAC,aAAa,CAAC;IACvD,MAAM,qBAAqB,GAAG,IAAI,CAAC,wCAAwC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IACxF,MAAM,sBAAsB,GAAG,aAAa,IAAI,qBAAqB,EAAE,aAAa,CAAC;IACrF,MAAM,MAAM,GAA0C,aAAa,KAAK,SAAS,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,qBAAqB,EAAE,MAAM,CAAC;IAC9H,MAAM,MAAM,GAAG,IAAI,CAAC,+BAA+B,CAAC,IAAI,CAAC,KAAK,EAAE,sBAAsB,EAAE,MAAM,CAAC,CAAC;IAChG,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,KAAK,GAAG,MAAM,CAAC,KAAK,CAAC;IACtC,IAAI,qBAAqB,KAAK,MAAM,CAAC,aAAa,EAAE,CAAC;QACpD,IAAI,CAAC,KAAK,CAAC,EAAE,IAAI,EAAE,wBAAwB,EAAE,aAAa,EAAE,MAAM,CAAC,aAAa,EAAE,CAAC,CAAC;IACrF,CAAC;AACF,CAAC;AAGD,MAAM,UAAU,mCAAmC,CAClD,aAAqC,EACrC,SAAqB;IAErB,MAAM,MAAM,GAAG,IAAI,CAAC,qCAAqC,CAAC,SAAS,CAAC,CAAC;IACrE,IAAI,CAAC,MAAM,CAAC,SAAS,IAAI,SAAS,CAAC,aAAa,KAAK,MAAM,CAAC,aAAa;QAAE,OAAO;IAClF,IAAI,CAAC,cAAc,CAAC,yBAAyB,CAAC,SAAS,CAAC,aAAa,CAAC,CAAC;IACvE,IAAI,aAAa,EAAE,aAAa,KAAK,SAAS,CAAC,aAAa,EAAE,CAAC;QAC9D,IAAI,CAAC,KAAK,CAAC,EAAE,IAAI,EAAE,wBAAwB,EAAE,aAAa,EAAE,SAAS,CAAC,aAAa,EAAE,CAAC,CAAC;IACxF,CAAC;AACF,CAAC;AAED,4EAA4E;AAC5E,wBAAwB;AACxB,4EAA4E;AAE5E;;;GAGG;AAEH,MAAM,CAAC,MAAM,yBAAyB,GAAG;IACxC,uBAAuB;IACvB,iBAAiB;IACjB,gBAAgB;IAChB,QAAQ;IACR,UAAU;IACV,iBAAiB;IACjB,oBAAoB;IACpB,gBAAgB;IAChB,kBAAkB;IAClB,0BAA0B;IAC1B,gBAAgB;IAChB,+BAA+B;IAC/B,mBAAmB;IACnB,0BAA0B;IAC1B,8BAA8B;IAC9B,gBAAgB;IAChB,gCAAgC;IAChC,8CAA8C;IAC9C,wCAAwC;IACxC,+BAA+B;IAC/B,sCAAsC;IACtC,qCAAqC;IACrC,yBAAyB;IACzB,mCAAmC;CACnC,CAAC","sourcesContent":["import type { ThinkingLevel } from \"@earendil-works/pi-agent-core\";\nimport type { ProviderHeaders } from \"@earendil-works/pi-ai\";\nimport type { Api, Model } from \"@earendil-works/pi-ai/compat\";\nimport { clampThinkingLevel, getSupportedThinkingLevels, modelsAreEqual } from \"@earendil-works/pi-ai/compat\";\nimport { getModelDefaultContextWindow, getSupportedContextWindows, selectContextWindow } from \"./context-window.ts\";\nimport { formatNoApiKeyFoundMessage } from \"./auth-guidance.ts\";\nimport { DEFAULT_THINKING_LEVEL } from \"./defaults.ts\";\nimport type { AgentSessionInternalSurface as AgentSession } from \"./agent-session-methods.ts\";\nimport { COPILOT_CONTEXT_WINDOW_SELECTION_OPTIONS, THINKING_LEVELS, type ContextWindowReplayRequest, type ContextWindowReplaySource, type ModelCycleResult } from \"./agent-session-types.ts\";\n\nexport async function _getRequiredRequestAuth(this: AgentSession, model: Model<Api>): Promise<{\n\tapiKey: string;\n\theaders?: ProviderHeaders;\n\tbaseUrl?: string;\n}> {\n\tconst result = await this._modelRegistry.getApiKeyAndHeaders(model);\n\tif (!result.ok) {\n\t\tif (result.error.startsWith(\"No API key found\")) {\n\t\t\tthrow new Error(formatNoApiKeyFoundMessage(model.provider));\n\t\t}\n\t\tthrow new Error(result.error);\n\t}\n\tif (result.apiKey) {\n\t\treturn { apiKey: result.apiKey, headers: result.headers, baseUrl: result.baseUrl };\n\t}\n\n\tconst isOAuth = this._modelRegistry.isUsingOAuth(model);\n\tif (isOAuth) {\n\t\tthrow new Error(\n\t\t\t`Authentication failed for \"${model.provider}\". ` +\n\t\t\t\t`Credentials may have expired or network is unavailable. ` +\n\t\t\t\t`Run '/login ${model.provider}' to re-authenticate.`,\n\t\t);\n\t}\n\tthrow new Error(formatNoApiKeyFoundMessage(model.provider));\n}\n\n/**\n * Install tool hooks once on the Agent instance.\n *\n * The callbacks read `this._extensionRunner` at execution time, so extension reload swaps in the\n * new runner without reinstalling hooks. Extension-specific tool wrappers are still used to adapt\n * registered tool execution to the extension context. Tool call and tool result interception now\n * happens here instead of in wrappers.\n */\n\nexport function _emitModelChanged(this: AgentSession, \n\tnextModel: Model<Api>,\n\tpreviousModel: Model<Api> | undefined,\n\tsource: \"set\" | \"cycle\" | \"restore\" | \"fallback\",\n): void {\n\tif (modelsAreEqual(previousModel, nextModel)) return;\n\tthis._emit({\n\t\ttype: \"model_changed\",\n\t\tmodel: nextModel,\n\t\tpreviousModel,\n\t\tsource,\n\t});\n}\n\n\nexport async function _emitModelSelect(this: AgentSession, \n\tnextModel: Model<Api>,\n\tpreviousModel: Model<Api> | undefined,\n\tsource: \"set\" | \"cycle\" | \"restore\" | \"fallback\",\n): Promise<void> {\n\tif (modelsAreEqual(previousModel, nextModel)) return;\n\tawait this._extensionRunner.emit({\n\t\ttype: \"model_select\",\n\t\tmodel: nextModel,\n\t\tpreviousModel,\n\t\tsource,\n\t});\n}\n\n/**\n * Set model directly.\n * Validates that auth is configured, saves to session and settings.\n * @throws Error if no auth is configured for the model\n */\n\nexport async function setModel(this: AgentSession, model: Model<Api>): Promise<void> {\n\tif (!this._modelRegistry.hasConfiguredAuth(model)) {\n\t\tthrow new Error(`No API key for ${model.provider}/${model.id}`);\n\t}\n\n\tconst previousModel = this.model;\n\tconst thinkingLevel = this._getThinkingLevelForModelSwitch();\n\tconst nextModel = this._withContextWindowForModelSwitch(model);\n\tthis.agent.state.model = nextModel;\n\tthis.sessionManager.appendModelChange(nextModel.provider, nextModel.id);\n\tthis._appendContextWindowChangeIfChanged(previousModel, nextModel);\n\tthis.settingsManager.setDefaultModelAndProvider(nextModel.provider, nextModel.id);\n\n\t// Re-clamp thinking level for new model's capabilities\n\tthis.setThinkingLevel(thinkingLevel);\n\tthis._refreshBaseSystemPromptFromActiveTools();\n\n\tthis._emitModelChanged(nextModel, previousModel, \"set\");\n\tawait this._emitModelSelect(nextModel, previousModel, \"set\");\n}\n\n/**\n * Cycle to next/previous model.\n * Uses scoped models (from --models flag) if available, otherwise all available models.\n * @param direction - \"forward\" (default) or \"backward\"\n * @returns The new model info, or undefined if only one model available\n */\n\nexport async function cycleModel(this: AgentSession, direction: \"forward\" | \"backward\" = \"forward\"): Promise<ModelCycleResult | undefined> {\n\tif (this._scopedModels.length > 0) {\n\t\treturn this._cycleScopedModel(direction);\n\t}\n\treturn this._cycleAvailableModel(direction);\n}\n\n\nexport async function _cycleScopedModel(this: AgentSession, direction: \"forward\" | \"backward\"): Promise<ModelCycleResult | undefined> {\n\tconst scopedModels = this._scopedModels.filter((scoped) => this._modelRegistry.hasConfiguredAuth(scoped.model));\n\tif (scopedModels.length <= 1) return undefined;\n\n\tconst currentModel = this.model;\n\tlet currentIndex = scopedModels.findIndex((sm) => modelsAreEqual(sm.model, currentModel));\n\n\tif (currentIndex === -1) currentIndex = 0;\n\tconst len = scopedModels.length;\n\tconst nextIndex = direction === \"forward\" ? (currentIndex + 1) % len : (currentIndex - 1 + len) % len;\n\tconst next = scopedModels[nextIndex];\n\tconst thinkingLevel = this._getThinkingLevelForModelSwitch(next.thinkingLevel);\n\tconst nextModel = this._withContextWindowForModelSwitch(next.model);\n\n\t// Apply model\n\tthis.agent.state.model = nextModel;\n\tthis.sessionManager.appendModelChange(nextModel.provider, nextModel.id);\n\tthis._appendContextWindowChangeIfChanged(currentModel, nextModel);\n\tthis.settingsManager.setDefaultModelAndProvider(nextModel.provider, nextModel.id);\n\n\t// Apply thinking level.\n\t// - Explicit scoped model thinking level overrides current session level\n\t// - Undefined scoped model thinking level inherits the current session preference\n\t// setThinkingLevel clamps to model capabilities.\n\tthis.setThinkingLevel(thinkingLevel);\n\tthis._refreshBaseSystemPromptFromActiveTools();\n\n\tthis._emitModelChanged(nextModel, currentModel, \"cycle\");\n\tawait this._emitModelSelect(nextModel, currentModel, \"cycle\");\n\n\treturn { model: nextModel, thinkingLevel: this.thinkingLevel, isScoped: true };\n}\n\n\nexport async function _cycleAvailableModel(this: AgentSession, direction: \"forward\" | \"backward\"): Promise<ModelCycleResult | undefined> {\n\tconst availableModels = await this._modelRegistry.getAvailable();\n\tif (availableModels.length <= 1) return undefined;\n\n\tconst currentModel = this.model;\n\tlet currentIndex = availableModels.findIndex((m) => modelsAreEqual(m, currentModel));\n\n\tif (currentIndex === -1) currentIndex = 0;\n\tconst len = availableModels.length;\n\tconst nextIndex = direction === \"forward\" ? (currentIndex + 1) % len : (currentIndex - 1 + len) % len;\n\tconst selectedModel = this._withContextWindowForModelSwitch(availableModels[nextIndex]);\n\n\tconst thinkingLevel = this._getThinkingLevelForModelSwitch();\n\tthis.agent.state.model = selectedModel;\n\tthis.sessionManager.appendModelChange(selectedModel.provider, selectedModel.id);\n\tthis._appendContextWindowChangeIfChanged(currentModel, selectedModel);\n\tthis.settingsManager.setDefaultModelAndProvider(selectedModel.provider, selectedModel.id);\n\n\t// Re-clamp thinking level for new model's capabilities\n\tthis.setThinkingLevel(thinkingLevel);\n\tthis._refreshBaseSystemPromptFromActiveTools();\n\n\tthis._emitModelChanged(selectedModel, currentModel, \"cycle\");\n\tawait this._emitModelSelect(selectedModel, currentModel, \"cycle\");\n\n\treturn { model: selectedModel, thinkingLevel: this.thinkingLevel, isScoped: false };\n}\n\n// =========================================================================\n// Thinking Level Management\n// =========================================================================\n\n/**\n * Set thinking level.\n * Clamps to model capabilities based on available thinking levels.\n * Saves to session and settings only if the level actually changes.\n */\n\nexport function setThinkingLevel(this: AgentSession, level: ThinkingLevel): void {\n\tconst availableLevels = this.getAvailableThinkingLevels();\n\tconst effectiveLevel = availableLevels.includes(level) ? level : this._clampThinkingLevel(level, availableLevels);\n\n\t// Only persist if actually changing\n\tconst previousLevel = this.agent.state.thinkingLevel;\n\tconst isChanging = effectiveLevel !== previousLevel;\n\n\tthis.agent.state.thinkingLevel = effectiveLevel;\n\n\tif (isChanging) {\n\t\tthis.sessionManager.appendThinkingLevelChange(effectiveLevel);\n\t\tthis._refreshBaseSystemPromptFromActiveTools();\n\t\tif (this.supportsThinking() || effectiveLevel !== \"off\") {\n\t\t\tthis.settingsManager.setDefaultThinkingLevel(effectiveLevel);\n\t\t}\n\t\tthis._emit({ type: \"thinking_level_changed\", level: effectiveLevel });\n\t\tvoid this._extensionRunner.emit({\n\t\t\ttype: \"thinking_level_select\",\n\t\t\tlevel: effectiveLevel,\n\t\t\tpreviousLevel,\n\t\t});\n\t}\n}\n\n/**\n * Cycle to next thinking level.\n * @returns New level, or undefined if model doesn't support thinking\n */\n\nexport function cycleThinkingLevel(this: AgentSession): ThinkingLevel | undefined {\n\tif (!this.supportsThinking()) return undefined;\n\n\tconst levels = this.getAvailableThinkingLevels();\n\tconst currentIndex = levels.indexOf(this.thinkingLevel);\n\tconst nextIndex = (currentIndex + 1) % levels.length;\n\tconst nextLevel = levels[nextIndex];\n\n\tthis.setThinkingLevel(nextLevel);\n\treturn nextLevel;\n}\n\n/**\n * Get available thinking levels for current model.\n * The provider will clamp to what the specific model supports internally.\n */\n\nexport function getAvailableThinkingLevels(this: AgentSession): ThinkingLevel[] {\n\tif (!this.model) return THINKING_LEVELS;\n\treturn getSupportedThinkingLevels(this.model) as ThinkingLevel[];\n}\n\n/**\n * Check if current model supports thinking/reasoning.\n */\n\nexport function supportsThinking(this: AgentSession): boolean {\n\treturn !!this.model?.reasoning;\n}\n\n\nexport function _getThinkingLevelForModelSwitch(this: AgentSession, explicitLevel?: ThinkingLevel): ThinkingLevel {\n\tif (explicitLevel !== undefined) {\n\t\treturn explicitLevel;\n\t}\n\tif (!this.supportsThinking()) {\n\t\treturn this.settingsManager.getDefaultThinkingLevel() ?? DEFAULT_THINKING_LEVEL;\n\t}\n\treturn this.thinkingLevel;\n}\n\n\nexport function _clampThinkingLevel(this: AgentSession, level: ThinkingLevel, _availableLevels: ThinkingLevel[]): ThinkingLevel {\n\treturn this.model ? (clampThinkingLevel(this.model, level) as ThinkingLevel) : \"off\";\n}\n\n// =========================================================================\n// Context Window Management\n// =========================================================================\n\n\nexport function getAvailableContextWindows(this: AgentSession): number[] {\n\treturn this.model ? getSupportedContextWindows(this.model) : [];\n}\n\n\nexport function supportsContextWindowSelection(this: AgentSession): boolean {\n\treturn this.getAvailableContextWindows().length > 1;\n}\n\n\nexport function setContextWindow(this: AgentSession, contextWindow: number, options: { persistDefault?: boolean } = {}): void {\n\tif (!this.model) {\n\t\tthrow new Error(\"No model selected\");\n\t}\n\tconst selected = selectContextWindow(this.model, contextWindow, COPILOT_CONTEXT_WINDOW_SELECTION_OPTIONS);\n\tif (\"error\" in selected) {\n\t\tthrow new Error(selected.error);\n\t}\n\n\tconst previousContextWindow = this.model.contextWindow;\n\tconst isChanging = previousContextWindow !== selected.contextWindow;\n\tthis.agent.state.model = selected.model;\n\n\tif (isChanging) {\n\t\tthis.sessionManager.appendContextWindowChange(selected.contextWindow);\n\t\tthis._emit({ type: \"context_window_changed\", contextWindow: selected.contextWindow });\n\t}\n\tif (options.persistDefault === true) {\n\t\tthis.settingsManager.setDefaultContextWindowForModel(selected.model.provider, selected.model.id, selected.contextWindow);\n\t}\n}\n\n\nexport function _withContextWindowForModelSwitch(this: AgentSession, model: Model<Api>): Model<Api> {\n\t// A source model's scalar contextWindow can be its natural default (for example a 1m-default\n\t// model). Do not treat that alone as an opt-in to larger windows on a 400k-default target.\n\tconst settingsDefaultContextWindow = this._getSettingsContextWindowRequestForModel(model)?.contextWindow;\n\tconst candidates: number[] = [];\n\tconst targetDefaultContextWindow = getModelDefaultContextWindow(model);\n\tif (model.contextWindow !== targetDefaultContextWindow) {\n\t\t// Preserve an explicit context-window selection already applied to the target model\n\t\t// (for example a caller passing selectContextWindow(target, 1m).model).\n\t\tcandidates.push(model.contextWindow);\n\t}\n\tconst currentModel = this.model;\n\tif (currentModel && this._shouldCarryCurrentContextWindowForModelSwitch(currentModel, settingsDefaultContextWindow)) {\n\t\tcandidates.push(currentModel.contextWindow);\n\t}\n\tif (settingsDefaultContextWindow !== undefined) {\n\t\tcandidates.push(settingsDefaultContextWindow);\n\t}\n\tcandidates.push(targetDefaultContextWindow);\n\n\tfor (const candidate of candidates) {\n\t\tconst selected = selectContextWindow(model, candidate, COPILOT_CONTEXT_WINDOW_SELECTION_OPTIONS);\n\t\tif (!(\"error\" in selected)) return selected.model;\n\t}\n\treturn model;\n}\n\n\nexport function _shouldCarryCurrentContextWindowForModelSwitch(this: AgentSession, \n\tcurrentModel: Model<Api>,\n\tsettingsDefaultContextWindow: number | undefined,\n): boolean {\n\tif (currentModel.contextWindow !== getModelDefaultContextWindow(currentModel)) {\n\t\treturn true;\n\t}\n\tif (this.sessionManager.getBranch().some((entry) => entry.type === \"context_window_change\")) {\n\t\treturn true;\n\t}\n\treturn (\n\t\tsettingsDefaultContextWindow !== undefined &&\n\t\tcurrentModel.contextWindow === settingsDefaultContextWindow &&\n\t\tgetSupportedContextWindows(currentModel).includes(settingsDefaultContextWindow)\n\t);\n}\n\n\nexport function _getSettingsContextWindowRequestForModel(this: AgentSession, model: Model<Api>): ContextWindowReplayRequest | undefined {\n\tconst modelContextWindow = this.settingsManager.getDefaultContextWindowForModel(model.provider, model.id);\n\tif (modelContextWindow !== undefined) {\n\t\treturn { contextWindow: modelContextWindow, source: \"model-settings\" };\n\t}\n\tconst globalContextWindow = this.settingsManager.getDefaultContextWindow();\n\treturn globalContextWindow === undefined\n\t\t? undefined\n\t\t: { contextWindow: globalContextWindow, source: \"global-settings\" };\n}\n\n\nexport function _getContextWindowReplayForModel(this: AgentSession, \n\tmodel: Model<Api>,\n\trequestedContextWindow: number | undefined,\n\tsource: ContextWindowReplaySource | undefined,\n): { model: Model<Api>; contextWindow: number; wouldWarn: boolean } {\n\tif (requestedContextWindow !== undefined) {\n\t\tconst selected = selectContextWindow(model, requestedContextWindow, COPILOT_CONTEXT_WINDOW_SELECTION_OPTIONS);\n\t\tif (!(\"error\" in selected)) {\n\t\t\treturn { model: selected.model, contextWindow: selected.contextWindow, wouldWarn: false };\n\t\t}\n\t\treturn this._getDefaultContextWindowReplayForModel(model, source !== \"global-settings\");\n\t}\n\n\treturn this._getDefaultContextWindowReplayForModel(model, false);\n}\n\n\nexport function _getDefaultContextWindowReplayForModel(this: AgentSession, \n\tmodel: Model<Api>,\n\twouldWarn: boolean,\n): { model: Model<Api>; contextWindow: number; wouldWarn: boolean } {\n\tconst defaultContextWindow = getModelDefaultContextWindow(model);\n\tconst selected = selectContextWindow(model, defaultContextWindow, COPILOT_CONTEXT_WINDOW_SELECTION_OPTIONS);\n\tif (!(\"error\" in selected)) {\n\t\treturn { model: selected.model, contextWindow: selected.contextWindow, wouldWarn };\n\t}\n\treturn {\n\t\tmodel: { ...model, contextWindow: defaultContextWindow, defaultContextWindow },\n\t\tcontextWindow: defaultContextWindow,\n\t\twouldWarn,\n\t};\n}\n\n\nexport function _getResumeContextWindowReplayForModel(this: AgentSession, \n\tmodel: Model<Api>,\n): { model: Model<Api>; contextWindow: number; wouldWarn: boolean } {\n\tconst sessionContext = this.sessionManager.buildSessionContext();\n\tif (sessionContext.contextWindow !== undefined) {\n\t\treturn this._getContextWindowReplayForModel(model, sessionContext.contextWindow, \"session\");\n\t}\n\tconst settingsContextWindow = this._getSettingsContextWindowRequestForModel(model);\n\treturn this._getContextWindowReplayForModel(model, settingsContextWindow?.contextWindow, settingsContextWindow?.source);\n}\n\n\nexport function _applyContextWindowReplay(this: AgentSession, contextWindow: number | undefined): void {\n\tif (!this.model) return;\n\tconst previousContextWindow = this.model.contextWindow;\n\tconst settingsContextWindow = this._getSettingsContextWindowRequestForModel(this.model);\n\tconst requestedContextWindow = contextWindow ?? settingsContextWindow?.contextWindow;\n\tconst source: ContextWindowReplaySource | undefined = contextWindow !== undefined ? \"session\" : settingsContextWindow?.source;\n\tconst replay = this._getContextWindowReplayForModel(this.model, requestedContextWindow, source);\n\tthis.agent.state.model = replay.model;\n\tif (previousContextWindow !== replay.contextWindow) {\n\t\tthis._emit({ type: \"context_window_changed\", contextWindow: replay.contextWindow });\n\t}\n}\n\n\nexport function _appendContextWindowChangeIfChanged(this: AgentSession, \n\tpreviousModel: Model<Api> | undefined,\n\tnextModel: Model<Api>,\n): void {\n\tconst replay = this._getResumeContextWindowReplayForModel(nextModel);\n\tif (!replay.wouldWarn && nextModel.contextWindow === replay.contextWindow) return;\n\tthis.sessionManager.appendContextWindowChange(nextModel.contextWindow);\n\tif (previousModel?.contextWindow !== nextModel.contextWindow) {\n\t\tthis._emit({ type: \"context_window_changed\", contextWindow: nextModel.contextWindow });\n\t}\n}\n\n// =========================================================================\n// Queue Mode Management\n// =========================================================================\n\n/**\n * Set steering message mode.\n * Saves to settings.\n */\n\nexport const agentSessionModelsMethods = {\n\t_getRequiredRequestAuth,\n\t_emitModelChanged,\n\t_emitModelSelect,\n\tsetModel,\n\tcycleModel,\n\t_cycleScopedModel,\n\t_cycleAvailableModel,\n\tsetThinkingLevel,\n\tcycleThinkingLevel,\n\tgetAvailableThinkingLevels,\n\tsupportsThinking,\n\t_getThinkingLevelForModelSwitch,\n\t_clampThinkingLevel,\n\tgetAvailableContextWindows,\n\tsupportsContextWindowSelection,\n\tsetContextWindow,\n\t_withContextWindowForModelSwitch,\n\t_shouldCarryCurrentContextWindowForModelSwitch,\n\t_getSettingsContextWindowRequestForModel,\n\t_getContextWindowReplayForModel,\n\t_getDefaultContextWindowReplayForModel,\n\t_getResumeContextWindowReplayForModel,\n\t_applyContextWindowReplay,\n\t_appendContextWindowChangeIfChanged,\n};\n"]}
+{"version":3,"file":"agent-session-models.js","sourceRoot":"","sources":["../../src/core/agent-session-models.ts"],"names":[],"mappings":"AAGA,OAAO,EAAE,kBAAkB,EAAE,0BAA0B,EAAE,cAAc,EAAE,MAAM,8BAA8B,CAAC;AAC9G,OAAO,EAAE,0BAA0B,EAAE,MAAM,oBAAoB,CAAC;AAChE,OAAO,EAAE,sBAAsB,EAAE,MAAM,eAAe,CAAC;AAEvD,OAAO,EAAE,eAAe,EAAyB,MAAM,0BAA0B,CAAC;AAElF,MAAM,CAAC,KAAK,UAAU,uBAAuB,CAAqB,KAAiB;IAKlF,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,cAAc,CAAC,mBAAmB,CAAC,KAAK,CAAC,CAAC;IACpE,IAAI,CAAC,MAAM,CAAC,EAAE,EAAE,CAAC;QAChB,IAAI,MAAM,CAAC,KAAK,CAAC,UAAU,CAAC,kBAAkB,CAAC,EAAE,CAAC;YACjD,MAAM,IAAI,KAAK,CAAC,0BAA0B,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC,CAAC;QAC7D,CAAC;QACD,MAAM,IAAI,KAAK,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;IAC/B,CAAC;IACD,IAAI,MAAM,CAAC,MAAM,IAAI,MAAM,CAAC,OAAO,EAAE,CAAC;QACrC,OAAO,EAAE,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,EAAE,MAAM,CAAC,OAAO,EAAE,OAAO,EAAE,MAAM,CAAC,OAAO,EAAE,CAAC;IACpF,CAAC;IAED,MAAM,OAAO,GAAG,IAAI,CAAC,cAAc,CAAC,YAAY,CAAC,KAAK,CAAC,CAAC;IACxD,IAAI,OAAO,EAAE,CAAC;QACb,MAAM,IAAI,KAAK,CACd,8BAA8B,KAAK,CAAC,QAAQ,KAAK;YAChD,0DAA0D;YAC1D,eAAe,KAAK,CAAC,QAAQ,uBAAuB,CACrD,CAAC;IACH,CAAC;IACD,MAAM,IAAI,KAAK,CAAC,0BAA0B,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC,CAAC;AAC7D,CAAC;AAED;;;;;;;GAOG;AAEH,MAAM,UAAU,iBAAiB,CAChC,SAAqB,EACrB,aAAqC,EACrC,MAAgD;IAEhD,IAAI,cAAc,CAAC,aAAa,EAAE,SAAS,CAAC;QAAE,OAAO;IACrD,IAAI,CAAC,KAAK,CAAC;QACV,IAAI,EAAE,eAAe;QACrB,KAAK,EAAE,SAAS;QAChB,aAAa;QACb,MAAM;KACN,CAAC,CAAC;AACJ,CAAC;AAGD,MAAM,CAAC,KAAK,UAAU,gBAAgB,CACrC,SAAqB,EACrB,aAAqC,EACrC,MAAgD;IAEhD,IAAI,cAAc,CAAC,aAAa,EAAE,SAAS,CAAC;QAAE,OAAO;IACrD,MAAM,IAAI,CAAC,gBAAgB,CAAC,IAAI,CAAC;QAChC,IAAI,EAAE,cAAc;QACpB,KAAK,EAAE,SAAS;QAChB,aAAa;QACb,MAAM;KACN,CAAC,CAAC;AACJ,CAAC;AAED;;;;GAIG;AAEH,MAAM,CAAC,KAAK,UAAU,QAAQ,CAAqB,KAAiB;IACnE,IAAI,CAAC,IAAI,CAAC,cAAc,CAAC,iBAAiB,CAAC,KAAK,CAAC,EAAE,CAAC;QACnD,MAAM,IAAI,KAAK,CAAC,kBAAkB,KAAK,CAAC,QAAQ,IAAI,KAAK,CAAC,EAAE,EAAE,CAAC,CAAC;IACjE,CAAC;IAED,MAAM,aAAa,GAAG,IAAI,CAAC,KAAK,CAAC;IACjC,MAAM,aAAa,GAAG,IAAI,CAAC,+BAA+B,EAAE,CAAC;IAC7D,MAAM,SAAS,GAAG,KAAK,CAAC;IACxB,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,KAAK,GAAG,SAAS,CAAC;IACnC,IAAI,CAAC,cAAc,CAAC,iBAAiB,CAAC,SAAS,CAAC,QAAQ,EAAE,SAAS,CAAC,EAAE,CAAC,CAAC;IACxE,IAAI,CAAC,eAAe,CAAC,0BAA0B,CAAC,SAAS,CAAC,QAAQ,EAAE,SAAS,CAAC,EAAE,CAAC,CAAC;IAElF,uDAAuD;IACvD,IAAI,CAAC,gBAAgB,CAAC,aAAa,CAAC,CAAC;IACrC,IAAI,CAAC,uCAAuC,EAAE,CAAC;IAE/C,IAAI,CAAC,iBAAiB,CAAC,SAAS,EAAE,aAAa,EAAE,KAAK,CAAC,CAAC;IACxD,MAAM,IAAI,CAAC,gBAAgB,CAAC,SAAS,EAAE,aAAa,EAAE,KAAK,CAAC,CAAC;AAC9D,CAAC;AAED;;;;;GAKG;AAEH,MAAM,CAAC,KAAK,UAAU,UAAU,CAAqB,SAAS,GAA2B,SAAS;IACjG,IAAI,IAAI,CAAC,aAAa,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QACnC,OAAO,IAAI,CAAC,iBAAiB,CAAC,SAAS,CAAC,CAAC;IAC1C,CAAC;IACD,OAAO,IAAI,CAAC,oBAAoB,CAAC,SAAS,CAAC,CAAC;AAC7C,CAAC;AAGD,MAAM,CAAC,KAAK,UAAU,iBAAiB,CAAqB,SAAiC;IAC5F,MAAM,YAAY,GAAG,IAAI,CAAC,aAAa,CAAC,MAAM,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,IAAI,CAAC,cAAc,CAAC,iBAAiB,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC;IAChH,IAAI,YAAY,CAAC,MAAM,IAAI,CAAC;QAAE,OAAO,SAAS,CAAC;IAE/C,MAAM,YAAY,GAAG,IAAI,CAAC,KAAK,CAAC;IAChC,IAAI,YAAY,GAAG,YAAY,CAAC,SAAS,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,cAAc,CAAC,EAAE,CAAC,KAAK,EAAE,YAAY,CAAC,CAAC,CAAC;IAE1F,IAAI,YAAY,KAAK,CAAC,CAAC;QAAE,YAAY,GAAG,CAAC,CAAC;IAC1C,MAAM,GAAG,GAAG,YAAY,CAAC,MAAM,CAAC;IAChC,MAAM,SAAS,GAAG,SAAS,KAAK,SAAS,CAAC,CAAC,CAAC,CAAC,YAAY,GAAG,CAAC,CAAC,GAAG,GAAG,CAAC,CAAC,CAAC,CAAC,YAAY,GAAG,CAAC,GAAG,GAAG,CAAC,GAAG,GAAG,CAAC;IACtG,MAAM,IAAI,GAAG,YAAY,CAAC,SAAS,CAAC,CAAC;IACrC,MAAM,aAAa,GAAG,IAAI,CAAC,+BAA+B,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;IAC/E,MAAM,SAAS,GAAG,IAAI,CAAC,KAAK,CAAC;IAE7B,cAAc;IACd,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,KAAK,GAAG,SAAS,CAAC;IACnC,IAAI,CAAC,cAAc,CAAC,iBAAiB,CAAC,SAAS,CAAC,QAAQ,EAAE,SAAS,CAAC,EAAE,CAAC,CAAC;IACxE,IAAI,CAAC,eAAe,CAAC,0BAA0B,CAAC,SAAS,CAAC,QAAQ,EAAE,SAAS,CAAC,EAAE,CAAC,CAAC;IAElF,wBAAwB;IACxB,yEAAyE;IACzE,kFAAkF;IAClF,iDAAiD;IACjD,IAAI,CAAC,gBAAgB,CAAC,aAAa,CAAC,CAAC;IACrC,IAAI,CAAC,uCAAuC,EAAE,CAAC;IAE/C,IAAI,CAAC,iBAAiB,CAAC,SAAS,EAAE,YAAY,EAAE,OAAO,CAAC,CAAC;IACzD,MAAM,IAAI,CAAC,gBAAgB,CAAC,SAAS,EAAE,YAAY,EAAE,OAAO,CAAC,CAAC;IAE9D,OAAO,EAAE,KAAK,EAAE,SAAS,EAAE,aAAa,EAAE,IAAI,CAAC,aAAa,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC;AAChF,CAAC;AAGD,MAAM,CAAC,KAAK,UAAU,oBAAoB,CAAqB,SAAiC;IAC/F,MAAM,eAAe,GAAG,MAAM,IAAI,CAAC,cAAc,CAAC,YAAY,EAAE,CAAC;IACjE,IAAI,eAAe,CAAC,MAAM,IAAI,CAAC;QAAE,OAAO,SAAS,CAAC;IAElD,MAAM,YAAY,GAAG,IAAI,CAAC,KAAK,CAAC;IAChC,IAAI,YAAY,GAAG,eAAe,CAAC,SAAS,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,cAAc,CAAC,CAAC,EAAE,YAAY,CAAC,CAAC,CAAC;IAErF,IAAI,YAAY,KAAK,CAAC,CAAC;QAAE,YAAY,GAAG,CAAC,CAAC;IAC1C,MAAM,GAAG,GAAG,eAAe,CAAC,MAAM,CAAC;IACnC,MAAM,SAAS,GAAG,SAAS,KAAK,SAAS,CAAC,CAAC,CAAC,CAAC,YAAY,GAAG,CAAC,CAAC,GAAG,GAAG,CAAC,CAAC,CAAC,CAAC,YAAY,GAAG,CAAC,GAAG,GAAG,CAAC,GAAG,GAAG,CAAC;IACtG,MAAM,aAAa,GAAG,eAAe,CAAC,SAAS,CAAC,CAAC;IAEjD,MAAM,aAAa,GAAG,IAAI,CAAC,+BAA+B,EAAE,CAAC;IAC7D,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,KAAK,GAAG,aAAa,CAAC;IACvC,IAAI,CAAC,cAAc,CAAC,iBAAiB,CAAC,aAAa,CAAC,QAAQ,EAAE,aAAa,CAAC,EAAE,CAAC,CAAC;IAChF,IAAI,CAAC,eAAe,CAAC,0BAA0B,CAAC,aAAa,CAAC,QAAQ,EAAE,aAAa,CAAC,EAAE,CAAC,CAAC;IAE1F,uDAAuD;IACvD,IAAI,CAAC,gBAAgB,CAAC,aAAa,CAAC,CAAC;IACrC,IAAI,CAAC,uCAAuC,EAAE,CAAC;IAE/C,IAAI,CAAC,iBAAiB,CAAC,aAAa,EAAE,YAAY,EAAE,OAAO,CAAC,CAAC;IAC7D,MAAM,IAAI,CAAC,gBAAgB,CAAC,aAAa,EAAE,YAAY,EAAE,OAAO,CAAC,CAAC;IAElE,OAAO,EAAE,KAAK,EAAE,aAAa,EAAE,aAAa,EAAE,IAAI,CAAC,aAAa,EAAE,QAAQ,EAAE,KAAK,EAAE,CAAC;AACrF,CAAC;AAED,4EAA4E;AAC5E,4BAA4B;AAC5B,4EAA4E;AAE5E;;;;GAIG;AAEH,MAAM,UAAU,gBAAgB,CAAqB,KAAoB;IACxE,MAAM,eAAe,GAAG,IAAI,CAAC,0BAA0B,EAAE,CAAC;IAC1D,MAAM,cAAc,GAAG,eAAe,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,CAAC,mBAAmB,CAAC,KAAK,EAAE,eAAe,CAAC,CAAC;IAElH,oCAAoC;IACpC,MAAM,aAAa,GAAG,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,aAAa,CAAC;IACrD,MAAM,UAAU,GAAG,cAAc,KAAK,aAAa,CAAC;IAEpD,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,aAAa,GAAG,cAAc,CAAC;IAEhD,IAAI,UAAU,EAAE,CAAC;QAChB,IAAI,CAAC,cAAc,CAAC,yBAAyB,CAAC,cAAc,CAAC,CAAC;QAC9D,IAAI,CAAC,uCAAuC,EAAE,CAAC;QAC/C,IAAI,IAAI,CAAC,gBAAgB,EAAE,IAAI,cAAc,KAAK,KAAK,EAAE,CAAC;YACzD,IAAI,CAAC,eAAe,CAAC,uBAAuB,CAAC,cAAc,CAAC,CAAC;QAC9D,CAAC;QACD,IAAI,CAAC,KAAK,CAAC,EAAE,IAAI,EAAE,wBAAwB,EAAE,KAAK,EAAE,cAAc,EAAE,CAAC,CAAC;QACtE,KAAK,IAAI,CAAC,gBAAgB,CAAC,IAAI,CAAC;YAC/B,IAAI,EAAE,uBAAuB;YAC7B,KAAK,EAAE,cAAc;YACrB,aAAa;SACb,CAAC,CAAC;IACJ,CAAC;AACF,CAAC;AAED;;;GAGG;AAEH,MAAM,UAAU,kBAAkB;IACjC,IAAI,CAAC,IAAI,CAAC,gBAAgB,EAAE;QAAE,OAAO,SAAS,CAAC;IAE/C,MAAM,MAAM,GAAG,IAAI,CAAC,0BAA0B,EAAE,CAAC;IACjD,MAAM,YAAY,GAAG,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;IACxD,MAAM,SAAS,GAAG,CAAC,YAAY,GAAG,CAAC,CAAC,GAAG,MAAM,CAAC,MAAM,CAAC;IACrD,MAAM,SAAS,GAAG,MAAM,CAAC,SAAS,CAAC,CAAC;IAEpC,IAAI,CAAC,gBAAgB,CAAC,SAAS,CAAC,CAAC;IACjC,OAAO,SAAS,CAAC;AAClB,CAAC;AAED;;;GAGG;AAEH,MAAM,UAAU,0BAA0B;IACzC,IAAI,CAAC,IAAI,CAAC,KAAK;QAAE,OAAO,eAAe,CAAC;IACxC,OAAO,0BAA0B,CAAC,IAAI,CAAC,KAAK,CAAoB,CAAC;AAClE,CAAC;AAED;;GAEG;AAEH,MAAM,UAAU,gBAAgB;IAC/B,OAAO,CAAC,CAAC,IAAI,CAAC,KAAK,EAAE,SAAS,CAAC;AAChC,CAAC;AAGD,MAAM,UAAU,+BAA+B,CAAqB,aAA6B;IAChG,IAAI,aAAa,KAAK,SAAS,EAAE,CAAC;QACjC,OAAO,aAAa,CAAC;IACtB,CAAC;IACD,IAAI,CAAC,IAAI,CAAC,gBAAgB,EAAE,EAAE,CAAC;QAC9B,OAAO,IAAI,CAAC,eAAe,CAAC,uBAAuB,EAAE,IAAI,sBAAsB,CAAC;IACjF,CAAC;IACD,OAAO,IAAI,CAAC,aAAa,CAAC;AAC3B,CAAC;AAGD,MAAM,UAAU,mBAAmB,CAAqB,KAAoB,EAAE,gBAAiC;IAC9G,OAAO,IAAI,CAAC,KAAK,CAAC,CAAC,CAAE,kBAAkB,CAAC,IAAI,CAAC,KAAK,EAAE,KAAK,CAAmB,CAAC,CAAC,CAAC,KAAK,CAAC;AACtF,CAAC;AAGD,4EAA4E;AAC5E,wBAAwB;AACxB,4EAA4E;AAE5E;;;GAGG;AAEH,MAAM,CAAC,MAAM,yBAAyB,GAAG;IACxC,uBAAuB;IACvB,iBAAiB;IACjB,gBAAgB;IAChB,QAAQ;IACR,UAAU;IACV,iBAAiB;IACjB,oBAAoB;IACpB,gBAAgB;IAChB,kBAAkB;IAClB,0BAA0B;IAC1B,gBAAgB;IAChB,+BAA+B;IAC/B,mBAAmB;CACnB,CAAC","sourcesContent":["import type { ThinkingLevel } from \"@earendil-works/pi-agent-core\";\nimport type { ProviderHeaders } from \"@earendil-works/pi-ai\";\nimport type { Api, Model } from \"@earendil-works/pi-ai/compat\";\nimport { clampThinkingLevel, getSupportedThinkingLevels, modelsAreEqual } from \"@earendil-works/pi-ai/compat\";\nimport { formatNoApiKeyFoundMessage } from \"./auth-guidance.ts\";\nimport { DEFAULT_THINKING_LEVEL } from \"./defaults.ts\";\nimport type { AgentSessionInternalSurface as AgentSession } from \"./agent-session-methods.ts\";\nimport { THINKING_LEVELS, type ModelCycleResult } from \"./agent-session-types.ts\";\n\nexport async function _getRequiredRequestAuth(this: AgentSession, model: Model<Api>): Promise<{\n\tapiKey?: string;\n\theaders?: ProviderHeaders;\n\tbaseUrl?: string;\n}> {\n\tconst result = await this._modelRegistry.getApiKeyAndHeaders(model);\n\tif (!result.ok) {\n\t\tif (result.error.startsWith(\"No API key found\")) {\n\t\t\tthrow new Error(formatNoApiKeyFoundMessage(model.provider));\n\t\t}\n\t\tthrow new Error(result.error);\n\t}\n\tif (result.apiKey || result.headers) {\n\t\treturn { apiKey: result.apiKey, headers: result.headers, baseUrl: result.baseUrl };\n\t}\n\n\tconst isOAuth = this._modelRegistry.isUsingOAuth(model);\n\tif (isOAuth) {\n\t\tthrow new Error(\n\t\t\t`Authentication failed for \"${model.provider}\". ` +\n\t\t\t\t`Credentials may have expired or network is unavailable. ` +\n\t\t\t\t`Run '/login ${model.provider}' to re-authenticate.`,\n\t\t);\n\t}\n\tthrow new Error(formatNoApiKeyFoundMessage(model.provider));\n}\n\n/**\n * Install tool hooks once on the Agent instance.\n *\n * The callbacks read `this._extensionRunner` at execution time, so extension reload swaps in the\n * new runner without reinstalling hooks. Extension-specific tool wrappers are still used to adapt\n * registered tool execution to the extension context. Tool call and tool result interception now\n * happens here instead of in wrappers.\n */\n\nexport function _emitModelChanged(this: AgentSession, \n\tnextModel: Model<Api>,\n\tpreviousModel: Model<Api> | undefined,\n\tsource: \"set\" | \"cycle\" | \"restore\" | \"fallback\",\n): void {\n\tif (modelsAreEqual(previousModel, nextModel)) return;\n\tthis._emit({\n\t\ttype: \"model_changed\",\n\t\tmodel: nextModel,\n\t\tpreviousModel,\n\t\tsource,\n\t});\n}\n\n\nexport async function _emitModelSelect(this: AgentSession, \n\tnextModel: Model<Api>,\n\tpreviousModel: Model<Api> | undefined,\n\tsource: \"set\" | \"cycle\" | \"restore\" | \"fallback\",\n): Promise<void> {\n\tif (modelsAreEqual(previousModel, nextModel)) return;\n\tawait this._extensionRunner.emit({\n\t\ttype: \"model_select\",\n\t\tmodel: nextModel,\n\t\tpreviousModel,\n\t\tsource,\n\t});\n}\n\n/**\n * Set model directly.\n * Validates that auth is configured, saves to session and settings.\n * @throws Error if no auth is configured for the model\n */\n\nexport async function setModel(this: AgentSession, model: Model<Api>): Promise<void> {\n\tif (!this._modelRegistry.hasConfiguredAuth(model)) {\n\t\tthrow new Error(`No API key for ${model.provider}/${model.id}`);\n\t}\n\n\tconst previousModel = this.model;\n\tconst thinkingLevel = this._getThinkingLevelForModelSwitch();\n\tconst nextModel = model;\n\tthis.agent.state.model = nextModel;\n\tthis.sessionManager.appendModelChange(nextModel.provider, nextModel.id);\n\tthis.settingsManager.setDefaultModelAndProvider(nextModel.provider, nextModel.id);\n\n\t// Re-clamp thinking level for new model's capabilities\n\tthis.setThinkingLevel(thinkingLevel);\n\tthis._refreshBaseSystemPromptFromActiveTools();\n\n\tthis._emitModelChanged(nextModel, previousModel, \"set\");\n\tawait this._emitModelSelect(nextModel, previousModel, \"set\");\n}\n\n/**\n * Cycle to next/previous model.\n * Uses scoped models (from --models flag) if available, otherwise all available models.\n * @param direction - \"forward\" (default) or \"backward\"\n * @returns The new model info, or undefined if only one model available\n */\n\nexport async function cycleModel(this: AgentSession, direction: \"forward\" | \"backward\" = \"forward\"): Promise<ModelCycleResult | undefined> {\n\tif (this._scopedModels.length > 0) {\n\t\treturn this._cycleScopedModel(direction);\n\t}\n\treturn this._cycleAvailableModel(direction);\n}\n\n\nexport async function _cycleScopedModel(this: AgentSession, direction: \"forward\" | \"backward\"): Promise<ModelCycleResult | undefined> {\n\tconst scopedModels = this._scopedModels.filter((scoped) => this._modelRegistry.hasConfiguredAuth(scoped.model));\n\tif (scopedModels.length <= 1) return undefined;\n\n\tconst currentModel = this.model;\n\tlet currentIndex = scopedModels.findIndex((sm) => modelsAreEqual(sm.model, currentModel));\n\n\tif (currentIndex === -1) currentIndex = 0;\n\tconst len = scopedModels.length;\n\tconst nextIndex = direction === \"forward\" ? (currentIndex + 1) % len : (currentIndex - 1 + len) % len;\n\tconst next = scopedModels[nextIndex];\n\tconst thinkingLevel = this._getThinkingLevelForModelSwitch(next.thinkingLevel);\n\tconst nextModel = next.model;\n\n\t// Apply model\n\tthis.agent.state.model = nextModel;\n\tthis.sessionManager.appendModelChange(nextModel.provider, nextModel.id);\n\tthis.settingsManager.setDefaultModelAndProvider(nextModel.provider, nextModel.id);\n\n\t// Apply thinking level.\n\t// - Explicit scoped model thinking level overrides current session level\n\t// - Undefined scoped model thinking level inherits the current session preference\n\t// setThinkingLevel clamps to model capabilities.\n\tthis.setThinkingLevel(thinkingLevel);\n\tthis._refreshBaseSystemPromptFromActiveTools();\n\n\tthis._emitModelChanged(nextModel, currentModel, \"cycle\");\n\tawait this._emitModelSelect(nextModel, currentModel, \"cycle\");\n\n\treturn { model: nextModel, thinkingLevel: this.thinkingLevel, isScoped: true };\n}\n\n\nexport async function _cycleAvailableModel(this: AgentSession, direction: \"forward\" | \"backward\"): Promise<ModelCycleResult | undefined> {\n\tconst availableModels = await this._modelRegistry.getAvailable();\n\tif (availableModels.length <= 1) return undefined;\n\n\tconst currentModel = this.model;\n\tlet currentIndex = availableModels.findIndex((m) => modelsAreEqual(m, currentModel));\n\n\tif (currentIndex === -1) currentIndex = 0;\n\tconst len = availableModels.length;\n\tconst nextIndex = direction === \"forward\" ? (currentIndex + 1) % len : (currentIndex - 1 + len) % len;\n\tconst selectedModel = availableModels[nextIndex];\n\n\tconst thinkingLevel = this._getThinkingLevelForModelSwitch();\n\tthis.agent.state.model = selectedModel;\n\tthis.sessionManager.appendModelChange(selectedModel.provider, selectedModel.id);\n\tthis.settingsManager.setDefaultModelAndProvider(selectedModel.provider, selectedModel.id);\n\n\t// Re-clamp thinking level for new model's capabilities\n\tthis.setThinkingLevel(thinkingLevel);\n\tthis._refreshBaseSystemPromptFromActiveTools();\n\n\tthis._emitModelChanged(selectedModel, currentModel, \"cycle\");\n\tawait this._emitModelSelect(selectedModel, currentModel, \"cycle\");\n\n\treturn { model: selectedModel, thinkingLevel: this.thinkingLevel, isScoped: false };\n}\n\n// =========================================================================\n// Thinking Level Management\n// =========================================================================\n\n/**\n * Set thinking level.\n * Clamps to model capabilities based on available thinking levels.\n * Saves to session and settings only if the level actually changes.\n */\n\nexport function setThinkingLevel(this: AgentSession, level: ThinkingLevel): void {\n\tconst availableLevels = this.getAvailableThinkingLevels();\n\tconst effectiveLevel = availableLevels.includes(level) ? level : this._clampThinkingLevel(level, availableLevels);\n\n\t// Only persist if actually changing\n\tconst previousLevel = this.agent.state.thinkingLevel;\n\tconst isChanging = effectiveLevel !== previousLevel;\n\n\tthis.agent.state.thinkingLevel = effectiveLevel;\n\n\tif (isChanging) {\n\t\tthis.sessionManager.appendThinkingLevelChange(effectiveLevel);\n\t\tthis._refreshBaseSystemPromptFromActiveTools();\n\t\tif (this.supportsThinking() || effectiveLevel !== \"off\") {\n\t\t\tthis.settingsManager.setDefaultThinkingLevel(effectiveLevel);\n\t\t}\n\t\tthis._emit({ type: \"thinking_level_changed\", level: effectiveLevel });\n\t\tvoid this._extensionRunner.emit({\n\t\t\ttype: \"thinking_level_select\",\n\t\t\tlevel: effectiveLevel,\n\t\t\tpreviousLevel,\n\t\t});\n\t}\n}\n\n/**\n * Cycle to next thinking level.\n * @returns New level, or undefined if model doesn't support thinking\n */\n\nexport function cycleThinkingLevel(this: AgentSession): ThinkingLevel | undefined {\n\tif (!this.supportsThinking()) return undefined;\n\n\tconst levels = this.getAvailableThinkingLevels();\n\tconst currentIndex = levels.indexOf(this.thinkingLevel);\n\tconst nextIndex = (currentIndex + 1) % levels.length;\n\tconst nextLevel = levels[nextIndex];\n\n\tthis.setThinkingLevel(nextLevel);\n\treturn nextLevel;\n}\n\n/**\n * Get available thinking levels for current model.\n * The provider will clamp to what the specific model supports internally.\n */\n\nexport function getAvailableThinkingLevels(this: AgentSession): ThinkingLevel[] {\n\tif (!this.model) return THINKING_LEVELS;\n\treturn getSupportedThinkingLevels(this.model) as ThinkingLevel[];\n}\n\n/**\n * Check if current model supports thinking/reasoning.\n */\n\nexport function supportsThinking(this: AgentSession): boolean {\n\treturn !!this.model?.reasoning;\n}\n\n\nexport function _getThinkingLevelForModelSwitch(this: AgentSession, explicitLevel?: ThinkingLevel): ThinkingLevel {\n\tif (explicitLevel !== undefined) {\n\t\treturn explicitLevel;\n\t}\n\tif (!this.supportsThinking()) {\n\t\treturn this.settingsManager.getDefaultThinkingLevel() ?? DEFAULT_THINKING_LEVEL;\n\t}\n\treturn this.thinkingLevel;\n}\n\n\nexport function _clampThinkingLevel(this: AgentSession, level: ThinkingLevel, _availableLevels: ThinkingLevel[]): ThinkingLevel {\n\treturn this.model ? (clampThinkingLevel(this.model, level) as ThinkingLevel) : \"off\";\n}\n\n\n// =========================================================================\n// Queue Mode Management\n// =========================================================================\n\n/**\n * Set steering message mode.\n * Saves to settings.\n */\n\nexport const agentSessionModelsMethods = {\n\t_getRequiredRequestAuth,\n\t_emitModelChanged,\n\t_emitModelSelect,\n\tsetModel,\n\tcycleModel,\n\t_cycleScopedModel,\n\t_cycleAvailableModel,\n\tsetThinkingLevel,\n\tcycleThinkingLevel,\n\tgetAvailableThinkingLevels,\n\tsupportsThinking,\n\t_getThinkingLevelForModelSwitch,\n\t_clampThinkingLevel,\n};\n"]}

package/dist/core/agent-session-persistent-custom-messages.d.ts

@@ -0,0 +1,47 @@
+import type { DrainedAgentQueues } from "./agent-session-types.ts";
+import type { AgentSessionInternalSurface as AgentSession } from "./agent-session-methods.ts";
+import type { CustomMessage } from "./messages.ts";
+type ProtectedDelivery = "steer" | "followUp";
+type ProtectedPhase = "queued" | "consumed-unpersisted" | "persistence-failed";
+export declare const PROTECTED_RECONCILIATION_CUSTOM_TYPE = "atomic:protected-streaming-reconciliation";
+export interface ProtectedStreamingCustomMessage {
+    message: CustomMessage;
+    delivery: ProtectedDelivery;
+    phase: ProtectedPhase;
+}
+/** Requeue each durable card intent that has no persisted hidden completion. */
+export declare function recoverProtectedStreamingCustomMessages(session: AgentSession): number;
+/** Persist one visible card and register its hidden model-facing turn. */
+export declare function admitProtectedStreamingCustomMessage(session: AgentSession, message: CustomMessage, delivery: ProtectedDelivery): Promise<CustomMessage>;
+/** Queue one hidden turn before public listeners can transfer its ownership. */
+export declare function queueProtectedStreamingCustomMessage(session: AgentSession, message: CustomMessage, delivery: ProtectedDelivery): Promise<void>;
+/** Queue a whole hidden batch before public listeners can transfer ownership. */
+export declare function queueProtectedStreamingCustomMessages(session: AgentSession, messages: CustomMessage[], delivery: ProtectedDelivery): Promise<void>;
+/** Start or queue an idle protected batch before exposing its display cards. */
+export declare function triggerProtectedStreamingCustomMessages(session: AgentSession, messages: CustomMessage[], delivery: ProtectedDelivery): Promise<void>;
+/** Record that agent-core drained and consumed the hidden reconciliation. */
+export declare function markProtectedStreamingCustomMessageConsumed(session: AgentSession, message: CustomMessage): boolean;
+export declare function markProtectedStreamingCustomMessagePersistenceFailed(session: AgentSession, message: CustomMessage): void;
+export declare function isProtectedStreamingCustomMessage(session: AgentSession, message: CustomMessage): boolean;
+/**
+ * Persist a consumed hidden reconciliation exactly once. Its visible card was
+ * already committed at admission, so a transient failure never re-adds a card
+ * or re-injects another provider message.
+ */
+export declare function persistProtectedStreamingCustomMessage(session: AgentSession, message: CustomMessage): boolean;
+/** Retry only hidden reconciliation persistence; never queue another alias. */
+export declare function retryConsumedProtectedStreamingCustomMessages(session: AgentSession): void;
+/** Flush consumed reconciliations before session state can be discarded. */
+export declare function flushConsumedProtectedStreamingCustomMessages(session: AgentSession): void;
+/**
+ * Persist only reconciliation input already sealed inside an accepted pause
+ * hold. Other queued input can still race an active protocol turn and must keep
+ * the historical fail-closed disposal behavior.
+ */
+export declare function prepareProtectedStreamingCustomMessagesForDisposal(session: AgentSession): void;
+/** Restore only protected references actually removed from native queues/hold. */
+export declare function restoreProtectedStreamingCustomMessages(session: AgentSession, removed: DrainedAgentQueues): void;
+/** Move protection only for message references that transfer with native queues. */
+export declare function transferProtectedStreamingCustomMessages(source: AgentSession, target: AgentSession, transferred: DrainedAgentQueues): void;
+export {};
+//# sourceMappingURL=agent-session-persistent-custom-messages.d.ts.map

package/dist/core/agent-session-persistent-custom-messages.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"agent-session-persistent-custom-messages.d.ts","sourceRoot":"","sources":["../../src/core/agent-session-persistent-custom-messages.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,kBAAkB,EAAE,MAAM,0BAA0B,CAAC;AACnE,OAAO,KAAK,EAAE,2BAA2B,IAAI,YAAY,EAAE,MAAM,4BAA4B,CAAC;AAE9F,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,eAAe,CAAC;AAGnD,KAAK,iBAAiB,GAAG,OAAO,GAAG,UAAU,CAAC;AAC9C,KAAK,cAAc,GAAG,QAAQ,GAAG,sBAAsB,GAAG,oBAAoB,CAAC;AAE/E,eAAO,MAAM,oCAAoC,8CAA8C,CAAC;AAMhG,MAAM,WAAW,+BAA+B;IAC/C,OAAO,EAAE,aAAa,CAAC;IACvB,QAAQ,EAAE,iBAAiB,CAAC;IAC5B,KAAK,EAAE,cAAc,CAAC;CACtB;AA6DD,gFAAgF;AAChF,wBAAgB,uCAAuC,CAAC,OAAO,EAAE,YAAY,GAAG,MAAM,CA+BrF;AAsCD,0EAA0E;AAC1E,wBAAsB,oCAAoC,CACzD,OAAO,EAAE,YAAY,EACrB,OAAO,EAAE,aAAa,EACtB,QAAQ,EAAE,iBAAiB,GACzB,OAAO,CAAC,aAAa,CAAC,CAIxB;AAED,gFAAgF;AAChF,wBAAsB,oCAAoC,CACzD,OAAO,EAAE,YAAY,EACrB,OAAO,EAAE,aAAa,EACtB,QAAQ,EAAE,iBAAiB,GACzB,OAAO,CAAC,IAAI,CAAC,CAIf;AAED,iFAAiF;AACjF,wBAAsB,qCAAqC,CAC1D,OAAO,EAAE,YAAY,EACrB,QAAQ,EAAE,aAAa,EAAE,EACzB,QAAQ,EAAE,iBAAiB,GACzB,OAAO,CAAC,IAAI,CAAC,CAMf;AAED,gFAAgF;AAChF,wBAAsB,uCAAuC,CAC5D,OAAO,EAAE,YAAY,EACrB,QAAQ,EAAE,aAAa,EAAE,EACzB,QAAQ,EAAE,iBAAiB,GACzB,OAAO,CAAC,IAAI,CAAC,CAYf;AAED,6EAA6E;AAC7E,wBAAgB,2CAA2C,CAC1D,OAAO,EAAE,YAAY,EACrB,OAAO,EAAE,aAAa,GACpB,OAAO,CAKT;AAED,wBAAgB,oDAAoD,CACnE,OAAO,EAAE,YAAY,EACrB,OAAO,EAAE,aAAa,GACpB,IAAI,CAGN;AAED,wBAAgB,iCAAiC,CAAC,OAAO,EAAE,YAAY,EAAE,OAAO,EAAE,aAAa,GAAG,OAAO,CAExG;AAED;;;;GAIG;AACH,wBAAgB,sCAAsC,CACrD,OAAO,EAAE,YAAY,EACrB,OAAO,EAAE,aAAa,GACpB,OAAO,CAeT;AAED,+EAA+E;AAC/E,wBAAgB,6CAA6C,CAAC,OAAO,EAAE,YAAY,GAAG,IAAI,CAUzF;AAED,4EAA4E;AAC5E,wBAAgB,6CAA6C,CAAC,OAAO,EAAE,YAAY,GAAG,IAAI,CAOzF;AA6BD;;;;GAIG;AACH,wBAAgB,kDAAkD,CAAC,OAAO,EAAE,YAAY,GAAG,IAAI,CAmB9F;AAED,kFAAkF;AAClF,wBAAgB,uCAAuC,CACtD,OAAO,EAAE,YAAY,EACrB,OAAO,EAAE,kBAAkB,GACzB,IAAI,CAON;AAED,oFAAoF;AACpF,wBAAgB,wCAAwC,CACvD,MAAM,EAAE,YAAY,EACpB,MAAM,EAAE,YAAY,EACpB,WAAW,EAAE,kBAAkB,GAC7B,IAAI,CAUN"}

package/dist/core/agent-session-persistent-custom-messages.js

@@ -0,0 +1,267 @@
+import { customMessageExcludesContext } from "./agent-session-types.js";
+import { resolveWorkflowStageDeliveryTarget } from "./agent-session-delivery-forwarding.js";
+export const PROTECTED_RECONCILIATION_CUSTOM_TYPE = "atomic:protected-streaming-reconciliation";
+function protectedMessages(session) {
+    return session._protectedStreamingCustomMessages ??= [];
+}
+function appendDurableDisplayCard(session, message, delivery) {
+    const card = { ...message, excludeFromContext: true };
+    // The card and recovery marker share one append, so either both survive a
+    // process exit or neither does.
+    const intentEntryId = session.sessionManager.appendCustomMessageEntry(card.customType, card.content, card.display, card.details, true, { delivery });
+    session.agent.state.messages.push(card);
+    return { card, intentEntryId };
+}
+function emitDurableDisplayCard(session, card) {
+    // The card is already live and file-backed, and its protected reconciliation
+    // is registered. Public listeners remain fallible, but their errors cannot
+    // revoke admission or make the producer retry this committed occurrence.
+    try {
+        session._emit({ type: "message_start", message: card });
+    }
+    catch { }
+    try {
+        session._emit({ type: "message_end", message: card });
+    }
+    catch { }
+}
+function createProtectedReconciliation(message, intentEntryId) {
+    return {
+        role: "custom",
+        customType: PROTECTED_RECONCILIATION_CUSTOM_TYPE,
+        content: message.content,
+        display: false,
+        details: { protectedReconciliationOf: intentEntryId },
+        timestamp: message.timestamp,
+    };
+}
+function completedReconciliationIntent(details) {
+    if (details === null || typeof details !== "object")
+        return undefined;
+    const intent = details.protectedReconciliationOf;
+    return typeof intent === "string" ? intent : undefined;
+}
+function protectedDelivery(marker) {
+    if (marker === null || typeof marker !== "object")
+        return undefined;
+    const delivery = marker.delivery;
+    return delivery === "steer" || delivery === "followUp" ? delivery : undefined;
+}
+/** Requeue each durable card intent that has no persisted hidden completion. */
+export function recoverProtectedStreamingCustomMessages(session) {
+    const branch = session.sessionManager.getBranch();
+    const completed = new Set(branch.flatMap((entry) => entry.type === "custom_message" && entry.customType === PROTECTED_RECONCILIATION_CUSTOM_TYPE
+        ? [completedReconciliationIntent(entry.details)].filter((intent) => intent !== undefined)
+        : []));
+    const scheduled = new Set(protectedMessages(session).flatMap((entry) => {
+        const intent = completedReconciliationIntent(entry.message.details);
+        return intent === undefined ? [] : [intent];
+    }));
+    let recovered = 0;
+    for (const entry of branch) {
+        if (entry.type !== "custom_message")
+            continue;
+        const delivery = protectedDelivery(entry.protectedReconciliation);
+        if (delivery === undefined || completed.has(entry.id) || scheduled.has(entry.id))
+            continue;
+        const parsedTimestamp = Date.parse(entry.timestamp);
+        const card = {
+            role: "custom",
+            customType: entry.customType,
+            content: entry.content,
+            display: entry.display,
+            details: entry.details,
+            timestamp: Number.isFinite(parsedTimestamp) ? parsedTimestamp : Date.now(),
+        };
+        const reconciliation = createProtectedReconciliation(card, entry.id);
+        protectedMessages(session).push({ message: reconciliation, delivery, phase: "queued" });
+        session._queueAgentMessage(reconciliation, delivery);
+        recovered += 1;
+    }
+    return recovered;
+}
+async function prepareProtectedAdmission(session, messages, delivery) {
+    // While the initiating workflow tool is still executing, its assistant call
+    // may be ahead of SessionManager's async event writer. Wait only in that
+    // protocol-sensitive phase; at ordinary turn boundaries the hidden steer must
+    // be queued synchronously so agent-core's next native poll observes it.
+    if (session.agent.state.pendingToolCalls.size > 0)
+        await session._agentEventQueue;
+    const owner = resolveWorkflowStageDeliveryTarget(session);
+    const prepared = messages.map((message) => {
+        const { card, intentEntryId } = appendDurableDisplayCard(owner, message, delivery);
+        return { card, reconciliation: createProtectedReconciliation(message, intentEntryId) };
+    });
+    const cards = prepared.map((entry) => entry.card);
+    const reconciliations = prepared.map((entry) => entry.reconciliation);
+    protectedMessages(owner).push(...reconciliations.map((message) => ({ message, delivery, phase: "queued" })));
+    return { owner, cards, reconciliations };
+}
+function emitProtectedAdmission(admission) {
+    let owner = admission.owner;
+    for (const card of admission.cards) {
+        owner = resolveWorkflowStageDeliveryTarget(owner);
+        emitDurableDisplayCard(owner, card);
+    }
+    return resolveWorkflowStageDeliveryTarget(owner);
+}
+/** Persist one visible card and register its hidden model-facing turn. */
+export async function admitProtectedStreamingCustomMessage(session, message, delivery) {
+    const admission = await prepareProtectedAdmission(session, [message], delivery);
+    emitProtectedAdmission(admission);
+    return admission.reconciliations[0];
+}
+/** Queue one hidden turn before public listeners can transfer its ownership. */
+export async function queueProtectedStreamingCustomMessage(session, message, delivery) {
+    const admission = await prepareProtectedAdmission(session, [message], delivery);
+    admission.owner._queueAgentMessage(admission.reconciliations[0], delivery);
+    emitProtectedAdmission(admission);
+}
+/** Queue a whole hidden batch before public listeners can transfer ownership. */
+export async function queueProtectedStreamingCustomMessages(session, messages, delivery) {
+    const admission = await prepareProtectedAdmission(session, messages, delivery);
+    for (const reconciliation of admission.reconciliations) {
+        admission.owner._queueAgentMessage(reconciliation, delivery);
+    }
+    emitProtectedAdmission(admission);
+}
+/** Start or queue an idle protected batch before exposing its display cards. */
+export async function triggerProtectedStreamingCustomMessages(session, messages, delivery) {
+    const admission = await prepareProtectedAdmission(session, messages, delivery);
+    for (const reconciliation of admission.reconciliations) {
+        admission.owner._queueAgentMessage(reconciliation, delivery);
+    }
+    const owner = emitProtectedAdmission(admission);
+    if (owner.isStreaming || owner._queuedMessagesPaused)
+        return;
+    for (const reconciliation of admission.reconciliations) {
+        removeQueuedProtectedReference(owner, reconciliation);
+    }
+    const turn = owner._runAgentPrompt(admission.reconciliations);
+    void turn.catch(() => { });
+}
+/** Record that agent-core drained and consumed the hidden reconciliation. */
+export function markProtectedStreamingCustomMessageConsumed(session, message) {
+    const entry = protectedMessages(session).find((candidate) => candidate.message === message);
+    if (!entry)
+        return false;
+    entry.phase = "consumed-unpersisted";
+    return true;
+}
+export function markProtectedStreamingCustomMessagePersistenceFailed(session, message) {
+    const entry = protectedMessages(session).find((candidate) => candidate.message === message);
+    if (entry?.phase === "consumed-unpersisted")
+        entry.phase = "persistence-failed";
+}
+export function isProtectedStreamingCustomMessage(session, message) {
+    return protectedMessages(session).some((entry) => entry.message === message);
+}
+/**
+ * Persist a consumed hidden reconciliation exactly once. Its visible card was
+ * already committed at admission, so a transient failure never re-adds a card
+ * or re-injects another provider message.
+ */
+export function persistProtectedStreamingCustomMessage(session, message) {
+    const pending = protectedMessages(session);
+    const index = pending.findIndex((entry) => entry.message === message);
+    if (index === -1)
+        return false;
+    const entry = pending[index];
+    if (entry.phase !== "consumed-unpersisted" && entry.phase !== "persistence-failed")
+        return true;
+    session.sessionManager.appendCustomMessageEntry(message.customType, message.content, message.display, message.details, customMessageExcludesContext(message));
+    pending.splice(index, 1);
+    return true;
+}
+/** Retry only hidden reconciliation persistence; never queue another alias. */
+export function retryConsumedProtectedStreamingCustomMessages(session) {
+    for (const entry of [...protectedMessages(session)]) {
+        if (entry.phase !== "persistence-failed")
+            continue;
+        try {
+            persistProtectedStreamingCustomMessage(session, entry.message);
+        }
+        catch {
+            // The visible lifecycle card is already durable. Keep this consumed phase
+            // for a later event rather than duplicating either the card or model turn.
+        }
+    }
+}
+/** Flush consumed reconciliations before session state can be discarded. */
+export function flushConsumedProtectedStreamingCustomMessages(session) {
+    for (const entry of [...protectedMessages(session)]) {
+        if (entry.phase === "queued")
+            continue;
+        // Do not swallow the final write failure: callers must keep this session
+        // alive rather than discard the only remaining recovery state.
+        persistProtectedStreamingCustomMessage(session, entry.message);
+    }
+}
+function removeReference(items, message) {
+    let index = items.indexOf(message);
+    while (index >= 0) {
+        items.splice(index, 1);
+        index = items.indexOf(message);
+    }
+}
+function removeQueuedProtectedReference(session, message) {
+    const hold = session._activeInterruptQueueHold;
+    if (hold !== undefined) {
+        removeReference(hold.steering, message);
+        removeReference(hold.followUp, message);
+    }
+    const queues = session._drainQueuedAgentMessages();
+    session._restoreQueuedAgentMessages({
+        steering: queues.steering.filter((candidate) => candidate !== message),
+        followUp: queues.followUp.filter((candidate) => candidate !== message),
+    });
+}
+function isPausedHeldProtectedReference(session, message) {
+    const hold = session._activeInterruptQueueHold;
+    return session._queuedMessagesPaused && hold !== undefined &&
+        (hold.steering.includes(message) || hold.followUp.includes(message));
+}
+/**
+ * Persist only reconciliation input already sealed inside an accepted pause
+ * hold. Other queued input can still race an active protocol turn and must keep
+ * the historical fail-closed disposal behavior.
+ */
+export function prepareProtectedStreamingCustomMessagesForDisposal(session) {
+    const pending = protectedMessages(session);
+    const queued = pending.filter((entry) => entry.phase === "queued");
+    if (queued.some((entry) => !isPausedHeldProtectedReference(session, entry.message))) {
+        throw new Error("Cannot dispose a session with a queued protected reconciliation");
+    }
+    for (const entry of queued) {
+        session.sessionManager.appendCustomMessageEntry(entry.message.customType, entry.message.content, entry.message.display, entry.message.details, customMessageExcludesContext(entry.message));
+        removeQueuedProtectedReference(session, entry.message);
+        const index = pending.indexOf(entry);
+        if (index >= 0)
+            pending.splice(index, 1);
+    }
+    flushConsumedProtectedStreamingCustomMessages(session);
+}
+/** Restore only protected references actually removed from native queues/hold. */
+export function restoreProtectedStreamingCustomMessages(session, removed) {
+    const removedReferences = new Set([...removed.steering, ...removed.followUp]);
+    for (const entry of protectedMessages(session)) {
+        if (entry.phase === "queued" && removedReferences.has(entry.message)) {
+            session._queueAgentMessage(entry.message, entry.delivery);
+        }
+    }
+}
+/** Move protection only for message references that transfer with native queues. */
+export function transferProtectedStreamingCustomMessages(source, target, transferred) {
+    const transferredReferences = new Set([...transferred.steering, ...transferred.followUp]);
+    const retained = [];
+    const moved = [];
+    for (const entry of protectedMessages(source)) {
+        if (entry.phase === "queued" && transferredReferences.has(entry.message))
+            moved.push(entry);
+        else
+            retained.push(entry);
+    }
+    source._protectedStreamingCustomMessages = retained;
+    protectedMessages(target).unshift(...moved);
+}
+//# sourceMappingURL=agent-session-persistent-custom-messages.js.map