@outputai/core 0.2.1-next.380dff4.0 → 0.2.1-next.7bd2d4e.0

package/src/tracing/tools/build_trace_tree.js

@@ -1,3 +1,4 @@
+import { EventAction } from '../trace_consts.js';
 /**
  * @typedef {object} NodeEntry
  * @property {string} id
@@ -27,19 +28,21 @@ const createEntry = id => ( {
   input: undefined,
   output: undefined,
   error: undefined,
-  children: []
+  children: [],
+  attributes: {}
 } );
 
 /**
- * Build a tree of nodes from a list of entries
+ * Builds a tree of nodes from a list of entries.
  *
  * Each node will have: id, name, kind, children, input, output or error, startedAt, endedAt.
  *
- * Entries with same id will be combined according to their phase (start, end OR error).
- * - The details of the start phase becomes input, timestamp becomes startedAt;
- * - The details of the end phase become output, timestamp becomes endedAt;
- * - The details of the error phase become error, timestamp becomes endedAt;
- * - Only start phase's kind and name are used;
+ * Entries with the same id are combined according to their actions.
+ * - The details of the START action become input, and timestamp becomes startedAt;
+ * - The details of the END action become output, timestamp becomes endedAt;
+ * - The details of the ERROR action become error, timestamp becomes endedAt;
+ * - The details of the ADD_ATTR action are attached to `.attributes`;
+ * - Only the START action's `kind` and `name` fields are used;
  *
  *
  * Children are added according to the parentId of each entry.
@@ -53,18 +56,20 @@ export default entries => {
   const ensureNode = id => nodes.get( id ) ?? nodes.set( id, createEntry( id ) ).get( id );
 
   for ( const entry of entries ) {
-    const { kind, id, name, parentId, details, phase, timestamp } = entry;
+    const { kind, id, name, parentId, details, action, timestamp } = entry;
     const node = ensureNode( id );
 
-    if ( phase === 'start' ) {
+    if ( action === EventAction.START ) {
       Object.assign( node, { input: details, startedAt: timestamp, kind, name } );
-    } else if ( phase === 'end' ) {
+    } else if ( action === EventAction.ADD_ATTR ) {
+      node.attributes[details.name] = details.value;
+    } else if ( action === EventAction.END ) {
       Object.assign( node, { output: details, endedAt: timestamp } );
-    } else if ( phase === 'error' ) {
+    } else if ( action === EventAction.ERROR ) {
       Object.assign( node, { error: details, endedAt: timestamp } );
     }
 
-    if ( parentId && phase === 'start' ) {
+    if ( parentId && action === EventAction.START ) {
       const parent = ensureNode( parentId );
       parent.children.push( node );
       parent.children.sort( ( a, b ) => a.startedAt - b.startedAt );

package/src/tracing/tools/build_trace_tree.spec.js

@@ -1,4 +1,5 @@
 import { describe, it, expect } from 'vitest';
+import { EventAction } from '../trace_consts.js';
 import buildTraceTree from './build_trace_tree.js';
 
 describe( 'build_trace_tree', () => {
@@ -6,9 +7,9 @@ describe( 'build_trace_tree', () => {
     expect( buildTraceTree( [] ) ).toBeNull();
   } );
 
-  it( 'sets root output with a fixed message when workflow has no end/error phase yet', () => {
+  it( 'sets root output with a fixed message when workflow has no end/error action yet', () => {
     const entries = [
-      { kind: 'workflow', id: 'wf', parentId: undefined, phase: 'start', name: 'wf', details: {}, timestamp: 1000 }
+      { kind: 'workflow', id: 'wf', parentId: undefined, action: EventAction.START, name: 'wf', details: {}, timestamp: 1000 }
     ];
     const result = buildTraceTree( entries );
     expect( result ).not.toBeNull();
@@ -19,17 +20,52 @@ this can indicate it timed out or was interrupted.>>' );
 
   it( 'returns null when there is no root (all entries have parentId)', () => {
     const entries = [
-      { id: 'a', parentId: 'x', phase: 'start', name: 'a', timestamp: 1 },
-      { id: 'b', parentId: 'a', phase: 'start', name: 'b', timestamp: 2 }
+      { id: 'a', parentId: 'x', action: EventAction.START, name: 'a', timestamp: 1 },
+      { id: 'b', parentId: 'a', action: EventAction.START, name: 'b', timestamp: 2 }
     ];
     expect( buildTraceTree( entries ) ).toBeNull();
   } );
 
-  it( 'error phase sets error and endedAt on node', () => {
+  it( 'add_attr action merges details.name and details.value into node.attributes', () => {
     const entries = [
-      { kind: 'wf', id: 'r', parentId: undefined, phase: 'start', name: 'root', details: {}, timestamp: 100 },
-      { kind: 'step', id: 's', parentId: 'r', phase: 'start', name: 'step', details: {}, timestamp: 200 },
-      { id: 's', phase: 'error', details: { message: 'failed' }, timestamp: 300 }
+      { kind: 'workflow', id: 'wf', parentId: undefined, action: EventAction.START, name: 'wf', details: {}, timestamp: 100 },
+      { kind: 'step', id: 's', parentId: 'wf', action: EventAction.START, name: 'step', details: {}, timestamp: 200 },
+      { id: 's', action: EventAction.ADD_ATTR, details: { name: 'latency_ms', value: 42 }, timestamp: 250 },
+      { id: 's', action: EventAction.ADD_ATTR, details: { name: 'retries', value: 1 }, timestamp: 260 },
+      { id: 'wf', action: EventAction.END, details: {}, timestamp: 300 }
+    ];
+    const result = buildTraceTree( entries );
+    expect( result ).not.toBeNull();
+    expect( result.children[0].attributes ).toEqual( { latency_ms: 42, retries: 1 } );
+  } );
+
+  it( 'add_attr action overwrites prior value for the same attribute name', () => {
+    const entries = [
+      { kind: 'workflow', id: 'wf', parentId: undefined, action: EventAction.START, name: 'wf', details: {}, timestamp: 1 },
+      { id: 'wf', action: EventAction.ADD_ATTR, details: { name: 'x', value: 1 }, timestamp: 2 },
+      { id: 'wf', action: EventAction.ADD_ATTR, details: { name: 'x', value: 2 }, timestamp: 3 },
+      { id: 'wf', action: EventAction.END, details: {}, timestamp: 4 }
+    ];
+    const result = buildTraceTree( entries );
+    expect( result.attributes ).toEqual( { x: 2 } );
+  } );
+
+  it( 'add_attr does not attach nodes as children (only start does)', () => {
+    const entries = [
+      { kind: 'workflow', id: 'wf', parentId: undefined, action: EventAction.START, name: 'wf', details: {}, timestamp: 1 },
+      { id: 'orphan', parentId: 'wf', action: EventAction.ADD_ATTR, details: { name: 'k', value: 'v' }, timestamp: 2 },
+      { id: 'wf', action: EventAction.END, details: {}, timestamp: 3 }
+    ];
+    const result = buildTraceTree( entries );
+    expect( result.children ).toHaveLength( 0 );
+    expect( result.attributes ).toEqual( {} );
+  } );
+
+  it( 'error action sets error and endedAt on node', () => {
+    const entries = [
+      { kind: 'wf', id: 'r', parentId: undefined, action: EventAction.START, name: 'root', details: {}, timestamp: 100 },
+      { kind: 'step', id: 's', parentId: 'r', action: EventAction.START, name: 'step', details: {}, timestamp: 200 },
+      { id: 's', action: EventAction.ERROR, details: { message: 'failed' }, timestamp: 300 }
     ];
     const result = buildTraceTree( entries );
     expect( result ).not.toBeNull();
@@ -41,27 +77,28 @@ this can indicate it timed out or was interrupted.>>' );
   it( 'builds a tree from workflow/step/IO entries with grouping and sorting', () => {
     const entries = [
       // workflow start
-      { kind: 'workflow', phase: 'start', name: 'wf', id: 'wf', parentId: undefined, details: { a: 1 }, timestamp: 1000 },
+      { kind: 'workflow', action: EventAction.START, name: 'wf', id: 'wf', parentId: undefined, details: { a: 1 }, timestamp: 1000 },
       // evaluator start/stop
-      { kind: 'evaluator', phase: 'start', name: 'eval', id: 'eval', parentId: 'wf', details: { z: 0 }, timestamp: 1500 },
-      { id: 'eval', phase: 'end', details: { z: 1 }, timestamp: 1600 },
+      { kind: 'evaluator', action: EventAction.START, name: 'eval', id: 'eval', parentId: 'wf', details: { z: 0 }, timestamp: 1500 },
+      { id: 'eval', action: EventAction.END, details: { z: 1 }, timestamp: 1600 },
       // step1 start
-      { kind: 'step', phase: 'start', name: 'step-1', id: 's1', parentId: 'wf', details: { x: 1 }, timestamp: 2000 },
+      { kind: 'step', action: EventAction.START, name: 'step-1', id: 's1', parentId: 'wf', details: { x: 1 }, timestamp: 2000 },
+      { id: 's1', action: EventAction.ADD_ATTR, details: { name: 'step_tag', value: 'alpha' }, timestamp: 2050 },
       // IO under step1
-      { kind: 'IO', phase: 'start', name: 'test-1', id: 'io1', parentId: 's1', details: { y: 2 }, timestamp: 2300 },
+      { kind: 'IO', action: EventAction.START, name: 'test-1', id: 'io1', parentId: 's1', details: { y: 2 }, timestamp: 2300 },
       // step2 start
-      { kind: 'step', phase: 'start', name: 'step-2', id: 's2', parentId: 'wf', details: { x: 2 }, timestamp: 2400 },
+      { kind: 'step', action: EventAction.START, name: 'step-2', id: 's2', parentId: 'wf', details: { x: 2 }, timestamp: 2400 },
       // IO under step2
-      { kind: 'IO', phase: 'start', name: 'test-2', id: 'io2', parentId: 's2', details: { y: 3 }, timestamp: 2500 },
-      { id: 'io2', phase: 'end', details: { y: 4 }, timestamp: 2600 },
+      { kind: 'IO', action: EventAction.START, name: 'test-2', id: 'io2', parentId: 's2', details: { y: 3 }, timestamp: 2500 },
+      { id: 'io2', action: EventAction.END, details: { y: 4 }, timestamp: 2600 },
       // IO under step1 ends
-      { id: 'io1', phase: 'end', details: { y: 5 }, timestamp: 2700 },
+      { id: 'io1', action: EventAction.END, details: { y: 5 }, timestamp: 2700 },
       // step1 end
-      { id: 's1', phase: 'end', details: { done: true }, timestamp: 2800 },
+      { id: 's1', action: EventAction.END, details: { done: true }, timestamp: 2800 },
       // step2 end
-      { id: 's2', phase: 'end', details: { done: true }, timestamp: 2900 },
+      { id: 's2', action: EventAction.END, details: { done: true }, timestamp: 2900 },
       // workflow end
-      { id: 'wf', phase: 'end', details: { ok: true }, timestamp: 3000 }
+      { id: 'wf', action: EventAction.END, details: { ok: true }, timestamp: 3000 }
     ];
 
     const result = buildTraceTree( entries );
@@ -74,6 +111,7 @@ this can indicate it timed out or was interrupted.>>' );
       endedAt: 3000,
       input: { a: 1 },
       output: { ok: true },
+      attributes: {},
       children: [
         {
           id: 'eval',
@@ -83,6 +121,7 @@ this can indicate it timed out or was interrupted.>>' );
           endedAt: 1600,
           input: { z: 0 },
           output: { z: 1 },
+          attributes: {},
           children: []
         },
         {
@@ -93,6 +132,7 @@ this can indicate it timed out or was interrupted.>>' );
           endedAt: 2800,
           input: { x: 1 },
           output: { done: true },
+          attributes: { step_tag: 'alpha' },
           children: [
             {
               id: 'io1',
@@ -102,6 +142,7 @@ this can indicate it timed out or was interrupted.>>' );
               endedAt: 2700,
               input: { y: 2 },
               output: { y: 5 },
+              attributes: {},
               children: []
             }
           ]
@@ -114,6 +155,7 @@ this can indicate it timed out or was interrupted.>>' );
           endedAt: 2900,
           input: { x: 2 },
           output: { done: true },
+          attributes: {},
           children: [
             {
               id: 'io2',
@@ -123,6 +165,7 @@ this can indicate it timed out or was interrupted.>>' );
               endedAt: 2600,
               input: { y: 3 },
               output: { y: 4 },
+              attributes: {},
               children: []
             }
           ]

package/src/tracing/tools/utils.js

@@ -19,3 +19,31 @@ export const serializeError = error =>
     message: error.message,
     stack: error.stack
   };
+
+/**
+ * Tries to stringify an object to an indented JSON string.
+ * If its byte size is bigger than threshold returns a plain JSON string without formatting.
+ *
+ * @param {object|array} content
+ * @param {*} [threshold] - The max allowed size to try to stringify with formatting (in bytes). Default is 50mb
+ * @returns {string} String representation of the object
+ */
+export const safeFormatJSON = ( content, threshold = 50 * 1024 * 1024 /* 50mb */ ) => {
+  const plainString = JSON.stringify( content );
+  const plainStringSize = Buffer.byteLength( plainString, 'utf8' );
+
+  if ( plainStringSize > threshold ) {
+    return plainString;
+  }
+  try {
+    return JSON.stringify( content, undefined, 2 );
+  } catch ( error ) {
+    // Only handles this specific error because other common parsing errors like:
+    // "TypeError: cyclic object value" and "RangeError: Maximum call stack size exceeded"
+    // would have been thrown on the first parsing.
+    if ( error instanceof RangeError && error.message === 'Invalid string length' ) {
+      return plainString;
+    }
+    throw error;
+  }
+};

package/src/tracing/tools/utils.spec.js

@@ -1,5 +1,15 @@
-import { describe, it, expect } from 'vitest';
-import { serializeError } from './utils.js';
+import { describe, it, expect, vi } from 'vitest';
+import { safeFormatJSON, serializeError } from './utils.js';
+
+const isPrettyStringifyCall = args => args.length >= 3 && args[2] === 2;
+
+/** @param {number} targetBytes UTF-8 size of compact JSON.stringify( { a: "<xs>" } ) */
+const objectWithCompactByteLength = targetBytes => {
+  const sample = { a: '' };
+  const overhead = Buffer.byteLength( JSON.stringify( sample ), 'utf8' );
+  const repeat = Math.max( 0, targetBytes - overhead );
+  return { a: 'x'.repeat( repeat ) };
+};
 
 describe( 'tracing/utils', () => {
   it( 'serializeError unwraps causes and keeps message/stack', () => {
@@ -11,4 +21,126 @@ describe( 'tracing/utils', () => {
     expect( out.message ).toBe( 'inner' );
     expect( typeof out.stack ).toBe( 'string' );
   } );
+
+  describe( 'safeFormatJSON', () => {
+    it( 'formats small objects with indentation when under threshold', () => {
+      const content = { a: 1, b: [ 2, 3 ] };
+      const out = safeFormatJSON( content, 10_000 );
+
+      expect( out ).toContain( '\n' );
+      expect( out ).toMatch( /^\{\n/ );
+      expect( JSON.parse( out ) ).toEqual( content );
+    } );
+
+    it( 'formats small arrays with indentation when under threshold', () => {
+      const content = [ 1, { nested: true } ];
+      const out = safeFormatJSON( content, 10_000 );
+
+      expect( out ).toContain( '\n' );
+      expect( out.trimStart() ).toMatch( /^\[/ );
+      expect( JSON.parse( out ) ).toEqual( content );
+    } );
+
+    it( 'returns compact JSON when compact UTF-8 size is strictly greater than threshold', () => {
+      const content = objectWithCompactByteLength( 40 );
+      const compact = JSON.stringify( content );
+      expect( Buffer.byteLength( compact, 'utf8' ) ).toBe( 40 );
+
+      const out = safeFormatJSON( content, 39 );
+      expect( out ).toBe( compact );
+      expect( out ).not.toContain( '\n  ' );
+      expect( JSON.parse( out ) ).toEqual( content );
+    } );
+
+    it( 'uses pretty JSON when compact UTF-8 size equals threshold', () => {
+      const content = objectWithCompactByteLength( 40 );
+      const compact = JSON.stringify( content );
+      expect( Buffer.byteLength( compact, 'utf8' ) ).toBe( 40 );
+
+      const out = safeFormatJSON( content, 40 );
+      expect( out ).not.toBe( compact );
+      expect( out ).toContain( '\n' );
+      expect( JSON.parse( out ) ).toEqual( content );
+    } );
+
+    it( 'uses UTF-8 byte length for threshold, not JavaScript string length', () => {
+      const content = { label: 'éclair' };
+      const compact = JSON.stringify( content );
+      expect( compact.length ).toBeLessThan( Buffer.byteLength( compact, 'utf8' ) );
+
+      const bytes = Buffer.byteLength( compact, 'utf8' );
+      const outCompact = safeFormatJSON( content, bytes - 1 );
+      expect( outCompact ).toBe( compact );
+
+      const outPretty = safeFormatJSON( content, bytes + 100 );
+      expect( outPretty ).toContain( '\n' );
+      expect( JSON.parse( outPretty ) ).toEqual( content );
+    } );
+
+    it( 'round-trips empty object and primitives for both branches', () => {
+      const tiny = {};
+      const pretty = safeFormatJSON( tiny, 100 );
+      expect( JSON.parse( pretty ) ).toEqual( tiny );
+
+      const forcedCompact = safeFormatJSON( tiny, 0 );
+      expect( JSON.parse( forcedCompact ) ).toEqual( tiny );
+    } );
+
+    it( 'returns compact JSON when pretty stringify throws Invalid string length', () => {
+      const content = { a: 1 };
+      const compact = JSON.stringify( content );
+      const origStringify = JSON.stringify.bind( JSON );
+
+      const spy = vi.spyOn( JSON, 'stringify' ).mockImplementation( ( ...args ) => {
+        if ( isPrettyStringifyCall( args ) ) {
+          throw new RangeError( 'Invalid string length' );
+        }
+        return origStringify( ...args );
+      } );
+
+      try {
+        const out = safeFormatJSON( content, 10_000 );
+        expect( out ).toBe( compact );
+        expect( JSON.parse( out ) ).toEqual( content );
+      } finally {
+        spy.mockRestore();
+      }
+    } );
+
+    it( 'rethrows RangeError when message is not Invalid string length', () => {
+      const content = { a: 1 };
+      const origStringify = JSON.stringify.bind( JSON );
+
+      const spy = vi.spyOn( JSON, 'stringify' ).mockImplementation( ( ...args ) => {
+        if ( isPrettyStringifyCall( args ) ) {
+          throw new RangeError( 'not the string length error' );
+        }
+        return origStringify( ...args );
+      } );
+
+      try {
+        expect( () => safeFormatJSON( content, 10_000 ) ).toThrow( RangeError );
+      } finally {
+        spy.mockRestore();
+      }
+    } );
+
+    it( 'rethrows non-RangeError from pretty stringify', () => {
+      const content = { a: 1 };
+      const origStringify = JSON.stringify.bind( JSON );
+
+      const spy = vi.spyOn( JSON, 'stringify' ).mockImplementation( ( ...args ) => {
+        if ( isPrettyStringifyCall( args ) ) {
+          throw new TypeError( 'cyclic structure' );
+        }
+        return origStringify( ...args );
+      } );
+
+      try {
+        expect( () => safeFormatJSON( content, 10_000 ) ).toThrow( TypeError );
+      } finally {
+        spy.mockRestore();
+      }
+    } );
+  } );
 } );

package/src/tracing/trace_consts.js

@@ -0,0 +1,9 @@
+/**
+ * Each possible action of a trace event
+ */
+export const EventAction = {
+  START: 'start',
+  END: 'end',
+  ERROR: 'error',
+  ADD_ATTR: 'add_attr'
+};

package/src/tracing/trace_engine.js

@@ -64,34 +64,34 @@ export const init = async () => {
 const serializeDetails = details => details instanceof Error ? serializeError( details ) : details;
 
 /**
- * Creates a new trace event phase and sends it to be written
+ * Emits an event action to the event bus.
  *
- * @param {string} phase - The phase
+ * @param {string} action - The action
  * @param {object} fields - All the trace fields
  * @returns {void}
  */
-export const addEventPhase = ( phase, { kind, name, id, parentId, details, executionContext } ) => {
+export const addEventAction = ( action, { kind, name, id, parentId, details, executionContext } ) => {
   // Ignores internal steps in the actual trace files, ignore trace if the flag is true
   if ( kind !== ComponentType.INTERNAL_STEP && !executionContext.disableTrace ) {
     traceBus.emit( 'entry', {
       executionContext,
-      entry: { kind, phase, name, id, parentId, timestamp: Date.now(), details: serializeDetails( details ) }
+      entry: { kind, action, name, id, parentId, timestamp: Date.now(), details: serializeDetails( details ) }
     } );
   }
 };
 
 /**
- * Adds an Event Phase, complementing the options with parentId and executionContext from the async storage.
+ * Attaches contextual information to an event action before calling the method to emit it to the bus.
  *
- * This function will have no effect if called from outside an Temporal Workflow/Activity environment,
- * so it is safe to be used on unit tests or any dependencies that might be used elsewhere
+ * This function has no effect if called outside a Temporal Workflow/Activity environment,
+ * so it is safe to use in unit tests or dependencies that might be used elsewhere.
  *
  * @param {object} options - The common trace configurations
  */
-export function addEventPhaseWithContext( phase, options ) {
+export function addEventActionWithContext( action, options ) {
   const storeContent = Storage.load();
-  if ( storeContent ) { // If there is no storageContext this was not called from an Temporal Environment
+  if ( storeContent ) { // If there is no storageContext this was not called from a Temporal environment
     const { parentId, executionContext } = storeContent;
-    addEventPhase( phase, { ...options, parentId, executionContext } );
+    addEventAction( action, { ...options, parentId, executionContext } );
   }
 };

package/src/tracing/trace_engine.spec.js

@@ -39,7 +39,7 @@ describe( 'tracing/trace_engine', () => {
   it( 'init() starts only enabled processors and attaches listeners', async () => {
     process.env.OUTPUT_TRACE_LOCAL_ON = '1';
     process.env.OUTPUT_TRACE_REMOTE_ON = '0';
-    const { init, addEventPhase } = await loadTraceEngine();
+    const { init, addEventAction } = await loadTraceEngine();
 
     await init();
 
@@ -47,96 +47,96 @@ describe( 'tracing/trace_engine', () => {
     expect( s3InitMock ).not.toHaveBeenCalled();
 
     const executionContext = { disableTrace: false };
-    addEventPhase( 'start', {
+    addEventAction( 'start', {
       kind: 'step', name: 'N', id: '1', parentId: 'p', details: { ok: true }, executionContext
     } );
     expect( localExecMock ).toHaveBeenCalledTimes( 1 );
     const payload = localExecMock.mock.calls[0][0];
     expect( payload.entry.name ).toBe( 'N' );
     expect( payload.entry.kind ).toBe( 'step' );
-    expect( payload.entry.phase ).toBe( 'start' );
+    expect( payload.entry.action ).toBe( 'start' );
     expect( payload.entry.details ).toEqual( { ok: true } );
     expect( payload.executionContext ).toBe( executionContext );
   } );
 
-  it( 'addEventPhase() emits an entry consumed by processors', async () => {
+  it( 'addEventAction() emits an entry consumed by processors', async () => {
     process.env.OUTPUT_TRACE_LOCAL_ON = 'on';
-    const { init, addEventPhase } = await loadTraceEngine();
+    const { init, addEventAction } = await loadTraceEngine();
     await init();
 
-    addEventPhase( 'end', {
+    addEventAction( 'end', {
       kind: 'workflow', name: 'W', id: '2', parentId: 'p2', details: 'done',
       executionContext: { disableTrace: false }
     } );
     expect( localExecMock ).toHaveBeenCalledTimes( 1 );
     const payload = localExecMock.mock.calls[0][0];
     expect( payload.entry.name ).toBe( 'W' );
-    expect( payload.entry.phase ).toBe( 'end' );
+    expect( payload.entry.action ).toBe( 'end' );
     expect( payload.entry.details ).toBe( 'done' );
   } );
 
-  it( 'addEventPhase() does not emit when executionContext.disableTrace is true', async () => {
+  it( 'addEventAction() does not emit when executionContext.disableTrace is true', async () => {
     process.env.OUTPUT_TRACE_LOCAL_ON = '1';
-    const { init, addEventPhase } = await loadTraceEngine();
+    const { init, addEventAction } = await loadTraceEngine();
     await init();
 
-    addEventPhase( 'start', {
+    addEventAction( 'start', {
       kind: 'step', name: 'X', id: '1', parentId: 'p', details: {},
       executionContext: { disableTrace: true }
     } );
     expect( localExecMock ).not.toHaveBeenCalled();
   } );
 
-  it( 'addEventPhase() does not emit when kind is INTERNAL_STEP', async () => {
+  it( 'addEventAction() does not emit when kind is INTERNAL_STEP', async () => {
     process.env.OUTPUT_TRACE_LOCAL_ON = '1';
-    const { init, addEventPhase } = await loadTraceEngine();
+    const { init, addEventAction } = await loadTraceEngine();
     await init();
 
-    addEventPhase( 'start', {
+    addEventAction( 'start', {
       kind: 'internal_step', name: 'Internal', id: '1', parentId: 'p', details: {},
       executionContext: { disableTrace: false }
     } );
     expect( localExecMock ).not.toHaveBeenCalled();
   } );
 
-  it( 'addEventPhaseWithContext() uses storage when available', async () => {
+  it( 'addEventActionWithContext() uses storage when available', async () => {
     process.env.OUTPUT_TRACE_LOCAL_ON = 'true';
     storageLoadMock.mockReturnValue( {
       parentId: 'ctx-p',
       executionContext: { runId: 'r1', disableTrace: false }
     } );
-    const { init, addEventPhaseWithContext } = await loadTraceEngine();
+    const { init, addEventActionWithContext } = await loadTraceEngine();
     await init();
 
-    addEventPhaseWithContext( 'tick', { kind: 'step', name: 'S', id: '3', details: 1 } );
+    addEventActionWithContext( 'tick', { kind: 'step', name: 'S', id: '3', details: 1 } );
     expect( localExecMock ).toHaveBeenCalledTimes( 1 );
     const payload = localExecMock.mock.calls[0][0];
     expect( payload.executionContext ).toEqual( { runId: 'r1', disableTrace: false } );
     expect( payload.entry.parentId ).toBe( 'ctx-p' );
     expect( payload.entry.name ).toBe( 'S' );
-    expect( payload.entry.phase ).toBe( 'tick' );
+    expect( payload.entry.action ).toBe( 'tick' );
   } );
 
-  it( 'addEventPhaseWithContext() does not emit when storage executionContext.disableTrace is true', async () => {
+  it( 'addEventActionWithContext() does not emit when storage executionContext.disableTrace is true', async () => {
     process.env.OUTPUT_TRACE_LOCAL_ON = '1';
     storageLoadMock.mockReturnValue( {
       parentId: 'ctx-p',
       executionContext: { runId: 'r1', disableTrace: true }
     } );
-    const { init, addEventPhaseWithContext } = await loadTraceEngine();
+    const { init, addEventActionWithContext } = await loadTraceEngine();
     await init();
 
-    addEventPhaseWithContext( 'tick', { kind: 'step', name: 'S', id: '3', details: 1 } );
+    addEventActionWithContext( 'tick', { kind: 'step', name: 'S', id: '3', details: 1 } );
     expect( localExecMock ).not.toHaveBeenCalled();
   } );
 
-  it( 'addEventPhaseWithContext() is a no-op when storage is absent', async () => {
+  it( 'addEventActionWithContext() is a no-op when storage is absent', async () => {
     process.env.OUTPUT_TRACE_LOCAL_ON = '1';
     storageLoadMock.mockReturnValue( undefined );
-    const { init, addEventPhaseWithContext } = await loadTraceEngine();
+    const { init, addEventActionWithContext } = await loadTraceEngine();
     await init();
 
-    addEventPhaseWithContext( 'noop', { kind: 'step', name: 'X', id: '4', details: null } );
+    addEventActionWithContext( 'noop', { kind: 'step', name: 'X', id: '4', details: null } );
     expect( localExecMock ).not.toHaveBeenCalled();
   } );
 

package/src/worker/configs.js

@@ -26,7 +26,14 @@ const envVarSchema = z.object( {
   // Whether to send activity heartbeats (enabled by default)
   OUTPUT_ACTIVITY_HEARTBEAT_ENABLED: z.transform( v => v === undefined ? true : isStringboolTrue( v ) ),
   // Time to allow for hooks to flush before shutdown
-  OUTPUT_PROCESS_FAILURE_SHUTDOWN_DELAY: z.preprocess( coalesceEmptyString, z.coerce.number().int().positive().default( 3000 ) )
+  OUTPUT_PROCESS_FAILURE_SHUTDOWN_DELAY: z.preprocess( coalesceEmptyString, z.coerce.number().int().positive().default( 3000 ) ),
+  // HTTP CONNECT proxy for Temporal gRPC connections (e.g. "proxy-host:8080").
+  // Must be a bare host:port — no scheme (Temporal's native HTTP CONNECT
+  // option is not a URL).
+  TEMPORAL_GRPC_PROXY: z.string().optional().refine(
+    v => !v || !v.includes( '://' ),
+    'TEMPORAL_GRPC_PROXY must be host:port without a scheme (e.g. "proxy:8080", not "http://proxy:8080")'
+  )
 } );
 
 const { data: envVars, error } = envVarSchema.safeParse( process.env );
@@ -47,3 +54,4 @@ export const catalogId = envVars.OUTPUT_CATALOG_ID;
 export const activityHeartbeatIntervalMs = envVars.OUTPUT_ACTIVITY_HEARTBEAT_INTERVAL_MS;
 export const activityHeartbeatEnabled = envVars.OUTPUT_ACTIVITY_HEARTBEAT_ENABLED;
 export const processFailureShutdownDelay = envVars.OUTPUT_PROCESS_FAILURE_SHUTDOWN_DELAY;
+export const grpcProxy = envVars.TEMPORAL_GRPC_PROXY;

package/src/worker/index.js

@@ -9,6 +9,7 @@ import { initInterceptors } from './interceptors.js';
 import { createChildLogger } from '#logger';
 import { registerShutdown } from './shutdown.js';
 import { startCatalog } from './start_catalog.js';
+import { bootstrapFetchProxy } from './proxy.js';
 import { messageBus } from '#bus';
 import './log_hooks.js';
 import { BusEventType } from '#consts';
@@ -24,6 +25,7 @@ const callerDir = process.argv[2];
     apiKey,
     namespace,
     taskQueue,
+    grpcProxy,
     maxConcurrentWorkflowTaskExecutions,
     maxConcurrentActivityTaskExecutions,
     maxCachedWorkflows,
@@ -41,6 +43,7 @@ const callerDir = process.argv[2];
   const activities = await loadActivities( callerDir, workflows );
 
   messageBus.emit( BusEventType.WORKER_BEFORE_START );
+  bootstrapFetchProxy();
 
   log.info( 'Creating worker entry point...' );
   const workflowsPath = createWorkflowsEntryPoint( workflows );
@@ -52,8 +55,11 @@ const callerDir = process.argv[2];
   const catalog = createCatalog( { workflows, activities } );
 
   log.info( 'Connecting Temporal...' );
-  // Enable TLS when connecting to remote Temporal (API key present)
-  const connection = await NativeConnection.connect( { address, tls: Boolean( apiKey ), apiKey } );
+  const proxy = grpcProxy ? { type: 'http-connect', targetHost: grpcProxy } : undefined;
+  if ( proxy ) {
+    log.info( 'Using gRPC proxy', { targetHost: grpcProxy } );
+  }
+  const connection = await NativeConnection.connect( { address, tls: Boolean( apiKey ), apiKey, proxy } );
 
   log.info( 'Creating worker...' );
   const worker = await Worker.create( {