@agentswf/engine 0.1.0-beta.4 → 0.1.0

package/src/operation-events.ts

@@ -1,10 +1,11 @@
 import { appendFile, mkdir } from "node:fs/promises";
-import { dirname, join } from "node:path";
+import { dirname } from "node:path";
 import {
   OPERATION_LIVENESS_KINDS,
   OPERATION_LIVENESS_RECORD_VERSION,
   type OperationLivenessRecord,
 } from "@agentswf/contract/records";
+import { liveFile } from "./ops";
 
 export type OperationEvent = Pick<
   OperationLivenessRecord,
@@ -30,7 +31,7 @@ export function createOperationEvents(
   record(event: OperationEvent): void;
   close(): Promise<void>;
 } {
-  const path = eventFile(runDir, operationId);
+  const path = liveFile(runDir, operationId);
   const write = options.write ?? append;
   const schedule =
     options.schedule ??
@@ -144,7 +145,7 @@ export function createOperationEvents(
   };
 }
 
-function validEvent(value: OperationEvent): boolean {
+export function validEvent(value: OperationEvent): boolean {
   return (
     typeof value.kind === "string" &&
     (OPERATION_LIVENESS_KINDS as readonly string[]).includes(value.kind) &&
@@ -173,9 +174,6 @@ function bound(value: number | undefined, fallback: number, minimum: number): nu
     throw new Error(`diagnostic bound must be an integer >= ${minimum}`);
   return actual;
 }
-function eventFile(runDir: string, operationId: string): string {
-  return join(runDir, "calls", operationId, "liveness.jsonl");
-}
 async function append(path: string, line: string): Promise<void> {
   await mkdir(dirname(path), { recursive: true });
   await appendFile(path, line, { mode: 0o600 });

package/src/operation-liveness.ts

@@ -79,7 +79,7 @@ export type SupervisedOutcome = {
   value?: unknown;
   reason: string;
   cleanupUnresolved: boolean;
-  charges: number[];
+  reportedCosts: number[];
   deliveredAt?: number;
   settledAt: number;
   /** A failed turn whose harness showed it cannot sign in. */
@@ -91,6 +91,8 @@ export type OperationStop = { kind: "cancelled" | "timed-out"; reason: string };
 type Options = {
   operationId: string;
   agentId: string;
+  /** Why its `op.json` wasn't written: the operation fails as it opens, before anything runs. */
+  unrecorded?: string;
   question: string;
   schema: JsonSchema;
   /** Absent when nothing bounds it. */
@@ -113,6 +115,8 @@ type Options = {
   messages?: OperationMessages;
   /** Once, when the harness first accepts the operation. */
   delivered?(at: number): void;
+  /** Keeps what its turn shows when it ends without an answer, before the turn is released. */
+  keepScreen?(turn: HarnessTurn): Promise<void>;
   policy?: LivenessPolicy;
   now?: () => number;
   schedule?: (at: number, action: () => void) => () => void;
@@ -177,7 +181,7 @@ export async function superviseOperation(options: Options): Promise<SupervisedOu
     | Promise<import("@agentswf/harness/adapter").HarnessReleaseDisposition>
     | undefined;
   let answerReleaseDeadline: number | undefined;
-  const charges: number[] = [];
+  const reportedCosts: number[] = [];
   const emit = (
     kind: OperationLivenessKind,
     reason?: string,
@@ -262,6 +266,7 @@ export async function superviseOperation(options: Options): Promise<SupervisedOu
   });
   let slot: Awaited<ReturnType<ResultSlotRegistry["open"]>>;
   try {
+    if (options.unrecorded !== undefined) throw new Error(options.unrecorded);
     slot = await options.slots.open({
       operationId: options.operationId,
       agentId: options.agentId,
@@ -327,7 +332,7 @@ export async function superviseOperation(options: Options): Promise<SupervisedOu
           (outcome) => {
             if (!valid()) return;
             native = outcome;
-            charges.push(...outcome.chargesUsd);
+            reportedCosts.push(...outcome.reportedCostsUsd);
             if (
               result === undefined &&
               outcome.state === "completed" &&
@@ -608,6 +613,16 @@ export async function superviseOperation(options: Options): Promise<SupervisedOu
       }
     }
     if (result.kind === "answered" && !held) result.cleanupUnresolved = true;
+    if (
+      held &&
+      options.keepScreen &&
+      result.kind !== "answered" &&
+      (stop?.kind ?? result.kind) !== "cancelled"
+    ) {
+      await options.keepScreen(held);
+      // The read took from the release's own time; it gets it back.
+      releaseDeadline = now() + Math.min(policy.releaseMs, 5000);
+    }
     if (held) {
       const stopping = stop !== undefined;
       if (stopping)
@@ -690,7 +705,7 @@ export async function superviseOperation(options: Options): Promise<SupervisedOu
       kind,
       reason,
       cleanupUnresolved: false,
-      charges,
+      reportedCosts,
       ...(deliveredAt === undefined ? {} : { deliveredAt }),
       settledAt:
         kind === "timed-out"