open-multi-agent-kit 0.90.6 → 0.90.8

package/dist/core/agent-session.js

@@ -25,6 +25,7 @@ import { executeBashWithOperations } from "./bash-executor.js";
 import { classifyShellCommand } from "./command-safety.js";
 import { calculateContextTokens, collectEntriesForBranchSummary, compact, estimateContextTokens, estimateProjectedContextTokens, generateBranchSummary, prepareCompaction, resolveCompactionModel, shouldCompact, } from "./compaction/index.js";
 import { compactionEmitWillRetry } from "./compaction/resume-policy.js";
+import { createMemoryContextBudgetCacheProviderV2 } from "./context-budget-v2-cache-provider.js";
 import { DEFAULT_THINKING_LEVEL } from "./defaults.js";
 import { exportSessionToHtml } from "./export-html/index.js";
 import { createToolHtmlRenderer } from "./export-html/tool-renderer.js";
@@ -34,6 +35,7 @@ import { emitSessionShutdownEvent } from "./extensions/runner.js";
 import { assertTextChatModelForCompletion } from "./grok-harness.js";
 import { grokPlaybookAppendForProvider } from "./grok-playbook.js";
 import { assertLoadoutAccess, decideLoadoutAccess } from "./loadout-access-policy.js";
+import { findExactModelReferenceMatch } from "./model-resolver.js";
 import { expandPromptTemplate } from "./prompt-templates.js";
 import { getBiasStepsForCell, getDefaultRouterBiasSnapshotPath, parseRouterBiasSnapshot, } from "./reasoning-router-bias.js";
 import { classifyTaskV4, resolveThinkingLevelV4WithUncertainty } from "./reasoning-router-v4.js";
@@ -171,6 +173,8 @@ export class AgentSession {
     _extensionErrorUnsubscriber;
     // Model registry for API key resolution
     _modelRegistry;
+    /** Lazy, in-memory only; never shared across sessions or persisted. */
+    _contextBudgetCacheProvider;
     // Tool registry for extension getTools/setTools
     _toolRegistry = new Map();
     _toolDefinitions = new Map();
@@ -744,7 +748,11 @@ export class AgentSession {
         return undefined;
     }
     _getContextBudgetOptions() {
-        if (process.env.OMK_CONTEXT_GOVERNOR !== "1") {
+        const contextGovernorOverride = process.env.OMK_CONTEXT_GOVERNOR;
+        if (contextGovernorOverride === "0") {
+            return undefined;
+        }
+        if (contextGovernorOverride !== "1" && !this.settingsManager.getContextBudgetEnabled()) {
             return undefined;
         }
         const contextWindow = this.model?.contextWindow ?? 0;
@@ -782,6 +790,8 @@ export class AgentSession {
             responseReserveTokens =
                 parsePositiveIntegerEnv("OMK_CONTEXT_GOVERNOR_RESPONSE_RESERVE_TOKENS") ?? LEGACY_RESPONSE_RESERVE_TOKENS;
         }
+        const cacheProvider = this._contextBudgetCacheProvider ?? createMemoryContextBudgetCacheProviderV2("session");
+        this._contextBudgetCacheProvider = cacheProvider;
         // Enforce floor
         if (maxPromptTokens < MIN_PROMPT_TOKENS) {
             maxPromptTokens = MIN_PROMPT_TOKENS;
@@ -797,6 +807,7 @@ export class AgentSession {
             tokenizerMode: parseTokenizerModeEnv(process.env.OMK_CONTEXT_GOVERNOR_TOKENIZER),
             activeSkillNames: parseCommaSeparatedEnv(process.env.OMK_CONTEXT_GOVERNOR_ACTIVE_SKILLS),
             queryContext: this._extractCurrentQuery(),
+            cacheProvider,
         };
     }
     /**
@@ -1649,6 +1660,18 @@ export class AgentSession {
     // =========================================================================
     // Compaction
     // =========================================================================
+    _resolveCompactionModel(sessionModel) {
+        const configuredModel = this.settingsManager.getCompactionModel();
+        const availableModels = this._modelRegistry.getAvailable();
+        if (configuredModel) {
+            const model = findExactModelReferenceMatch(configuredModel, availableModels);
+            if (!model) {
+                throw new Error(`Configured compaction model "${configuredModel}" is unavailable or unauthenticated.`);
+            }
+            return model;
+        }
+        return resolveCompactionModel(sessionModel, availableModels);
+    }
     /**
      * Manually compact the session context.
      * Aborts current agent operation first.
@@ -1663,7 +1686,7 @@ export class AgentSession {
             if (!this.model) {
                 throw new Error(formatNoModelSelectedMessage());
             }
-            const compactionModel = resolveCompactionModel(this.model, this._modelRegistry.getAvailable());
+            const compactionModel = this._resolveCompactionModel(this.model);
             const { apiKey, headers } = await this._getCompactionRequestAuth(compactionModel);
             const pathEntries = this.sessionManager.getBranch();
             const settings = this.settingsManager.getCompactionSettings();
@@ -1897,7 +1920,7 @@ export class AgentSession {
                 });
                 return false;
             }
-            const compactionModel = resolveCompactionModel(this.model, this._modelRegistry.getAvailable());
+            const compactionModel = this._resolveCompactionModel(this.model);
             let apiKey;
             let headers;
             if (this.agent.streamFn === streamSimple) {