@amodalai/runtime 0.1.26 → 0.2.0

package/dist/src/agent/snapshot-server.js

@@ -1,22 +1,24 @@
 /**
  * @license
- * Copyright 2025 Amodal Labs, Inc.
+ * Copyright 2026 Amodal Labs, Inc.
  * SPDX-License-Identifier: MIT
  */
+/**
+ * Snapshot server.
+ *
+ * Creates an Express server from an immutable deploy snapshot. Used for
+ * local testing before deploying to the platform.
+ */
 import express from 'express';
 import { loadSnapshotFromFile, snapshotToBundle } from '@amodalai/core';
-import { SessionManager } from '../session/session-manager.js';
+import { StandaloneSessionManager } from '../session/manager.js';
+import { buildSessionComponents } from '../session/session-builder.js';
 import { LocalToolExecutor } from './tool-executor-local.js';
 import { createChatStreamRouter } from '../routes/chat-stream.js';
 import { createTaskRouter } from './routes/task.js';
 import { errorHandler } from '../middleware/error-handler.js';
-import { log } from '../logger.js';
-/**
- * Creates an Express server that runs from an immutable deploy snapshot.
- *
- * This is the local testing path: the CLI builds a snapshot and tests it
- * locally before deploying to the platform.
- */
+import { ConfigError } from '../errors.js';
+import { log, createLogger } from '../logger.js';
 export async function createSnapshotServer(config) {
     let bundle;
     let deployId;
@@ -34,26 +36,44 @@ export async function createSnapshotServer(config) {
         deployId = snapshot.deployId;
     }
     else {
-        throw new Error('One of snapshotPath, snapshot, or bundle must be provided');
+        throw new ConfigError('One of snapshotPath, snapshot, or bundle must be provided', { key: 'snapshotSource' });
     }
-    // Set up tool executor — use injected executor if provided, otherwise local
     let toolExecutor = config.toolExecutor;
     if (!toolExecutor && bundle.tools.length > 0) {
         toolExecutor = new LocalToolExecutor();
     }
-    const sessionManager = new SessionManager({
-        baseParams: {
-            sessionId: 'snapshot-init',
-            interactive: false,
-            noBrowser: true,
-            cwd: process.cwd(),
-            targetDir: process.cwd(),
-        },
+    const sessionLogger = createLogger({ component: 'snapshot-session' });
+    const sessionManager = new StandaloneSessionManager({
+        logger: sessionLogger,
         ttlMs: config.sessionTtlMs,
-        bundle,
-        toolExecutor,
-        shellExecutor: config.shellExecutor,
     });
+    sessionManager.start();
+    const shared = {
+        storeBackend: null,
+        mcpManager: null,
+        logger: log,
+        toolExecutor,
+    };
+    const createTaskSession = () => {
+        const components = buildSessionComponents({
+            bundle,
+            storeBackend: null,
+            mcpManager: null,
+            logger: log,
+            toolExecutor,
+        });
+        const session = sessionManager.create({
+            tenantId: 'snapshot',
+            userId: 'snapshot',
+            provider: components.provider,
+            toolRegistry: components.toolRegistry,
+            permissionChecker: components.permissionChecker,
+            systemPrompt: components.systemPrompt,
+            userRoles: components.userRoles,
+            toolContextFactory: components.toolContextFactory,
+        });
+        return { session, toolContextFactory: components.toolContextFactory };
+    };
     const app = express();
     // CORS
     const corsOrigin = config.corsOrigin ?? '*';
@@ -83,8 +103,12 @@ export async function createSnapshotServer(config) {
         });
     });
     // Routes
-    app.use(createChatStreamRouter({ sessionManager }));
-    app.use(createTaskRouter({ sessionManager }));
+    app.use(createChatStreamRouter({
+        sessionManager,
+        bundleResolver: { staticBundle: bundle },
+        shared,
+    }));
+    app.use(createTaskRouter({ sessionManager, createTaskSession }));
     // Error handler (must be last)
     app.use(errorHandler);
     let server = null;
@@ -95,8 +119,7 @@ export async function createSnapshotServer(config) {
         async start() {
             return new Promise((resolve) => {
                 const httpServer = app.listen(port, host, () => {
-                    log.info(`Snapshot server listening on ${host}:${port}`);
-                    log.info(`Deploy: ${deployId}, Agent: ${bundle.config.name}`);
+                    log.info('snapshot_server_started', { host, port, deployId, agent: bundle.config.name });
                     resolve(httpServer);
                 });
                 server = httpServer;
@@ -116,7 +139,7 @@ export async function createSnapshotServer(config) {
                 server = null;
             }
             await sessionManager.shutdown();
-            log.info('Snapshot server stopped');
+            log.info('snapshot_server_stopped', {});
         },
     };
 }

package/dist/src/agent/snapshot-server.js.map

@@ -1 +1 @@
-{"version":3,"file":"snapshot-server.js","sourceRoot":"","sources":["../../../src/agent/snapshot-server.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,OAAO,MAAM,SAAS,CAAC;AAE9B,OAAO,EAAC,oBAAoB,EAAE,gBAAgB,EAAC,MAAM,gBAAgB,CAAC;AAEtE,OAAO,EAAC,cAAc,EAAC,MAAM,+BAA+B,CAAC;AAC7D,OAAO,EAAC,iBAAiB,EAAC,MAAM,0BAA0B,CAAC;AAC3D,OAAO,EAAC,sBAAsB,EAAC,MAAM,0BAA0B,CAAC;AAChE,OAAO,EAAC,gBAAgB,EAAC,MAAM,kBAAkB,CAAC;AAClD,OAAO,EAAC,YAAY,EAAC,MAAM,gCAAgC,CAAC;AAE5D,OAAO,EAAC,GAAG,EAAC,MAAM,cAAc,CAAC;AA2BjC;;;;;GAKG;AACH,MAAM,CAAC,KAAK,UAAU,oBAAoB,CAAC,MAA4B;IACrE,IAAI,MAAmB,CAAC;IACxB,IAAI,QAAgB,CAAC;IAErB,IAAI,MAAM,CAAC,MAAM,EAAE,CAAC;QAClB,MAAM,GAAG,MAAM,CAAC,MAAM,CAAC;QACvB,QAAQ,GAAG,MAAM,CAAC,MAAM,CAAC;IAC3B,CAAC;SAAM,IAAI,MAAM,CAAC,QAAQ,EAAE,CAAC;QAC3B,MAAM,GAAG,gBAAgB,CAAC,MAAM,CAAC,QAAQ,EAAE,YAAY,MAAM,CAAC,QAAQ,CAAC,QAAQ,EAAE,CAAC,CAAC;QACnF,QAAQ,GAAG,MAAM,CAAC,QAAQ,CAAC,QAAQ,CAAC;IACtC,CAAC;SAAM,IAAI,MAAM,CAAC,YAAY,EAAE,CAAC;QAC/B,MAAM,QAAQ,GAAG,MAAM,oBAAoB,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC;QACjE,MAAM,GAAG,gBAAgB,CAAC,QAAQ,EAAE,MAAM,CAAC,YAAY,CAAC,CAAC;QACzD,QAAQ,GAAG,QAAQ,CAAC,QAAQ,CAAC;IAC/B,CAAC;SAAM,CAAC;QACN,MAAM,IAAI,KAAK,CAAC,2DAA2D,CAAC,CAAC;IAC/E,CAAC;IAED,4EAA4E;IAC5E,IAAI,YAAY,GAAmC,MAAM,CAAC,YAAY,CAAC;IACvE,IAAI,CAAC,YAAY,IAAI,MAAM,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QAC7C,YAAY,GAAG,IAAI,iBAAiB,EAAE,CAAC;IACzC,CAAC;IAED,MAAM,cAAc,GAAG,IAAI,cAAc,CAAC;QACxC,UAAU,EAAE;YACV,SAAS,EAAE,eAAe;YAC1B,WAAW,EAAE,KAAK;YAClB,SAAS,EAAE,IAAI;YACf,GAAG,EAAE,OAAO,CAAC,GAAG,EAAE;YAClB,SAAS,EAAE,OAAO,CAAC,GAAG,EAAE;SACzB;QACD,KAAK,EAAE,MAAM,CAAC,YAAY;QAC1B,MAAM;QACN,YAAY;QACZ,aAAa,EAAE,MAAM,CAAC,aAAa;KACpC,CAAC,CAAC;IAEH,MAAM,GAAG,GAAG,OAAO,EAAE,CAAC;IAEtB,OAAO;IACP,MAAM,UAAU,GAAG,MAAM,CAAC,UAAU,IAAI,GAAG,CAAC;IAC5C,GAAG,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,GAAG,EAAE,IAAI,EAAE,EAAE;QAC1B,GAAG,CAAC,MAAM,CAAC,6BAA6B,EAAE,UAAU,CAAC,CAAC;QACtD,GAAG,CAAC,MAAM,CACR,8BAA8B,EAC9B,+DAA+D,CAChE,CAAC;QACF,GAAG,CAAC,MAAM,CACR,8BAA8B,EAC9B,wCAAwC,CACzC,CAAC;QACF,IAAI,IAAI,CAAC,MAAM,KAAK,SAAS,EAAE,CAAC;YAC9B,GAAG,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC;YACpB,OAAO;QACT,CAAC;QACD,IAAI,EAAE,CAAC;IACT,CAAC,CAAC,CAAC;IAEH,GAAG,CAAC,GAAG,CAAC,OAAO,CAAC,IAAI,EAAE,CAAC,CAAC;IAExB,SAAS;IACT,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;IAC7B,GAAG,CAAC,GAAG,CAAC,SAAS,EAAE,CAAC,IAAI,EAAE,GAAG,EAAE,EAAE;QAC/B,GAAG,CAAC,IAAI,CAAC;YACP,MAAM,EAAE,IAAI;YACZ,IAAI,EAAE,UAAU;YAChB,SAAS,EAAE,QAAQ;YACnB,UAAU,EAAE,MAAM,CAAC,MAAM,CAAC,IAAI;YAC9B,WAAW,EAAE,MAAM,CAAC,WAAW,CAAC,IAAI;YACpC,MAAM,EAAE,MAAM,CAAC,MAAM,CAAC,MAAM;YAC5B,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS;YACjC,eAAe,EAAE,cAAc,CAAC,IAAI;SACrC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,SAAS;IACT,GAAG,CAAC,GAAG,CAAC,sBAAsB,CAAC,EAAC,cAAc,EAAC,CAAC,CAAC,CAAC;IAClD,GAAG,CAAC,GAAG,CAAC,gBAAgB,CAAC,EAAC,cAAc,EAAC,CAAC,CAAC,CAAC;IAE5C,+BAA+B;IAC/B,GAAG,CAAC,GAAG,CAAC,YAAY,CAAC,CAAC;IAEtB,IAAI,MAAM,GAAuB,IAAI,CAAC;IACtC,MAAM,IAAI,GAAG,MAAM,CAAC,IAAI,IAAI,SAAS,CAAC;IACtC,MAAM,IAAI,GAAG,MAAM,CAAC,IAAI,CAAC;IAEzB,OAAO;QACL,GAAG;QAEH,KAAK,CAAC,KAAK;YACT,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE;gBAC7B,MAAM,UAAU,GAAG,GAAG,CAAC,MAAM,CAAC,IAAI,EAAE,IAAI,EAAE,GAAG,EAAE;oBAC7C,GAAG,CAAC,IAAI,CAAC,gCAAgC,IAAI,IAAI,IAAI,EAAE,CAAC,CAAC;oBACzD,GAAG,CAAC,IAAI,CAAC,WAAW,QAAQ,YAAY,MAAM,CAAC,MAAM,CAAC,IAAI,EAAE,CAAC,CAAC;oBAC9D,OAAO,CAAC,UAAU,CAAC,CAAC;gBACtB,CAAC,CAAC,CAAC;gBACH,MAAM,GAAG,UAAU,CAAC;YACtB,CAAC,CAAC,CAAC;QACL,CAAC;QAED,KAAK,CAAC,IAAI;YACR,IAAI,MAAM,EAAE,CAAC;gBACX,MAAM,CAAC,GAAG,MAAM,CAAC;gBACjB,MAAM,IAAI,OAAO,CAAO,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;oBAC1C,CAAC,CAAC,KAAK,CAAC,CAAC,GAAG,EAAE,EAAE;wBACd,IAAI,GAAG;4BAAE,MAAM,CAAC,GAAG,CAAC,CAAC;;4BAChB,OAAO,EAAE,CAAC;oBACjB,CAAC,CAAC,CAAC;gBACL,CAAC,CAAC,CAAC;gBACH,MAAM,GAAG,IAAI,CAAC;YAChB,CAAC;YAED,MAAM,cAAc,CAAC,QAAQ,EAAE,CAAC;YAEhC,GAAG,CAAC,IAAI,CAAC,yBAAyB,CAAC,CAAC;QACtC,CAAC;KACF,CAAC;AACJ,CAAC"}
+{"version":3,"file":"snapshot-server.js","sourceRoot":"","sources":["../../../src/agent/snapshot-server.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH;;;;;GAKG;AAEH,OAAO,OAAO,MAAM,SAAS,CAAC;AAE9B,OAAO,EAAC,oBAAoB,EAAE,gBAAgB,EAAC,MAAM,gBAAgB,CAAC;AAEtE,OAAO,EAAC,wBAAwB,EAAC,MAAM,uBAAuB,CAAC;AAC/D,OAAO,EAAC,sBAAsB,EAAC,MAAM,+BAA+B,CAAC;AACrE,OAAO,EAAC,iBAAiB,EAAC,MAAM,0BAA0B,CAAC;AAC3D,OAAO,EAAC,sBAAsB,EAAC,MAAM,0BAA0B,CAAC;AAChE,OAAO,EAAC,gBAAgB,EAAC,MAAM,kBAAkB,CAAC;AAClD,OAAO,EAAC,YAAY,EAAC,MAAM,gCAAgC,CAAC;AAE5D,OAAO,EAAC,WAAW,EAAC,MAAM,cAAc,CAAC;AACzC,OAAO,EAAC,GAAG,EAAE,YAAY,EAAC,MAAM,cAAc,CAAC;AAa/C,MAAM,CAAC,KAAK,UAAU,oBAAoB,CAAC,MAA4B;IACrE,IAAI,MAAmB,CAAC;IACxB,IAAI,QAAgB,CAAC;IAErB,IAAI,MAAM,CAAC,MAAM,EAAE,CAAC;QAClB,MAAM,GAAG,MAAM,CAAC,MAAM,CAAC;QACvB,QAAQ,GAAG,MAAM,CAAC,MAAM,CAAC;IAC3B,CAAC;SAAM,IAAI,MAAM,CAAC,QAAQ,EAAE,CAAC;QAC3B,MAAM,GAAG,gBAAgB,CAAC,MAAM,CAAC,QAAQ,EAAE,YAAY,MAAM,CAAC,QAAQ,CAAC,QAAQ,EAAE,CAAC,CAAC;QACnF,QAAQ,GAAG,MAAM,CAAC,QAAQ,CAAC,QAAQ,CAAC;IACtC,CAAC;SAAM,IAAI,MAAM,CAAC,YAAY,EAAE,CAAC;QAC/B,MAAM,QAAQ,GAAG,MAAM,oBAAoB,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC;QACjE,MAAM,GAAG,gBAAgB,CAAC,QAAQ,EAAE,MAAM,CAAC,YAAY,CAAC,CAAC;QACzD,QAAQ,GAAG,QAAQ,CAAC,QAAQ,CAAC;IAC/B,CAAC;SAAM,CAAC;QACN,MAAM,IAAI,WAAW,CAAC,2DAA2D,EAAE,EAAC,GAAG,EAAE,gBAAgB,EAAC,CAAC,CAAC;IAC9G,CAAC;IAED,IAAI,YAAY,GAAmC,MAAM,CAAC,YAAY,CAAC;IACvE,IAAI,CAAC,YAAY,IAAI,MAAM,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QAC7C,YAAY,GAAG,IAAI,iBAAiB,EAAE,CAAC;IACzC,CAAC;IAED,MAAM,aAAa,GAAG,YAAY,CAAC,EAAC,SAAS,EAAE,kBAAkB,EAAC,CAAC,CAAC;IACpE,MAAM,cAAc,GAAG,IAAI,wBAAwB,CAAC;QAClD,MAAM,EAAE,aAAa;QACrB,KAAK,EAAE,MAAM,CAAC,YAAY;KAC3B,CAAC,CAAC;IACH,cAAc,CAAC,KAAK,EAAE,CAAC;IAEvB,MAAM,MAAM,GAAG;QACb,YAAY,EAAE,IAAI;QAClB,UAAU,EAAE,IAAI;QAChB,MAAM,EAAE,GAAG;QACX,YAAY;KACb,CAAC;IAEF,MAAM,iBAAiB,GAAG,GAAG,EAAE;QAC7B,MAAM,UAAU,GAAG,sBAAsB,CAAC;YACxC,MAAM;YACN,YAAY,EAAE,IAAI;YAClB,UAAU,EAAE,IAAI;YAChB,MAAM,EAAE,GAAG;YACX,YAAY;SACb,CAAC,CAAC;QACH,MAAM,OAAO,GAAG,cAAc,CAAC,MAAM,CAAC;YACpC,QAAQ,EAAE,UAAU;YACpB,MAAM,EAAE,UAAU;YAClB,QAAQ,EAAE,UAAU,CAAC,QAAQ;YAC7B,YAAY,EAAE,UAAU,CAAC,YAAY;YACrC,iBAAiB,EAAE,UAAU,CAAC,iBAAiB;YAC/C,YAAY,EAAE,UAAU,CAAC,YAAY;YACrC,SAAS,EAAE,UAAU,CAAC,SAAS;YAC/B,kBAAkB,EAAE,UAAU,CAAC,kBAAkB;SAClD,CAAC,CAAC;QACH,OAAO,EAAC,OAAO,EAAE,kBAAkB,EAAE,UAAU,CAAC,kBAAkB,EAAC,CAAC;IACtE,CAAC,CAAC;IAEF,MAAM,GAAG,GAAG,OAAO,EAAE,CAAC;IAEtB,OAAO;IACP,MAAM,UAAU,GAAG,MAAM,CAAC,UAAU,IAAI,GAAG,CAAC;IAC5C,GAAG,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,GAAG,EAAE,IAAI,EAAE,EAAE;QAC1B,GAAG,CAAC,MAAM,CAAC,6BAA6B,EAAE,UAAU,CAAC,CAAC;QACtD,GAAG,CAAC,MAAM,CAAC,8BAA8B,EAAE,+DAA+D,CAAC,CAAC;QAC5G,GAAG,CAAC,MAAM,CAAC,8BAA8B,EAAE,wCAAwC,CAAC,CAAC;QACrF,IAAI,IAAI,CAAC,MAAM,KAAK,SAAS,EAAE,CAAC;YAC9B,GAAG,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC;YACpB,OAAO;QACT,CAAC;QACD,IAAI,EAAE,CAAC;IACT,CAAC,CAAC,CAAC;IAEH,GAAG,CAAC,GAAG,CAAC,OAAO,CAAC,IAAI,EAAE,CAAC,CAAC;IAExB,SAAS;IACT,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;IAC7B,GAAG,CAAC,GAAG,CAAC,SAAS,EAAE,CAAC,IAAI,EAAE,GAAG,EAAE,EAAE;QAC/B,GAAG,CAAC,IAAI,CAAC;YACP,MAAM,EAAE,IAAI;YACZ,IAAI,EAAE,UAAU;YAChB,SAAS,EAAE,QAAQ;YACnB,UAAU,EAAE,MAAM,CAAC,MAAM,CAAC,IAAI;YAC9B,WAAW,EAAE,MAAM,CAAC,WAAW,CAAC,IAAI;YACpC,MAAM,EAAE,MAAM,CAAC,MAAM,CAAC,MAAM;YAC5B,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS;YACjC,eAAe,EAAE,cAAc,CAAC,IAAI;SACrC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,SAAS;IACT,GAAG,CAAC,GAAG,CAAC,sBAAsB,CAAC;QAC7B,cAAc;QACd,cAAc,EAAE,EAAC,YAAY,EAAE,MAAM,EAAC;QACtC,MAAM;KACP,CAAC,CAAC,CAAC;IACJ,GAAG,CAAC,GAAG,CAAC,gBAAgB,CAAC,EAAC,cAAc,EAAE,iBAAiB,EAAC,CAAC,CAAC,CAAC;IAE/D,+BAA+B;IAC/B,GAAG,CAAC,GAAG,CAAC,YAAY,CAAC,CAAC;IAEtB,IAAI,MAAM,GAAuB,IAAI,CAAC;IACtC,MAAM,IAAI,GAAG,MAAM,CAAC,IAAI,IAAI,SAAS,CAAC;IACtC,MAAM,IAAI,GAAG,MAAM,CAAC,IAAI,CAAC;IAEzB,OAAO;QACL,GAAG;QAEH,KAAK,CAAC,KAAK;YACT,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE;gBAC7B,MAAM,UAAU,GAAG,GAAG,CAAC,MAAM,CAAC,IAAI,EAAE,IAAI,EAAE,GAAG,EAAE;oBAC7C,GAAG,CAAC,IAAI,CAAC,yBAAyB,EAAE,EAAC,IAAI,EAAE,IAAI,EAAE,QAAQ,EAAE,KAAK,EAAE,MAAM,CAAC,MAAM,CAAC,IAAI,EAAC,CAAC,CAAC;oBACvF,OAAO,CAAC,UAAU,CAAC,CAAC;gBACtB,CAAC,CAAC,CAAC;gBACH,MAAM,GAAG,UAAU,CAAC;YACtB,CAAC,CAAC,CAAC;QACL,CAAC;QAED,KAAK,CAAC,IAAI;YACR,IAAI,MAAM,EAAE,CAAC;gBACX,MAAM,CAAC,GAAG,MAAM,CAAC;gBACjB,MAAM,IAAI,OAAO,CAAO,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;oBAC1C,CAAC,CAAC,KAAK,CAAC,CAAC,GAAG,EAAE,EAAE;wBACd,IAAI,GAAG;4BAAE,MAAM,CAAC,GAAG,CAAC,CAAC;;4BAChB,OAAO,EAAE,CAAC;oBACjB,CAAC,CAAC,CAAC;gBACL,CAAC,CAAC,CAAC;gBACH,MAAM,GAAG,IAAI,CAAC;YAChB,CAAC;YACD,MAAM,cAAc,CAAC,QAAQ,EAAE,CAAC;YAChC,GAAG,CAAC,IAAI,CAAC,yBAAyB,EAAE,EAAE,CAAC,CAAC;QAC1C,CAAC;KACF,CAAC;AACJ,CAAC"}

package/dist/src/agent/states/compacting.d.ts

@@ -0,0 +1,14 @@
+/**
+ * @license
+ * Copyright 2026 Amodal Labs, Inc.
+ * SPDX-License-Identifier: MIT
+ */
+import type { CompactingState, AgentContext, TransitionResult } from '../loop-types.js';
+/**
+ * Handle the COMPACTING state.
+ *
+ * Splits messages into "old" (to be summarized) and "recent" (kept verbatim).
+ * Calls generateText with a cheap model to produce a structured summary.
+ * Replaces old messages with a single summary message.
+ */
+export declare function handleCompacting(state: CompactingState, ctx: AgentContext): Promise<TransitionResult>;

package/dist/src/agent/states/compacting.js

@@ -0,0 +1,258 @@
+/**
+ * @license
+ * Copyright 2026 Amodal Labs, Inc.
+ * SPDX-License-Identifier: MIT
+ */
+/**
+ * COMPACTING state handler.
+ *
+ * Summarizes older conversation turns to free context space:
+ * 1. Keep the last N turns verbatim (active conversation)
+ * 2. Send older turns to a cheap model for structured summarization
+ * 3. Replace old turns with the summary
+ * 4. Circuit breaker: after N consecutive failures, give up and continue
+ *
+ * The summary is structured as a handoff document with sections for
+ * current state, original task, key data, actions taken, errors, and
+ * next steps. This preserves the information the agent needs to
+ * continue working without the full conversation history.
+ */
+import { SSEEventType } from '../../types.js';
+import { CompactionError } from '../../errors.js';
+import { estimateTokenCount } from '../token-estimate.js';
+/** Prefix for the summary message injected after compaction. */
+const COMPACTION_SUMMARY_PREFIX = '[Conversation Summary — older messages compacted]';
+/** Timeout for the compaction generateText call (30 seconds). */
+const COMPACTION_TIMEOUT_MS = 30_000;
+const COMPACTION_PROMPT = `Summarize this conversation as a structured handoff into these sections:
+
+1. **Current State** — what we're working on right now
+2. **Original Task** — what the user originally asked for
+3. **Key Data** — important values, IDs, file paths, or facts that must be preserved
+4. **Actions Taken** — tools called and what they returned (summarized, not verbatim)
+5. **Errors & Corrections** — what went wrong and what we tried instead
+6. **Next Steps** — what should happen next
+
+Keep each section under 500 tokens. Focus on information the agent needs to continue working. Omit sections that have no content.`;
+/**
+ * Handle the COMPACTING state.
+ *
+ * Splits messages into "old" (to be summarized) and "recent" (kept verbatim).
+ * Calls generateText with a cheap model to produce a structured summary.
+ * Replaces old messages with a single summary message.
+ */
+export async function handleCompacting(state, ctx) {
+    const effects = [];
+    const { keepRecentTurns, maxSummaryTokens, compactionCircuitBreaker } = ctx.config;
+    // Circuit breaker — if we've failed too many times, skip compaction
+    if (ctx.compactionFailures >= compactionCircuitBreaker) {
+        ctx.logger.warn('compaction_circuit_breaker', {
+            session: ctx.sessionId,
+            failures: ctx.compactionFailures,
+            threshold: compactionCircuitBreaker,
+        });
+        return {
+            next: { type: 'thinking', messages: state.messages },
+            effects,
+        };
+    }
+    // Split messages: keep last N turns verbatim, summarize the rest.
+    // A "turn" is a user message + assistant response (+ any tool results).
+    // We count from the end by counting user messages.
+    const splitIndex = findSplitIndex(state.messages, keepRecentTurns);
+    if (splitIndex <= 0) {
+        // Not enough messages to compact — nothing older than the recent window
+        ctx.logger.debug('compaction_skipped_too_few', {
+            session: ctx.sessionId,
+            messageCount: state.messages.length,
+            keepRecentTurns,
+        });
+        return {
+            next: { type: 'thinking', messages: state.messages },
+            effects,
+        };
+    }
+    const oldMessages = state.messages.slice(0, splitIndex);
+    const recentMessages = state.messages.slice(splitIndex);
+    const tokensBefore = state.estimatedTokens;
+    effects.push({
+        type: SSEEventType.CompactionStart,
+        estimated_tokens: tokensBefore,
+        threshold: ctx.config.compactThreshold,
+        timestamp: new Date().toISOString(),
+    });
+    ctx.logger.info('compaction_start', {
+        session: ctx.sessionId,
+        totalMessages: state.messages.length,
+        oldMessages: oldMessages.length,
+        recentMessages: recentMessages.length,
+        tokensBefore,
+    });
+    const summaryResult = await summarizeMessages(oldMessages, ctx, maxSummaryTokens);
+    if (!summaryResult.ok) {
+        ctx.compactionFailures++;
+        ctx.logger.error('compaction_failed', {
+            session: ctx.sessionId,
+            error: summaryResult.error.message,
+            failure: ctx.compactionFailures,
+            circuitBreakerAt: compactionCircuitBreaker,
+        });
+        // Continue without compaction — the agent can still work, just with
+        // a fuller context. Better than crashing the loop.
+        return {
+            next: { type: 'thinking', messages: state.messages },
+            effects,
+        };
+    }
+    // Build the compacted message list: summary + recent turns.
+    // Uses 'system' role — this is context injected by the runtime, not user input.
+    const summaryMessage = {
+        role: 'system',
+        content: `${COMPACTION_SUMMARY_PREFIX}\n\n${summaryResult.value.text}`,
+    };
+    const compactedMessages = [summaryMessage, ...recentMessages];
+    // Update context
+    ctx.messages = compactedMessages;
+    ctx.compactionFailures = 0; // Reset circuit breaker on success
+    const tokensAfter = estimateTokenCount(compactedMessages);
+    const compactionTokens = summaryResult.value.usage.inputTokens + summaryResult.value.usage.outputTokens;
+    ctx.logger.info('compaction_end', {
+        session: ctx.sessionId,
+        tokensBefore,
+        tokensAfter,
+        messagesBefore: state.messages.length,
+        messagesAfter: compactedMessages.length,
+    });
+    effects.push({
+        type: SSEEventType.CompactionEnd,
+        tokens_before: tokensBefore,
+        tokens_after: tokensAfter,
+        compaction_tokens: compactionTokens,
+        timestamp: new Date().toISOString(),
+    });
+    return {
+        next: { type: 'thinking', messages: compactedMessages },
+        effects,
+    };
+}
+/**
+ * Summarize old messages using generateText.
+ * Returns Result to avoid try/catch in the caller.
+ */
+async function summarizeMessages(messages, ctx, maxSummaryTokens) {
+    // Serialize old messages into a readable conversation format
+    const conversationText = messagesToText(messages);
+    // Per-call timeout composed with session abort signal — prevents a hung
+    // provider from blocking the loop until session-level abort.
+    const timeoutSignal = AbortSignal.timeout(COMPACTION_TIMEOUT_MS);
+    const combinedSignal = AbortSignal.any([ctx.signal, timeoutSignal]);
+    try {
+        const result = await ctx.provider.generateText({
+            messages: [
+                { role: 'user', content: `Here is a conversation to summarize:\n\n${conversationText}` },
+            ],
+            system: COMPACTION_PROMPT,
+            maxOutputTokens: maxSummaryTokens,
+            abortSignal: combinedSignal,
+        });
+        // Track compaction token usage
+        ctx.usage.inputTokens += result.usage.inputTokens;
+        ctx.usage.outputTokens += result.usage.outputTokens;
+        ctx.usage.totalTokens += result.usage.inputTokens + result.usage.outputTokens;
+        if (!result.text || result.text.trim().length === 0) {
+            return {
+                ok: false,
+                error: new CompactionError('Summarization returned empty text', {
+                    stage: 'summarize',
+                    context: { messageCount: messages.length },
+                }),
+            };
+        }
+        return {
+            ok: true,
+            value: {
+                text: result.text,
+                usage: { inputTokens: result.usage.inputTokens, outputTokens: result.usage.outputTokens },
+            },
+        };
+    }
+    catch (err) {
+        // Specific expected failure with specific handling (reason #3):
+        // Provider errors during compaction are non-fatal — the caller degrades
+        // gracefully by continuing without compaction.
+        return {
+            ok: false,
+            error: new CompactionError(`Summarization failed: ${err instanceof Error ? err.message : String(err)}`, { stage: 'summarize', cause: err, context: { messageCount: messages.length } }),
+        };
+    }
+}
+/**
+ * Convert messages to a readable text format for the summarizer.
+ */
+function messagesToText(messages) {
+    const lines = [];
+    for (const msg of messages) {
+        const role = msg.role.toUpperCase();
+        if (typeof msg.content === 'string') {
+            lines.push(`${role}: ${msg.content}`);
+            continue;
+        }
+        if (!Array.isArray(msg.content))
+            continue;
+        for (const part of msg.content) {
+            if (!('type' in part))
+                continue;
+            if (part.type === 'text' && 'text' in part) {
+                lines.push(`${role}: ${part.text}`);
+            }
+            else if (part.type === 'tool-call' && 'toolName' in part) {
+                const args = 'input' in part ? JSON.stringify(part.input) : '{}';
+                lines.push(`${role} [tool_call: ${part.toolName}(${args})]`);
+            }
+            else if (part.type === 'tool-result' && 'output' in part) {
+                const output = extractToolResultText(part.output);
+                // Truncate long tool results for the summarizer — it doesn't need 20K
+                const truncated = output.length > 2_000
+                    ? output.slice(0, 2_000) + '... [truncated for summarization]'
+                    : output;
+                lines.push(`TOOL_RESULT: ${truncated}`);
+            }
+        }
+    }
+    return lines.join('\n');
+}
+/**
+ * Extract text from a tool result output field.
+ */
+function extractToolResultText(output) {
+    if (typeof output === 'string')
+        return output;
+    if (typeof output === 'object' && output !== null && 'value' in output) {
+        // `in` narrows to `object & Record<'value', unknown>` — safe to access
+        return String(output.value);
+    }
+    return JSON.stringify(output);
+}
+// ---------------------------------------------------------------------------
+// Helpers
+// ---------------------------------------------------------------------------
+/**
+ * Find the index to split messages: everything before this index gets
+ * summarized, everything at or after is kept verbatim.
+ *
+ * Counts user messages from the end to find `keepTurns` turn boundaries.
+ */
+function findSplitIndex(messages, keepTurns) {
+    let userMessagesSeen = 0;
+    for (let i = messages.length - 1; i >= 0; i--) {
+        if (messages[i].role === 'user') {
+            userMessagesSeen++;
+            if (userMessagesSeen >= keepTurns) {
+                return i;
+            }
+        }
+    }
+    // Not enough turns to split — return 0 (don't compact)
+    return 0;
+}
+//# sourceMappingURL=compacting.js.map

package/dist/src/agent/states/compacting.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"compacting.js","sourceRoot":"","sources":["../../../../src/agent/states/compacting.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH;;;;;;;;;;;;;GAaG;AAEH,OAAO,EAAC,YAAY,EAAC,MAAM,gBAAgB,CAAC;AAG5C,OAAO,EAAC,eAAe,EAAC,MAAM,iBAAiB,CAAC;AAOhD,OAAO,EAAC,kBAAkB,EAAC,MAAM,sBAAsB,CAAC;AAExD,gEAAgE;AAChE,MAAM,yBAAyB,GAAG,mDAAmD,CAAC;AAEtF,iEAAiE;AACjE,MAAM,qBAAqB,GAAG,MAAM,CAAC;AAErC,MAAM,iBAAiB,GAAG;;;;;;;;;kIASwG,CAAC;AAEnI;;;;;;GAMG;AACH,MAAM,CAAC,KAAK,UAAU,gBAAgB,CACpC,KAAsB,EACtB,GAAiB;IAEjB,MAAM,OAAO,GAAe,EAAE,CAAC;IAC/B,MAAM,EAAC,eAAe,EAAE,gBAAgB,EAAE,wBAAwB,EAAC,GAAG,GAAG,CAAC,MAAM,CAAC;IAEjF,oEAAoE;IACpE,IAAI,GAAG,CAAC,kBAAkB,IAAI,wBAAwB,EAAE,CAAC;QACvD,GAAG,CAAC,MAAM,CAAC,IAAI,CAAC,4BAA4B,EAAE;YAC5C,OAAO,EAAE,GAAG,CAAC,SAAS;YACtB,QAAQ,EAAE,GAAG,CAAC,kBAAkB;YAChC,SAAS,EAAE,wBAAwB;SACpC,CAAC,CAAC;QACH,OAAO;YACL,IAAI,EAAE,EAAC,IAAI,EAAE,UAAU,EAAE,QAAQ,EAAE,KAAK,CAAC,QAAQ,EAAC;YAClD,OAAO;SACR,CAAC;IACJ,CAAC;IAED,kEAAkE;IAClE,wEAAwE;IACxE,mDAAmD;IACnD,MAAM,UAAU,GAAG,cAAc,CAAC,KAAK,CAAC,QAAQ,EAAE,eAAe,CAAC,CAAC;IAEnE,IAAI,UAAU,IAAI,CAAC,EAAE,CAAC;QACpB,wEAAwE;QACxE,GAAG,CAAC,MAAM,CAAC,KAAK,CAAC,4BAA4B,EAAE;YAC7C,OAAO,EAAE,GAAG,CAAC,SAAS;YACtB,YAAY,EAAE,KAAK,CAAC,QAAQ,CAAC,MAAM;YACnC,eAAe;SAChB,CAAC,CAAC;QACH,OAAO;YACL,IAAI,EAAE,EAAC,IAAI,EAAE,UAAU,EAAE,QAAQ,EAAE,KAAK,CAAC,QAAQ,EAAC;YAClD,OAAO;SACR,CAAC;IACJ,CAAC;IAED,MAAM,WAAW,GAAG,KAAK,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,EAAE,UAAU,CAAC,CAAC;IACxD,MAAM,cAAc,GAAG,KAAK,CAAC,QAAQ,CAAC,KAAK,CAAC,UAAU,CAAC,CAAC;IACxD,MAAM,YAAY,GAAG,KAAK,CAAC,eAAe,CAAC;IAE3C,OAAO,CAAC,IAAI,CAAC;QACX,IAAI,EAAE,YAAY,CAAC,eAAe;QAClC,gBAAgB,EAAE,YAAY;QAC9B,SAAS,EAAE,GAAG,CAAC,MAAM,CAAC,gBAAgB;QACtC,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;KACpC,CAAC,CAAC;IAEH,GAAG,CAAC,MAAM,CAAC,IAAI,CAAC,kBAAkB,EAAE;QAClC,OAAO,EAAE,GAAG,CAAC,SAAS;QACtB,aAAa,EAAE,KAAK,CAAC,QAAQ,CAAC,MAAM;QACpC,WAAW,EAAE,WAAW,CAAC,MAAM;QAC/B,cAAc,EAAE,cAAc,CAAC,MAAM;QACrC,YAAY;KACb,CAAC,CAAC;IAEH,MAAM,aAAa,GAAG,MAAM,iBAAiB,CAAC,WAAW,EAAE,GAAG,EAAE,gBAAgB,CAAC,CAAC;IAElF,IAAI,CAAC,aAAa,CAAC,EAAE,EAAE,CAAC;QACtB,GAAG,CAAC,kBAAkB,EAAE,CAAC;QACzB,GAAG,CAAC,MAAM,CAAC,KAAK,CAAC,mBAAmB,EAAE;YACpC,OAAO,EAAE,GAAG,CAAC,SAAS;YACtB,KAAK,EAAE,aAAa,CAAC,KAAK,CAAC,OAAO;YAClC,OAAO,EAAE,GAAG,CAAC,kBAAkB;YAC/B,gBAAgB,EAAE,wBAAwB;SAC3C,CAAC,CAAC;QACH,oEAAoE;QACpE,mDAAmD;QACnD,OAAO;YACL,IAAI,EAAE,EAAC,IAAI,EAAE,UAAU,EAAE,QAAQ,EAAE,KAAK,CAAC,QAAQ,EAAC;YAClD,OAAO;SACR,CAAC;IACJ,CAAC;IAED,4DAA4D;IAC5D,gFAAgF;IAChF,MAAM,cAAc,GAAiB;QACnC,IAAI,EAAE,QAAQ;QACd,OAAO,EAAE,GAAG,yBAAyB,OAAO,aAAa,CAAC,KAAK,CAAC,IAAI,EAAE;KACvE,CAAC;IACF,MAAM,iBAAiB,GAAG,CAAC,cAAc,EAAE,GAAG,cAAc,CAAC,CAAC;IAE9D,iBAAiB;IACjB,GAAG,CAAC,QAAQ,GAAG,iBAAiB,CAAC;IACjC,GAAG,CAAC,kBAAkB,GAAG,CAAC,CAAC,CAAC,mCAAmC;IAE/D,MAAM,WAAW,GAAG,kBAAkB,CAAC,iBAAiB,CAAC,CAAC;IAC1D,MAAM,gBAAgB,GAAG,aAAa,CAAC,KAAK,CAAC,KAAK,CAAC,WAAW,GAAG,aAAa,CAAC,KAAK,CAAC,KAAK,CAAC,YAAY,CAAC;IAExG,GAAG,CAAC,MAAM,CAAC,IAAI,CAAC,gBAAgB,EAAE;QAChC,OAAO,EAAE,GAAG,CAAC,SAAS;QACtB,YAAY;QACZ,WAAW;QACX,cAAc,EAAE,KAAK,CAAC,QAAQ,CAAC,MAAM;QACrC,aAAa,EAAE,iBAAiB,CAAC,MAAM;KACxC,CAAC,CAAC;IAEH,OAAO,CAAC,IAAI,CAAC;QACX,IAAI,EAAE,YAAY,CAAC,aAAa;QAChC,aAAa,EAAE,YAAY;QAC3B,YAAY,EAAE,WAAW;QACzB,iBAAiB,EAAE,gBAAgB;QACnC,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;KACpC,CAAC,CAAC;IAEH,OAAO;QACL,IAAI,EAAE,EAAC,IAAI,EAAE,UAAU,EAAE,QAAQ,EAAE,iBAAiB,EAAC;QACrD,OAAO;KACR,CAAC;AACJ,CAAC;AAWD;;;GAGG;AACH,KAAK,UAAU,iBAAiB,CAC9B,QAAwB,EACxB,GAAiB,EACjB,gBAAwB;IAExB,6DAA6D;IAC7D,MAAM,gBAAgB,GAAG,cAAc,CAAC,QAAQ,CAAC,CAAC;IAElD,wEAAwE;IACxE,6DAA6D;IAC7D,MAAM,aAAa,GAAG,WAAW,CAAC,OAAO,CAAC,qBAAqB,CAAC,CAAC;IACjE,MAAM,cAAc,GAAG,WAAW,CAAC,GAAG,CAAC,CAAC,GAAG,CAAC,MAAM,EAAE,aAAa,CAAC,CAAC,CAAC;IAEpE,IAAI,CAAC;QACH,MAAM,MAAM,GAAG,MAAM,GAAG,CAAC,QAAQ,CAAC,YAAY,CAAC;YAC7C,QAAQ,EAAE;gBACR,EAAC,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,2CAA2C,gBAAgB,EAAE,EAAC;aACvF;YACD,MAAM,EAAE,iBAAiB;YACzB,eAAe,EAAE,gBAAgB;YACjC,WAAW,EAAE,cAAc;SAC5B,CAAC,CAAC;QAEH,+BAA+B;QAC/B,GAAG,CAAC,KAAK,CAAC,WAAW,IAAI,MAAM,CAAC,KAAK,CAAC,WAAW,CAAC;QAClD,GAAG,CAAC,KAAK,CAAC,YAAY,IAAI,MAAM,CAAC,KAAK,CAAC,YAAY,CAAC;QACpD,GAAG,CAAC,KAAK,CAAC,WAAW,IAAI,MAAM,CAAC,KAAK,CAAC,WAAW,GAAG,MAAM,CAAC,KAAK,CAAC,YAAY,CAAC;QAE9E,IAAI,CAAC,MAAM,CAAC,IAAI,IAAI,MAAM,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACpD,OAAO;gBACL,EAAE,EAAE,KAAK;gBACT,KAAK,EAAE,IAAI,eAAe,CAAC,mCAAmC,EAAE;oBAC9D,KAAK,EAAE,WAAW;oBAClB,OAAO,EAAE,EAAC,YAAY,EAAE,QAAQ,CAAC,MAAM,EAAC;iBACzC,CAAC;aACH,CAAC;QACJ,CAAC;QAED,OAAO;YACL,EAAE,EAAE,IAAI;YACR,KAAK,EAAE;gBACL,IAAI,EAAE,MAAM,CAAC,IAAI;gBACjB,KAAK,EAAE,EAAC,WAAW,EAAE,MAAM,CAAC,KAAK,CAAC,WAAW,EAAE,YAAY,EAAE,MAAM,CAAC,KAAK,CAAC,YAAY,EAAC;aACxF;SACF,CAAC;IACJ,CAAC;IAAC,OAAO,GAAG,EAAE,CAAC;QACb,gEAAgE;QAChE,wEAAwE;QACxE,+CAA+C;QAC/C,OAAO;YACL,EAAE,EAAE,KAAK;YACT,KAAK,EAAE,IAAI,eAAe,CACxB,yBAAyB,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC,EAAE,EAC3E,EAAC,KAAK,EAAE,WAAW,EAAE,KAAK,EAAE,GAAG,EAAE,OAAO,EAAE,EAAC,YAAY,EAAE,QAAQ,CAAC,MAAM,EAAC,EAAC,CAC3E;SACF,CAAC;IACJ,CAAC;AACH,CAAC;AAED;;GAEG;AACH,SAAS,cAAc,CAAC,QAAwB;IAC9C,MAAM,KAAK,GAAa,EAAE,CAAC;IAE3B,KAAK,MAAM,GAAG,IAAI,QAAQ,EAAE,CAAC;QAC3B,MAAM,IAAI,GAAG,GAAG,CAAC,IAAI,CAAC,WAAW,EAAE,CAAC;QAEpC,IAAI,OAAO,GAAG,CAAC,OAAO,KAAK,QAAQ,EAAE,CAAC;YACpC,KAAK,CAAC,IAAI,CAAC,GAAG,IAAI,KAAK,GAAG,CAAC,OAAO,EAAE,CAAC,CAAC;YACtC,SAAS;QACX,CAAC;QAED,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,GAAG,CAAC,OAAO,CAAC;YAAE,SAAS;QAE1C,KAAK,MAAM,IAAI,IAAI,GAAG,CAAC,OAAO,EAAE,CAAC;YAC/B,IAAI,CAAC,CAAC,MAAM,IAAI,IAAI,CAAC;gBAAE,SAAS;YAEhC,IAAI,IAAI,CAAC,IAAI,KAAK,MAAM,IAAI,MAAM,IAAI,IAAI,EAAE,CAAC;gBAC3C,KAAK,CAAC,IAAI,CAAC,GAAG,IAAI,KAAK,IAAI,CAAC,IAAI,EAAE,CAAC,CAAC;YACtC,CAAC;iBAAM,IAAI,IAAI,CAAC,IAAI,KAAK,WAAW,IAAI,UAAU,IAAI,IAAI,EAAE,CAAC;gBAC3D,MAAM,IAAI,GAAG,OAAO,IAAI,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC;gBACjE,KAAK,CAAC,IAAI,CAAC,GAAG,IAAI,gBAAgB,IAAI,CAAC,QAAQ,IAAI,IAAI,IAAI,CAAC,CAAC;YAC/D,CAAC;iBAAM,IAAI,IAAI,CAAC,IAAI,KAAK,aAAa,IAAI,QAAQ,IAAI,IAAI,EAAE,CAAC;gBAC3D,MAAM,MAAM,GAAG,qBAAqB,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;gBAClD,sEAAsE;gBACtE,MAAM,SAAS,GAAG,MAAM,CAAC,MAAM,GAAG,KAAK;oBACrC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,EAAE,KAAK,CAAC,GAAG,mCAAmC;oBAC9D,CAAC,CAAC,MAAM,CAAC;gBACX,KAAK,CAAC,IAAI,CAAC,gBAAgB,SAAS,EAAE,CAAC,CAAC;YAC1C,CAAC;QACH,CAAC;IACH,CAAC;IAED,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;AAC1B,CAAC;AAED;;GAEG;AACH,SAAS,qBAAqB,CAAC,MAAe;IAC5C,IAAI,OAAO,MAAM,KAAK,QAAQ;QAAE,OAAO,MAAM,CAAC;IAC9C,IAAI,OAAO,MAAM,KAAK,QAAQ,IAAI,MAAM,KAAK,IAAI,IAAI,OAAO,IAAI,MAAM,EAAE,CAAC;QACvE,uEAAuE;QACvE,OAAO,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;IAC9B,CAAC;IACD,OAAO,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC;AAChC,CAAC;AAED,8EAA8E;AAC9E,UAAU;AACV,8EAA8E;AAE9E;;;;;GAKG;AACH,SAAS,cAAc,CAAC,QAAwB,EAAE,SAAiB;IACjE,IAAI,gBAAgB,GAAG,CAAC,CAAC;IAEzB,KAAK,IAAI,CAAC,GAAG,QAAQ,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC,IAAI,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC;QAC9C,IAAI,QAAQ,CAAC,CAAC,CAAC,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;YAChC,gBAAgB,EAAE,CAAC;YACnB,IAAI,gBAAgB,IAAI,SAAS,EAAE,CAAC;gBAClC,OAAO,CAAC,CAAC;YACX,CAAC;QACH,CAAC;IACH,CAAC;IAED,uDAAuD;IACvD,OAAO,CAAC,CAAC;AACX,CAAC"}

package/dist/src/agent/states/confirming.d.ts

@@ -0,0 +1,10 @@
+/**
+ * @license
+ * Copyright 2026 Amodal Labs, Inc.
+ * SPDX-License-Identifier: MIT
+ */
+import type { ConfirmingState, AgentContext, TransitionResult } from '../loop-types.js';
+/**
+ * Handle the CONFIRMING state.
+ */
+export declare function handleConfirming(state: ConfirmingState, ctx: AgentContext): Promise<TransitionResult>;

package/dist/src/agent/states/confirming.js

@@ -0,0 +1,76 @@
+/**
+ * @license
+ * Copyright 2026 Amodal Labs, Inc.
+ * SPDX-License-Identifier: MIT
+ */
+/**
+ * CONFIRMING state handler.
+ *
+ * Waits for user confirmation on a tool call (via ctx.waitForConfirmation).
+ * - Approved → resume EXECUTING with the confirmed call
+ * - Denied → tell the model the tool call was denied, back to THINKING
+ */
+import { SSEEventType } from '../../types.js';
+/**
+ * Handle the CONFIRMING state.
+ */
+export async function handleConfirming(state, ctx) {
+    const effects = [];
+    ctx.logger.info('tool_confirmation_waiting', {
+        tool: state.call.toolName,
+        callId: state.call.toolCallId,
+        session: ctx.sessionId,
+    });
+    // Race confirmation against a timeout — don't hang forever if user never responds
+    const timeoutMs = ctx.config.confirmationTimeoutMs;
+    const timeoutPromise = new Promise((resolve) => {
+        setTimeout(() => resolve(false), timeoutMs);
+    });
+    const approved = await Promise.race([
+        ctx.waitForConfirmation(state.call.toolCallId),
+        timeoutPromise,
+    ]);
+    if (approved) {
+        ctx.logger.info('tool_confirmation_approved', {
+            tool: state.call.toolName,
+            callId: state.call.toolCallId,
+            session: ctx.sessionId,
+        });
+        effects.push({
+            type: SSEEventType.Approved,
+            resource_type: 'tool_call',
+            preview_id: state.call.toolCallId,
+            timestamp: new Date().toISOString(),
+        });
+        return {
+            next: {
+                type: 'executing',
+                queue: state.remainingQueue,
+                current: state.call,
+                results: [],
+            },
+            effects,
+        };
+    }
+    // Denied — inject a tool result message telling the model
+    ctx.logger.info('tool_confirmation_denied', {
+        tool: state.call.toolName,
+        callId: state.call.toolCallId,
+        session: ctx.sessionId,
+    });
+    const denialMessage = {
+        role: 'tool',
+        content: [{
+                type: 'tool-result',
+                toolCallId: state.call.toolCallId,
+                toolName: state.call.toolName,
+                output: { type: 'text', value: 'Tool call denied by user. Do not retry this action without asking the user first.' },
+            }],
+    };
+    ctx.messages = [...ctx.messages, denialMessage];
+    return {
+        next: { type: 'thinking', messages: ctx.messages },
+        effects,
+    };
+}
+//# sourceMappingURL=confirming.js.map

package/dist/src/agent/states/confirming.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"confirming.js","sourceRoot":"","sources":["../../../../src/agent/states/confirming.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH;;;;;;GAMG;AAEH,OAAO,EAAC,YAAY,EAAC,MAAM,gBAAgB,CAAC;AAQ5C;;GAEG;AACH,MAAM,CAAC,KAAK,UAAU,gBAAgB,CACpC,KAAsB,EACtB,GAAiB;IAEjB,MAAM,OAAO,GAAe,EAAE,CAAC;IAE/B,GAAG,CAAC,MAAM,CAAC,IAAI,CAAC,2BAA2B,EAAE;QAC3C,IAAI,EAAE,KAAK,CAAC,IAAI,CAAC,QAAQ;QACzB,MAAM,EAAE,KAAK,CAAC,IAAI,CAAC,UAAU;QAC7B,OAAO,EAAE,GAAG,CAAC,SAAS;KACvB,CAAC,CAAC;IAEH,kFAAkF;IAClF,MAAM,SAAS,GAAG,GAAG,CAAC,MAAM,CAAC,qBAAqB,CAAC;IACnD,MAAM,cAAc,GAAG,IAAI,OAAO,CAAQ,CAAC,OAAO,EAAE,EAAE;QACpD,UAAU,CAAC,GAAG,EAAE,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE,SAAS,CAAC,CAAC;IAC9C,CAAC,CAAC,CAAC;IACH,MAAM,QAAQ,GAAG,MAAM,OAAO,CAAC,IAAI,CAAC;QAClC,GAAG,CAAC,mBAAmB,CAAC,KAAK,CAAC,IAAI,CAAC,UAAU,CAAC;QAC9C,cAAc;KACf,CAAC,CAAC;IAEH,IAAI,QAAQ,EAAE,CAAC;QACb,GAAG,CAAC,MAAM,CAAC,IAAI,CAAC,4BAA4B,EAAE;YAC5C,IAAI,EAAE,KAAK,CAAC,IAAI,CAAC,QAAQ;YACzB,MAAM,EAAE,KAAK,CAAC,IAAI,CAAC,UAAU;YAC7B,OAAO,EAAE,GAAG,CAAC,SAAS;SACvB,CAAC,CAAC;QAEH,OAAO,CAAC,IAAI,CAAC;YACX,IAAI,EAAE,YAAY,CAAC,QAAQ;YAC3B,aAAa,EAAE,WAAW;YAC1B,UAAU,EAAE,KAAK,CAAC,IAAI,CAAC,UAAU;YACjC,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;SACpC,CAAC,CAAC;QAEH,OAAO;YACL,IAAI,EAAE;gBACJ,IAAI,EAAE,WAAW;gBACjB,KAAK,EAAE,KAAK,CAAC,cAAc;gBAC3B,OAAO,EAAE,KAAK,CAAC,IAAI;gBACnB,OAAO,EAAE,EAAE;aACZ;YACD,OAAO;SACR,CAAC;IACJ,CAAC;IAED,0DAA0D;IAC1D,GAAG,CAAC,MAAM,CAAC,IAAI,CAAC,0BAA0B,EAAE;QAC1C,IAAI,EAAE,KAAK,CAAC,IAAI,CAAC,QAAQ;QACzB,MAAM,EAAE,KAAK,CAAC,IAAI,CAAC,UAAU;QAC7B,OAAO,EAAE,GAAG,CAAC,SAAS;KACvB,CAAC,CAAC;IAEH,MAAM,aAAa,GAA8B;QAC/C,IAAI,EAAE,MAAM;QACZ,OAAO,EAAE,CAAC;gBACR,IAAI,EAAE,aAAsB;gBAC5B,UAAU,EAAE,KAAK,CAAC,IAAI,CAAC,UAAU;gBACjC,QAAQ,EAAE,KAAK,CAAC,IAAI,CAAC,QAAQ;gBAC7B,MAAM,EAAE,EAAC,IAAI,EAAE,MAAe,EAAE,KAAK,EAAE,mFAAmF,EAAC;aAC5H,CAAC;KACH,CAAC;IAEF,GAAG,CAAC,QAAQ,GAAG,CAAC,GAAG,GAAG,CAAC,QAAQ,EAAE,aAAa,CAAC,CAAC;IAEhD,OAAO;QACL,IAAI,EAAE,EAAC,IAAI,EAAE,UAAU,EAAE,QAAQ,EAAE,GAAG,CAAC,QAAQ,EAAC;QAChD,OAAO;KACR,CAAC;AACJ,CAAC"}

package/dist/src/agent/states/dispatching.d.ts

@@ -0,0 +1,18 @@
+/**
+ * @license
+ * Copyright 2026 Amodal Labs, Inc.
+ * SPDX-License-Identifier: MIT
+ */
+import type { DispatchingState, AgentContext, TransitionResult } from '../loop-types.js';
+/**
+ * Handle the DISPATCHING state — run a child agent loop.
+ *
+ * Creates a child AgentContext with:
+ * - Same provider, logger, signal, tenant, user, permission checker
+ * - Tool registry subset (only the requested tools, never dispatch_task)
+ * - Same buildToolContext factory (child shares parent's connections/stores)
+ * - Reduced maxTurns (default 10) and maxContextTokens
+ * - Fresh messages (dispatch prompt as user message)
+ * - Fresh usage counters (merged back to parent after completion)
+ */
+export declare function handleDispatching(state: DispatchingState, ctx: AgentContext): Promise<TransitionResult>;