@wrongstack/core 0.284.1 → 0.286.0

package/dist/infrastructure/context-manager.d.ts

@@ -0,0 +1,92 @@
+import type { Compactor } from '../types/compactor.js';
+import type { Message } from '../types/messages.js';
+import type { Tool } from '../types/tool.js';
+/**
+ * Context introspection and management tool.
+ * Allows the model to:
+ *   - "check"  → see token budget and message counts
+ *   - "summary" → summarize and replace a range of messages
+ *   - "prune"  → remove specific message indices
+ *   - "add_note" → inject a summary note at a specific point
+ *   - "compact" → run compaction via the injected compactor
+ */
+export declare const CONTEXT_MANAGER_TOOL_NAME = "context_manager";
+export type ContextManagerAction = 'check' | 'summary' | 'prune' | 'add_note' | 'compact' | 'repair';
+export interface ContextManagerInput {
+    action: ContextManagerAction;
+    /** 0-based message indices for prune/summary (inclusive). */
+    from?: number | undefined;
+    to?: number | undefined;
+    /** Text for add_note / summary actions. For summary, this is the LLM-provided summary text. */
+    text?: string | undefined;
+    /** Inject after which index (for add_note). Defaults to prepend (0). */
+    afterIndex?: number | undefined;
+    /**
+     * System prompt blocks for accurate total token estimation in check action.
+     * When provided, check returns the full API request estimate
+     * (messages + system + tools) instead of just message tokens.
+     */
+    systemPrompt?: unknown | undefined;
+    /**
+     * Registered tools for accurate total token estimation in check action.
+     * Each tool's name + description + inputSchema is counted.
+     */
+    tools?: {
+        name: string;
+        description?: string | undefined;
+        inputSchema: unknown;
+    }[];
+}
+export interface ContextManagerResult {
+    action: ContextManagerAction;
+    beforeTokens: number;
+    afterTokens?: number | undefined;
+    removedCount?: number | undefined;
+    messageCount: number;
+    summary?: string | undefined;
+    notes?: string | undefined;
+    repaired?: {
+        removedToolUses: string[];
+        removedToolResults: string[];
+        removedMessages: number;
+    } | undefined;
+}
+/**
+ * Options for creating a context manager tool.
+ * `compactor` is required for the "compact" action; without it the action returns an error.
+ */
+export interface ContextManagerToolOptions {
+    compactor?: Compactor | undefined;
+    /**
+     * Optional sub-LLM summarizer. When provided, the "summary" action calls this
+     * to produce real summaries of message ranges instead of placeholder text.
+     * (signature matches Provider.complete — return the summary text in result.content[0].text)
+     */
+    summarizer?: (((messages: Message[]) => Promise<string>)) | undefined;
+    /**
+     * Minimum full-request token count before the compact action is allowed to run.
+     * Prevents unnecessary compaction calls when context is small.
+     * Default: 0 (always allow). Set to ~5000 for meaningful compaction targets.
+     */
+    minCompactThreshold?: number | undefined;
+    /**
+     * Minimum token growth required before retrying after a NOOP compaction.
+     * A NOOP is when compaction saved nothing (preserveK protects everything,
+     * no oversized tool_results). Default: 2000.
+     */
+    noopRetryDeltaTokens?: number | undefined;
+    /**
+     * Provider's max context window in tokens. Used to compute a relative
+     * threshold when `minCompactThreshold` is not set. Default: 128_000.
+     */
+    maxContext?: number | undefined;
+    /**
+     * Fraction of maxContext that triggers compaction. Only used when
+     * `minCompactThreshold` is not set. Default: 0.5 (50% of maxContext).
+     */
+    compactThresholdFraction?: number | undefined;
+}
+export declare function createContextManagerTool(opts?: ContextManagerToolOptions): Tool<ContextManagerInput, ContextManagerResult>;
+/** Pre-built instance with no compactor — compact action will return an error. */
+export declare const contextManagerTool: Tool<ContextManagerInput, ContextManagerResult>;
+//# sourceMappingURL=context-manager.d.ts.map

package/dist/infrastructure/context-manager.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"context-manager.d.ts","sourceRoot":"","sources":["../../src/infrastructure/context-manager.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,uBAAuB,CAAC;AACvD,OAAO,KAAK,EAAE,OAAO,EAAE,MAAM,sBAAsB,CAAC;AACpD,OAAO,KAAK,EAAE,IAAI,EAAE,MAAM,kBAAkB,CAAC;AAI7C;;;;;;;;GAQG;AACH,eAAO,MAAM,yBAAyB,oBAAoB,CAAC;AAE3D,MAAM,MAAM,oBAAoB,GAC5B,OAAO,GACP,SAAS,GACT,OAAO,GACP,UAAU,GACV,SAAS,GACT,QAAQ,CAAC;AAEb,MAAM,WAAW,mBAAmB;IAClC,MAAM,EAAE,oBAAoB,CAAC;IAC7B,6DAA6D;IAC7D,IAAI,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;IAC1B,EAAE,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;IACxB,+FAA+F;IAC/F,IAAI,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;IAC1B,wEAAwE;IACxE,UAAU,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;IAChC;;;;OAIG;IACH,YAAY,CAAC,EAAE,OAAO,GAAG,SAAS,CAAC;IACnC;;;OAGG;IACH,KAAK,CAAC,EAAE;QAAE,IAAI,EAAE,MAAM,CAAC;QAAC,WAAW,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;QAAC,WAAW,EAAE,OAAO,CAAA;KAAE,EAAE,CAAC;CACpF;AAED,MAAM,WAAW,oBAAoB;IACnC,MAAM,EAAE,oBAAoB,CAAC;IAC7B,YAAY,EAAE,MAAM,CAAC;IACrB,WAAW,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;IACjC,YAAY,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;IAClC,YAAY,EAAE,MAAM,CAAC;IACrB,OAAO,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;IAC7B,KAAK,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;IAC3B,QAAQ,CAAC,EAAE;QACT,eAAe,EAAE,MAAM,EAAE,CAAC;QAC1B,kBAAkB,EAAE,MAAM,EAAE,CAAC;QAC7B,eAAe,EAAE,MAAM,CAAC;KACzB,GAAG,SAAS,CAAC;CACf;AAED;;;GAGG;AACH,MAAM,WAAW,yBAAyB;IACxC,SAAS,CAAC,EAAE,SAAS,GAAG,SAAS,CAAC;IAClC;;;;OAIG;IACH,UAAU,CAAC,EAAE,CAAC,CAAC,CAAC,QAAQ,EAAE,OAAO,EAAE,KAAK,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,GAAG,SAAS,CAAC;IACtE;;;;OAIG;IACH,mBAAmB,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;IACzC;;;;OAIG;IACH,oBAAoB,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;IAC1C;;;OAGG;IACH,UAAU,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;IAChC;;;OAGG;IACH,wBAAwB,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;CAC/C;AAQD,wBAAgB,wBAAwB,CACtC,IAAI,GAAE,yBAA8B,GACnC,IAAI,CAAC,mBAAmB,EAAE,oBAAoB,CAAC,CAqUjD;AAED,kFAAkF;AAClF,eAAO,MAAM,kBAAkB,EAAE,IAAI,CAAC,mBAAmB,EAAE,oBAAoB,CACnD,CAAC"}

package/dist/infrastructure/index.d.ts

@@ -1,64 +1,6 @@
-export { C as ContextManagerAction, a as ContextManagerInput, b as ContextManagerResult, c as ContextManagerToolOptions, D as DefaultLogger, d as DefaultLoggerOptions, L as LogFormat, e as allServers, f as awsServer, g as blockServer, h as braveSearchServer, i as context7Server, j as contextManagerTool, k as createContextManagerTool, l as everArtServer, m as filesystemServer, n as githubServer, o as googleMapsServer, p as miniMaxVisionServer, q as playwrightServer, s as sentinelServer, r as slackServer, t as sshManagerServer, z as zaiVisionServer } from '../mcp-servers-DEV4s6Ks.js';
-export { D as DefaultPathResolver } from '../path-resolver-CMS5307d.js';
-import { E as EventBus } from '../events-Bgnh43r9.js';
-import { M as ModelsRegistry, R as ResolvedModel } from '../config-MRqn1V-u.js';
-import { T as TokenCounter, U as Usage, C as CacheStats } from '../tool-BNlnIJvo.js';
-import '../compactor-DQLL4HTo.js';
-import '../logger-B63L5bTg.js';
-import '../path-resolver-CPRj4bFY.js';
-import '../permission-ByDKXEcG.js';
-
-/**
- * Token counter that derives pricing from the ModelsRegistry instead of a
- * hardcoded table. If a model is unknown to the registry (or the registry is
- * unavailable) the counter still tracks token totals but reports zero cost.
- */
-declare class DefaultTokenCounter implements TokenCounter {
-    private input;
-    private output;
-    private cacheRead;
-    private cacheWrite;
-    private costInput;
-    private costOutput;
-    private readonly registry?;
-    private readonly providerId?;
-    private readonly events?;
-    private sessionId?;
-    private priceCache;
-    /** Most recently accounted request's tokens. Used for per-request context pressure. */
-    private lastInput;
-    private lastCacheRead;
-    private lastCacheWrite;
-    constructor(opts?: {
-        registry?: ModelsRegistry | undefined;
-        providerId?: string | undefined;
-        events?: EventBus | undefined;
-        sessionId?: string | (() => string | undefined) | undefined;
-    });
-    setSessionId(sessionId: string | (() => string | undefined) | undefined): void;
-    account(usage: Usage, model?: string): void;
-    /** Synchronous variant for code paths that have already resolved the model. */
-    accountWithModel(usage: Usage, resolved: ResolvedModel): void;
-    total(): Usage;
-    currentRequestTokens(): {
-        input: number;
-        cacheRead: number;
-        cacheWrite: number;
-    };
-    setCurrentRequestTokens(input: number, cacheRead?: number, cacheWrite?: number): void;
-    estimateCost(): {
-        input: number;
-        output: number;
-        total: number;
-        currency: 'USD';
-    };
-    cacheStats(): CacheStats;
-    /** Invalidate cached prices so the next account() call fetches fresh data. */
-    invalidateCache(): void;
-    private emitAccounted;
-    private currentSessionId;
-    reset(): void;
-    private applyPrice;
-}
-
-export { DefaultTokenCounter };
+export { DefaultLogger, type DefaultLoggerOptions, type LogFormat } from './logger.js';
+export { DefaultPathResolver } from './path-resolver.js';
+export { DefaultTokenCounter } from './token-counter.js';
+export { filesystemServer, githubServer, context7Server, braveSearchServer, blockServer, everArtServer, slackServer, awsServer, googleMapsServer, sentinelServer, zaiVisionServer, miniMaxVisionServer, playwrightServer, sshManagerServer, allServers, } from './mcp-servers.js';
+export { contextManagerTool, createContextManagerTool, type ContextManagerInput, type ContextManagerResult, type ContextManagerAction, type ContextManagerToolOptions, } from './context-manager.js';
+//# sourceMappingURL=index.d.ts.map

package/dist/infrastructure/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/infrastructure/index.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,aAAa,EAAE,KAAK,oBAAoB,EAAE,KAAK,SAAS,EAAE,MAAM,aAAa,CAAC;AACvF,OAAO,EAAE,mBAAmB,EAAE,MAAM,oBAAoB,CAAC;AACzD,OAAO,EAAE,mBAAmB,EAAE,MAAM,oBAAoB,CAAC;AACzD,OAAO,EACL,gBAAgB,EAChB,YAAY,EACZ,cAAc,EACd,iBAAiB,EACjB,WAAW,EACX,aAAa,EACb,WAAW,EACX,SAAS,EACT,gBAAgB,EAChB,cAAc,EACd,eAAe,EACf,mBAAmB,EACnB,gBAAgB,EAChB,gBAAgB,EAChB,UAAU,GACX,MAAM,kBAAkB,CAAC;AAC1B,OAAO,EACL,kBAAkB,EAClB,wBAAwB,EACxB,KAAK,mBAAmB,EACxB,KAAK,oBAAoB,EACzB,KAAK,oBAAoB,EACzB,KAAK,yBAAyB,GAC/B,MAAM,sBAAsB,CAAC"}

package/dist/infrastructure/index.js

@@ -1,21 +1,24 @@
-import * as fsp from 'fs/promises';
-import * as path2 from 'path';
-import * as fs from 'fs';
-import * as os from 'os';
-
 // src/infrastructure/logger.ts
+import * as fsp from "node:fs/promises";
+import * as path from "node:path";
 
 // src/utils/term.ts
 var hasStdout = () => typeof process !== "undefined" && !!process.stdout;
 function isStdoutTTY() {
   return hasStdout() && Boolean(process.stdout.isTTY);
 }
+var activeOutputGuard = null;
 function writeTo(s, stream) {
   if (!stream || typeof stream.write !== "function") return false;
-  {
+  const guard = activeOutputGuard;
+  if (!guard) {
     stream.write(s);
     return true;
   }
+  guard.suspend();
+  stream.write(s);
+  guard.resume();
+  return true;
 }
 function writeErr(s, stream = process.stderr) {
   return writeTo(s, stream);
@@ -117,7 +120,7 @@ var DefaultLogger = class _DefaultLogger {
     this.stderr = opts.stderr !== false;
     this.maxFileBytes = opts.maxFileBytes ?? 10 * 1024 * 1024;
     if (this.file) {
-      const dir = path2.dirname(this.file);
+      const dir = path.dirname(this.file);
       this._tail = this._tail.then(async () => {
         await fsp.mkdir(dir, { recursive: true });
       }).catch(() => void 0);
@@ -235,6 +238,11 @@ function formatCtx(ctx) {
     return color.dim(String(ctx));
   }
 }
+
+// src/infrastructure/path-resolver.ts
+import * as fs from "node:fs";
+import * as os from "node:os";
+import * as path2 from "node:path";
 var PROJECT_MARKERS = [
   ".git",
   "package.json",
@@ -642,7 +650,7 @@ var allServers = () => ({
 // src/utils/expect-defined.ts
 function expectDefined(value, label) {
   if (value === null || value === void 0) {
-    const err = new Error("Expected value to be defined");
+    const err = new Error(label ? `Expected ${label} to be defined` : "Expected value to be defined");
     err.name = "ExpectDefinedError";
     throw err;
   }
@@ -1040,7 +1048,7 @@ function createContextManagerTool(opts = {}) {
               notes: "No compactor registered. Use /compact aggressive via slash command instead."
             };
           }
-          const fullEstimate = input.systemPrompt != null && Array.isArray(input.tools) ? estimateRequestTokens(messages, input.systemPrompt, input.tools) : { total: beforeTokens};
+          const fullEstimate = input.systemPrompt != null && Array.isArray(input.tools) ? estimateRequestTokens(messages, input.systemPrompt, input.tools) : { total: beforeTokens, messages: beforeTokens, systemPrompt: 0, tools: 0 };
           const currentTokens = fullEstimate.total;
           const runtimeMaxContext = configuredMaxContext ?? ctx.provider?.capabilities?.maxContext ?? 128e3;
           const runtimeThreshold = minCompactThreshold > 0 ? minCompactThreshold : Math.floor(runtimeMaxContext * compactThresholdFraction);
@@ -1194,7 +1202,26 @@ function createContextManagerTool(opts = {}) {
   };
 }
 var contextManagerTool = createContextManagerTool();
-
-export { DefaultLogger, DefaultPathResolver, DefaultTokenCounter, allServers, awsServer, blockServer, braveSearchServer, context7Server, contextManagerTool, createContextManagerTool, everArtServer, filesystemServer, githubServer, googleMapsServer, miniMaxVisionServer, playwrightServer, sentinelServer, slackServer, sshManagerServer, zaiVisionServer };
+export {
+  DefaultLogger,
+  DefaultPathResolver,
+  DefaultTokenCounter,
+  allServers,
+  awsServer,
+  blockServer,
+  braveSearchServer,
+  context7Server,
+  contextManagerTool,
+  createContextManagerTool,
+  everArtServer,
+  filesystemServer,
+  githubServer,
+  googleMapsServer,
+  miniMaxVisionServer,
+  playwrightServer,
+  sentinelServer,
+  slackServer,
+  sshManagerServer,
+  zaiVisionServer
+};
 //# sourceMappingURL=index.js.map
-//# sourceMappingURL=index.js.map