@amaster.ai/employee-runtime-connector 0.1.1-beta.248 → 0.1.1-beta.249

package/dist/amaster-runtime-daemon.mjs

@@ -17397,6 +17397,28 @@ function createPiOutputRetentionCompactor() {
   };
 }
 
+// src/amaster-runtime-daemon/task-session-checkpoint.mjs
+function taskSessionCheckpointMatches(command, manifest) {
+  const payload = asRecord(command.payload);
+  const session = asRecord(payload.nativeSession);
+  const expected = {
+    companyId: readString(payload.companyId),
+    issueId: readString(command.issueId ?? payload.issueId),
+    agentId: readString(payload.agentId),
+    sourceRunId: readString(session.sourceRunId),
+    sourceCommandId: readString(session.sourceCommandId),
+    sessionId: readString(session.sessionId)
+  };
+  return session.mode === "task_session" && session.resume === true && manifest.kind === "task_session" && Object.entries(expected).every(([key, value]) => value !== null && value === manifest[key]);
+}
+function taskSessionCheckpointPaths(workspaceStatus) {
+  const artifacts = workspaceStatus.artifacts;
+  if (asRecord(workspaceStatus.metadata).artifactScanTruncated === true || !Array.isArray(artifacts) || artifacts.length > 20) {
+    throw new Error("native_session_checkpoint_unavailable: workspace inventory exceeds checkpoint limits");
+  }
+  return artifacts.map((file) => file.relativePath);
+}
+
 // src/amaster-runtime-daemon/pi-run-exit-closure-guard.mjs
 function readText(value) {
   return typeof value === "string" && value.trim() ? value.trim() : null;
@@ -30823,7 +30845,7 @@ function assertRequiredTaskAttachmentMaterialized(input, materialized) {
 }
 
 // src/amaster-runtime-daemon.mjs
-var CONNECTOR_VERSION = "0.1.1-beta.248";
+var CONNECTOR_VERSION = "0.1.1-beta.249";
 var CONNECTOR_CONTRACT_VERSION = "2026-06-04.v1";
 var daemonRequire = createRequire(import.meta.url);
 var MAX_CHECKPOINT_BYTES = 20 * 1024 * 1024;
@@ -35631,9 +35653,20 @@ async function materializeRequiredArtifactInputs(config, command, workspace) {
 function issueCheckpointDir(workspace) {
   return join17(dirname11(workspace.runDir), "checkpoint");
 }
-async function clearIssueCheckpoint(config, command, workspace, reason) {
+async function clearIssueCheckpoint(config, command, workspace, reason, consumedSource = null) {
   const checkpointDir = issueCheckpointDir(workspace);
   if (!existsSync17(checkpointDir)) return false;
+  if (consumedSource !== null) {
+    const sourceRunId = readString(consumedSource.sourceRunId);
+    if (!sourceRunId) return false;
+    let manifest;
+    try {
+      manifest = asRecord(JSON.parse(readFileSync18(join17(checkpointDir, "manifest.json"), "utf8")));
+    } catch {
+      return false;
+    }
+    if (readString(manifest.issueId) !== commandIssueId(command) || readString(manifest.sourceRunId) !== sourceRunId) return false;
+  }
   rmSync9(checkpointDir, { recursive: true, force: true });
   await ingestLog(config, command, "system", "info", "Cleared issue continuation checkpoint", { reason });
   return true;
@@ -35675,9 +35708,10 @@ async function materializeIssueCheckpoint(config, command, workspace) {
   const checkpointDir = issueCheckpointDir(workspace);
   const manifestPath = join17(checkpointDir, "manifest.json");
   const expectedContinuation = expectedContinuationCheckpoint(command);
+  const unavailable2 = { files: [], taskSessionRestored: false };
   if (!existsSync17(manifestPath)) {
     if (expectedContinuation) throw new Error("continuation_checkpoint_unavailable: manifest_missing");
-    return [];
+    return unavailable2;
   }
   let manifest;
   try {
@@ -35690,12 +35724,20 @@ async function materializeIssueCheckpoint(config, command, workspace) {
   if (!Number.isFinite(expiresAt) || expiresAt <= Date.now() || readString(manifest.issueId) !== commandIssueId(command)) {
     rmSync9(checkpointDir, { recursive: true, force: true });
     if (expectedContinuation) throw new Error("continuation_checkpoint_unavailable: manifest_expired_or_mismatched");
-    return [];
+    return unavailable2;
   }
   try {
+    if (!expectedContinuation && asRecord(asRecord(command.payload).nativeSession).mode === "task_session" && manifest.kind !== "task_session") {
+      return unavailable2;
+    }
     if (readString(manifest.kind) === "continue_task" && !expectedContinuation) {
-      return [];
+      return unavailable2;
+    }
+    const taskSessionRestored = !expectedContinuation && taskSessionCheckpointMatches(command, manifest) && resolveNativeSessionRequest(command, workspace).used;
+    if (manifest.kind === "task_session" && !taskSessionRestored && !expectedContinuation) {
+      return unavailable2;
     }
+    if (taskSessionRestored && !Array.isArray(manifest.files)) throw new Error("task session checkpoint file inventory is missing");
     const files = Array.isArray(manifest.files) ? manifest.files : [];
     if (files.length > 20) throw new Error("issue checkpoint manifest exceeds the limit of 20 files");
     if (expectedContinuation) {
@@ -35724,7 +35766,7 @@ async function materializeIssueCheckpoint(config, command, workspace) {
       }
       const expectedSha256 = readString(file.sha256);
       const expectedByteSize = readNumber(file.byteSize, null);
-      if (!expectedSha256 || expectedByteSize === null || expectedByteSize <= 0) {
+      if (!expectedSha256 || expectedByteSize === null || expectedByteSize < 0 || expectedByteSize === 0 && !taskSessionRestored) {
         throw new Error(`issue checkpoint evidence is incomplete: ${relativePath}`);
       }
       if (hashFileSha256(source) !== expectedSha256 || statSync11(source).size !== expectedByteSize) {
@@ -35779,7 +35821,7 @@ async function materializeIssueCheckpoint(config, command, workspace) {
         checkpointExpiresAt: new Date(expiresAt).toISOString()
       });
     }
-    return materialized;
+    return { files: materialized, taskSessionRestored };
   } catch (restoreErr) {
     try {
       await clearIssueCheckpoint(config, command, workspace, "restore_failed");
@@ -35806,7 +35848,7 @@ async function saveIssueCheckpoint(config, command, workspace, candidates, optio
     let totalBytes = 0;
     const workspaceRoot = realpathSync8(workspace.cwd);
     const requiredPaths = Array.isArray(options.requiredPaths) ? options.requiredPaths : [];
-    const selectedCandidates = requiredPaths.length > 0 ? requiredPaths.map((requiredPath) => {
+    const selectedCandidates = requiredPaths.length > 0 || options.taskSession ? requiredPaths.map((requiredPath) => {
       const normalized = safeCheckpointRelativePath(requiredPath);
       if (!normalized) {
         throw new Error(`continuation_checkpoint_file_unavailable: ${requiredPath}`);
@@ -35837,7 +35879,7 @@ async function saveIssueCheckpoint(config, command, workspace, candidates, optio
       const relativePath = safeCheckpointRelativePath(candidate.rawPath);
       const source = readString(candidate.filePath);
       if (!relativePath || !source || !existsSync17(source) || !statSync11(source).isFile()) {
-        if (requiredPaths.length > 0) throw new Error(`continuation_checkpoint_file_unavailable: ${candidate.rawPath}`);
+        if (requiredPaths.length > 0 || options.taskSession) throw new Error(`continuation_checkpoint_file_unavailable: ${candidate.rawPath}`);
         continue;
       }
       const ownedSource = realpathSync8(source);
@@ -35845,8 +35887,8 @@ async function saveIssueCheckpoint(config, command, workspace, candidates, optio
         throw new Error(`issue checkpoint source escapes its execution workspace: ${relativePath}`);
       }
       const byteSize = statSync11(ownedSource).size;
-      if (byteSize <= 0 || totalBytes + byteSize > MAX_CHECKPOINT_BYTES) {
-        if (requiredPaths.length > 0) throw new Error(`continuation_checkpoint_file_size_invalid: ${relativePath}`);
+      if (byteSize <= 0 && !options.taskSession || totalBytes + byteSize > MAX_CHECKPOINT_BYTES) {
+        if (requiredPaths.length > 0 || options.taskSession) throw new Error(`continuation_checkpoint_file_size_invalid: ${relativePath}`);
         continue;
       }
       const target = resolve16(filesDir, relativePath);
@@ -35860,15 +35902,22 @@ async function saveIssueCheckpoint(config, command, workspace, candidates, optio
       totalBytes += byteSize;
       files.push({ path: relativePath, byteSize, sha256: hashFileSha256(target) });
     }
-    if (files.length === 0) {
+    if (files.length === 0 && !options.taskSession) {
       rmSync9(stagingDir, { recursive: true, force: true });
       rmSync9(checkpointDir, { recursive: true, force: true });
       return null;
     }
     const manifest = {
-      version: options.continuation ? 2 : 1,
+      version: options.continuation || options.taskSession ? 2 : 1,
       issueId: commandIssueId(command),
       sourceRunId: commandRunId(command),
+      ...options.taskSession ? {
+        kind: "task_session",
+        companyId: readString(asRecord(command.payload).companyId),
+        agentId: readString(asRecord(command.payload).agentId),
+        sourceCommandId: command.commandId,
+        sessionId: options.taskSession.sessionId
+      } : {},
       ...options.continuation ? {
         kind: "continue_task",
         continuationChainId: options.continuation.chainId,
@@ -35970,7 +36019,19 @@ async function executeRunCommand(config, command) {
     agentInstructionFiles: [...finalAgentInstructionFiles.values()].sort((left, right) => left.path.localeCompare(right.path)),
     workspaceProjectInstructionProjection: workspaceProjectInstructionReceipt
   });
-  const materializedCheckpointFiles = await materializeIssueCheckpoint(config, command, workspace);
+  const restoredCheckpoint = await materializeIssueCheckpoint(config, command, workspace);
+  const materializedCheckpointFiles = restoredCheckpoint.files;
+  const requestedNativeSession = asRecord(asRecord(command.payload).nativeSession);
+  const taskSessionCheckpointUnavailable = Object.keys(governedMcp).length > 0 && requestedNativeSession.mode === "task_session" && !expectedContinuationCheckpoint(command) && !restoredCheckpoint.taskSessionRestored;
+  if (taskSessionCheckpointUnavailable && requestedNativeSession.required === true) {
+    throw new Error("native_session_required: exact task-session workspace checkpoint is unavailable");
+  }
+  const executionCommand = taskSessionCheckpointUnavailable ? { ...command, payload: { ...asRecord(command.payload), nativeSession: void 0 } } : command;
+  if (taskSessionCheckpointUnavailable) {
+    await ingestLog(config, command, "system", "info", "Starting fresh: no exact task-session workspace checkpoint is available", {
+      skippedReason: "native_session_checkpoint_unavailable"
+    });
+  }
   const requiredArtifactInputs = await materializeRequiredArtifactInputs(config, command, workspace);
   let materializedAttachments = requiredArtifactInputs;
   try {
@@ -36016,7 +36077,10 @@ async function executeRunCommand(config, command) {
           AMASTER_BROWSER_SESSION_USER_DATA_DIR: ownedProfile.userDataDir
         };
       }
-      invocation = buildExecutorInvocation(executor, command, workspace);
+      invocation = buildExecutorInvocation(executor, executionCommand, workspace);
+      if (taskSessionCheckpointUnavailable) {
+        invocation.nativeSession = { ...asRecord(invocation.nativeSession), requested: true, skippedReason: "native_session_checkpoint_unavailable" };
+      }
     });
     if (executor.kind === "codex" && Object.keys(governedMcp).length > 0) {
       managedMcpProfile = await runExecutionStage("managed_mcp", () => prepareManagedCodexMcpProfile({
@@ -36028,7 +36092,7 @@ async function executeRunCommand(config, command) {
         runDir: workspace.runDir,
         cwd,
         runtimeAuth: commandRuntimeAuth(command),
-        nativeSession: asRecord(asRecord(command.payload).nativeSession),
+        nativeSession: asRecord(asRecord(executionCommand.payload).nativeSession),
         baseEnv: executorEnv,
         commandEnv: commandExecutorEnv(command),
         extraArgs: splitExtraArgs(process.env.AMASTER_CODEX_EXTRA_ARGS)
@@ -36061,7 +36125,7 @@ async function executeRunCommand(config, command) {
           runtimeCacheRoot: join17(dirname11(config.daemonStateFile), "cache"),
           cwd,
           runtimeAuth: commandRuntimeAuth(command),
-          nativeSession: asRecord(asRecord(command.payload).nativeSession),
+          nativeSession: asRecord(asRecord(executionCommand.payload).nativeSession),
           baseEnv: executorEnv,
           commandEnv: piProfileCommandEnv,
           extraArgs: splitExtraArgs(process.env.AMASTER_PI_EXTRA_ARGS),
@@ -36211,7 +36275,7 @@ async function executeRunCommand(config, command) {
         };
       }
     );
-    const promptCompilation = await runExecutionStage("prompt_compilation", () => buildCommandPrompt(command, workspace, materializedAttachments, {
+    const promptCompilation = await runExecutionStage("prompt_compilation", () => buildCommandPrompt(executionCommand, workspace, materializedAttachments, {
       runnerKind: config.runnerKind,
       negotiatedCapabilities: command.requiredCapabilities,
       executorKind: executor.kind,
@@ -36604,7 +36668,10 @@ async function executeRunCommand(config, command) {
             rawPath: artifact.relativePath,
             filePath: resolve16(cwd, artifact.relativePath)
           }))
-        ) : null;
+        ) : readString(terminalRunDisposition?.disposition) !== "continuation_scheduled" ? await saveIssueCheckpoint(config, command, workspace, [], {
+          requiredPaths: taskSessionCheckpointPaths(workspaceStatus),
+          taskSession: { sessionId: preservedRollout.sessionId }
+        }) : null;
         nativeSessionRollout = {
           status: preservedRollout.status,
           sessionId: preservedRollout.sessionId,
@@ -36633,7 +36700,9 @@ async function executeRunCommand(config, command) {
     }
     const approvedContinuationSucceeded = nativeSessionRequest.mode === "governed_action_approval" && approvedMcpInvocationSucceeded(mcpToolResults, nativeSessionRequest.invocationId);
     if (approvedContinuationSucceeded) {
-      await clearIssueCheckpoint(config, command, workspace, "approved_action_succeeded");
+      await clearIssueCheckpoint(config, command, workspace, "approved_action_succeeded", {
+        sourceRunId: nativeSessionRequest.sourceRunId
+      });
     }
     const taskSessionContinuationSucceeded = nativeSessionRequest.mode === "task_session" && asRecord(invocation.nativeSession).used === true && shouldPreserveNativeSession && nativeSessionRollout?.status === "preserved";
     if (managedMcpProfile && (approvedContinuationSucceeded || taskSessionContinuationSucceeded)) {

package/dist/amaster-runtime.mjs

@@ -7,7 +7,7 @@ import { homedir, hostname } from "node:os";
 import { fileURLToPath } from "node:url";
 import { BUILD_SUPPORTED_CAPABILITIES } from "./amaster-runtime-daemon/capability-registry.mjs";
 
-const CONNECTOR_VERSION = "0.1.1-beta.248";
+const CONNECTOR_VERSION = "0.1.1-beta.249";
 
 const CONFIG_KEY_MAP = new Map([
   ["server_url", "AMASTER_EMPLOYEE_SERVER_URL"],

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@amaster.ai/employee-runtime-connector",
-  "version": "0.1.1-beta.248",
+  "version": "0.1.1-beta.249",
   "description": "MirrorX runtime connector CLI and daemon",
   "license": "MIT",
   "type": "module",