elasticdash-sdk 0.2.6 → 0.2.7-beta

package/src/workflow-runner-worker.ts

@@ -43,6 +43,13 @@ async function readStdin(): Promise<string> {
 
 const RESULT_PREFIX = '__ELASTICDASH_RESULT__:'
 
+const WORKER_START_MS = Date.now()
+function stage(name: string, extra?: Record<string, unknown>): void {
+  if (process.env.ELASTICDASH_DEBUG !== '1') return
+  const tail = extra ? ' ' + Object.entries(extra).map(([k, v]) => `${k}=${typeof v === 'string' ? v : JSON.stringify(v)}`).join(' ') : ''
+  process.stderr.write(`[elasticdash-worker workflow] stage=${name} pid=${process.pid} elapsedMs=${Date.now() - WORKER_START_MS}${tail}\n`)
+}
+
 const IS_DENO = typeof (globalThis as any).Deno !== 'undefined'
 
 /** Write the result JSON to fd3 pipe (Node.js) or a prefixed stdout line (Deno).
@@ -215,10 +222,12 @@ async function loadAndWrapTools(
 }
 
 async function main() {
+  stage('boot')
   // Keep a reference to the real process.exit so we can call it after flushing stdout.
   const originalExit = process.exit.bind(process)
 
   const raw = await readStdin()
+  stage('stdin-eof', { bytes: raw.length })
 
   let payload: {
     workflowsModulePath: string
@@ -249,6 +258,7 @@ async function main() {
     originalExit(1)
     return
   }
+  stage('payload-parsed')
 
   const { workflowsModulePath, toolsModulePath, workflowName, args, input, replayMode = false, checkpoint = 0, history = [], agentState, toolMockConfig, aiMockConfig, promptMockConfig, userPromptMockConfig, strict } = payload
 
@@ -273,6 +283,9 @@ async function main() {
       originalValues[name] = globals[name]
       globals[name] = fn
     }
+    stage('tools-wrapped', { count: Object.keys(wrappedTools).length })
+  } else {
+    stage('tools-wrapped-skipped')
   }
 
   // Intercept process.exit() so that workflows that call it internally (e.g. agent
@@ -305,20 +318,25 @@ async function main() {
     interceptFetch()
     interceptRandom()
     interceptDateNow()
+    stage('interceptors-installed')
 
     try {
       if (agentState) {
         // Agent mid-trace resumption path: load ed_agents and resume from saved state
         const agentsModulePath = workflowsModulePath.replace(/ed_workflows(\.[^.]+)?$/, 'ed_agents$1')
         const agentsMod = await import(pathToFileURL(agentsModulePath).href)
+        stage('agents-module-imported')
         if (typeof agentsMod.resumeAgentFromTrace !== 'function') {
           throw new Error(`"resumeAgentFromTrace" is not an exported function in ${agentsModulePath}`)
         }
+        stage('workflow-call-start', { mode: 'agent-resume' })
         currentOutput = await (agentsMod.resumeAgentFromTrace as (s: AgentState) => Promise<unknown>)(agentState)
+        stage('workflow-call-end', { mode: 'agent-resume' })
         console.error('[worker] resumeAgentFromTrace resolved, currentOutput:', currentOutput)
       } else {
         // Standard workflow path
         const workflowsMod = await import(pathToFileURL(workflowsModulePath).href)
+        stage('workflow-module-imported')
         const workflowFn = workflowsMod[workflowName]
         if (typeof workflowFn !== 'function') {
           ;(process as NodeJS.Process).exit = originalExit
@@ -328,7 +346,9 @@ async function main() {
         }
         // Standardize workflow argument resolution: always pass [input] if args is empty
         const callArgs = args.length ? args : [input]
+        stage('workflow-call-start', { workflow: workflowName })
         currentOutput = await (workflowFn as (...a: unknown[]) => unknown)(...callArgs)
+        stage('workflow-call-end', { workflow: workflowName })
         console.error('[worker] workflowFn resolved, currentOutput:', currentOutput)  // stderr so it's visible
       }
     } finally {
@@ -369,6 +389,7 @@ async function main() {
   }
 
   await recorder.flush()
+  stage('recorder-flushed')
 
   const traceData = {
     steps: context.trace.getSteps(),
@@ -380,9 +401,11 @@ async function main() {
 
   if (workflowError) {
     await writeResult({ ok: false, error: workflowError.message ?? String(workflowError), ...traceData })
+    stage('result-written', { ok: false })
     originalExit(pendingExitCode ?? 1)
   } else {
     await writeResult({ ok: true, currentOutput, ...traceData })
+    stage('result-written', { ok: true })
     originalExit(pendingExitCode ?? 0)
   }
 }