@geminixiang/mama 0.2.0-beta.0 → 0.2.0-beta.10

package/dist/runtime/conversation-orchestrator.js

@@ -0,0 +1,150 @@
+import { hasMaterializedSlackBranchSession, waitForSlackBranchBootstrap, } from "../adapters/slack/branch-manager.js";
+import { isPrivateConversation } from "../commands/utils.js";
+import * as log from "../log.js";
+import { addLifecycleBreadcrumb, applyRunScope } from "../sentry.js";
+import { formatStopped } from "../ui-copy.js";
+import * as Sentry from "@sentry/node";
+import { join } from "path";
+export class ConversationOrchestrator {
+    constructor(options) {
+        this.options = options;
+    }
+    async runSession({ event, bot, adapters, isSyntheticEvent, }) {
+        const conversationId = event.conversationId;
+        if (this.options.isShuttingDown()) {
+            log.logInfo(`[${conversationId}] Rejected event during shutdown: ${event.text.substring(0, 50)}`);
+            return;
+        }
+        const sessionKey = event.sessionKey ?? `${conversationId}:${event.thread_ts ?? event.ts}`;
+        const privateConversation = isPrivateConversation(event);
+        const handledCommand = await this.options.commandRegistry.handle({
+            bot,
+            responseCtx: adapters.responseCtx,
+            platform: adapters.platform.name,
+            platformUserId: event.user,
+            conversationId,
+            vaultConversationId: event.vaultConversationId,
+            sessionKey,
+            commandText: event.text,
+            privateConversation,
+            services: this.options.commandServices,
+        });
+        if (handledCommand)
+            return;
+        const conversationDir = join(this.options.workingDir, conversationId);
+        const waitedForParent = adapters.platform.name === "slack" && !isSyntheticEvent
+            ? await waitForSlackBranchBootstrap({
+                parentSessionKey: conversationId,
+                sessionKey,
+                hasThreadSession: () => hasMaterializedSlackBranchSession(conversationDir, sessionKey),
+                isParentRunning: () => this.options.getState(conversationId)?.running === true,
+            })
+            : false;
+        if (waitedForParent) {
+            log.logInfo(`[${conversationId}] Delayed thread bootstrap until parent session sealed: ${sessionKey}`);
+        }
+        const state = await this.options.getOrCreateState({
+            conversationId,
+            platformName: adapters.platform.name,
+            sessionKey,
+        });
+        state.running = true;
+        state.stopRequested = false;
+        state.startedAt = Date.now();
+        state.lastActivityAt = Date.now();
+        log.logInfo(`[${conversationId}] Starting run: ${event.text.substring(0, 50)}`);
+        const runPromise = (async () => {
+            try {
+                const result = await this.runWithInstrumentation(adapters, { conversationId, sessionKey, isSyntheticEvent, startedAt: state.startedAt }, async () => {
+                    await adapters.responseCtx.setTyping(true);
+                    await adapters.responseCtx.setWorking(true);
+                    const runnerResult = await state.runner.run(adapters.message, adapters.responseCtx, adapters.platform);
+                    await adapters.responseCtx.setWorking(false);
+                    return runnerResult;
+                });
+                if (result?.stopReason === "aborted" && state.stopRequested) {
+                    if (state.stopMessageTs) {
+                        await bot.updateMessage(conversationId, state.stopMessageTs, formatStopped(bot));
+                        state.stopMessageTs = undefined;
+                    }
+                    else {
+                        await bot.postMessage(conversationId, formatStopped(bot));
+                    }
+                }
+            }
+            finally {
+                state.running = false;
+                state.lastAccessedAt = Date.now();
+                this.options.onRunFinished();
+            }
+        })();
+        this.options.beforeRunTracked(runPromise);
+        try {
+            await runPromise;
+        }
+        finally {
+            this.options.afterRunTracked(runPromise);
+        }
+    }
+    async runWithInstrumentation(adapters, meta, body) {
+        const { conversationId, sessionKey, isSyntheticEvent, startedAt } = meta;
+        const { message, platform } = adapters;
+        Sentry.metrics.count("agent.run.started", 1, {
+            attributes: { channel: conversationId },
+        });
+        return Sentry.startSpan({ name: "agent.run", op: "agent", attributes: { conversationId, sessionKey } }, async () => Sentry.withScope(async (scope) => {
+            applyRunScope(scope, {
+                conversationId,
+                sessionKey,
+                messageId: message.id,
+                platform: platform.name,
+                userId: message.userId,
+                userName: message.userName,
+                threadTs: message.threadTs,
+                isSyntheticEvent,
+            });
+            addLifecycleBreadcrumb("agent.run.started", {
+                channel_id: conversationId,
+                platform: platform.name,
+                has_attachments: (message.attachments?.length ?? 0) > 0,
+            });
+            try {
+                const result = await body();
+                const durationMs = Date.now() - startedAt;
+                const completionAttrs = {
+                    channel: conversationId,
+                    platform: platform.name,
+                    stop_reason: result.stopReason,
+                };
+                Sentry.metrics.distribution("agent.run.duration", durationMs, {
+                    unit: "millisecond",
+                    attributes: completionAttrs,
+                });
+                Sentry.metrics.count("agent.run.completed", 1, { attributes: completionAttrs });
+                addLifecycleBreadcrumb("agent.run.completed", {
+                    channel_id: conversationId,
+                    platform: platform.name,
+                    stop_reason: result.stopReason,
+                    duration_ms: durationMs,
+                });
+                return result;
+            }
+            catch (err) {
+                scope.setContext("agent_run_error", {
+                    conversationId,
+                    sessionKey,
+                    platform: platform.name,
+                    messageId: message.id,
+                    threadTs: message.threadTs,
+                });
+                Sentry.captureException(err);
+                Sentry.metrics.count("agent.run.errors", 1, {
+                    attributes: { channel: conversationId, platform: platform.name },
+                });
+                log.logWarning(`[${conversationId}] Run error`, err instanceof Error ? err.message : String(err));
+                return undefined;
+            }
+        }));
+    }
+}
+//# sourceMappingURL=conversation-orchestrator.js.map

package/dist/runtime/conversation-orchestrator.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"conversation-orchestrator.js","sourceRoot":"","sources":["../../src/runtime/conversation-orchestrator.ts"],"names":[],"mappings":"AACA,OAAO,EACL,iCAAiC,EACjC,2BAA2B,GAC5B,MAAM,qCAAqC,CAAC;AAI7C,OAAO,EAAE,qBAAqB,EAAE,MAAM,sBAAsB,CAAC;AAC7D,OAAO,KAAK,GAAG,MAAM,WAAW,CAAC;AACjC,OAAO,EAAE,sBAAsB,EAAE,aAAa,EAAE,MAAM,cAAc,CAAC;AACrE,OAAO,EAAE,aAAa,EAAE,MAAM,eAAe,CAAC;AAC9C,OAAO,KAAK,MAAM,MAAM,cAAc,CAAC;AACvC,OAAO,EAAE,IAAI,EAAE,MAAM,MAAM,CAAC;AAmC5B,MAAM,OAAO,wBAAwB;IACnC,YAA6B,OAAwC;uBAAxC,OAAO;IAAoC,CAAC;IAEzE,KAAK,CAAC,UAAU,CAAC,EACf,KAAK,EACL,GAAG,EACH,QAAQ,EACR,gBAAgB,GACO;QACvB,MAAM,cAAc,GAAG,KAAK,CAAC,cAAc,CAAC;QAC5C,IAAI,IAAI,CAAC,OAAO,CAAC,cAAc,EAAE,EAAE,CAAC;YAClC,GAAG,CAAC,OAAO,CACT,IAAI,cAAc,qCAAqC,KAAK,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,EAAE,EAAE,CAAC,EAAE,CACrF,CAAC;YACF,OAAO;QACT,CAAC;QAED,MAAM,UAAU,GAAG,KAAK,CAAC,UAAU,IAAI,GAAG,cAAc,IAAI,KAAK,CAAC,SAAS,IAAI,KAAK,CAAC,EAAE,EAAE,CAAC;QAC1F,MAAM,mBAAmB,GAAG,qBAAqB,CAAC,KAAK,CAAC,CAAC;QACzD,MAAM,cAAc,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,eAAe,CAAC,MAAM,CAAC;YAC/D,GAAG;YACH,WAAW,EAAE,QAAQ,CAAC,WAAW;YACjC,QAAQ,EAAE,QAAQ,CAAC,QAAQ,CAAC,IAAoB;YAChD,cAAc,EAAE,KAAK,CAAC,IAAI;YAC1B,cAAc;YACd,mBAAmB,EAAE,KAAK,CAAC,mBAAmB;YAC9C,UAAU;YACV,WAAW,EAAE,KAAK,CAAC,IAAI;YACvB,mBAAmB;YACnB,QAAQ,EAAE,IAAI,CAAC,OAAO,CAAC,eAAe;SACvC,CAAC,CAAC;QACH,IAAI,cAAc;YAAE,OAAO;QAE3B,MAAM,eAAe,GAAG,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,UAAU,EAAE,cAAc,CAAC,CAAC;QACtE,MAAM,eAAe,GACnB,QAAQ,CAAC,QAAQ,CAAC,IAAI,KAAK,OAAO,IAAI,CAAC,gBAAgB;YACrD,CAAC,CAAC,MAAM,2BAA2B,CAAC;gBAChC,gBAAgB,EAAE,cAAc;gBAChC,UAAU;gBACV,gBAAgB,EAAE,GAAG,EAAE,CAAC,iCAAiC,CAAC,eAAe,EAAE,UAAU,CAAC;gBACtF,eAAe,EAAE,GAAG,EAAE,CAAC,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,cAAc,CAAC,EAAE,OAAO,KAAK,IAAI;aAC/E,CAAC;YACJ,CAAC,CAAC,KAAK,CAAC;QACZ,IAAI,eAAe,EAAE,CAAC;YACpB,GAAG,CAAC,OAAO,CACT,IAAI,cAAc,2DAA2D,UAAU,EAAE,CAC1F,CAAC;QACJ,CAAC;QAED,MAAM,KAAK,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,gBAAgB,CAAC;YAChD,cAAc;YACd,YAAY,EAAE,QAAQ,CAAC,QAAQ,CAAC,IAAI;YACpC,UAAU;SACX,CAAC,CAAC;QAEH,KAAK,CAAC,OAAO,GAAG,IAAI,CAAC;QACrB,KAAK,CAAC,aAAa,GAAG,KAAK,CAAC;QAC5B,KAAK,CAAC,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAC7B,KAAK,CAAC,cAAc,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAElC,GAAG,CAAC,OAAO,CAAC,IAAI,cAAc,mBAAmB,KAAK,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,EAAE,EAAE,CAAC,EAAE,CAAC,CAAC;QAEhF,MAAM,UAAU,GAAG,CAAC,KAAK,IAAI,EAAE;YAC7B,IAAI,CAAC;gBACH,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,sBAAsB,CAC9C,QAAQ,EACR,EAAE,cAAc,EAAE,UAAU,EAAE,gBAAgB,EAAE,SAAS,EAAE,KAAK,CAAC,SAAU,EAAE,EAC7E,KAAK,IAAI,EAAE;oBACT,MAAM,QAAQ,CAAC,WAAW,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC;oBAC3C,MAAM,QAAQ,CAAC,WAAW,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC;oBAC5C,MAAM,YAAY,GAAG,MAAM,KAAK,CAAC,MAAM,CAAC,GAAG,CACzC,QAAQ,CAAC,OAAO,EAChB,QAAQ,CAAC,WAAW,EACpB,QAAQ,CAAC,QAAQ,CAClB,CAAC;oBACF,MAAM,QAAQ,CAAC,WAAW,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC;oBAC7C,OAAO,YAAY,CAAC;gBACtB,CAAC,CACF,CAAC;gBAEF,IAAI,MAAM,EAAE,UAAU,KAAK,SAAS,IAAI,KAAK,CAAC,aAAa,EAAE,CAAC;oBAC5D,IAAI,KAAK,CAAC,aAAa,EAAE,CAAC;wBACxB,MAAM,GAAG,CAAC,aAAa,CAAC,cAAc,EAAE,KAAK,CAAC,aAAa,EAAE,aAAa,CAAC,GAAG,CAAC,CAAC,CAAC;wBACjF,KAAK,CAAC,aAAa,GAAG,SAAS,CAAC;oBAClC,CAAC;yBAAM,CAAC;wBACN,MAAM,GAAG,CAAC,WAAW,CAAC,cAAc,EAAE,aAAa,CAAC,GAAG,CAAC,CAAC,CAAC;oBAC5D,CAAC;gBACH,CAAC;YACH,CAAC;oBAAS,CAAC;gBACT,KAAK,CAAC,OAAO,GAAG,KAAK,CAAC;gBACtB,KAAK,CAAC,cAAc,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;gBAClC,IAAI,CAAC,OAAO,CAAC,aAAa,EAAE,CAAC;YAC/B,CAAC;QACH,CAAC,CAAC,EAAE,CAAC;QAEL,IAAI,CAAC,OAAO,CAAC,gBAAgB,CAAC,UAAU,CAAC,CAAC;QAC1C,IAAI,CAAC;YACH,MAAM,UAAU,CAAC;QACnB,CAAC;gBAAS,CAAC;YACT,IAAI,CAAC,OAAO,CAAC,eAAe,CAAC,UAAU,CAAC,CAAC;QAC3C,CAAC;IACH,CAAC;IAEO,KAAK,CAAC,sBAAsB,CAClC,QAAqB,EACrB,IAKC,EACD,IAAkE;QAElE,MAAM,EAAE,cAAc,EAAE,UAAU,EAAE,gBAAgB,EAAE,SAAS,EAAE,GAAG,IAAI,CAAC;QACzE,MAAM,EAAE,OAAO,EAAE,QAAQ,EAAE,GAAG,QAAQ,CAAC;QAEvC,MAAM,CAAC,OAAO,CAAC,KAAK,CAAC,mBAAmB,EAAE,CAAC,EAAE;YAC3C,UAAU,EAAE,EAAE,OAAO,EAAE,cAAc,EAAE;SACxC,CAAC,CAAC;QAEH,OAAO,MAAM,CAAC,SAAS,CACrB,EAAE,IAAI,EAAE,WAAW,EAAE,EAAE,EAAE,OAAO,EAAE,UAAU,EAAE,EAAE,cAAc,EAAE,UAAU,EAAE,EAAE,EAC9E,KAAK,IAAI,EAAE,CACT,MAAM,CAAC,SAAS,CAAC,KAAK,EAAE,KAAK,EAAE,EAAE;YAC/B,aAAa,CAAC,KAAK,EAAE;gBACnB,cAAc;gBACd,UAAU;gBACV,SAAS,EAAE,OAAO,CAAC,EAAE;gBACrB,QAAQ,EAAE,QAAQ,CAAC,IAAI;gBACvB,MAAM,EAAE,OAAO,CAAC,MAAM;gBACtB,QAAQ,EAAE,OAAO,CAAC,QAAQ;gBAC1B,QAAQ,EAAE,OAAO,CAAC,QAAQ;gBAC1B,gBAAgB;aACjB,CAAC,CAAC;YACH,sBAAsB,CAAC,mBAAmB,EAAE;gBAC1C,UAAU,EAAE,cAAc;gBAC1B,QAAQ,EAAE,QAAQ,CAAC,IAAI;gBACvB,eAAe,EAAE,CAAC,OAAO,CAAC,WAAW,EAAE,MAAM,IAAI,CAAC,CAAC,GAAG,CAAC;aACxD,CAAC,CAAC;YAEH,IAAI,CAAC;gBACH,MAAM,MAAM,GAAG,MAAM,IAAI,EAAE,CAAC;gBAC5B,MAAM,UAAU,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,CAAC;gBAC1C,MAAM,eAAe,GAAG;oBACtB,OAAO,EAAE,cAAc;oBACvB,QAAQ,EAAE,QAAQ,CAAC,IAAI;oBACvB,WAAW,EAAE,MAAM,CAAC,UAAU;iBAC/B,CAAC;gBACF,MAAM,CAAC,OAAO,CAAC,YAAY,CAAC,oBAAoB,EAAE,UAAU,EAAE;oBAC5D,IAAI,EAAE,aAAa;oBACnB,UAAU,EAAE,eAAe;iBAC5B,CAAC,CAAC;gBACH,MAAM,CAAC,OAAO,CAAC,KAAK,CAAC,qBAAqB,EAAE,CAAC,EAAE,EAAE,UAAU,EAAE,eAAe,EAAE,CAAC,CAAC;gBAChF,sBAAsB,CAAC,qBAAqB,EAAE;oBAC5C,UAAU,EAAE,cAAc;oBAC1B,QAAQ,EAAE,QAAQ,CAAC,IAAI;oBACvB,WAAW,EAAE,MAAM,CAAC,UAAU;oBAC9B,WAAW,EAAE,UAAU;iBACxB,CAAC,CAAC;gBACH,OAAO,MAAM,CAAC;YAChB,CAAC;YAAC,OAAO,GAAG,EAAE,CAAC;gBACb,KAAK,CAAC,UAAU,CAAC,iBAAiB,EAAE;oBAClC,cAAc;oBACd,UAAU;oBACV,QAAQ,EAAE,QAAQ,CAAC,IAAI;oBACvB,SAAS,EAAE,OAAO,CAAC,EAAE;oBACrB,QAAQ,EAAE,OAAO,CAAC,QAAQ;iBAC3B,CAAC,CAAC;gBACH,MAAM,CAAC,gBAAgB,CAAC,GAAG,CAAC,CAAC;gBAC7B,MAAM,CAAC,OAAO,CAAC,KAAK,CAAC,kBAAkB,EAAE,CAAC,EAAE;oBAC1C,UAAU,EAAE,EAAE,OAAO,EAAE,cAAc,EAAE,QAAQ,EAAE,QAAQ,CAAC,IAAI,EAAE;iBACjE,CAAC,CAAC;gBACH,GAAG,CAAC,UAAU,CACZ,IAAI,cAAc,aAAa,EAC/B,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC,CACjD,CAAC;gBACF,OAAO,SAAS,CAAC;YACnB,CAAC;QACH,CAAC,CAAC,CACL,CAAC;IACJ,CAAC;CACF","sourcesContent":["import type { Bot, BotAdapters, BotEvent, PlatformName } from \"../adapter.js\";\nimport {\n  hasMaterializedSlackBranchSession,\n  waitForSlackBranchBootstrap,\n} from \"../adapters/slack/branch-manager.js\";\nimport type { AgentRunner } from \"../agent.js\";\nimport type { CommandRegistry } from \"../commands/index.js\";\nimport type { CommandServices } from \"../commands/index.js\";\nimport { isPrivateConversation } from \"../commands/utils.js\";\nimport * as log from \"../log.js\";\nimport { addLifecycleBreadcrumb, applyRunScope } from \"../sentry.js\";\nimport { formatStopped } from \"../ui-copy.js\";\nimport * as Sentry from \"@sentry/node\";\nimport { join } from \"path\";\n\nexport interface ConversationRuntimeState {\n  running: boolean;\n  runner: AgentRunner;\n  stopRequested: boolean;\n  stopMessageTs?: string;\n  lastAccessedAt: number;\n  startedAt?: number;\n  lastActivityAt?: number;\n}\n\nexport interface RunConversationOptions {\n  event: BotEvent;\n  bot: Bot;\n  adapters: BotAdapters;\n  isSyntheticEvent?: boolean;\n}\n\ninterface ConversationOrchestratorOptions {\n  workingDir: string;\n  commandRegistry: CommandRegistry;\n  commandServices: CommandServices;\n  isShuttingDown: () => boolean;\n  getState: (sessionKey: string) => ConversationRuntimeState | undefined;\n  getOrCreateState: (options: {\n    conversationId: string;\n    platformName: string;\n    sessionKey: string;\n  }) => Promise<ConversationRuntimeState>;\n  beforeRunTracked: (runPromise: Promise<void>) => void;\n  afterRunTracked: (runPromise: Promise<void>) => void;\n  onRunFinished: () => void;\n}\n\nexport class ConversationOrchestrator {\n  constructor(private readonly options: ConversationOrchestratorOptions) {}\n\n  async runSession({\n    event,\n    bot,\n    adapters,\n    isSyntheticEvent,\n  }: RunConversationOptions): Promise<void> {\n    const conversationId = event.conversationId;\n    if (this.options.isShuttingDown()) {\n      log.logInfo(\n        `[${conversationId}] Rejected event during shutdown: ${event.text.substring(0, 50)}`,\n      );\n      return;\n    }\n\n    const sessionKey = event.sessionKey ?? `${conversationId}:${event.thread_ts ?? event.ts}`;\n    const privateConversation = isPrivateConversation(event);\n    const handledCommand = await this.options.commandRegistry.handle({\n      bot,\n      responseCtx: adapters.responseCtx,\n      platform: adapters.platform.name as PlatformName,\n      platformUserId: event.user,\n      conversationId,\n      vaultConversationId: event.vaultConversationId,\n      sessionKey,\n      commandText: event.text,\n      privateConversation,\n      services: this.options.commandServices,\n    });\n    if (handledCommand) return;\n\n    const conversationDir = join(this.options.workingDir, conversationId);\n    const waitedForParent =\n      adapters.platform.name === \"slack\" && !isSyntheticEvent\n        ? await waitForSlackBranchBootstrap({\n            parentSessionKey: conversationId,\n            sessionKey,\n            hasThreadSession: () => hasMaterializedSlackBranchSession(conversationDir, sessionKey),\n            isParentRunning: () => this.options.getState(conversationId)?.running === true,\n          })\n        : false;\n    if (waitedForParent) {\n      log.logInfo(\n        `[${conversationId}] Delayed thread bootstrap until parent session sealed: ${sessionKey}`,\n      );\n    }\n\n    const state = await this.options.getOrCreateState({\n      conversationId,\n      platformName: adapters.platform.name,\n      sessionKey,\n    });\n\n    state.running = true;\n    state.stopRequested = false;\n    state.startedAt = Date.now();\n    state.lastActivityAt = Date.now();\n\n    log.logInfo(`[${conversationId}] Starting run: ${event.text.substring(0, 50)}`);\n\n    const runPromise = (async () => {\n      try {\n        const result = await this.runWithInstrumentation(\n          adapters,\n          { conversationId, sessionKey, isSyntheticEvent, startedAt: state.startedAt! },\n          async () => {\n            await adapters.responseCtx.setTyping(true);\n            await adapters.responseCtx.setWorking(true);\n            const runnerResult = await state.runner.run(\n              adapters.message,\n              adapters.responseCtx,\n              adapters.platform,\n            );\n            await adapters.responseCtx.setWorking(false);\n            return runnerResult;\n          },\n        );\n\n        if (result?.stopReason === \"aborted\" && state.stopRequested) {\n          if (state.stopMessageTs) {\n            await bot.updateMessage(conversationId, state.stopMessageTs, formatStopped(bot));\n            state.stopMessageTs = undefined;\n          } else {\n            await bot.postMessage(conversationId, formatStopped(bot));\n          }\n        }\n      } finally {\n        state.running = false;\n        state.lastAccessedAt = Date.now();\n        this.options.onRunFinished();\n      }\n    })();\n\n    this.options.beforeRunTracked(runPromise);\n    try {\n      await runPromise;\n    } finally {\n      this.options.afterRunTracked(runPromise);\n    }\n  }\n\n  private async runWithInstrumentation(\n    adapters: BotAdapters,\n    meta: {\n      conversationId: string;\n      sessionKey: string;\n      isSyntheticEvent?: boolean;\n      startedAt: number;\n    },\n    body: () => Promise<{ stopReason: string; errorMessage?: string }>,\n  ): Promise<{ stopReason: string; errorMessage?: string } | undefined> {\n    const { conversationId, sessionKey, isSyntheticEvent, startedAt } = meta;\n    const { message, platform } = adapters;\n\n    Sentry.metrics.count(\"agent.run.started\", 1, {\n      attributes: { channel: conversationId },\n    });\n\n    return Sentry.startSpan(\n      { name: \"agent.run\", op: \"agent\", attributes: { conversationId, sessionKey } },\n      async () =>\n        Sentry.withScope(async (scope) => {\n          applyRunScope(scope, {\n            conversationId,\n            sessionKey,\n            messageId: message.id,\n            platform: platform.name,\n            userId: message.userId,\n            userName: message.userName,\n            threadTs: message.threadTs,\n            isSyntheticEvent,\n          });\n          addLifecycleBreadcrumb(\"agent.run.started\", {\n            channel_id: conversationId,\n            platform: platform.name,\n            has_attachments: (message.attachments?.length ?? 0) > 0,\n          });\n\n          try {\n            const result = await body();\n            const durationMs = Date.now() - startedAt;\n            const completionAttrs = {\n              channel: conversationId,\n              platform: platform.name,\n              stop_reason: result.stopReason,\n            };\n            Sentry.metrics.distribution(\"agent.run.duration\", durationMs, {\n              unit: \"millisecond\",\n              attributes: completionAttrs,\n            });\n            Sentry.metrics.count(\"agent.run.completed\", 1, { attributes: completionAttrs });\n            addLifecycleBreadcrumb(\"agent.run.completed\", {\n              channel_id: conversationId,\n              platform: platform.name,\n              stop_reason: result.stopReason,\n              duration_ms: durationMs,\n            });\n            return result;\n          } catch (err) {\n            scope.setContext(\"agent_run_error\", {\n              conversationId,\n              sessionKey,\n              platform: platform.name,\n              messageId: message.id,\n              threadTs: message.threadTs,\n            });\n            Sentry.captureException(err);\n            Sentry.metrics.count(\"agent.run.errors\", 1, {\n              attributes: { channel: conversationId, platform: platform.name },\n            });\n            log.logWarning(\n              `[${conversationId}] Run error`,\n              err instanceof Error ? err.message : String(err),\n            );\n            return undefined;\n          }\n        }),\n    );\n  }\n}\n"]}

package/dist/runtime/index.d.ts

@@ -0,0 +1,2 @@
+export { createSessionRuntime, type CreateSessionSandboxOptions, type RunSessionOptions, type SessionRuntime, type SessionRuntimeOptions, } from "./session-runtime.js";
+//# sourceMappingURL=index.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/runtime/index.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,oBAAoB,EACpB,KAAK,2BAA2B,EAChC,KAAK,iBAAiB,EACtB,KAAK,cAAc,EACnB,KAAK,qBAAqB,GAC3B,MAAM,sBAAsB,CAAC","sourcesContent":["export {\n  createSessionRuntime,\n  type CreateSessionSandboxOptions,\n  type RunSessionOptions,\n  type SessionRuntime,\n  type SessionRuntimeOptions,\n} from \"./session-runtime.js\";\n"]}

package/dist/runtime/index.js

@@ -0,0 +1,2 @@
+export { createSessionRuntime, } from "./session-runtime.js";
+//# sourceMappingURL=index.js.map

package/dist/runtime/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/runtime/index.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,oBAAoB,GAKrB,MAAM,sBAAsB,CAAC","sourcesContent":["export {\n  createSessionRuntime,\n  type CreateSessionSandboxOptions,\n  type RunSessionOptions,\n  type SessionRuntime,\n  type SessionRuntimeOptions,\n} from \"./session-runtime.js\";\n"]}

package/dist/runtime/session-runtime.d.ts

@@ -0,0 +1,27 @@
+import type { Bot, BotAdapters, BotEvent, BotHandler } from "../adapter.js";
+import { type AgentRunner } from "../agent.js";
+import { CommandRegistry } from "../commands/index.js";
+import type { CommandServices } from "../commands/index.js";
+export interface RunSessionOptions {
+    event: BotEvent;
+    bot: Bot;
+    adapters: BotAdapters;
+    isSyntheticEvent?: boolean;
+}
+export interface CreateSessionSandboxOptions {
+    conversationId: string;
+    platformName: string;
+    sessionKey: string;
+}
+export interface SessionRuntimeOptions extends CommandServices {
+    /** Override the default command registry (e.g., to add /help, /status). */
+    commandRegistry?: CommandRegistry;
+}
+export interface SessionRuntime extends BotHandler {
+    runSession(options: RunSessionOptions): Promise<void>;
+    createSessionSandbox(options: CreateSessionSandboxOptions): Promise<AgentRunner>;
+    switchConversationModel(conversationId: string, provider: string, model: string): boolean;
+    shutdown(timeoutMs?: number): Promise<void>;
+}
+export declare function createSessionRuntime(options: SessionRuntimeOptions): SessionRuntime;
+//# sourceMappingURL=session-runtime.d.ts.map

package/dist/runtime/session-runtime.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"session-runtime.d.ts","sourceRoot":"","sources":["../../src/runtime/session-runtime.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,GAAG,EAAE,WAAW,EAAE,QAAQ,EAAE,UAAU,EAAkB,MAAM,eAAe,CAAC;AAE5F,OAAO,EAAE,KAAK,WAAW,EAAgB,MAAM,aAAa,CAAC;AAC7D,OAAO,EAAE,eAAe,EAAgC,MAAM,sBAAsB,CAAC;AACrF,OAAO,KAAK,EAAE,eAAe,EAAE,MAAM,sBAAsB,CAAC;AAqB5D,MAAM,WAAW,iBAAiB;IAChC,KAAK,EAAE,QAAQ,CAAC;IAChB,GAAG,EAAE,GAAG,CAAC;IACT,QAAQ,EAAE,WAAW,CAAC;IACtB,gBAAgB,CAAC,EAAE,OAAO,CAAC;CAC5B;AAED,MAAM,WAAW,2BAA2B;IAC1C,cAAc,EAAE,MAAM,CAAC;IACvB,YAAY,EAAE,MAAM,CAAC;IACrB,UAAU,EAAE,MAAM,CAAC;CACpB;AAED,MAAM,WAAW,qBAAsB,SAAQ,eAAe;IAC5D,2EAA2E;IAC3E,eAAe,CAAC,EAAE,eAAe,CAAC;CACnC;AAED,MAAM,WAAW,cAAe,SAAQ,UAAU;IAChD,UAAU,CAAC,OAAO,EAAE,iBAAiB,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;IACtD,oBAAoB,CAAC,OAAO,EAAE,2BAA2B,GAAG,OAAO,CAAC,WAAW,CAAC,CAAC;IACjF,uBAAuB,CAAC,cAAc,EAAE,MAAM,EAAE,QAAQ,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,GAAG,OAAO,CAAC;IAC1F,QAAQ,CAAC,SAAS,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;CAC7C;AAiBD,wBAAgB,oBAAoB,CAAC,OAAO,EAAE,qBAAqB,GAAG,cAAc,CAEnF","sourcesContent":["import type { Bot, BotAdapters, BotEvent, BotHandler, RunningSession } from \"../adapter.js\";\nimport { resolveSlackSessionScope } from \"../adapters/slack/branch-manager.js\";\nimport { type AgentRunner, createRunner } from \"../agent.js\";\nimport { CommandRegistry, createDefaultCommandRegistry } from \"../commands/index.js\";\nimport type { CommandServices } from \"../commands/index.js\";\nimport * as log from \"../log.js\";\nimport {\n  createManagedSessionFile,\n  createManagedSessionFileAtPath,\n  getChannelSessionDir,\n  getThreadSessionFile,\n  resolveGenericSessionScope,\n  type ResolvedSessionScope,\n} from \"../session-store.js\";\nimport { formatNothingRunning, formatStopping } from \"../ui-copy.js\";\nimport {\n  ConversationOrchestrator,\n  type ConversationRuntimeState,\n} from \"./conversation-orchestrator.js\";\nimport * as Sentry from \"@sentry/node\";\nimport { join } from \"path\";\nimport { getUnresolvedSandboxPathContext } from \"../agent.js\";\n\ntype ConversationState = ConversationRuntimeState;\n\nexport interface RunSessionOptions {\n  event: BotEvent;\n  bot: Bot;\n  adapters: BotAdapters;\n  isSyntheticEvent?: boolean;\n}\n\nexport interface CreateSessionSandboxOptions {\n  conversationId: string;\n  platformName: string;\n  sessionKey: string;\n}\n\nexport interface SessionRuntimeOptions extends CommandServices {\n  /** Override the default command registry (e.g., to add /help, /status). */\n  commandRegistry?: CommandRegistry;\n}\n\nexport interface SessionRuntime extends BotHandler {\n  runSession(options: RunSessionOptions): Promise<void>;\n  createSessionSandbox(options: CreateSessionSandboxOptions): Promise<AgentRunner>;\n  switchConversationModel(conversationId: string, provider: string, model: string): boolean;\n  shutdown(timeoutMs?: number): Promise<void>;\n}\n\nconst MAX_SESSIONS = 500;\nconst IDLE_TIMEOUT_MS = 3_600_000;\n\nfunction runtimeCwdForSandbox(\n  sandbox: SessionRuntimeOptions[\"sandbox\"],\n  hostWorkspaceRoot: string,\n  conversationId: string,\n): string {\n  const runtimeWorkspaceRoot = getUnresolvedSandboxPathContext(\n    sandbox,\n    hostWorkspaceRoot,\n  ).runtimeWorkspaceRoot;\n  return `${runtimeWorkspaceRoot.replace(/\\/+$/, \"\")}/${conversationId}`;\n}\n\nexport function createSessionRuntime(options: SessionRuntimeOptions): SessionRuntime {\n  return new MamaSessionRuntime(options);\n}\n\nclass MamaSessionRuntime implements SessionRuntime {\n  private readonly conversationStates = new Map<string, ConversationState>();\n  private readonly sessionQueues = new Map<string, Promise<void>>();\n  private readonly inFlightRuns = new Set<Promise<void>>();\n  private readonly commandRegistry: CommandRegistry;\n  private readonly orchestrator: ConversationOrchestrator;\n  private isShuttingDown = false;\n\n  constructor(private readonly options: SessionRuntimeOptions) {\n    this.options.runtime = this;\n    this.commandRegistry = options.commandRegistry ?? createDefaultCommandRegistry();\n    this.orchestrator = new ConversationOrchestrator({\n      workingDir: options.workingDir,\n      commandRegistry: this.commandRegistry,\n      commandServices: this.options,\n      isShuttingDown: () => this.isShuttingDown,\n      getState: (sessionKey) => this.conversationStates.get(sessionKey),\n      getOrCreateState: (createOptions) => this.getOrCreateState(createOptions),\n      beforeRunTracked: (runPromise) => {\n        this.inFlightRuns.add(runPromise);\n        Sentry.metrics.gauge(\"agent.sessions.active\", this.inFlightRuns.size);\n      },\n      afterRunTracked: (runPromise) => {\n        this.inFlightRuns.delete(runPromise);\n      },\n      onRunFinished: () => {\n        Sentry.metrics.gauge(\"agent.sessions.active\", this.inFlightRuns.size - 1);\n        this.evictIdleSessions();\n      },\n    });\n  }\n\n  isRunning(sessionKey: string): boolean {\n    const state = this.conversationStates.get(sessionKey);\n    return !!state?.running;\n  }\n\n  getRunningSessions(): RunningSession[] {\n    const sessions: RunningSession[] = [];\n    for (const [sessionKey, state] of this.conversationStates) {\n      if (state.running && state.startedAt) {\n        const currentStep = state.runner.getCurrentStep();\n        sessions.push({\n          sessionKey,\n          startedAt: state.startedAt,\n          lastActivityAt: state.lastActivityAt,\n          currentTool: currentStep?.label || currentStep?.toolName,\n        });\n      }\n    }\n    return sessions;\n  }\n\n  async handleStop(sessionKey: string, conversationId: string, bot: Bot): Promise<void> {\n    const state = this.conversationStates.get(sessionKey);\n    if (state?.running) {\n      state.stopRequested = true;\n      state.runner.abort();\n      const ts = await bot.postMessage(conversationId, formatStopping(bot));\n      state.stopMessageTs = ts;\n    } else {\n      await bot.postMessage(conversationId, formatNothingRunning(bot));\n    }\n  }\n\n  forceStop(sessionKey: string): void {\n    const state = this.conversationStates.get(sessionKey);\n    if (state?.running) {\n      log.logInfo(`[Force Stop] Force stopping session: ${sessionKey}`);\n      state.stopRequested = true;\n      state.runner.abort();\n      state.running = false;\n    }\n  }\n\n  async handleNewCommand(sessionKey: string, conversationId: string, bot: Bot): Promise<void> {\n    const state = this.conversationStates.get(sessionKey);\n    if (state?.running) {\n      state.stopRequested = true;\n      state.runner.abort();\n    }\n\n    const conversationDir = join(this.options.workingDir, conversationId);\n    const runtimeCwd = runtimeCwdForSandbox(\n      this.options.sandbox,\n      this.options.workingDir,\n      conversationId,\n    );\n    if (sessionKey.includes(\":\")) {\n      createManagedSessionFileAtPath(getThreadSessionFile(conversationDir, sessionKey), runtimeCwd);\n    } else {\n      createManagedSessionFile(getChannelSessionDir(conversationDir), runtimeCwd);\n    }\n\n    this.conversationStates.delete(sessionKey);\n\n    log.logInfo(`[${conversationId}] Session reset: ${sessionKey}`);\n    await bot.postMessage(conversationId, \"Conversation reset. Send a new message to start fresh.\");\n  }\n\n  async handleEvent(\n    event: BotEvent,\n    bot: Bot,\n    adapters: BotAdapters,\n    isSyntheticEvent?: boolean,\n  ): Promise<void> {\n    const sessionKey = event.sessionKey ?? `${event.conversationId}:${event.thread_ts ?? event.ts}`;\n    const previous = this.sessionQueues.get(sessionKey) ?? Promise.resolve();\n    const next = previous\n      .catch(() => {})\n      .then(() => this.runSession({ event, bot, adapters, isSyntheticEvent }));\n    this.sessionQueues.set(sessionKey, next);\n    try {\n      await next;\n    } finally {\n      if (this.sessionQueues.get(sessionKey) === next) {\n        this.sessionQueues.delete(sessionKey);\n      }\n    }\n  }\n\n  async runSession({ event, bot, adapters, isSyntheticEvent }: RunSessionOptions): Promise<void> {\n    await this.orchestrator.runSession({ event, bot, adapters, isSyntheticEvent });\n  }\n\n  async createSessionSandbox(options: CreateSessionSandboxOptions): Promise<AgentRunner> {\n    const state = await this.getOrCreateState(options);\n    return state.runner;\n  }\n\n  switchConversationModel(conversationId: string, _provider: string, _model: string): boolean {\n    for (const [sessionKey, state] of this.conversationStates) {\n      if (this.isConversationSession(sessionKey, conversationId) && state.running) {\n        return false;\n      }\n    }\n\n    for (const sessionKey of Array.from(this.conversationStates.keys())) {\n      if (this.isConversationSession(sessionKey, conversationId)) {\n        this.conversationStates.delete(sessionKey);\n      }\n    }\n    log.logInfo(`[${conversationId}] Model switched; cleared cached session runners`);\n    return true;\n  }\n\n  private isConversationSession(sessionKey: string, conversationId: string): boolean {\n    return sessionKey === conversationId || sessionKey.startsWith(`${conversationId}:`);\n  }\n\n  private async getOrCreateState({\n    conversationId,\n    platformName,\n    sessionKey,\n  }: CreateSessionSandboxOptions): Promise<ConversationState> {\n    const existing = this.conversationStates.get(sessionKey);\n    if (existing) {\n      existing.lastAccessedAt = Date.now();\n      return existing;\n    }\n\n    const conversationDir = join(this.options.workingDir, conversationId);\n    const runtimeCwd = runtimeCwdForSandbox(\n      this.options.sandbox,\n      this.options.workingDir,\n      conversationId,\n    );\n    const sessionScope = await this.resolveSessionScope(\n      platformName,\n      conversationDir,\n      sessionKey,\n      runtimeCwd,\n    );\n    const state: ConversationState = {\n      running: false,\n      runner: await createRunner(\n        this.options.sandbox,\n        sessionKey,\n        conversationId,\n        conversationDir,\n        this.options.workingDir,\n        sessionScope,\n        this.options.vaultManager,\n        this.options.provisioner,\n      ),\n      stopRequested: false,\n      lastAccessedAt: Date.now(),\n    };\n    this.conversationStates.set(sessionKey, state);\n    return state;\n  }\n\n  async shutdown(timeoutMs = 30_000): Promise<void> {\n    if (this.isShuttingDown) return;\n    this.isShuttingDown = true;\n    log.logInfo(\"Shutting down gracefully...\");\n\n    const timeout = Date.now() + timeoutMs;\n    while (this.inFlightRuns.size > 0 && Date.now() < timeout) {\n      await new Promise((resolve) => setTimeout(resolve, 500));\n    }\n\n    if (this.inFlightRuns.size > 0) {\n      log.logWarning(`Forcing exit with ${this.inFlightRuns.size} runs still in progress`);\n    }\n  }\n\n  private async resolveSessionScope(\n    platformName: string,\n    conversationDir: string,\n    sessionKey: string,\n    cwd: string,\n  ): Promise<ResolvedSessionScope> {\n    if (sessionKey.includes(\":event:\")) {\n      const sessionDir = getChannelSessionDir(conversationDir);\n      const contextFile = createManagedSessionFileAtPath(\n        getThreadSessionFile(conversationDir, sessionKey),\n        cwd,\n      );\n      return { sessionDir, contextFile, threadRootMessage: null };\n    }\n\n    if (platformName === \"slack\") {\n      return resolveSlackSessionScope({ conversationDir, sessionKey, cwd });\n    }\n    return resolveGenericSessionScope({ conversationDir, sessionKey, cwd });\n  }\n\n  private evictIdleSessions(): void {\n    const now = Date.now();\n\n    for (const [key, state] of this.conversationStates) {\n      if (!state.running && now - state.lastAccessedAt > IDLE_TIMEOUT_MS) {\n        this.conversationStates.delete(key);\n      }\n    }\n\n    if (this.conversationStates.size > MAX_SESSIONS) {\n      const idleSessions: Array<{ key: string; lastAccessedAt: number }> = [];\n      for (const [key, state] of this.conversationStates) {\n        if (!state.running) {\n          idleSessions.push({ key, lastAccessedAt: state.lastAccessedAt });\n        }\n      }\n\n      idleSessions.sort((a, b) => a.lastAccessedAt - b.lastAccessedAt);\n\n      const toEvict = this.conversationStates.size - MAX_SESSIONS;\n      for (let i = 0; i < toEvict && i < idleSessions.length; i++) {\n        this.conversationStates.delete(idleSessions[i].key);\n      }\n    }\n  }\n}\n"]}

package/dist/runtime/session-runtime.js

@@ -0,0 +1,211 @@
+import { resolveSlackSessionScope } from "../adapters/slack/branch-manager.js";
+import { createRunner } from "../agent.js";
+import { createDefaultCommandRegistry } from "../commands/index.js";
+import * as log from "../log.js";
+import { createManagedSessionFile, createManagedSessionFileAtPath, getChannelSessionDir, getThreadSessionFile, resolveGenericSessionScope, } from "../session-store.js";
+import { formatNothingRunning, formatStopping } from "../ui-copy.js";
+import { ConversationOrchestrator, } from "./conversation-orchestrator.js";
+import * as Sentry from "@sentry/node";
+import { join } from "path";
+import { getUnresolvedSandboxPathContext } from "../agent.js";
+const MAX_SESSIONS = 500;
+const IDLE_TIMEOUT_MS = 3_600_000;
+function runtimeCwdForSandbox(sandbox, hostWorkspaceRoot, conversationId) {
+    const runtimeWorkspaceRoot = getUnresolvedSandboxPathContext(sandbox, hostWorkspaceRoot).runtimeWorkspaceRoot;
+    return `${runtimeWorkspaceRoot.replace(/\/+$/, "")}/${conversationId}`;
+}
+export function createSessionRuntime(options) {
+    return new MamaSessionRuntime(options);
+}
+class MamaSessionRuntime {
+    constructor(options) {
+        this.options = options;
+        this.conversationStates = new Map();
+        this.sessionQueues = new Map();
+        this.inFlightRuns = new Set();
+        this.isShuttingDown = false;
+        this.options.runtime = this;
+        this.commandRegistry = options.commandRegistry ?? createDefaultCommandRegistry();
+        this.orchestrator = new ConversationOrchestrator({
+            workingDir: options.workingDir,
+            commandRegistry: this.commandRegistry,
+            commandServices: this.options,
+            isShuttingDown: () => this.isShuttingDown,
+            getState: (sessionKey) => this.conversationStates.get(sessionKey),
+            getOrCreateState: (createOptions) => this.getOrCreateState(createOptions),
+            beforeRunTracked: (runPromise) => {
+                this.inFlightRuns.add(runPromise);
+                Sentry.metrics.gauge("agent.sessions.active", this.inFlightRuns.size);
+            },
+            afterRunTracked: (runPromise) => {
+                this.inFlightRuns.delete(runPromise);
+            },
+            onRunFinished: () => {
+                Sentry.metrics.gauge("agent.sessions.active", this.inFlightRuns.size - 1);
+                this.evictIdleSessions();
+            },
+        });
+    }
+    isRunning(sessionKey) {
+        const state = this.conversationStates.get(sessionKey);
+        return !!state?.running;
+    }
+    getRunningSessions() {
+        const sessions = [];
+        for (const [sessionKey, state] of this.conversationStates) {
+            if (state.running && state.startedAt) {
+                const currentStep = state.runner.getCurrentStep();
+                sessions.push({
+                    sessionKey,
+                    startedAt: state.startedAt,
+                    lastActivityAt: state.lastActivityAt,
+                    currentTool: currentStep?.label || currentStep?.toolName,
+                });
+            }
+        }
+        return sessions;
+    }
+    async handleStop(sessionKey, conversationId, bot) {
+        const state = this.conversationStates.get(sessionKey);
+        if (state?.running) {
+            state.stopRequested = true;
+            state.runner.abort();
+            const ts = await bot.postMessage(conversationId, formatStopping(bot));
+            state.stopMessageTs = ts;
+        }
+        else {
+            await bot.postMessage(conversationId, formatNothingRunning(bot));
+        }
+    }
+    forceStop(sessionKey) {
+        const state = this.conversationStates.get(sessionKey);
+        if (state?.running) {
+            log.logInfo(`[Force Stop] Force stopping session: ${sessionKey}`);
+            state.stopRequested = true;
+            state.runner.abort();
+            state.running = false;
+        }
+    }
+    async handleNewCommand(sessionKey, conversationId, bot) {
+        const state = this.conversationStates.get(sessionKey);
+        if (state?.running) {
+            state.stopRequested = true;
+            state.runner.abort();
+        }
+        const conversationDir = join(this.options.workingDir, conversationId);
+        const runtimeCwd = runtimeCwdForSandbox(this.options.sandbox, this.options.workingDir, conversationId);
+        if (sessionKey.includes(":")) {
+            createManagedSessionFileAtPath(getThreadSessionFile(conversationDir, sessionKey), runtimeCwd);
+        }
+        else {
+            createManagedSessionFile(getChannelSessionDir(conversationDir), runtimeCwd);
+        }
+        this.conversationStates.delete(sessionKey);
+        log.logInfo(`[${conversationId}] Session reset: ${sessionKey}`);
+        await bot.postMessage(conversationId, "Conversation reset. Send a new message to start fresh.");
+    }
+    async handleEvent(event, bot, adapters, isSyntheticEvent) {
+        const sessionKey = event.sessionKey ?? `${event.conversationId}:${event.thread_ts ?? event.ts}`;
+        const previous = this.sessionQueues.get(sessionKey) ?? Promise.resolve();
+        const next = previous
+            .catch(() => { })
+            .then(() => this.runSession({ event, bot, adapters, isSyntheticEvent }));
+        this.sessionQueues.set(sessionKey, next);
+        try {
+            await next;
+        }
+        finally {
+            if (this.sessionQueues.get(sessionKey) === next) {
+                this.sessionQueues.delete(sessionKey);
+            }
+        }
+    }
+    async runSession({ event, bot, adapters, isSyntheticEvent }) {
+        await this.orchestrator.runSession({ event, bot, adapters, isSyntheticEvent });
+    }
+    async createSessionSandbox(options) {
+        const state = await this.getOrCreateState(options);
+        return state.runner;
+    }
+    switchConversationModel(conversationId, _provider, _model) {
+        for (const [sessionKey, state] of this.conversationStates) {
+            if (this.isConversationSession(sessionKey, conversationId) && state.running) {
+                return false;
+            }
+        }
+        for (const sessionKey of Array.from(this.conversationStates.keys())) {
+            if (this.isConversationSession(sessionKey, conversationId)) {
+                this.conversationStates.delete(sessionKey);
+            }
+        }
+        log.logInfo(`[${conversationId}] Model switched; cleared cached session runners`);
+        return true;
+    }
+    isConversationSession(sessionKey, conversationId) {
+        return sessionKey === conversationId || sessionKey.startsWith(`${conversationId}:`);
+    }
+    async getOrCreateState({ conversationId, platformName, sessionKey, }) {
+        const existing = this.conversationStates.get(sessionKey);
+        if (existing) {
+            existing.lastAccessedAt = Date.now();
+            return existing;
+        }
+        const conversationDir = join(this.options.workingDir, conversationId);
+        const runtimeCwd = runtimeCwdForSandbox(this.options.sandbox, this.options.workingDir, conversationId);
+        const sessionScope = await this.resolveSessionScope(platformName, conversationDir, sessionKey, runtimeCwd);
+        const state = {
+            running: false,
+            runner: await createRunner(this.options.sandbox, sessionKey, conversationId, conversationDir, this.options.workingDir, sessionScope, this.options.vaultManager, this.options.provisioner),
+            stopRequested: false,
+            lastAccessedAt: Date.now(),
+        };
+        this.conversationStates.set(sessionKey, state);
+        return state;
+    }
+    async shutdown(timeoutMs = 30_000) {
+        if (this.isShuttingDown)
+            return;
+        this.isShuttingDown = true;
+        log.logInfo("Shutting down gracefully...");
+        const timeout = Date.now() + timeoutMs;
+        while (this.inFlightRuns.size > 0 && Date.now() < timeout) {
+            await new Promise((resolve) => setTimeout(resolve, 500));
+        }
+        if (this.inFlightRuns.size > 0) {
+            log.logWarning(`Forcing exit with ${this.inFlightRuns.size} runs still in progress`);
+        }
+    }
+    async resolveSessionScope(platformName, conversationDir, sessionKey, cwd) {
+        if (sessionKey.includes(":event:")) {
+            const sessionDir = getChannelSessionDir(conversationDir);
+            const contextFile = createManagedSessionFileAtPath(getThreadSessionFile(conversationDir, sessionKey), cwd);
+            return { sessionDir, contextFile, threadRootMessage: null };
+        }
+        if (platformName === "slack") {
+            return resolveSlackSessionScope({ conversationDir, sessionKey, cwd });
+        }
+        return resolveGenericSessionScope({ conversationDir, sessionKey, cwd });
+    }
+    evictIdleSessions() {
+        const now = Date.now();
+        for (const [key, state] of this.conversationStates) {
+            if (!state.running && now - state.lastAccessedAt > IDLE_TIMEOUT_MS) {
+                this.conversationStates.delete(key);
+            }
+        }
+        if (this.conversationStates.size > MAX_SESSIONS) {
+            const idleSessions = [];
+            for (const [key, state] of this.conversationStates) {
+                if (!state.running) {
+                    idleSessions.push({ key, lastAccessedAt: state.lastAccessedAt });
+                }
+            }
+            idleSessions.sort((a, b) => a.lastAccessedAt - b.lastAccessedAt);
+            const toEvict = this.conversationStates.size - MAX_SESSIONS;
+            for (let i = 0; i < toEvict && i < idleSessions.length; i++) {
+                this.conversationStates.delete(idleSessions[i].key);
+            }
+        }
+    }
+}
+//# sourceMappingURL=session-runtime.js.map

package/dist/runtime/session-runtime.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"session-runtime.js","sourceRoot":"","sources":["../../src/runtime/session-runtime.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,wBAAwB,EAAE,MAAM,qCAAqC,CAAC;AAC/E,OAAO,EAAoB,YAAY,EAAE,MAAM,aAAa,CAAC;AAC7D,OAAO,EAAmB,4BAA4B,EAAE,MAAM,sBAAsB,CAAC;AAErF,OAAO,KAAK,GAAG,MAAM,WAAW,CAAC;AACjC,OAAO,EACL,wBAAwB,EACxB,8BAA8B,EAC9B,oBAAoB,EACpB,oBAAoB,EACpB,0BAA0B,GAE3B,MAAM,qBAAqB,CAAC;AAC7B,OAAO,EAAE,oBAAoB,EAAE,cAAc,EAAE,MAAM,eAAe,CAAC;AACrE,OAAO,EACL,wBAAwB,GAEzB,MAAM,gCAAgC,CAAC;AACxC,OAAO,KAAK,MAAM,MAAM,cAAc,CAAC;AACvC,OAAO,EAAE,IAAI,EAAE,MAAM,MAAM,CAAC;AAC5B,OAAO,EAAE,+BAA+B,EAAE,MAAM,aAAa,CAAC;AA6B9D,MAAM,YAAY,GAAG,GAAG,CAAC;AACzB,MAAM,eAAe,GAAG,SAAS,CAAC;AAElC,SAAS,oBAAoB,CAC3B,OAAyC,EACzC,iBAAyB,EACzB,cAAsB;IAEtB,MAAM,oBAAoB,GAAG,+BAA+B,CAC1D,OAAO,EACP,iBAAiB,CAClB,CAAC,oBAAoB,CAAC;IACvB,OAAO,GAAG,oBAAoB,CAAC,OAAO,CAAC,MAAM,EAAE,EAAE,CAAC,IAAI,cAAc,EAAE,CAAC;AACzE,CAAC;AAED,MAAM,UAAU,oBAAoB,CAAC,OAA8B;IACjE,OAAO,IAAI,kBAAkB,CAAC,OAAO,CAAC,CAAC;AACzC,CAAC;AAED,MAAM,kBAAkB;IAQtB,YAA6B,OAA8B;QAA9B,YAAO,GAAP,OAAO,CAAuB;QAP1C,uBAAkB,GAAG,IAAI,GAAG,EAA6B,CAAC;QAC1D,kBAAa,GAAG,IAAI,GAAG,EAAyB,CAAC;QACjD,iBAAY,GAAG,IAAI,GAAG,EAAiB,CAAC;QAGjD,mBAAc,GAAG,KAAK,CAAC;QAG7B,IAAI,CAAC,OAAO,CAAC,OAAO,GAAG,IAAI,CAAC;QAC5B,IAAI,CAAC,eAAe,GAAG,OAAO,CAAC,eAAe,IAAI,4BAA4B,EAAE,CAAC;QACjF,IAAI,CAAC,YAAY,GAAG,IAAI,wBAAwB,CAAC;YAC/C,UAAU,EAAE,OAAO,CAAC,UAAU;YAC9B,eAAe,EAAE,IAAI,CAAC,eAAe;YACrC,eAAe,EAAE,IAAI,CAAC,OAAO;YAC7B,cAAc,EAAE,GAAG,EAAE,CAAC,IAAI,CAAC,cAAc;YACzC,QAAQ,EAAE,CAAC,UAAU,EAAE,EAAE,CAAC,IAAI,CAAC,kBAAkB,CAAC,GAAG,CAAC,UAAU,CAAC;YACjE,gBAAgB,EAAE,CAAC,aAAa,EAAE,EAAE,CAAC,IAAI,CAAC,gBAAgB,CAAC,aAAa,CAAC;YACzE,gBAAgB,EAAE,CAAC,UAAU,EAAE,EAAE;gBAC/B,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;gBAClC,MAAM,CAAC,OAAO,CAAC,KAAK,CAAC,uBAAuB,EAAE,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,CAAC;YACxE,CAAC;YACD,eAAe,EAAE,CAAC,UAAU,EAAE,EAAE;gBAC9B,IAAI,CAAC,YAAY,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC;YACvC,CAAC;YACD,aAAa,EAAE,GAAG,EAAE;gBAClB,MAAM,CAAC,OAAO,CAAC,KAAK,CAAC,uBAAuB,EAAE,IAAI,CAAC,YAAY,CAAC,IAAI,GAAG,CAAC,CAAC,CAAC;gBAC1E,IAAI,CAAC,iBAAiB,EAAE,CAAC;YAC3B,CAAC;SACF,CAAC,CAAC;IACL,CAAC;IAED,SAAS,CAAC,UAAkB;QAC1B,MAAM,KAAK,GAAG,IAAI,CAAC,kBAAkB,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;QACtD,OAAO,CAAC,CAAC,KAAK,EAAE,OAAO,CAAC;IAC1B,CAAC;IAED,kBAAkB;QAChB,MAAM,QAAQ,GAAqB,EAAE,CAAC;QACtC,KAAK,MAAM,CAAC,UAAU,EAAE,KAAK,CAAC,IAAI,IAAI,CAAC,kBAAkB,EAAE,CAAC;YAC1D,IAAI,KAAK,CAAC,OAAO,IAAI,KAAK,CAAC,SAAS,EAAE,CAAC;gBACrC,MAAM,WAAW,GAAG,KAAK,CAAC,MAAM,CAAC,cAAc,EAAE,CAAC;gBAClD,QAAQ,CAAC,IAAI,CAAC;oBACZ,UAAU;oBACV,SAAS,EAAE,KAAK,CAAC,SAAS;oBAC1B,cAAc,EAAE,KAAK,CAAC,cAAc;oBACpC,WAAW,EAAE,WAAW,EAAE,KAAK,IAAI,WAAW,EAAE,QAAQ;iBACzD,CAAC,CAAC;YACL,CAAC;QACH,CAAC;QACD,OAAO,QAAQ,CAAC;IAClB,CAAC;IAED,KAAK,CAAC,UAAU,CAAC,UAAkB,EAAE,cAAsB,EAAE,GAAQ;QACnE,MAAM,KAAK,GAAG,IAAI,CAAC,kBAAkB,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;QACtD,IAAI,KAAK,EAAE,OAAO,EAAE,CAAC;YACnB,KAAK,CAAC,aAAa,GAAG,IAAI,CAAC;YAC3B,KAAK,CAAC,MAAM,CAAC,KAAK,EAAE,CAAC;YACrB,MAAM,EAAE,GAAG,MAAM,GAAG,CAAC,WAAW,CAAC,cAAc,EAAE,cAAc,CAAC,GAAG,CAAC,CAAC,CAAC;YACtE,KAAK,CAAC,aAAa,GAAG,EAAE,CAAC;QAC3B,CAAC;aAAM,CAAC;YACN,MAAM,GAAG,CAAC,WAAW,CAAC,cAAc,EAAE,oBAAoB,CAAC,GAAG,CAAC,CAAC,CAAC;QACnE,CAAC;IACH,CAAC;IAED,SAAS,CAAC,UAAkB;QAC1B,MAAM,KAAK,GAAG,IAAI,CAAC,kBAAkB,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;QACtD,IAAI,KAAK,EAAE,OAAO,EAAE,CAAC;YACnB,GAAG,CAAC,OAAO,CAAC,wCAAwC,UAAU,EAAE,CAAC,CAAC;YAClE,KAAK,CAAC,aAAa,GAAG,IAAI,CAAC;YAC3B,KAAK,CAAC,MAAM,CAAC,KAAK,EAAE,CAAC;YACrB,KAAK,CAAC,OAAO,GAAG,KAAK,CAAC;QACxB,CAAC;IACH,CAAC;IAED,KAAK,CAAC,gBAAgB,CAAC,UAAkB,EAAE,cAAsB,EAAE,GAAQ;QACzE,MAAM,KAAK,GAAG,IAAI,CAAC,kBAAkB,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;QACtD,IAAI,KAAK,EAAE,OAAO,EAAE,CAAC;YACnB,KAAK,CAAC,aAAa,GAAG,IAAI,CAAC;YAC3B,KAAK,CAAC,MAAM,CAAC,KAAK,EAAE,CAAC;QACvB,CAAC;QAED,MAAM,eAAe,GAAG,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,UAAU,EAAE,cAAc,CAAC,CAAC;QACtE,MAAM,UAAU,GAAG,oBAAoB,CACrC,IAAI,CAAC,OAAO,CAAC,OAAO,EACpB,IAAI,CAAC,OAAO,CAAC,UAAU,EACvB,cAAc,CACf,CAAC;QACF,IAAI,UAAU,CAAC,QAAQ,CAAC,GAAG,CAAC,EAAE,CAAC;YAC7B,8BAA8B,CAAC,oBAAoB,CAAC,eAAe,EAAE,UAAU,CAAC,EAAE,UAAU,CAAC,CAAC;QAChG,CAAC;aAAM,CAAC;YACN,wBAAwB,CAAC,oBAAoB,CAAC,eAAe,CAAC,EAAE,UAAU,CAAC,CAAC;QAC9E,CAAC;QAED,IAAI,CAAC,kBAAkB,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC;QAE3C,GAAG,CAAC,OAAO,CAAC,IAAI,cAAc,oBAAoB,UAAU,EAAE,CAAC,CAAC;QAChE,MAAM,GAAG,CAAC,WAAW,CAAC,cAAc,EAAE,wDAAwD,CAAC,CAAC;IAClG,CAAC;IAED,KAAK,CAAC,WAAW,CACf,KAAe,EACf,GAAQ,EACR,QAAqB,EACrB,gBAA0B;QAE1B,MAAM,UAAU,GAAG,KAAK,CAAC,UAAU,IAAI,GAAG,KAAK,CAAC,cAAc,IAAI,KAAK,CAAC,SAAS,IAAI,KAAK,CAAC,EAAE,EAAE,CAAC;QAChG,MAAM,QAAQ,GAAG,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,UAAU,CAAC,IAAI,OAAO,CAAC,OAAO,EAAE,CAAC;QACzE,MAAM,IAAI,GAAG,QAAQ;aAClB,KAAK,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC;aACf,IAAI,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC,UAAU,CAAC,EAAE,KAAK,EAAE,GAAG,EAAE,QAAQ,EAAE,gBAAgB,EAAE,CAAC,CAAC,CAAC;QAC3E,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,UAAU,EAAE,IAAI,CAAC,CAAC;QACzC,IAAI,CAAC;YACH,MAAM,IAAI,CAAC;QACb,CAAC;gBAAS,CAAC;YACT,IAAI,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,UAAU,CAAC,KAAK,IAAI,EAAE,CAAC;gBAChD,IAAI,CAAC,aAAa,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC;YACxC,CAAC;QACH,CAAC;IACH,CAAC;IAED,KAAK,CAAC,UAAU,CAAC,EAAE,KAAK,EAAE,GAAG,EAAE,QAAQ,EAAE,gBAAgB,EAAqB;QAC5E,MAAM,IAAI,CAAC,YAAY,CAAC,UAAU,CAAC,EAAE,KAAK,EAAE,GAAG,EAAE,QAAQ,EAAE,gBAAgB,EAAE,CAAC,CAAC;IACjF,CAAC;IAED,KAAK,CAAC,oBAAoB,CAAC,OAAoC;QAC7D,MAAM,KAAK,GAAG,MAAM,IAAI,CAAC,gBAAgB,CAAC,OAAO,CAAC,CAAC;QACnD,OAAO,KAAK,CAAC,MAAM,CAAC;IACtB,CAAC;IAED,uBAAuB,CAAC,cAAsB,EAAE,SAAiB,EAAE,MAAc;QAC/E,KAAK,MAAM,CAAC,UAAU,EAAE,KAAK,CAAC,IAAI,IAAI,CAAC,kBAAkB,EAAE,CAAC;YAC1D,IAAI,IAAI,CAAC,qBAAqB,CAAC,UAAU,EAAE,cAAc,CAAC,IAAI,KAAK,CAAC,OAAO,EAAE,CAAC;gBAC5E,OAAO,KAAK,CAAC;YACf,CAAC;QACH,CAAC;QAED,KAAK,MAAM,UAAU,IAAI,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,kBAAkB,CAAC,IAAI,EAAE,CAAC,EAAE,CAAC;YACpE,IAAI,IAAI,CAAC,qBAAqB,CAAC,UAAU,EAAE,cAAc,CAAC,EAAE,CAAC;gBAC3D,IAAI,CAAC,kBAAkB,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC;YAC7C,CAAC;QACH,CAAC;QACD,GAAG,CAAC,OAAO,CAAC,IAAI,cAAc,kDAAkD,CAAC,CAAC;QAClF,OAAO,IAAI,CAAC;IACd,CAAC;IAEO,qBAAqB,CAAC,UAAkB,EAAE,cAAsB;QACtE,OAAO,UAAU,KAAK,cAAc,IAAI,UAAU,CAAC,UAAU,CAAC,GAAG,cAAc,GAAG,CAAC,CAAC;IACtF,CAAC;IAEO,KAAK,CAAC,gBAAgB,CAAC,EAC7B,cAAc,EACd,YAAY,EACZ,UAAU,GACkB;QAC5B,MAAM,QAAQ,GAAG,IAAI,CAAC,kBAAkB,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;QACzD,IAAI,QAAQ,EAAE,CAAC;YACb,QAAQ,CAAC,cAAc,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;YACrC,OAAO,QAAQ,CAAC;QAClB,CAAC;QAED,MAAM,eAAe,GAAG,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,UAAU,EAAE,cAAc,CAAC,CAAC;QACtE,MAAM,UAAU,GAAG,oBAAoB,CACrC,IAAI,CAAC,OAAO,CAAC,OAAO,EACpB,IAAI,CAAC,OAAO,CAAC,UAAU,EACvB,cAAc,CACf,CAAC;QACF,MAAM,YAAY,GAAG,MAAM,IAAI,CAAC,mBAAmB,CACjD,YAAY,EACZ,eAAe,EACf,UAAU,EACV,UAAU,CACX,CAAC;QACF,MAAM,KAAK,GAAsB;YAC/B,OAAO,EAAE,KAAK;YACd,MAAM,EAAE,MAAM,YAAY,CACxB,IAAI,CAAC,OAAO,CAAC,OAAO,EACpB,UAAU,EACV,cAAc,EACd,eAAe,EACf,IAAI,CAAC,OAAO,CAAC,UAAU,EACvB,YAAY,EACZ,IAAI,CAAC,OAAO,CAAC,YAAY,EACzB,IAAI,CAAC,OAAO,CAAC,WAAW,CACzB;YACD,aAAa,EAAE,KAAK;YACpB,cAAc,EAAE,IAAI,CAAC,GAAG,EAAE;SAC3B,CAAC;QACF,IAAI,CAAC,kBAAkB,CAAC,GAAG,CAAC,UAAU,EAAE,KAAK,CAAC,CAAC;QAC/C,OAAO,KAAK,CAAC;IACf,CAAC;IAED,KAAK,CAAC,QAAQ,CAAC,SAAS,GAAG,MAAM;QAC/B,IAAI,IAAI,CAAC,cAAc;YAAE,OAAO;QAChC,IAAI,CAAC,cAAc,GAAG,IAAI,CAAC;QAC3B,GAAG,CAAC,OAAO,CAAC,6BAA6B,CAAC,CAAC;QAE3C,MAAM,OAAO,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,CAAC;QACvC,OAAO,IAAI,CAAC,YAAY,CAAC,IAAI,GAAG,CAAC,IAAI,IAAI,CAAC,GAAG,EAAE,GAAG,OAAO,EAAE,CAAC;YAC1D,MAAM,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,GAAG,CAAC,CAAC,CAAC;QAC3D,CAAC;QAED,IAAI,IAAI,CAAC,YAAY,CAAC,IAAI,GAAG,CAAC,EAAE,CAAC;YAC/B,GAAG,CAAC,UAAU,CAAC,qBAAqB,IAAI,CAAC,YAAY,CAAC,IAAI,yBAAyB,CAAC,CAAC;QACvF,CAAC;IACH,CAAC;IAEO,KAAK,CAAC,mBAAmB,CAC/B,YAAoB,EACpB,eAAuB,EACvB,UAAkB,EAClB,GAAW;QAEX,IAAI,UAAU,CAAC,QAAQ,CAAC,SAAS,CAAC,EAAE,CAAC;YACnC,MAAM,UAAU,GAAG,oBAAoB,CAAC,eAAe,CAAC,CAAC;YACzD,MAAM,WAAW,GAAG,8BAA8B,CAChD,oBAAoB,CAAC,eAAe,EAAE,UAAU,CAAC,EACjD,GAAG,CACJ,CAAC;YACF,OAAO,EAAE,UAAU,EAAE,WAAW,EAAE,iBAAiB,EAAE,IAAI,EAAE,CAAC;QAC9D,CAAC;QAED,IAAI,YAAY,KAAK,OAAO,EAAE,CAAC;YAC7B,OAAO,wBAAwB,CAAC,EAAE,eAAe,EAAE,UAAU,EAAE,GAAG,EAAE,CAAC,CAAC;QACxE,CAAC;QACD,OAAO,0BAA0B,CAAC,EAAE,eAAe,EAAE,UAAU,EAAE,GAAG,EAAE,CAAC,CAAC;IAC1E,CAAC;IAEO,iBAAiB;QACvB,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAEvB,KAAK,MAAM,CAAC,GAAG,EAAE,KAAK,CAAC,IAAI,IAAI,CAAC,kBAAkB,EAAE,CAAC;YACnD,IAAI,CAAC,KAAK,CAAC,OAAO,IAAI,GAAG,GAAG,KAAK,CAAC,cAAc,GAAG,eAAe,EAAE,CAAC;gBACnE,IAAI,CAAC,kBAAkB,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;YACtC,CAAC;QACH,CAAC;QAED,IAAI,IAAI,CAAC,kBAAkB,CAAC,IAAI,GAAG,YAAY,EAAE,CAAC;YAChD,MAAM,YAAY,GAAmD,EAAE,CAAC;YACxE,KAAK,MAAM,CAAC,GAAG,EAAE,KAAK,CAAC,IAAI,IAAI,CAAC,kBAAkB,EAAE,CAAC;gBACnD,IAAI,CAAC,KAAK,CAAC,OAAO,EAAE,CAAC;oBACnB,YAAY,CAAC,IAAI,CAAC,EAAE,GAAG,EAAE,cAAc,EAAE,KAAK,CAAC,cAAc,EAAE,CAAC,CAAC;gBACnE,CAAC;YACH,CAAC;YAED,YAAY,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,cAAc,GAAG,CAAC,CAAC,cAAc,CAAC,CAAC;YAEjE,MAAM,OAAO,GAAG,IAAI,CAAC,kBAAkB,CAAC,IAAI,GAAG,YAAY,CAAC;YAC5D,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,OAAO,IAAI,CAAC,GAAG,YAAY,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;gBAC5D,IAAI,CAAC,kBAAkB,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC;YACtD,CAAC;QACH,CAAC;IACH,CAAC;CACF","sourcesContent":["import type { Bot, BotAdapters, BotEvent, BotHandler, RunningSession } from \"../adapter.js\";\nimport { resolveSlackSessionScope } from \"../adapters/slack/branch-manager.js\";\nimport { type AgentRunner, createRunner } from \"../agent.js\";\nimport { CommandRegistry, createDefaultCommandRegistry } from \"../commands/index.js\";\nimport type { CommandServices } from \"../commands/index.js\";\nimport * as log from \"../log.js\";\nimport {\n  createManagedSessionFile,\n  createManagedSessionFileAtPath,\n  getChannelSessionDir,\n  getThreadSessionFile,\n  resolveGenericSessionScope,\n  type ResolvedSessionScope,\n} from \"../session-store.js\";\nimport { formatNothingRunning, formatStopping } from \"../ui-copy.js\";\nimport {\n  ConversationOrchestrator,\n  type ConversationRuntimeState,\n} from \"./conversation-orchestrator.js\";\nimport * as Sentry from \"@sentry/node\";\nimport { join } from \"path\";\nimport { getUnresolvedSandboxPathContext } from \"../agent.js\";\n\ntype ConversationState = ConversationRuntimeState;\n\nexport interface RunSessionOptions {\n  event: BotEvent;\n  bot: Bot;\n  adapters: BotAdapters;\n  isSyntheticEvent?: boolean;\n}\n\nexport interface CreateSessionSandboxOptions {\n  conversationId: string;\n  platformName: string;\n  sessionKey: string;\n}\n\nexport interface SessionRuntimeOptions extends CommandServices {\n  /** Override the default command registry (e.g., to add /help, /status). */\n  commandRegistry?: CommandRegistry;\n}\n\nexport interface SessionRuntime extends BotHandler {\n  runSession(options: RunSessionOptions): Promise<void>;\n  createSessionSandbox(options: CreateSessionSandboxOptions): Promise<AgentRunner>;\n  switchConversationModel(conversationId: string, provider: string, model: string): boolean;\n  shutdown(timeoutMs?: number): Promise<void>;\n}\n\nconst MAX_SESSIONS = 500;\nconst IDLE_TIMEOUT_MS = 3_600_000;\n\nfunction runtimeCwdForSandbox(\n  sandbox: SessionRuntimeOptions[\"sandbox\"],\n  hostWorkspaceRoot: string,\n  conversationId: string,\n): string {\n  const runtimeWorkspaceRoot = getUnresolvedSandboxPathContext(\n    sandbox,\n    hostWorkspaceRoot,\n  ).runtimeWorkspaceRoot;\n  return `${runtimeWorkspaceRoot.replace(/\\/+$/, \"\")}/${conversationId}`;\n}\n\nexport function createSessionRuntime(options: SessionRuntimeOptions): SessionRuntime {\n  return new MamaSessionRuntime(options);\n}\n\nclass MamaSessionRuntime implements SessionRuntime {\n  private readonly conversationStates = new Map<string, ConversationState>();\n  private readonly sessionQueues = new Map<string, Promise<void>>();\n  private readonly inFlightRuns = new Set<Promise<void>>();\n  private readonly commandRegistry: CommandRegistry;\n  private readonly orchestrator: ConversationOrchestrator;\n  private isShuttingDown = false;\n\n  constructor(private readonly options: SessionRuntimeOptions) {\n    this.options.runtime = this;\n    this.commandRegistry = options.commandRegistry ?? createDefaultCommandRegistry();\n    this.orchestrator = new ConversationOrchestrator({\n      workingDir: options.workingDir,\n      commandRegistry: this.commandRegistry,\n      commandServices: this.options,\n      isShuttingDown: () => this.isShuttingDown,\n      getState: (sessionKey) => this.conversationStates.get(sessionKey),\n      getOrCreateState: (createOptions) => this.getOrCreateState(createOptions),\n      beforeRunTracked: (runPromise) => {\n        this.inFlightRuns.add(runPromise);\n        Sentry.metrics.gauge(\"agent.sessions.active\", this.inFlightRuns.size);\n      },\n      afterRunTracked: (runPromise) => {\n        this.inFlightRuns.delete(runPromise);\n      },\n      onRunFinished: () => {\n        Sentry.metrics.gauge(\"agent.sessions.active\", this.inFlightRuns.size - 1);\n        this.evictIdleSessions();\n      },\n    });\n  }\n\n  isRunning(sessionKey: string): boolean {\n    const state = this.conversationStates.get(sessionKey);\n    return !!state?.running;\n  }\n\n  getRunningSessions(): RunningSession[] {\n    const sessions: RunningSession[] = [];\n    for (const [sessionKey, state] of this.conversationStates) {\n      if (state.running && state.startedAt) {\n        const currentStep = state.runner.getCurrentStep();\n        sessions.push({\n          sessionKey,\n          startedAt: state.startedAt,\n          lastActivityAt: state.lastActivityAt,\n          currentTool: currentStep?.label || currentStep?.toolName,\n        });\n      }\n    }\n    return sessions;\n  }\n\n  async handleStop(sessionKey: string, conversationId: string, bot: Bot): Promise<void> {\n    const state = this.conversationStates.get(sessionKey);\n    if (state?.running) {\n      state.stopRequested = true;\n      state.runner.abort();\n      const ts = await bot.postMessage(conversationId, formatStopping(bot));\n      state.stopMessageTs = ts;\n    } else {\n      await bot.postMessage(conversationId, formatNothingRunning(bot));\n    }\n  }\n\n  forceStop(sessionKey: string): void {\n    const state = this.conversationStates.get(sessionKey);\n    if (state?.running) {\n      log.logInfo(`[Force Stop] Force stopping session: ${sessionKey}`);\n      state.stopRequested = true;\n      state.runner.abort();\n      state.running = false;\n    }\n  }\n\n  async handleNewCommand(sessionKey: string, conversationId: string, bot: Bot): Promise<void> {\n    const state = this.conversationStates.get(sessionKey);\n    if (state?.running) {\n      state.stopRequested = true;\n      state.runner.abort();\n    }\n\n    const conversationDir = join(this.options.workingDir, conversationId);\n    const runtimeCwd = runtimeCwdForSandbox(\n      this.options.sandbox,\n      this.options.workingDir,\n      conversationId,\n    );\n    if (sessionKey.includes(\":\")) {\n      createManagedSessionFileAtPath(getThreadSessionFile(conversationDir, sessionKey), runtimeCwd);\n    } else {\n      createManagedSessionFile(getChannelSessionDir(conversationDir), runtimeCwd);\n    }\n\n    this.conversationStates.delete(sessionKey);\n\n    log.logInfo(`[${conversationId}] Session reset: ${sessionKey}`);\n    await bot.postMessage(conversationId, \"Conversation reset. Send a new message to start fresh.\");\n  }\n\n  async handleEvent(\n    event: BotEvent,\n    bot: Bot,\n    adapters: BotAdapters,\n    isSyntheticEvent?: boolean,\n  ): Promise<void> {\n    const sessionKey = event.sessionKey ?? `${event.conversationId}:${event.thread_ts ?? event.ts}`;\n    const previous = this.sessionQueues.get(sessionKey) ?? Promise.resolve();\n    const next = previous\n      .catch(() => {})\n      .then(() => this.runSession({ event, bot, adapters, isSyntheticEvent }));\n    this.sessionQueues.set(sessionKey, next);\n    try {\n      await next;\n    } finally {\n      if (this.sessionQueues.get(sessionKey) === next) {\n        this.sessionQueues.delete(sessionKey);\n      }\n    }\n  }\n\n  async runSession({ event, bot, adapters, isSyntheticEvent }: RunSessionOptions): Promise<void> {\n    await this.orchestrator.runSession({ event, bot, adapters, isSyntheticEvent });\n  }\n\n  async createSessionSandbox(options: CreateSessionSandboxOptions): Promise<AgentRunner> {\n    const state = await this.getOrCreateState(options);\n    return state.runner;\n  }\n\n  switchConversationModel(conversationId: string, _provider: string, _model: string): boolean {\n    for (const [sessionKey, state] of this.conversationStates) {\n      if (this.isConversationSession(sessionKey, conversationId) && state.running) {\n        return false;\n      }\n    }\n\n    for (const sessionKey of Array.from(this.conversationStates.keys())) {\n      if (this.isConversationSession(sessionKey, conversationId)) {\n        this.conversationStates.delete(sessionKey);\n      }\n    }\n    log.logInfo(`[${conversationId}] Model switched; cleared cached session runners`);\n    return true;\n  }\n\n  private isConversationSession(sessionKey: string, conversationId: string): boolean {\n    return sessionKey === conversationId || sessionKey.startsWith(`${conversationId}:`);\n  }\n\n  private async getOrCreateState({\n    conversationId,\n    platformName,\n    sessionKey,\n  }: CreateSessionSandboxOptions): Promise<ConversationState> {\n    const existing = this.conversationStates.get(sessionKey);\n    if (existing) {\n      existing.lastAccessedAt = Date.now();\n      return existing;\n    }\n\n    const conversationDir = join(this.options.workingDir, conversationId);\n    const runtimeCwd = runtimeCwdForSandbox(\n      this.options.sandbox,\n      this.options.workingDir,\n      conversationId,\n    );\n    const sessionScope = await this.resolveSessionScope(\n      platformName,\n      conversationDir,\n      sessionKey,\n      runtimeCwd,\n    );\n    const state: ConversationState = {\n      running: false,\n      runner: await createRunner(\n        this.options.sandbox,\n        sessionKey,\n        conversationId,\n        conversationDir,\n        this.options.workingDir,\n        sessionScope,\n        this.options.vaultManager,\n        this.options.provisioner,\n      ),\n      stopRequested: false,\n      lastAccessedAt: Date.now(),\n    };\n    this.conversationStates.set(sessionKey, state);\n    return state;\n  }\n\n  async shutdown(timeoutMs = 30_000): Promise<void> {\n    if (this.isShuttingDown) return;\n    this.isShuttingDown = true;\n    log.logInfo(\"Shutting down gracefully...\");\n\n    const timeout = Date.now() + timeoutMs;\n    while (this.inFlightRuns.size > 0 && Date.now() < timeout) {\n      await new Promise((resolve) => setTimeout(resolve, 500));\n    }\n\n    if (this.inFlightRuns.size > 0) {\n      log.logWarning(`Forcing exit with ${this.inFlightRuns.size} runs still in progress`);\n    }\n  }\n\n  private async resolveSessionScope(\n    platformName: string,\n    conversationDir: string,\n    sessionKey: string,\n    cwd: string,\n  ): Promise<ResolvedSessionScope> {\n    if (sessionKey.includes(\":event:\")) {\n      const sessionDir = getChannelSessionDir(conversationDir);\n      const contextFile = createManagedSessionFileAtPath(\n        getThreadSessionFile(conversationDir, sessionKey),\n        cwd,\n      );\n      return { sessionDir, contextFile, threadRootMessage: null };\n    }\n\n    if (platformName === \"slack\") {\n      return resolveSlackSessionScope({ conversationDir, sessionKey, cwd });\n    }\n    return resolveGenericSessionScope({ conversationDir, sessionKey, cwd });\n  }\n\n  private evictIdleSessions(): void {\n    const now = Date.now();\n\n    for (const [key, state] of this.conversationStates) {\n      if (!state.running && now - state.lastAccessedAt > IDLE_TIMEOUT_MS) {\n        this.conversationStates.delete(key);\n      }\n    }\n\n    if (this.conversationStates.size > MAX_SESSIONS) {\n      const idleSessions: Array<{ key: string; lastAccessedAt: number }> = [];\n      for (const [key, state] of this.conversationStates) {\n        if (!state.running) {\n          idleSessions.push({ key, lastAccessedAt: state.lastAccessedAt });\n        }\n      }\n\n      idleSessions.sort((a, b) => a.lastAccessedAt - b.lastAccessedAt);\n\n      const toEvict = this.conversationStates.size - MAX_SESSIONS;\n      for (let i = 0; i < toEvict && i < idleSessions.length; i++) {\n        this.conversationStates.delete(idleSessions[i].key);\n      }\n    }\n  }\n}\n"]}

package/dist/sandbox/cloudflare.d.ts

@@ -0,0 +1,15 @@
+import type { CloudflareSandboxConfig, ExecOptions, ExecResult, Executor, RuntimePathContext, SandboxAdapter } from "./types.js";
+export declare function parseCloudflareSandboxArg(value: string): CloudflareSandboxConfig | undefined;
+export declare function validateCloudflareSandbox(_config: CloudflareSandboxConfig): Promise<void>;
+export declare class CloudflareSandboxExecutor implements Executor {
+    private readonly sandboxId;
+    private readonly env?;
+    private readonly cwd;
+    constructor(sandboxId: string, env?: Record<string, string> | undefined, _ensureReady?: () => Promise<void>);
+    exec(command: string, options?: ExecOptions): Promise<ExecResult>;
+    getWorkspacePath(_hostPath: string): string;
+    getPathContext(hostWorkspaceRoot: string): RuntimePathContext;
+    getSandboxConfig(): CloudflareSandboxConfig;
+}
+export declare const cloudflareSandboxAdapter: SandboxAdapter<CloudflareSandboxConfig>;
+//# sourceMappingURL=cloudflare.d.ts.map

package/dist/sandbox/cloudflare.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"cloudflare.d.ts","sourceRoot":"","sources":["../../src/sandbox/cloudflare.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EACV,uBAAuB,EACvB,WAAW,EACX,UAAU,EACV,QAAQ,EACR,kBAAkB,EAClB,cAAc,EACf,MAAM,YAAY,CAAC;AAoBpB,wBAAgB,yBAAyB,CAAC,KAAK,EAAE,MAAM,GAAG,uBAAuB,GAAG,SAAS,CAa5F;AAED,wBAAsB,yBAAyB,CAAC,OAAO,EAAE,uBAAuB,GAAG,OAAO,CAAC,IAAI,CAAC,CAsB/F;AAED,qBAAa,yBAA0B,YAAW,QAAQ;IAItD,OAAO,CAAC,QAAQ,CAAC,SAAS;IAC1B,OAAO,CAAC,QAAQ,CAAC,GAAG,CAAC;IAJvB,OAAO,CAAC,QAAQ,CAAC,GAAG,CAAS;IAE7B,YACmB,SAAS,EAAE,MAAM,EACjB,GAAG,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,YAAA,EAC7C,YAAY,CAAC,EAAE,MAAM,OAAO,CAAC,IAAI,CAAC,EAGnC;IAEK,IAAI,CAAC,OAAO,EAAE,MAAM,EAAE,OAAO,CAAC,EAAE,WAAW,GAAG,OAAO,CAAC,UAAU,CAAC,CAiEtE;IAED,gBAAgB,CAAC,SAAS,EAAE,MAAM,GAAG,MAAM,CAE1C;IAED,cAAc,CAAC,iBAAiB,EAAE,MAAM,GAAG,kBAAkB,CAK5D;IAED,gBAAgB,IAAI,uBAAuB,CAE1C;CACF;AAED,eAAO,MAAM,wBAAwB,EAAE,cAAc,CAAC,uBAAuB,CAM5E,CAAC","sourcesContent":["import type {\n  CloudflareSandboxConfig,\n  ExecOptions,\n  ExecResult,\n  Executor,\n  RuntimePathContext,\n  SandboxAdapter,\n} from \"./types.js\";\nimport { SandboxError } from \"./errors.js\";\n\nconst DEFAULT_CLOUDFLARE_CWD = \"/workspace\";\n\ninterface CloudflareExecPayload {\n  sandboxId: string;\n  command: string;\n  timeoutSeconds?: number;\n  cwd?: string;\n  env?: Record<string, string>;\n}\n\ninterface CloudflareExecResponse {\n  stdout?: string;\n  stderr?: string;\n  code?: number;\n  error?: string;\n}\n\nexport function parseCloudflareSandboxArg(value: string): CloudflareSandboxConfig | undefined {\n  if (!value.startsWith(\"cloudflare:\")) {\n    return undefined;\n  }\n\n  const sandboxId = value.slice(\"cloudflare:\".length).trim();\n  if (!sandboxId) {\n    throw new SandboxError(\n      \"Error: cloudflare sandbox requires sandbox id (e.g., cloudflare:slack-u123)\",\n    );\n  }\n\n  return { type: \"cloudflare\", sandboxId };\n}\n\nexport async function validateCloudflareSandbox(_config: CloudflareSandboxConfig): Promise<void> {\n  const url = resolveCloudflareSandboxUrl();\n  try {\n    const response = await fetch(new URL(\"/health\", url), {\n      headers: buildCloudflareHeaders(),\n    });\n    if (!response.ok) {\n      throw new SandboxError(\n        `Error: Cloudflare sandbox bridge health check failed with HTTP ${response.status}`,\n      );\n    }\n  } catch (error) {\n    if (error instanceof SandboxError) {\n      throw error;\n    }\n    const detail = error instanceof Error ? error.message : String(error);\n    throw new SandboxError(`Error: Cloudflare sandbox bridge is not reachable: ${detail}`);\n  }\n\n  console.log(\n    `  Cloudflare sandbox bridge enabled. Base URL: ${url.toString().replace(/\\/$/, \"\")}`,\n  );\n}\n\nexport class CloudflareSandboxExecutor implements Executor {\n  private readonly cwd: string;\n\n  constructor(\n    private readonly sandboxId: string,\n    private readonly env?: Record<string, string>,\n    _ensureReady?: () => Promise<void>,\n  ) {\n    this.cwd = process.env.MAMA_CLOUDFLARE_SANDBOX_CWD?.trim() || DEFAULT_CLOUDFLARE_CWD;\n  }\n\n  async exec(command: string, options?: ExecOptions): Promise<ExecResult> {\n    const controller = new AbortController();\n    const timeoutHandle =\n      options?.timeout && options.timeout > 0\n        ? setTimeout(() => controller.abort(), options.timeout * 1000)\n        : undefined;\n\n    const onAbort = () => controller.abort();\n    if (options?.signal) {\n      if (options.signal.aborted) {\n        controller.abort();\n      } else {\n        options.signal.addEventListener(\"abort\", onAbort, { once: true });\n      }\n    }\n\n    try {\n      const payload: CloudflareExecPayload = {\n        sandboxId: this.sandboxId,\n        command,\n        cwd: this.cwd,\n      };\n      if (options?.timeout) payload.timeoutSeconds = options.timeout;\n      if (this.env && Object.keys(this.env).length > 0) payload.env = this.env;\n\n      const response = await fetch(new URL(\"/exec\", resolveCloudflareSandboxUrl()), {\n        method: \"POST\",\n        headers: {\n          \"content-type\": \"application/json\",\n          ...buildCloudflareHeaders(),\n        },\n        body: JSON.stringify(payload),\n        signal: controller.signal,\n      });\n\n      const raw = (await response.text()).trim();\n      const parsed = raw ? (JSON.parse(raw) as CloudflareExecResponse) : {};\n\n      if (!response.ok) {\n        throw new Error(\n          parsed.error ||\n            parsed.stderr ||\n            `Cloudflare sandbox bridge returned HTTP ${response.status}`,\n        );\n      }\n\n      return {\n        stdout: parsed.stdout || \"\",\n        stderr: parsed.stderr || \"\",\n        code: parsed.code ?? 0,\n      };\n    } catch (error) {\n      if (controller.signal.aborted) {\n        if (options?.signal?.aborted) {\n          throw new Error(\"Command aborted\", { cause: error });\n        }\n        throw new Error(`Command timed out after ${options?.timeout} seconds`, { cause: error });\n      }\n      throw error;\n    } finally {\n      if (timeoutHandle) clearTimeout(timeoutHandle);\n      if (options?.signal) {\n        options.signal.removeEventListener(\"abort\", onAbort);\n      }\n    }\n  }\n\n  getWorkspacePath(_hostPath: string): string {\n    return this.cwd;\n  }\n\n  getPathContext(hostWorkspaceRoot: string): RuntimePathContext {\n    return {\n      hostWorkspaceRoot,\n      runtimeWorkspaceRoot: this.cwd,\n    };\n  }\n\n  getSandboxConfig(): CloudflareSandboxConfig {\n    return { type: \"cloudflare\", sandboxId: this.sandboxId };\n  }\n}\n\nexport const cloudflareSandboxAdapter: SandboxAdapter<CloudflareSandboxConfig> = {\n  type: \"cloudflare\",\n  parse: parseCloudflareSandboxArg,\n  validate: validateCloudflareSandbox,\n  createExecutor: (config, env, ensureReady) =>\n    new CloudflareSandboxExecutor(config.sandboxId, env, ensureReady),\n};\n\nfunction resolveCloudflareSandboxUrl(): URL {\n  const raw = process.env.MAMA_CLOUDFLARE_SANDBOX_URL?.trim();\n  if (!raw) {\n    throw new SandboxError(\n      \"Error: MAMA_CLOUDFLARE_SANDBOX_URL is required for cloudflare sandbox mode\",\n    );\n  }\n\n  try {\n    return new URL(raw);\n  } catch (error) {\n    const detail = error instanceof Error ? error.message : String(error);\n    throw new SandboxError(`Error: invalid MAMA_CLOUDFLARE_SANDBOX_URL: ${detail}`);\n  }\n}\n\nfunction buildCloudflareHeaders(): Record<string, string> {\n  const token = process.env.MAMA_CLOUDFLARE_SANDBOX_TOKEN?.trim();\n  return token ? { authorization: `Bearer ${token}` } : {};\n}\n"]}