@outputai/core 0.8.2-next.4b5c049.0 → 0.8.2-next.57bc8d6.0

package/src/tracing/trace_engine.js

@@ -1,7 +1,7 @@
 import { Storage } from '#async_storage';
 import { EventEmitter } from 'node:events';
 import { serializeError } from './tools/utils.js';
-import { isStringboolTrue } from '#utils';
+import { isStringboolTrue } from '#helpers/string';
 import * as localProcessor from './processors/local/index.js';
 import * as s3Processor from './processors/s3/index.js';
 import { ComponentType } from '#consts';

package/src/worker/catalog_workflow/catalog_job.js

@@ -3,7 +3,7 @@ import { WorkflowExecutionAlreadyStartedError, WorkflowIdConflictPolicy } from '
 import { WORKFLOW_CATALOG } from '#consts';
 import { catalogId, taskQueue } from '../configs.js';
 import { createChildLogger } from '#logger';
-import { CancellablePromise } from '#utils';
+import { CancellablePromise } from '#helpers/promise';
 
 const log = createChildLogger( 'Catalog' );
 

package/src/worker/catalog_workflow/index.spec.js

@@ -7,9 +7,6 @@ vi.mock( '#consts', () => ( {
   METADATA_ACCESS_SYMBOL
 } ) );
 
-const setMetadata = ( target, values ) =>
-  Object.defineProperty( target, METADATA_ACCESS_SYMBOL, { value: values, writable: false, enumerable: false, configurable: false } );
-
 describe( 'createCatalog', () => {
   it( 'builds catalog with activities grouped by workflow path and returns Catalog with CatalogWorkflow entries', async () => {
     const { createCatalog } = await import( './index.js' );
@@ -32,40 +29,40 @@ describe( 'createCatalog', () => {
     ];
 
     const activity1 = () => {};
-    setMetadata( activity1, {
+    activity1[METADATA_ACCESS_SYMBOL] = {
       name: 'A1',
       path: '/flows/flow1#A1',
       description: 'desc-a1',
       inputSchema: z.object( { in: z.literal( 'a1' ) } ),
       outputSchema: z.object( { out: z.literal( 'a1' ) } )
-    } );
+    };
 
     const activity2 = () => {};
-    setMetadata( activity2, {
+    activity2[METADATA_ACCESS_SYMBOL] = {
       name: 'A2',
       path: '/flows/flow1#A2',
       description: 'desc-a2',
       inputSchema: z.object( { in: z.literal( 'a2' ) } ),
       outputSchema: z.object( { out: z.literal( 'a2' ) } )
-    } );
+    };
 
     const activity3 = () => {};
-    setMetadata( activity3, {
+    activity3[METADATA_ACCESS_SYMBOL] = {
       name: 'B1',
       path: '/flows/flow2#B1',
       description: 'desc-b1',
       inputSchema: z.object( { in: z.literal( 'b1' ) } ),
       outputSchema: z.object( { out: z.literal( 'b1' ) } )
-    } );
+    };
 
     const activity4 = () => {};
-    setMetadata( activity4, {
+    activity4[METADATA_ACCESS_SYMBOL] = {
       name: 'X',
       path: '/other#X',
       description: 'desc-x',
       inputSchema: z.object( { in: z.literal( 'x' ) } ),
       outputSchema: z.object( { out: z.literal( 'x' ) } )
-    } );
+    };
 
     const activities = {
       '/flows/flow1#A1': activity1,

package/src/worker/configs.js

@@ -1,5 +1,5 @@
 import * as z from 'zod';
-import { isStringboolTrue } from '#utils';
+import { isStringboolTrue } from '#helpers/string';
 
 class InvalidEnvVarsErrors extends Error { }
 

package/src/worker/connection_monitor.js

@@ -1,6 +1,6 @@
 import { createChildLogger } from '#logger';
 import { setTimeout as delay } from 'node:timers/promises';
-import { CancellablePromise } from '#utils';
+import { CancellablePromise } from '#helpers/promise';
 
 const log = createChildLogger( 'Connection' );
 

package/src/worker/global_functions.js

@@ -1,20 +1,14 @@
 import { messageBus } from '#bus';
 import { ACTIVITY_LOGGER_SYMBOL, BusEventType } from '#consts';
 import { activityInfo as activityInfoFn } from '@temporalio/activity';
-
-const setFunction = ( key, fn ) => Object.defineProperty( globalThis, key, {
-  value: fn,
-  enumerable: false,
-  configurable: false,
-  writable: false
-} );
+import { assignImmutableProperty } from '#helpers/object';
 
 /**
  * Sets global functions on globalThis
  */
 export const bindGlobalFunctions = () => {
   /** Defines the activity logger function, accessible in activity context via logger interface */
-  setFunction( ACTIVITY_LOGGER_SYMBOL, ( { level, message, metadata } ) =>
+  assignImmutableProperty( globalThis, ACTIVITY_LOGGER_SYMBOL, ( { level, message, metadata } ) =>
     messageBus.emit( BusEventType.ACTIVITY_LOG, { level, message, metadata, activityInfo: activityInfoFn() } )
   );
 };

package/src/worker/index.js

@@ -18,7 +18,7 @@ import { BusEventType } from '#consts';
 import { setupTelemetry } from './telemetry.js';
 import { TemporalConnectionMonitor } from './connection_monitor.js';
 import { bindGlobalFunctions } from './global_functions.js';
-import { runOnce } from '#utils';
+import { runOnce } from '#helpers/function';
 
 import './log_hooks.js';
 

package/src/worker/interceptors/activity.js

@@ -5,8 +5,8 @@ import { headersToObject } from './headers.js';
 import { ACTIVITY_WRAPPER_VERSION_FIELD, BusEventType, METADATA_ACCESS_SYMBOL } from '#consts';
 import { activityHeartbeatEnabled, activityHeartbeatIntervalMs } from '../configs.js';
 import { messageBus } from '#bus';
-import { aggregateAttributes } from '#internal_utils/aggregations';
-import { buildApplicationFailureWithDetails } from '#internal_utils/errors';
+import { aggregateAttributes } from '#helpers/aggregations';
+import { buildApplicationFailureWithDetails } from '#helpers/errors';
 
 /*
   This interceptor wraps every activity execution with cross-cutting concerns:
@@ -77,20 +77,17 @@ export class ActivityExecutionInterceptor {
 
       const output = await Storage.runWithContext( async _ => next( input ), storageContext );
 
-      messageBus.emit( BusEventType.ACTIVITY_END, { activityInfo, workflowDetails, outputActivityKind } );
+      const aggregations = state.attributes.length > 0 ? aggregateAttributes( state.attributes ) : null;
+
+      messageBus.emit( BusEventType.ACTIVITY_END, { activityInfo, aggregations, workflowDetails, outputActivityKind } );
       Tracing.addEventEnd( { id: activityId, details: output, traceInfo } );
 
-      return {
-        [ACTIVITY_WRAPPER_VERSION_FIELD]: 1,
-        output,
-        aggregations: state.attributes.length > 0 ? aggregateAttributes( state.attributes ) : null
-      };
+      return { [ACTIVITY_WRAPPER_VERSION_FIELD]: 1, output, aggregations };
 
     } catch ( error ) {
-      messageBus.emit( BusEventType.ACTIVITY_ERROR, { activityInfo, workflowDetails, outputActivityKind, error } );
-      Tracing.addEventError( { id: activityId, details: error, traceInfo } );
-
       const aggregations = state.attributes.length > 0 ? aggregateAttributes( state.attributes ) : null;
+      messageBus.emit( BusEventType.ACTIVITY_ERROR, { activityInfo, aggregations, workflowDetails, outputActivityKind, error } );
+      Tracing.addEventError( { id: activityId, details: error, traceInfo } );
 
       throw aggregations ? buildApplicationFailureWithDetails( error, { aggregations } ) : error;
     } finally {

package/src/worker/interceptors/activity.spec.js

@@ -100,6 +100,12 @@ const httpRequestAttribute = {
   requestId: 'req-1'
 };
 
+const httpRequestAggregations = {
+  cost: { total: 0 },
+  tokens: { total: 0 },
+  httpRequests: { total: 1 }
+};
+
 describe( 'ActivityExecutionInterceptor', () => {
   beforeEach( () => {
     vi.clearAllMocks();
@@ -136,7 +142,7 @@ describe( 'ActivityExecutionInterceptor', () => {
     );
     expect( messageBusEmitMock ).toHaveBeenCalledWith(
       BusEventType.ACTIVITY_END,
-      { activityInfo: activityInfoMock, workflowDetails: workflowDetailsMock, outputActivityKind: 'step' }
+      { activityInfo: activityInfoMock, aggregations: null, workflowDetails: workflowDetailsMock, outputActivityKind: 'step' }
     );
     expect( addEventStartMock ).toHaveBeenCalledWith( {
       id: 'act-1',
@@ -189,14 +195,16 @@ describe( 'ActivityExecutionInterceptor', () => {
 
     await expect( interceptor.execute( makeInput(), next ) ).resolves.toEqual( {
       output: { result: 'ok' },
-      aggregations: {
-        cost: { total: 0 },
-        tokens: { total: 0 },
-        httpRequests: { total: 1 }
-      },
+      aggregations: httpRequestAggregations,
       [ACTIVITY_WRAPPER_VERSION_FIELD]: 1
     } );
 
+    expect( messageBusEmitMock ).toHaveBeenCalledWith( BusEventType.ACTIVITY_END, {
+      activityInfo: activityInfoMock,
+      aggregations: httpRequestAggregations,
+      workflowDetails: workflowDetailsMock,
+      outputActivityKind: 'step'
+    } );
   } );
 
   it( 'stores collected aggregations in ApplicationFailure details after failed execution', async () => {
@@ -215,15 +223,17 @@ describe( 'ActivityExecutionInterceptor', () => {
       message: 'step failed',
       type: 'Error',
       details: [ {
-        aggregations: {
-          cost: { total: 0 },
-          tokens: { total: 0 },
-          httpRequests: { total: 1 }
-        }
+        aggregations: httpRequestAggregations
       } ],
       cause: error
     } );
-    expect( messageBusEmitMock ).toHaveBeenCalledWith( BusEventType.ACTIVITY_ERROR, expect.objectContaining( { error } ) );
+    expect( messageBusEmitMock ).toHaveBeenCalledWith( BusEventType.ACTIVITY_ERROR, {
+      activityInfo: activityInfoMock,
+      aggregations: httpRequestAggregations,
+      workflowDetails: workflowDetailsMock,
+      outputActivityKind: 'step',
+      error
+    } );
     expect( addEventErrorMock ).toHaveBeenCalledOnce();
     expect( addEventEndMock ).not.toHaveBeenCalled();
   } );
@@ -243,13 +253,7 @@ describe( 'ActivityExecutionInterceptor', () => {
 
     expect( thrown.details ).toEqual( [
       { domain: { reason: 'bad-input' } },
-      {
-        aggregations: {
-          cost: { total: 0 },
-          tokens: { total: 0 },
-          httpRequests: { total: 1 }
-        }
-      }
+      { aggregations: httpRequestAggregations }
     ] );
   } );
 
@@ -300,13 +304,7 @@ describe( 'ActivityExecutionInterceptor', () => {
       nonRetryable: true,
       details: [
         { domain: { reason: 'bad-input' } },
-        {
-          aggregations: {
-            cost: { total: 0 },
-            tokens: { total: 0 },
-            httpRequests: { total: 1 }
-          }
-        }
+        { aggregations: httpRequestAggregations }
       ]
     } );
   } );
@@ -324,6 +322,7 @@ describe( 'ActivityExecutionInterceptor', () => {
     expect( messageBusEmitMock ).toHaveBeenCalledWith( BusEventType.ACTIVITY_START, expect.any( Object ) );
     expect( messageBusEmitMock ).toHaveBeenCalledWith( BusEventType.ACTIVITY_ERROR, {
       activityInfo: activityInfoMock,
+      aggregations: null,
       workflowDetails: workflowDetailsMock,
       outputActivityKind: 'step',
       error

package/src/worker/interceptors/workflow.js

@@ -1,10 +1,10 @@
 // THIS RUNS IN THE TEMPORAL'S SANDBOX ENVIRONMENT
 import { workflowInfo, proxySinks, ContinueAsNew, isCancellation } from '@temporalio/workflow';
 import { memoToHeaders } from './headers.js';
-import { deepMerge } from '#utils';
-import { buildApplicationFailureWithDetails } from '#internal_utils/errors';
+import { deepMerge } from '#helpers/object';
+import { buildApplicationFailureWithDetails } from '#helpers/errors';
 import { METADATA_ACCESS_SYMBOL, WorkflowSpecialOutput } from '#consts';
-import { createWorkflowDetails } from '#internal_utils/temporal_context';
+import { createWorkflowDetails } from '#helpers/temporal_context';
 
 // this is a dynamic generated file with activity configs overwrites
 import stepOptions from '../temp/__activity_options.js';

package/src/worker/interceptors/workflow.spec.js

@@ -54,7 +54,7 @@ const memoToHeadersMock = vi.fn( memo => ( memo ? { ...memo, __asHeaders: true }
 vi.mock( './headers.js', () => ( { memoToHeaders: ( ...args ) => memoToHeadersMock( ...args ) } ) );
 
 const deepMergeMock = vi.fn( ( a, b ) => ( { ...( a || {} ), ...( b || {} ) } ) );
-vi.mock( '#utils', () => ( { deepMerge: ( ...args ) => deepMergeMock( ...args ) } ) );
+vi.mock( '#helpers/object', () => ( { deepMerge: ( ...args ) => deepMergeMock( ...args ) } ) );
 
 const stepOptionsDefault = {};
 vi.mock( '../temp/__activity_options.js', () => ( { default: stepOptionsDefault } ) );

package/src/worker/loader/matchers.js

@@ -1,5 +1,5 @@
 import { sep } from 'node:path';
-import { rxEscape } from '#utils';
+import { rxEscape } from '#helpers/string';
 
 /**
  * Creates a matcher function that based on "path", matches:

package/src/worker/sinks.js

@@ -1,7 +1,7 @@
 import { BusEventType, ComponentType } from '#consts';
 import * as Tracing from '#tracing';
 import { messageBus } from '#bus';
-import { createWorkflowDetails } from '#internal_utils/temporal_context';
+import { createWorkflowDetails } from '#helpers/temporal_context';
 
 // This sink allow for sandbox Temporal environment to send trace logs back to the main thread.
 export const sinks = {

package/src/worker/sinks.spec.js

@@ -21,7 +21,7 @@ vi.mock( '#tracing', () => ( {
   addEventEnd: addEventEndMock,
   addEventError: addEventErrorMock
 } ) );
-vi.mock( '#internal_utils/temporal_context', () => ( {
+vi.mock( '#helpers/temporal_context', () => ( {
   createWorkflowDetails: createWorkflowDetailsMock
 } ) );
 

package/src/internal_utils/component.js

@@ -1,9 +0,0 @@
-import { METADATA_ACCESS_SYMBOL } from '#consts';
-
-/**
- * Add metadata "values" property to a given object
- * @param {object} target
- * @param {object} values
- */
-export const setMetadata = ( target, values ) =>
-  Object.defineProperty( target, METADATA_ACCESS_SYMBOL, { value: values, writable: false, enumerable: false, configurable: false } );

package/src/utils/index.d.ts

@@ -1,148 +0,0 @@
-/**
- * > [!WARNING]
- * > **Internal use only.** Not part of the public API; may change without notice.
- *
- * @packageDocumentation
- */
-
-/**
- * Node safe clone implementation that doesn't use global structuredClone().
- *
- * Returns a cloned version of the object.
- *
- * Only clones static properties. Getters become static properties.
- *
- * @param object
- */
-export function clone( object: object ): object;
-
-/** Represents a {Response} serialized to plain object  */
-export type SerializedFetchResponse = {
-  /** The response url */
-  url: string,
-
-  /** The response status code */
-  status: number,
-
-  /** The response status text */
-  statusText: string,
-
-  /** Flag indicating if the request succeeded */
-  ok: boolean,
-
-  /** Object with response headers */
-  headers: Record<string, string>,
-
-  /** Response body, either JSON, text or arrayBuffer converter to base64 */
-  body: object | string
-};
-
-/**
- * Consumes an HTTP `Response` and serializes it to a plain object.
- *
- * @param response - The response to serialize.
- * @returns SerializedFetchResponse
- */
-export function serializeFetchResponse( response: Response ): SerializedFetchResponse;
-
-export type SerializedBodyAndContentType = {
-  /** The body as a string when possible; otherwise the original value */
-  body: string | unknown,
-  /** The inferred `Content-Type` header value, if any */
-  contentType: string | undefined
-};
-
-/**
- * Serializes a payload for use as a fetch POST body and infers its `Content-Type`.
- *
- * @param body - The payload to serialize.
- * @returns The serialized body and inferred `Content-Type`.
- */
-export function serializeBodyAndInferContentType( body: unknown ): SerializedBodyAndContentType;
-
-/**
- * Returns true if the value is a plain object:
- * - `{}`
- * - `new Object()`
- * - `Object.create(null)`
- *
- * @param object - The value to check.
- * @returns Whether the value is a plain object.
- */
-export function isPlainObject( object: unknown ): boolean;
-
-/**
- * Returns a copy of an array with its content shuffled.
- *
- * @param arr - The array to shuffle
- * @returns A shuffled array copy
- */
-export function shuffleArray( arr: unknown[] ): unknown[];
-
-/**
- * Creates a new object by merging object `b` onto object `a`, biased toward `b`:
- * - Fields in `b` overwrite fields in `a`.
- * - Fields in `b` that don't exist in `a` are created.
- * - Fields in `a` that don't exist in `b` are left unchanged.
- *
- * @param a - The base object.
- * @param b - The overriding object.
- * @throws {Error} If either `a` or `b` is not a plain object.
- * @returns A new merged object.
- */
-export function deepMerge( a: object, b: object | null | undefined ): object;
-
-/**
- * Creates a new object by merging object `b` onto object `a`, biased toward `b`:
- * - Fields in `b` that don't exist in `a` are created.
- * - Fields in `a` that don't exist in `b` are left unchanged.
- * - Fields in `a` and `b` are passed as arguments to the resolve function (a,b) and its return assigns the new value.
- *
- * @param a - The base object.
- * @param b - The overriding object.
- * @param resolver - The resolver function.
- * @throws {Error} If either `a` or `b` is not a plain object.
- * @returns A new merged object.
- */
-export function deepMergeWithResolver( a: object, b: object | null | undefined, resolver: function ): object;
-
-/**
- * Shortens a UUID to a url-safe base64-like string (custom 64-char alphabet).
- * Temporal-friendly: no Buffer or crypto; safe to use inside workflows.
- *
- * @param uuid - Standard UUID (e.g. `xxxxxxxx-xxxx-xxxx-xxxx-xxxxxxxxxxxx`).
- * @returns Short string using A–Z, a–z, 0–9, `_`, `-` (typically 21–22 chars).
- */
-export function toUrlSafeBase64( uuid: string ): string;
-
-/**
- * Similar to native Promise.allSettled, but rejects with `{ isTimeout: true }`
- * if the execution exceeds the given timeout.
- *
- * @param promises - Values or promises to wait for.
- * @param timeoutMs - Maximum wait time in milliseconds.
- * @returns Native Promise.allSettled-style results.
- */
-export function allSettledWithTimeout<T>(
-  promises: Array<T | PromiseLike<T>>,
-  timeoutMs: number
-): Promise<PromiseSettledResult<Awaited<T>>[]>;
-
-/**
- * Promise wrapper that can be resolved externally.
- */
-export class CancellablePromise {
-  /** The internal promise */
-  readonly promise: Promise<void>;
-  /** Whether the promise is already resolved or not */
-  readonly completed: boolean;
-  /** Resolves the promise */
-  complete(): void;
-}
-
-/**
- * Returns a function that invokes the wrapped function once.
- */
-export function runOnce<Args extends unknown[], Return>(
-  fn: ( ...args: Args ) => Return
-): ( ...args: Args ) => Return;

package/src/utils/index.js

@@ -1 +0,0 @@
-export * from './utils.js';