@agimon-ai/po-core 0.0.0-alpha.4 → 0.0.0-alpha.5

package/dist/src/server/directHarnessRuntime.cjs

@@ -126,6 +126,17 @@ async function createDirectHarnessRuntime(options) {
 		const listeners = /* @__PURE__ */ new Set();
 		const eventListeners = /* @__PURE__ */ new Set();
 		const checkpointBoundaries = /* @__PURE__ */ new Set();
+		const toolBoundaryQueue = [];
+		let toolBoundaryDrain;
+		const flushToolBoundary = () => {
+			if (toolBoundaryDrain) return toolBoundaryDrain;
+			toolBoundaryDrain = (async () => {
+				while (toolBoundaryQueue.length && !disposed && !disposePromise) await toolBoundaryQueue.shift().run();
+			})().finally(() => {
+				toolBoundaryDrain = void 0;
+			});
+			return toolBoundaryDrain;
+		};
 		const emit = (frame) => {
 			for (const listener of listeners) try {
 				listener(frame);
@@ -356,6 +367,7 @@ async function createDirectHarnessRuntime(options) {
 					async afterTools(_assistant, _results, api, ctx) {
 						const lane = conversation;
 						if (lane.id !== api.conversationId) return;
+						await flushToolBoundary();
 						for (const registration of checkpointBoundaries) {
 							if (!registration.active || disposed || ctx.abortSignal?.aborted || external) continue;
 							let open = true;
@@ -593,6 +605,31 @@ async function createDirectHarnessRuntime(options) {
 			kind: "run",
 			status: "open"
 		} : null);
+		const runAtToolBoundary = (callback) => {
+			if (disposed || disposePromise) return Promise.reject(/* @__PURE__ */ new Error("Direct harness is disposed"));
+			return new Promise((resolve, reject) => {
+				const pending = {
+					async run() {
+						try {
+							resolve(await callback());
+						} catch (error) {
+							reject(error);
+						}
+					},
+					reject
+				};
+				toolBoundaryQueue.push(pending);
+				nativeExecution().then(async (active) => {
+					if (!active && !external) await flushToolBoundary();
+				}).catch((error) => {
+					const index = toolBoundaryQueue.indexOf(pending);
+					if (index !== -1) {
+						toolBoundaryQueue.splice(index, 1);
+						reject(error);
+					}
+				});
+			});
+		};
 		const readLifecycle = async () => {
 			const record = await readRecord();
 			return {
@@ -849,6 +886,7 @@ async function createDirectHarnessRuntime(options) {
 							}
 						};
 						eventsSettled = eventsSettled.then(async () => {
+							if (!await nativeExecution() && !external) await flushToolBoundary();
 							await reconcile();
 							try {
 								await writable(() => harness.commit((tx) => tx.appendEntry(conversation.id, {
@@ -1538,6 +1576,7 @@ async function createDirectHarnessRuntime(options) {
 					} finally {
 						if (!admitted) {
 							external = void 0;
+							if (!await nativeExecution()) await flushToolBoundary();
 							drain().catch((error) => emit({
 								type: "error",
 								error: String(error)
@@ -1564,6 +1603,7 @@ async function createDirectHarnessRuntime(options) {
 			} finally {
 				external = void 0;
 				externalAbort = void 0;
+				await flushToolBoundary();
 				await publish();
 				drain().catch((error) => emit({
 					type: "error",
@@ -1666,6 +1706,7 @@ async function createDirectHarnessRuntime(options) {
 		};
 		let disposePromise;
 		const dispose = () => disposePromise ??= (async () => {
+			for (const pending of toolBoundaryQueue.splice(0)) pending.reject(/* @__PURE__ */ new Error("Direct harness is disposed"));
 			const failures = [];
 			const attempt = async (cleanup) => {
 				try {
@@ -2147,6 +2188,7 @@ async function createDirectHarnessRuntime(options) {
 				return resources;
 			},
 			runExternalOperation,
+			runAtToolBoundary,
 			appendCustomEntry: async (customType, data) => {
 				const lane = conversation;
 				return String((await writable(() => lane.commit((tx) => tx.appendEntry(lane.id, {
@@ -2354,6 +2396,7 @@ async function createDirectHarnessRuntime(options) {
 				} finally {
 					settings = previous;
 					compactionAdmission = false;
+					if (!await nativeExecution()) await flushToolBoundary();
 				}
 			},
 			async admitResume() {