@hatchet-dev/typescript-sdk 1.19.0 → 1.19.2-alpha.0

package/clients/event/event-client.js

@@ -14,6 +14,7 @@ const events_1 = require("../../protoc/events/events");
 const hatchet_error_1 = require("../../util/errors/hatchet-error");
 const retrier_1 = require("../../util/retrier");
 const apply_namespace_1 = require("../../util/apply-namespace");
+const parent_run_context_vars_1 = require("../../v1/parent-run-context-vars");
 var LogLevel;
 (function (LogLevel) {
     LogLevel["INFO"] = "INFO";
@@ -21,6 +22,13 @@ var LogLevel;
     LogLevel["ERROR"] = "ERROR";
     LogLevel["DEBUG"] = "DEBUG";
 })(LogLevel || (exports.LogLevel = LogLevel = {}));
+function injectSourceInfo(metadata) {
+    const ctx = parent_run_context_vars_1.parentRunContextManager.getContext();
+    if (!(ctx === null || ctx === void 0 ? void 0 : ctx.parentId) || !(ctx === null || ctx === void 0 ? void 0 : ctx.parentTaskRunExternalId)) {
+        return metadata;
+    }
+    return Object.assign(Object.assign({}, metadata), { hatchet__source_workflow_run_id: ctx.parentId, hatchet__source_step_run_id: ctx.parentTaskRunExternalId });
+}
 class EventClient {
     constructor(config, channel, factory, api) {
         this.config = config;
@@ -35,14 +43,14 @@ class EventClient {
      * Keep the signature in sync with the instrumentor wrapper.
      */
     push(type, input, options = {}) {
+        var _a;
         const namespacedType = (0, apply_namespace_1.applyNamespace)(type, this.config.namespace);
+        const enhancedMetadata = injectSourceInfo((_a = options.additionalMetadata) !== null && _a !== void 0 ? _a : {});
         const req = {
             key: namespacedType,
             payload: JSON.stringify(input),
             eventTimestamp: new Date(),
-            additionalMetadata: options.additionalMetadata
-                ? JSON.stringify(options.additionalMetadata)
-                : undefined,
+            additionalMetadata: Object.keys(enhancedMetadata).length > 0 ? JSON.stringify(enhancedMetadata) : undefined,
             priority: options.priority,
             scope: options.scope,
         };
@@ -62,19 +70,14 @@ class EventClient {
     bulkPush(type, inputs, options = {}) {
         const namespacedType = (0, apply_namespace_1.applyNamespace)(type, this.config.namespace);
         const events = inputs.map((input) => {
+            var _a, _b;
+            const baseMeta = (_b = (_a = input.additionalMetadata) !== null && _a !== void 0 ? _a : options.additionalMetadata) !== null && _b !== void 0 ? _b : {};
+            const enhanced = injectSourceInfo(baseMeta);
             return {
                 key: namespacedType,
                 payload: JSON.stringify(input.payload),
                 eventTimestamp: new Date(),
-                additionalMetadata: (() => {
-                    if (input.additionalMetadata) {
-                        return JSON.stringify(input.additionalMetadata);
-                    }
-                    if (options.additionalMetadata) {
-                        return JSON.stringify(options.additionalMetadata);
-                    }
-                    return undefined;
-                })(),
+                additionalMetadata: Object.keys(enhanced).length > 0 ? JSON.stringify(enhanced) : undefined,
                 priority: input.priority,
                 scope: input.scope,
             };

package/opentelemetry/hatchet-exporter.js

@@ -96,9 +96,19 @@ class HatchetExporterWrapper {
  */
 class HatchetAttributeSpanProcessor extends BatchSpanProcessor {
     onStart(span) {
+        var _a;
         const attrs = hatchet_span_context_1.hatchetSpanAttributes.getStore();
         if (attrs) {
-            span.setAttributes(attrs);
+            const existing = (_a = span.attributes) !== null && _a !== void 0 ? _a : {};
+            const filtered = {};
+            for (const [key, value] of Object.entries(attrs)) {
+                if (!(key in existing)) {
+                    filtered[key] = value;
+                }
+            }
+            if (Object.keys(filtered).length > 0) {
+                span.setAttributes(filtered);
+            }
         }
         super.onStart(span, undefined);
     }
@@ -108,10 +118,14 @@ class HatchetAttributeSpanProcessor extends BatchSpanProcessor {
 }
 function createHatchetExporter(config) {
     const insecure = config.tls_config.tls_strategy === 'none';
+    // eslint-disable-next-line @typescript-eslint/no-require-imports
+    const grpc = require('@grpc/grpc-js');
+    const metadata = new grpc.Metadata();
+    metadata.set('authorization', `Bearer ${config.token}`);
     // eslint-disable-next-line @typescript-eslint/no-explicit-any
     const opts = {
         url: `${insecure ? 'http' : 'https'}://${config.host_port}`,
-        metadata: { authorization: `Bearer ${config.token}` },
+        metadata,
     };
     if (!insecure && config.tls_config.ca_file) {
         try {

package/opentelemetry/instrumentor.js

@@ -46,6 +46,17 @@ function extractContext(carrier) {
 function injectContext(carrier) {
     propagation.inject(context.active(), carrier);
 }
+function injectSourceInfo(carrier) {
+    const store = hatchet_span_context_1.hatchetSpanAttributes.getStore();
+    if (!store)
+        return;
+    const wfRunId = store['hatchet.workflow_run_id'];
+    const stepRunId = store['hatchet.step_run_id'];
+    if (typeof wfRunId === 'string' && typeof stepRunId === 'string') {
+        carrier['hatchet__source_workflow_run_id'] = wfRunId;
+        carrier['hatchet__source_step_run_id'] = stepRunId;
+    }
+}
 function getActionOtelAttributes(action, excludedAttributes = [], workerId) {
     const attributes = {
         [opentelemetry_1.OTelAttribute.TENANT_ID]: action.tenantId,
@@ -215,6 +226,7 @@ class HatchetInstrumentor extends InstrumentationBase {
                     var _a;
                     const enhancedMetadata = Object.assign({}, ((_a = options.additionalMetadata) !== null && _a !== void 0 ? _a : {}));
                     injectContext(enhancedMetadata);
+                    injectSourceInfo(enhancedMetadata);
                     const enhancedOptions = Object.assign(Object.assign({}, options), { additionalMetadata: enhancedMetadata });
                     const result = original.call(this, type, input, enhancedOptions);
                     return result.finally(() => {
@@ -247,6 +259,7 @@ class HatchetInstrumentor extends InstrumentationBase {
                         var _a;
                         const enhancedMetadata = Object.assign({}, ((_a = input.additionalMetadata) !== null && _a !== void 0 ? _a : {}));
                         injectContext(enhancedMetadata);
+                        injectSourceInfo(enhancedMetadata);
                         return Object.assign(Object.assign({}, input), { additionalMetadata: enhancedMetadata });
                     });
                     const result = original.call(this, type, enhancedInputs, options);

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@hatchet-dev/typescript-sdk",
-  "version": "1.19.0",
+  "version": "1.19.2-alpha.0",
   "description": "Background task orchestration & visibility for developers",
   "types": "dist/index.d.ts",
   "files": [
@@ -48,7 +48,7 @@
   "dependencies": {
     "@bufbuild/protobuf": "^2.11.0",
     "@types/qs": "^6.15.0",
-    "abort-controller-x": "^0.4.3",
+    "abort-controller-x": "^0.5.0",
     "axios": "^1.13.5",
     "long": "^5.3.1",
     "nice-grpc": "^2.1.14",
@@ -61,10 +61,11 @@
     "zod-to-json-schema": "^3.24.1"
   },
   "optionalDependencies": {
+    "@grpc/grpc-js": "^1.14.3",
     "@opentelemetry/api": "^1.9.0",
     "@opentelemetry/core": "^2.0.0",
-    "@opentelemetry/exporter-trace-otlp-grpc": "^0.208.0",
-    "@opentelemetry/instrumentation": "^0.208.0",
+    "@opentelemetry/exporter-trace-otlp-grpc": "^0.214.0",
+    "@opentelemetry/instrumentation": "^0.214.0",
     "@opentelemetry/sdk-trace-base": "^2.0.0",
     "prom-client": "^15.1.3"
   },

package/v1/client/worker/worker-internal.js

@@ -95,10 +95,10 @@ class InternalWorker {
         this.labels = options.labels || {};
         this.enableHealthServer = (_b = (_a = client.config.healthcheck) === null || _a === void 0 ? void 0 : _a.enabled) !== null && _b !== void 0 ? _b : false;
         this.healthPort = (_d = (_c = client.config.healthcheck) === null || _c === void 0 ? void 0 : _c.port) !== null && _d !== void 0 ? _d : 8001;
-        process.on('SIGTERM', () => this.exitGracefully(true));
-        process.on('SIGINT', () => this.exitGracefully(true));
         this.killing = false;
         this.handle_kill = options.handleKill === undefined ? true : options.handleKill;
+        process.on('SIGTERM', () => this.exitGracefully(this.handle_kill));
+        process.on('SIGINT', () => this.exitGracefully(this.handle_kill));
         this.logger = client.config.logger(`Worker/${this.name}`, this.client.config.log_level);
         if (this.enableHealthServer && this.healthPort) {
             this.initializeHealthServer();

package/v1/declaration.js

@@ -63,9 +63,12 @@ class BaseWorkflowDeclaration {
             if (!this.client) {
                 throw UNBOUND_ERR;
             }
-            // set the parent run context
             const parentRunContext = parent_run_context_vars_1.parentRunContextManager.getContext();
-            parent_run_context_vars_1.parentRunContextManager.incrementChildIndex(Array.isArray(input) ? input.length : 1);
+            // Snapshot the child index BEFORE incrementing — getContext() returns a mutable
+            // reference, so any later increment would corrupt this value.
+            const childIndex = parentRunContext === null || parentRunContext === void 0 ? void 0 : parentRunContext.childIndex;
+            const inputCount = Array.isArray(input) ? input.length : 1;
+            parent_run_context_vars_1.parentRunContextManager.incrementChildIndex(inputCount);
             if (!parentRunContext && ((options === null || options === void 0 ? void 0 : options.childKey) || (options === null || options === void 0 ? void 0 : options.sticky))) {
                 this.client.admin.logger.warn('ignoring childKey or sticky because run is not being spawned from a parent task');
             }
@@ -79,18 +82,17 @@ class BaseWorkflowDeclaration {
                     : undefined,
                 warn: (message) => this.client.admin.logger.warn(message),
             });
-            parent_run_context_vars_1.parentRunContextManager.incrementChildIndex(Array.isArray(input) ? input.length : 1);
-            const runOpts = Object.assign(Object.assign({}, (options !== null && options !== void 0 ? options : {})), { parentId: parentRunContext === null || parentRunContext === void 0 ? void 0 : parentRunContext.parentId, parentTaskRunExternalId: parentRunContext === null || parentRunContext === void 0 ? void 0 : parentRunContext.parentTaskRunExternalId, childIndex: parentRunContext === null || parentRunContext === void 0 ? void 0 : parentRunContext.childIndex, sticky: (options === null || options === void 0 ? void 0 : options.sticky) ? parentRunContext === null || parentRunContext === void 0 ? void 0 : parentRunContext.desiredWorkerId : undefined, childKey: options === null || options === void 0 ? void 0 : options.childKey });
+            const runOpts = Object.assign(Object.assign({}, (options !== null && options !== void 0 ? options : {})), { parentId: parentRunContext === null || parentRunContext === void 0 ? void 0 : parentRunContext.parentId, parentTaskRunExternalId: parentRunContext === null || parentRunContext === void 0 ? void 0 : parentRunContext.parentTaskRunExternalId, childIndex, sticky: (options === null || options === void 0 ? void 0 : options.sticky) ? parentRunContext === null || parentRunContext === void 0 ? void 0 : parentRunContext.desiredWorkerId : undefined, childKey: options === null || options === void 0 ? void 0 : options.childKey });
             if (Array.isArray(input)) {
                 let resp = [];
                 for (let i = 0; i < input.length; i += 500) {
                     const batch = input.slice(i, i + 500);
-                    const batchResp = yield this.client.admin.runWorkflows(batch.map((inp) => {
+                    const batchResp = yield this.client.admin.runWorkflows(batch.map((inp, batchIdx) => {
                         var _a;
                         return ({
                             workflowName: this.definition.name,
                             input: inp,
-                            options: Object.assign(Object.assign({}, runOpts), { childIndex: ((_a = runOpts.childIndex) !== null && _a !== void 0 ? _a : 0) + i }),
+                            options: Object.assign(Object.assign({}, runOpts), { childIndex: ((_a = runOpts.childIndex) !== null && _a !== void 0 ? _a : 0) + i + batchIdx }),
                         });
                     }));
                     resp = resp.concat(batchResp);

package/v1/parent-run-context-vars.js

@@ -16,18 +16,16 @@ class ParentRunContextManager {
         var _a;
         const parentRunContext = this.getContext();
         if (parentRunContext) {
-            parentRunContext.parentId = opts.parentId;
-            parentRunContext.parentTaskRunExternalId = opts.parentTaskRunExternalId;
-            parentRunContext.childIndex = ((_a = parentRunContext.childIndex) !== null && _a !== void 0 ? _a : 0) + 1;
-            this.setContext(parentRunContext);
+            this.setContext(Object.assign(Object.assign({}, parentRunContext), { parentId: opts.parentId, parentTaskRunExternalId: opts.parentTaskRunExternalId, childIndex: ((_a = parentRunContext.childIndex) !== null && _a !== void 0 ? _a : 0) + 1 }));
         }
     }
     incrementChildIndex(n) {
         var _a;
         const parentRunContext = this.getContext();
         if (parentRunContext) {
-            parentRunContext.childIndex = ((_a = parentRunContext.childIndex) !== null && _a !== void 0 ? _a : 0) + n;
-            this.setContext(parentRunContext);
+            // Build a fresh object to avoid mutating the live store reference, which
+            // would corrupt any values snapshotted before this call.
+            this.setContext(Object.assign(Object.assign({}, parentRunContext), { childIndex: ((_a = parentRunContext.childIndex) !== null && _a !== void 0 ? _a : 0) + n }));
         }
     }
     getContext() {

package/version.d.ts

@@ -1 +1 @@
-export declare const HATCHET_VERSION = "1.19.0";
+export declare const HATCHET_VERSION = "1.19.2-alpha.0";

package/version.js

@@ -1,4 +1,4 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.HATCHET_VERSION = void 0;
-exports.HATCHET_VERSION = '1.19.0';
+exports.HATCHET_VERSION = '1.19.2-alpha.0';