@osolmaz/pi-workflows 0.16.4 → 0.16.6

package/src/resource-managers/types.ts

@@ -1,9 +1,8 @@
+import type { JsonObject } from "../state/json.js";
 import type { JsonPatch } from "../workflows/json-patch.js";
 
+export type { JsonPrimitive, JsonValue, JsonObject } from "../state/json.js";
 export type MaybePromise<T> = T | Promise<T>;
-export type JsonPrimitive = string | number | boolean | null;
-export type JsonValue = JsonPrimitive | JsonValue[] | { [key: string]: JsonValue };
-export type JsonObject = { [key: string]: JsonValue };
 
 export type ManagedResourceConditionStatus = true | false | "unknown";
 

package/src/server/resource-runner-entry.ts

@@ -145,7 +145,7 @@ class ServerBackedResourceManagerEffects implements ResourceManagerEffects {
     try {
       applied = await definition.apply(this.signal);
     } catch (error) {
-      applied = { state: "indeterminate", error: boundedError(error) };
+      applied = { state: "indeterminate", error: errorMessage(error) };
     }
     return await this.transport.request<EffectRecord>("effect.settle", {
       key: definition.key,
@@ -246,7 +246,7 @@ export async function runResourceRunner(): Promise<number> {
       validateResult(result);
       await transport.request("runner.finished", result as unknown as JsonValue);
     } catch (error) {
-      await transport.request("runner.failed", { error: boundedError(error) });
+      await transport.request("runner.failed", { error: errorMessage(error) });
     } finally {
       clearTimeout(timer);
     }
@@ -296,11 +296,6 @@ function raceWithAbort<T>(operation: Promise<T>, signal: AbortSignal): Promise<T
   });
 }
 
-function boundedError(error: unknown): string {
-  const message = errorMessage(error);
-  return message.length <= 8_192 ? message : `${message.slice(0, 8_192)}…`;
-}
-
 async function main(): Promise<void> {
   try {
     process.exitCode = await runResourceRunner();

package/src/server/rpc-bridge.ts

@@ -19,7 +19,7 @@ export default function piWorkflowsRpcBridge(pi: ExtensionAPI) {
     description: [
       "Publish an update or submit the output for the pending workflow step.",
       "Only call this tool when a workflow step contract in the conversation asks you to.",
-      "Pass the exact step and attempt ids from the contract.",
+      "Pass the exact requestId from the contract.",
     ].join(" "),
     parameters: WorkflowSubmissionToolParameters,
     async execute(toolCallId, rawParams) {

package/src/server/rpc-executor.ts

@@ -18,15 +18,13 @@ const DEFAULT_ABORT_GRACE_MS = 3_000;
 
 type StepSubmission = {
   action: "submit";
-  step: string;
-  attempt: string;
+  requestId: string;
   output: unknown;
 };
 
 type StepUpdate = {
   action: "update";
-  step: string;
-  attempt: string;
+  requestId: string;
   update: { type: string; key: string; data: Record<string, unknown> };
   idempotencyKey?: string;
 };
@@ -203,8 +201,7 @@ export class RpcStepExecutor implements AgentStepExecutor {
           const parsed = JSON.parse(line.slice(RPC_SUBMISSION_PREFIX.length)) as StepAction;
           if (
             (parsed.action === "submit" || parsed.action === "update") &&
-            typeof parsed.step === "string" &&
-            typeof parsed.attempt === "string"
+            typeof parsed.requestId === "string"
           ) {
             this.actions.push(parsed);
           }
@@ -289,9 +286,7 @@ export class RpcStepExecutor implements AgentStepExecutor {
 
   private takeMatchingAction(request: AgentStepRequest): StepAction | undefined {
     const index = this.actions.findIndex(
-      (candidate) =>
-        candidate.step === request.contract.nodeId &&
-        candidate.attempt === request.contract.attemptId,
+      (candidate) => candidate.requestId === request.contract.requestId,
     );
     return index === -1 ? undefined : (this.actions.splice(index, 1)[0] as StepAction);
   }

package/src/server/server-entry.ts

@@ -7,6 +7,9 @@ async function main(): Promise<void> {
   const databasePath = databaseIndex < 0 ? undefined : process.argv[databaseIndex + 1];
   const server = new WorkflowServer({
     ...(databasePath === undefined ? {} : { databasePath }),
+    ...(process.env.PI_WORKFLOWS_MAX_WORKERS === undefined
+      ? {}
+      : { maxWorkers: Number(process.env.PI_WORKFLOWS_MAX_WORKERS) }),
     onLog: (message) => process.stderr.write(`[pi-workflows server] ${message}\n`),
   });
   const shutdown = () => void server.stop();