@agimon-ai/doompi-core 0.0.1-alpha.100 → 0.0.1-alpha.102

package/dist/src/server/directHarnessRuntime.cjs

@@ -989,6 +989,14 @@ async function createDirectHarnessRuntime(options) {
 			return await operation();
 		} finally {
 			externalOperationActive = false;
+			if (drainRequested && !disposed) {
+				drainRequested = false;
+				drainAutomatic().catch((error) => emitFrame({
+					type: FRAME_ERROR,
+					code: "queue_drain",
+					error: require_harnessErrorMessage.harnessErrorMessage(error)
+				}));
+			}
 		}
 	};
 	let admissionLine = Promise.resolve();
@@ -1007,12 +1015,13 @@ async function createDirectHarnessRuntime(options) {
 	};
 	let drainingAutomatic = false;
 	let drainRequested = false;
-	const drive = async (operationId) => {
+	const drive = async (operationId, pollDeferred = false) => {
 		settlingOperationId = operationId;
 		let failure;
 		try {
 			const result = await writable(() => lane.drive({
 				operationId,
+				pollDeferred,
 				waitForRetry: true
 			}, context));
 			if (!result.ok) resultError(result);
@@ -1056,14 +1065,22 @@ async function createDirectHarnessRuntime(options) {
 			return;
 		}
 		drainingAutomatic = true;
+		let releaseAutomatic;
 		try {
 			while (!disposed) {
+				releaseAutomatic?.();
+				releaseAutomatic = void 0;
 				if ((await lane.inspectExecution(context)).current !== null) return;
 				const operationId = (0, node_crypto.randomUUID)();
 				const claimed = await changeRecord((record) => {
 					if (record.paused || record.abortOperationId !== void 0) return { result: void 0 };
 					const item = record.queue.find((candidate) => candidate.disposition === "pending" && candidate.scheduling === "automatic" && candidate.delivery !== "steer");
 					if (!item) return { result: void 0 };
+					if (externalOperationActive) {
+						drainRequested = true;
+						return { result: void 0 };
+					}
+					releaseAutomatic = acquireAgentOperation();
 					return {
 						record: {
 							...record,
@@ -1145,8 +1162,9 @@ async function createDirectHarnessRuntime(options) {
 				}
 			}
 		} finally {
+			releaseAutomatic?.();
 			drainingAutomatic = false;
-			if (drainRequested && !disposed) {
+			if (drainRequested && !externalOperationActive && !disposed) {
 				drainRequested = false;
 				drainAutomatic().catch((error) => emitFrame({
 					type: FRAME_ERROR,
@@ -1747,21 +1765,46 @@ async function createDirectHarnessRuntime(options) {
 		await reconcileNativeQueue();
 		await publishLifecycle();
 	};
-	const resume = () => runAgentOperation(async () => {
-		await recover();
-		if ((await readRecord()).paused) return false;
-		const result = await writable(() => lane.resume(context));
-		if (!result.ok && result.error._tag === "NothingToResume") {
-			drainAutomatic().catch((error) => emitFrame({
-				type: FRAME_ERROR,
-				code: "queue_drain",
-				error: require_harnessErrorMessage.harnessErrorMessage(error)
-			}));
-			return false;
+	const admitResume = async () => {
+		const release = acquireAgentOperation();
+		try {
+			await recover();
+			const submission = await serializeAdmission(async () => {
+				guardWritable();
+				if ((await readRecord()).paused) return {
+					resumed: false,
+					settled: Promise.resolve()
+				};
+				const execution = await lane.inspectExecution(context);
+				if (execution.current === null) return {
+					resumed: false,
+					settled: Promise.resolve()
+				};
+				return {
+					resumed: true,
+					settled: drive(execution.current.id, true)
+				};
+			});
+			if (submission.resumed) submission.settled.then(release, release);
+			else {
+				release();
+				drainAutomatic().catch((error) => emitFrame({
+					type: FRAME_ERROR,
+					code: "queue_drain",
+					error: require_harnessErrorMessage.harnessErrorMessage(error)
+				}));
+			}
+			return submission;
+		} catch (error) {
+			release();
+			throw error;
 		}
-		if (!result.ok) resultError(result);
-		return true;
-	});
+	};
+	const resume = async () => {
+		const submission = await admitResume();
+		await submission.settled;
+		return submission.resumed;
+	};
 	const readState = async () => {
 		const [lifecycle, retained, persisted, stats, thinking] = await Promise.all([
 			readLifecycle(),
@@ -1951,6 +1994,7 @@ async function createDirectHarnessRuntime(options) {
 		abort,
 		interrupt,
 		compact,
+		admitResume,
 		resume,
 		dispose
 	};