@atolis-hq/wake 0.3.64 → 0.3.65

package/dist/src/bootstrap/composition-root.js

@@ -1,6 +1,6 @@
 import { createPullRequestService } from '../activities/index.js';
 import { ControlStreamKind, DispatchPolicy, createAdvanceOnce, createControlPlaneService, createRunnerControlService, ineligibleRunners, } from '../control-plane/index.js';
-import { ExecutionCancellationReason, ExternalExecutionState, GitWorkspaceProvider, RecoveryService, TranscriptStore, createExecutionService, } from '../execution/index.js';
+import { ExecutionCancellationReason, ExternalExecutionState, GitWorkspaceProvider, RecoveryService, TranscriptStore, createExecutionService, createRunnerMemoryProfileDecorator, } from '../execution/index.js';
 import { createGitHubAgentContextReader, gitHubProviderDefinition, resolveGitHubResourceUrl, } from '../integrations/github/index.js';
 import { SystemClock, UlidIdGenerator, } from '../kernel/index.js';
 import { compileWorkflow, createOrchestrationService } from '../orchestration/index.js';
@@ -59,10 +59,22 @@ export async function createCompositionRoot(wakeRoot, options = {}) {
     const transcriptStore = config.transcripts.enabled
         ? (options.transcriptStore ?? new TranscriptStore(paths.transcriptsRoot))
         : undefined;
+    const profileDecorator = process.env.WAKE_MEMORY_PROFILE === 'runner'
+        ? createRunnerMemoryProfileDecorator({
+            write: (line) => process.stderr.write(line),
+            now: () => clock.now().toISOString(),
+            memoryUsage: () => process.memoryUsage(),
+        })
+        : undefined;
+    const decorateRunner = profileDecorator === undefined
+        ? options.decorateRunner
+        : options.decorateRunner === undefined
+            ? profileDecorator
+            : (runner, name) => profileDecorator(options.decorateRunner(runner, name), name);
     const execution = createExecutionService(journal, activities, config.execution, {
         clock,
         ids,
-        runners: createRunnerRegistry(config.execution, fakeScenarios, options.decorateRunner),
+        runners: createRunnerRegistry(config.execution, fakeScenarios, decorateRunner),
         reportRunnerQuota: createRunnerQuotaReporter(journal, clock, ids),
         ...(transcriptStore !== undefined
             ? {

package/dist/src/bootstrap/surface-cli-applications.js

@@ -179,6 +179,7 @@ function sandboxDockerOptions(root, overrides) {
         containerHomeMountPath: root.config.host.sandbox.containerHomeMountPath,
         extraMounts: root.config.host.sandbox.extraMounts,
         startEnabled: root.config.host.sandbox.start.enabled,
+        ...(process.env.WAKE_MEMORY_PROFILE === 'runner' ? { memoryProfile: 'runner' } : {}),
         inspect: createDockerInspection(root.paths.wakeRoot),
         resolveBuildVersion: () => resolveSandboxBuildVersion(root),
     };

package/dist/src/bootstrap/version.js

@@ -108,4 +108,4 @@ export function resolveWakeVersion(options = {}) {
         return `g${headHash.slice(0, 7)}`;
     return '0.1.0-dev';
 }
-export const wakeVersion = "gfe73f33";
+export const wakeVersion = "gcf37d8a";

package/dist/src/execution/index.js

@@ -26,6 +26,7 @@ export * from './infrastructure/runners/cursor.js';
 export * from './infrastructure/runners/fake.js';
 export * from './infrastructure/runners/fake-scenarios.js';
 export * from './infrastructure/runners/registry.js';
+export * from './infrastructure/runner-memory-profile.js';
 export * from './infrastructure/transcripts.js';
 export * from './infrastructure/transcript-store.js';
 export * from './infrastructure/workspace/fake-workspace.js';

package/dist/src/execution/infrastructure/runner-memory-profile.js

@@ -0,0 +1,34 @@
+export function createRunnerMemoryProfileDecorator(options) {
+    return (runner, name) => ({
+        ...(runner.supportsSessionResume === undefined
+            ? {}
+            : { supportsSessionResume: runner.supportsSessionResume }),
+        start: async (request, signal) => {
+            writeSample(options, 'runner.start.before', name, request.runId);
+            const execution = await runner.start(request, signal);
+            writeSample(options, 'runner.start.returned', name, request.runId);
+            return {
+                ...execution,
+                result: execution.result.finally(() => {
+                    writeSample(options, 'runner.result.settled', name, request.runId);
+                }),
+            };
+        },
+    });
+}
+function writeSample(options, phase, runner, runId) {
+    const memory = options.memoryUsage();
+    options.write(`${JSON.stringify({
+        type: 'wake.runner-memory',
+        phase,
+        at: options.now(),
+        pid: process.pid,
+        runner,
+        runId,
+        rss: memory.rss,
+        heapTotal: memory.heapTotal,
+        heapUsed: memory.heapUsed,
+        external: memory.external,
+        arrayBuffers: memory.arrayBuffers,
+    })}\n`);
+}

package/dist/src/surfaces/cli/infrastructure/docker-cli.js

@@ -197,6 +197,9 @@ async function createContainer(docker, options) {
                 `WAKE_HOME_INIT_DIRS=${homeInitDirectories.join('\n')}`,
             ]),
         ...(options.startEnabled === true ? ['-e', 'WAKE_START_ENABLED=true'] : []),
+        ...(options.memoryProfile === undefined
+            ? []
+            : ['-e', `WAKE_MEMORY_PROFILE=${options.memoryProfile}`]),
         options.image,
     ]);
 }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@atolis-hq/wake",
-  "version": "0.3.64",
+  "version": "0.3.65",
   "description": "Local autonomous agent control plane for software development",
   "license": "Apache-2.0",
   "repository": {