@pikku/core 0.12.20 → 0.12.21

package/src/wirings/workflow/graph/graph-runner.test.ts

@@ -526,4 +526,102 @@ describe('graph-runner bugs', () => {
 
     delete metaState['testInlineMetaGraph']
   })
+
+  test('queueGraphNode forwards node retries to queue attempts and backoff', async () => {
+    const ws = new InMemoryWorkflowService()
+    const enqueued: Array<{ queueName: string; data: any; options: any }> = []
+
+    pikkuState(null, 'package', 'singletonServices', {
+      queueService: {
+        add: async (queueName: string, data: any, options: any) => {
+          enqueued.push({ queueName, data, options })
+        },
+      },
+    } as any)
+
+    const meta: WorkflowRuntimeMeta = {
+      name: 'testRetries',
+      pikkuFuncId: 'testRetries',
+      source: 'graph',
+      entryNodeIds: ['a'],
+      graphHash: 'retries-hash',
+      nodes: {
+        a: {
+          nodeId: 'a',
+          rpcName: 'doA',
+          retries: 3,
+          retryDelay: 250,
+        },
+        b: {
+          nodeId: 'b',
+          rpcName: 'doB',
+          retries: 2,
+          retryDelay: 'exponential',
+        },
+        c: {
+          nodeId: 'c',
+          rpcName: 'doC',
+        },
+      },
+    }
+
+    const { runId } = await runWorkflowGraph(
+      ws,
+      'testRetries',
+      {},
+      { rpcWithWire: async () => ({}) },
+      false,
+      undefined,
+      undefined,
+      meta
+    )
+
+    const aJob = enqueued.find((e) => e.data?.stepName === 'a')
+    assert.ok(aJob, 'node a should have been enqueued')
+    assert.equal(aJob!.options?.attempts, 4, 'retries=3 → attempts=4')
+    assert.deepEqual(aJob!.options?.backoff, { type: 'fixed', delay: 250 })
+
+    const stepStateA = await ws.getStepState(runId, 'a')
+    assert.equal(
+      stepStateA.retries,
+      3,
+      'insertStepState should record node retries'
+    )
+    assert.equal(stepStateA.retryDelay, 250)
+
+    enqueued.length = 0
+    await runWorkflowGraph(
+      ws,
+      'testRetries',
+      {},
+      { rpcWithWire: async () => ({}) },
+      false,
+      'b',
+      undefined,
+      meta
+    )
+    const bJob = enqueued.find((e) => e.data?.stepName === 'b')
+    assert.ok(bJob, 'node b should have been enqueued')
+    assert.equal(bJob!.options?.attempts, 3, 'retries=2 → attempts=3')
+    assert.equal(bJob!.options?.backoff, 'exponential')
+
+    enqueued.length = 0
+    await runWorkflowGraph(
+      ws,
+      'testRetries',
+      {},
+      { rpcWithWire: async () => ({}) },
+      false,
+      'c',
+      undefined,
+      meta
+    )
+    const cJob = enqueued.find((e) => e.data?.stepName === 'c')
+    assert.ok(cJob, 'node c should have been enqueued')
+    assert.equal(
+      cJob!.options,
+      undefined,
+      'no retries → no queue options (preserves prior queue defaults)'
+    )
+  })
 })

package/src/wirings/workflow/graph/graph-runner.ts

@@ -324,12 +324,27 @@ async function queueGraphNode(
   _graphName: string,
   nodeId: string,
   rpcName: string,
-  input: any
+  input: any,
+  nodeConfig?: { retries?: number; retryDelay?: string | number }
 ): Promise<void> {
-  await workflowService.insertStepState(runId, nodeId, rpcName, input, {
-    retries: 0,
-  })
-  await workflowService.queueStepWorker(runId, nodeId, rpcName, input)
+  const stepOptions = {
+    retries: nodeConfig?.retries ?? 0,
+    retryDelay: nodeConfig?.retryDelay,
+  }
+  await workflowService.insertStepState(
+    runId,
+    nodeId,
+    rpcName,
+    input,
+    stepOptions
+  )
+  await workflowService.queueStepWorker(
+    runId,
+    nodeId,
+    rpcName,
+    input,
+    stepOptions
+  )
 }
 
 export async function continueGraph(
@@ -442,7 +457,8 @@ export async function continueGraph(
       graphName,
       nodeId,
       node.rpcName,
-      resolvedInput
+      resolvedInput,
+      node
     )
   }
 }
@@ -553,7 +569,8 @@ export async function executeGraphStep(
               graphName,
               errorNodeId,
               errorNode.rpcName,
-              { error: { message: (error as Error).message } }
+              { error: { message: (error as Error).message } },
+              errorNode
             )
           }
         }
@@ -600,7 +617,7 @@ async function executeGraphNodeInline(
     nodeId,
     rpcName,
     input,
-    { retries: 3 }
+    { retries: node.retries ?? 0, retryDelay: node.retryDelay }
   )
 
   await workflowService.setStepRunning(stepState.stepId)
@@ -677,7 +694,7 @@ async function executeGraphNodeInline(
         message: `RPC '${rpcName}' not found. Deploy the missing function and resume.`,
         code: 'RPC_NOT_FOUND',
       })
-      return
+      throw new WorkflowSuspendedException(runId, 'RPC_NOT_FOUND')
     }
     await workflowService.setStepError(stepState.stepId, error as Error)
 
@@ -932,7 +949,8 @@ export async function runWorkflowGraph(
         graphName,
         nodeId,
         node.rpcName,
-        resolvedInput
+        resolvedInput,
+        node
       )
     }
     if (inline) {

package/src/wirings/workflow/graph/workflow-graph.types.ts

@@ -65,6 +65,10 @@ export interface GraphNodeConfig<NodeIds extends string = string> {
   next?: NextConfig<NodeIds>
   /** Error routing - node(s) to execute on error */
   onError?: NodeIds | NodeIds[]
+  /** Maximum retry attempts allowed (excludes the initial attempt) */
+  retries?: number
+  /** Delay between retries — milliseconds, duration string, or 'exponential' */
+  retryDelay?: string | number
 }
 
 /**

package/src/wirings/workflow/index.ts

@@ -51,6 +51,7 @@ export type {
   WorkflowRunStatus,
   StepState,
   WorkflowRunService,
+  WorkflowRunMirror,
   CoreWorkflow,
   PikkuWorkflow,
   ContextVariable,

package/src/wirings/workflow/pikku-workflow-service.test.ts

@@ -2,13 +2,30 @@ import { describe, test } from 'node:test'
 import assert from 'node:assert/strict'
 
 import { InMemoryWorkflowService } from '../../services/in-memory-workflow-service.js'
-import { pikkuState } from '../../pikku-state.js'
+import { pikkuState, resetPikkuState } from '../../pikku-state.js'
 import { addWorkflow } from './dsl/workflow-runner.js'
 import {
   WorkflowSuspendedException,
   type PikkuWorkflowWire,
 } from './pikku-workflow-service.js'
 
+describe('pikku-workflow-service worker registration', () => {
+  test('registers sleeper function metadata', () => {
+    resetPikkuState()
+
+    new InMemoryWorkflowService()
+
+    const functionMeta = pikkuState(null, 'function', 'meta')
+    assert.deepEqual(functionMeta.pikkuWorkflowSleeper, {
+      pikkuFuncId: 'pikkuWorkflowSleeper',
+      sessionless: true,
+      functionType: 'helper',
+      inputSchemaName: null,
+      outputSchemaName: null,
+    })
+  })
+})
+
 describe('pikku-workflow-service inline meta flag', () => {
   test('workflow with inline: true in meta should create inline run and not queue', async () => {
     const ws = new InMemoryWorkflowService()
@@ -53,7 +70,7 @@ describe('pikku-workflow-service inline meta flag', () => {
       },
     })
 
-    const { runId } = await ws.startWorkflow(workflowName, {}, {})
+    const { runId } = await ws.startWorkflow(workflowName, {}, {}, {})
 
     // The run should be created as inline
     const run = await ws.getRun(runId)