@osolmaz/pi-workflows 0.12.1 → 0.13.1

package/src/workflows/index.ts

@@ -1,5 +1,6 @@
 export {
   agent,
+  assistantMessage,
   action,
   checkpoint,
   compute,
@@ -38,7 +39,7 @@ export {
   digest,
   humanDecision,
   humanDecisionEdge,
-  humanDecisionStateRoot,
+  humanDecisionDatabasePath,
   textInput,
   validateHumanDecisionResponse,
   validateHumanDecisionSubmission,
@@ -112,42 +113,37 @@ export {
   validateWorkflowUpdate,
 } from "./updates.js";
 export { sanitizeText, stripAnsi } from "./text.js";
-export {
-  ARTIFACT_THRESHOLD_BYTES,
-  ArtifactWriter,
-  decodeValueWith,
-  encodeValue,
-  isArtifactValue,
-  resolveArtifacts,
-} from "./artifacts.js";
 export {
   DEFINITION_SNAPSHOT_SCHEMA,
-  RUN_BUNDLE_SCHEMA,
   RUN_STATE_SCHEMA,
   SESSION_BINDING_SCHEMA,
-  TRACE_EVENT_SCHEMA,
   WorkflowRunStore,
   createDefinitionSnapshot,
   createRunId,
-  listRunBundles,
-  readRunBundle,
-  workflowRunsBaseDir,
-  type LoadedRunBundle,
-  type ReadRunBundleOptions,
+  listWorkflowRuns,
+  readLastTraceEvent,
+  readWorkflowRun,
+  workflowStateDatabasePath,
+  type LoadedWorkflowRun,
+  type ReadWorkflowRunOptions,
+  type RunWriteAuthority,
 } from "./store.js";
 export type {
   AcceptedHumanDecision,
   DefaultedHumanDecision,
   ResolvedHumanDecision,
   AgentNodeDefinition,
+  AgentExpectedOutput,
+  AgentStepCompletion,
   AgentStepContract,
+  AssistantAgentNodeDefinition,
+  AssistantMessageOutput,
+  AssistantMessageReceipt,
   AgentStepExecutor,
   AgentStepPresentation,
   AgentStepRequest,
   AgentStepSubmission,
   ActionNodeDefinition,
-  ArtifactRef,
-  ArtifactValue,
   CheckpointNodeDefinition,
   ConversationRange,
   ComputeNodeDefinition,
@@ -181,6 +177,7 @@ export type {
   NotifyNodeDefinition,
   ShellActionExecution,
   ShellActionNodeDefinition,
+  SubmittedAgentNodeDefinition,
   ShellActionResult,
   WorkflowActionContext,
   WorkflowActionReceipt,
@@ -210,7 +207,6 @@ export type {
   WorkflowPresentationContext,
   WorkflowProgressData,
   WorkflowProgressStatus,
-  WorkflowRunManifest,
   WorkflowRunResult,
   WorkflowRunState,
   WorkflowRunStatus,

package/src/workflows/schema.ts

@@ -51,7 +51,25 @@ export function assertValidAgentNode(node: AgentNodeDefinition, nodeId = "agent"
     fail(`node ${nodeId} requires a prompt function`);
   }
   if (node.expectedOutput !== undefined && typeof node.expectedOutput !== "string") {
-    fail(`node ${nodeId} expectedOutput must be a string`);
+    const output = node.expectedOutput as { kind?: unknown; maxChars?: unknown };
+    if (output === null || typeof output !== "object" || output.kind !== "assistant-message") {
+      fail(`node ${nodeId} expectedOutput must be a string or assistantMessage()`);
+    }
+    const unknown = Object.keys(output).filter((key) => key !== "kind" && key !== "maxChars");
+    if (unknown.length > 0) {
+      fail(`node ${nodeId} assistantMessage has unknown field ${JSON.stringify(unknown[0])}`);
+    }
+    if (
+      output.maxChars !== undefined &&
+      (typeof output.maxChars !== "number" ||
+        !Number.isInteger(output.maxChars) ||
+        output.maxChars <= 0)
+    ) {
+      fail(`node ${nodeId} assistantMessage maxChars must be a positive integer`);
+    }
+    if (node.validate !== undefined) {
+      fail(`node ${nodeId} cannot use validate with assistantMessage()`);
+    }
   }
   assertOptionalFunction(node.validate, `node ${nodeId} validate`);
   assertCommonNodeFields(node, nodeId);