@hatchet-dev/typescript-sdk 1.13.1 → 1.15.0

package/docs/doc_types.d.ts

@@ -0,0 +1,12 @@
+export interface Document {
+    sourcePath: string;
+    readableSourcePath: string;
+    mdxOutputPath: string;
+    mdxOutputMetaJsPath: string;
+    isIndex: boolean;
+    directory: string;
+    basename: string;
+    title: string;
+    metaJsEntry: string;
+}
+export declare function documentFromPath(filePath: string): Document;

package/docs/doc_types.js

@@ -0,0 +1,62 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.documentFromPath = documentFromPath;
+const path_1 = __importDefault(require("path"));
+const shared_1 = require("./shared");
+const FILENAME_REMAP = {
+    'Hatchet-TypeScript-SDK-Reference.mdx': 'client.mdx',
+};
+function remapFilename(filename) {
+    var _a;
+    const remapped = (_a = FILENAME_REMAP[filename]) !== null && _a !== void 0 ? _a : filename;
+    if (remapped.startsWith('client.features.')) {
+        const leafName = remapped.replace('client.features.', '');
+        return {
+            outRelative: 'feature-clients/' + leafName,
+            basename: path_1.default.basename(leafName, shared_1.MDX_EXTENSION),
+            directory: '/feature-clients',
+        };
+    }
+    return {
+        outRelative: remapped,
+        basename: path_1.default.basename(remapped, shared_1.MDX_EXTENSION),
+        directory: '',
+    };
+}
+function toTitle(basename) {
+    return basename
+        .replace(/[-_.]/g, ' ')
+        .replace(/[^0-9a-zA-Z ]+/g, '')
+        .trim()
+        .replace(/\s+/g, ' ')
+        .split(' ')
+        .map((word) => word.charAt(0).toUpperCase() + word.slice(1).toLowerCase())
+        .join(' ');
+}
+function documentFromPath(filePath) {
+    const filename = path_1.default.basename(filePath);
+    const { outRelative, basename, directory } = remapFilename(filename);
+    const title = toTitle(basename);
+    const outFull = path_1.default.join(shared_1.FRONTEND_DOCS_RELATIVE_PATH, outRelative);
+    const outDir = path_1.default.dirname(outFull);
+    const metaJsEntry = `  "${basename}": {
+    title: "${title}",
+    theme: {
+      toc: true,
+    },
+  },`;
+    return {
+        directory,
+        basename,
+        title,
+        metaJsEntry,
+        sourcePath: filePath,
+        readableSourcePath: filename,
+        mdxOutputPath: outFull,
+        mdxOutputMetaJsPath: path_1.default.join(outDir, '_meta.js'),
+        isIndex: basename === 'index' || basename === 'README',
+    };
+}

package/docs/generate.js

@@ -0,0 +1,87 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+const fs_1 = __importDefault(require("fs"));
+const path_1 = __importDefault(require("path"));
+const child_process_1 = require("child_process");
+const paths_1 = require("./paths");
+const shared_1 = require("./shared");
+function rmrf(target) {
+    if (fs_1.default.existsSync(target)) {
+        fs_1.default.rmSync(target, { recursive: true, force: true });
+    }
+}
+function metaEntry(key, title) {
+    return `  "${key}": {
+    title: "${title}",
+    theme: {
+      toc: true,
+    },
+  },`;
+}
+function generateMetaJs(docs, subDirs) {
+    const docEntries = docs.map((d) => d.metaJsEntry);
+    const subDirEntries = subDirs.map((dir) => {
+        const name = dir.replace(/^\//, '');
+        const title = name.replace(/-/g, ' ').replace(/\b\w/g, (c) => c.toUpperCase());
+        return metaEntry(name, title);
+    });
+    const all = [...docEntries, ...subDirEntries].sort((a, b) => {
+        const keyA = a.trim().split(':')[0].replace(/"/g, '').toLowerCase();
+        const keyB = b.trim().split(':')[0].replace(/"/g, '').toLowerCase();
+        return keyA.localeCompare(keyB);
+    });
+    return `export default {\n${all.join('\n\n')}\n};\n`;
+}
+function updateMetaJs(documents) {
+    const metaJsPaths = new Set(documents.map((d) => d.mdxOutputMetaJsPath));
+    for (const metaJsPath of metaJsPaths) {
+        const relevantDocs = documents.filter((d) => d.mdxOutputMetaJsPath === metaJsPath);
+        const thisDir = relevantDocs[0].directory;
+        const subDirs = [
+            ...new Set(documents
+                .map((d) => d.directory)
+                .filter((dir) => dir !== thisDir && dir.startsWith(thisDir) && dir.split('/').length === thisDir.split('/').filter(Boolean).length + 2)),
+        ];
+        const meta = generateMetaJs(relevantDocs, subDirs);
+        fs_1.default.mkdirSync(path_1.default.dirname(metaJsPath), { recursive: true });
+        fs_1.default.writeFileSync(metaJsPath, meta, 'utf-8');
+        console.log('Wrote', metaJsPath);
+    }
+}
+function fixBrokenFeatureAnchorLinks(content) {
+    return content.replace(/\(client\.features\.([^)\s#]+\.mdx)/g, '(feature-clients/$1');
+}
+function copyDoc(document) {
+    const content = fixBrokenFeatureAnchorLinks(fs_1.default.readFileSync(document.sourcePath, 'utf-8'));
+    fs_1.default.mkdirSync(path_1.default.dirname(document.mdxOutputPath), { recursive: true });
+    fs_1.default.writeFileSync(document.mdxOutputPath, content, 'utf-8');
+    console.log('Wrote', document.mdxOutputPath);
+}
+function run() {
+    rmrf(shared_1.TMP_GEN_PATH);
+    try {
+        console.log('Running typedoc...');
+        (0, child_process_1.execSync)('npx typedoc', { stdio: 'inherit' });
+        const documents = (0, paths_1.crawlDirectory)(shared_1.TMP_GEN_PATH);
+        console.log(`Found ${documents.length} documents`);
+        for (const doc of documents) {
+            copyDoc(doc);
+        }
+        updateMetaJs(documents);
+        console.log('Running prettier on frontend docs...');
+        (0, child_process_1.execSync)('pnpm lint:fix', {
+            stdio: 'inherit',
+            cwd: path_1.default.resolve(process.cwd(), '../../frontend/docs'),
+            shell: '/bin/bash',
+        });
+    }
+    finally {
+        rmrf(shared_1.TMP_GEN_PATH);
+    }
+}
+if (require.main === module) {
+    run();
+}

package/docs/paths.d.ts

@@ -0,0 +1,2 @@
+import { Document } from './doc_types';
+export declare function crawlDirectory(directory: string, onlyInclude?: string[]): Document[];

package/docs/paths.js

@@ -0,0 +1,16 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.crawlDirectory = crawlDirectory;
+const fs_1 = __importDefault(require("fs"));
+const path_1 = __importDefault(require("path"));
+const doc_types_1 = require("./doc_types");
+function crawlDirectory(directory, onlyInclude = []) {
+    return fs_1.default
+        .readdirSync(directory, { withFileTypes: true })
+        .filter((entry) => entry.isFile() && entry.name.endsWith('.mdx') && entry.name !== 'README.mdx')
+        .map((entry) => (0, doc_types_1.documentFromPath)(path_1.default.join(directory, entry.name)))
+        .filter((doc) => !onlyInclude.length || onlyInclude.includes(doc.readableSourcePath));
+}

package/docs/shared.d.ts

@@ -0,0 +1,3 @@
+export declare const TMP_GEN_PATH = "/tmp/hatchet-typescript/docs/gen";
+export declare const FRONTEND_DOCS_RELATIVE_PATH = "../../frontend/docs/pages/sdks/typescript";
+export declare const MDX_EXTENSION = ".mdx";

package/docs/shared.js

@@ -0,0 +1,6 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.MDX_EXTENSION = exports.FRONTEND_DOCS_RELATIVE_PATH = exports.TMP_GEN_PATH = void 0;
+exports.TMP_GEN_PATH = '/tmp/hatchet-typescript/docs/gen';
+exports.FRONTEND_DOCS_RELATIVE_PATH = '../../frontend/docs/pages/sdks/typescript';
+exports.MDX_EXTENSION = '.mdx';

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@hatchet-dev/typescript-sdk",
-  "version": "1.13.1",
+  "version": "1.15.0",
   "description": "Background task orchestration & visibility for developers",
   "types": "dist/index.d.ts",
   "files": [
@@ -49,8 +49,9 @@
     "ts-jest": "^29.3.1",
     "ts-node": "^10.9.2",
     "ts-proto": "^2.7.0",
-    "typedoc": "^0.28.0",
+    "typedoc": "^0.28.17",
     "typedoc-plugin-markdown": "^4.6.0",
+    "typedoc-plugin-no-inherit": "^1.6.1",
     "typescript": "^5.8.2"
   },
   "dependencies": {
@@ -77,7 +78,7 @@
     "dump-version": "node -e \"console.log('export const HATCHET_VERSION = \\'' + require('./package.json').version + '\\';');\" > src/version.ts",
     "tsc:build": "pnpm run dump-version && tsc && resolve-tspaths",
     "test:unit": "jest --testMatch='**/*.test.ts'",
-    "test:e2e": "jest --testMatch='**/*.e2e.ts'",
+    "test:e2e": "jest --config jest.e2e.config.ts --silent --forceExit --runInBand",
     "test:unit:watch": "jest --testMatch='**/*.test.ts' --watch",
     "generate": "pnpm run '/generate-.*/'",
     "generate-api": "npx swagger-typescript-api@13.2.7 generate -p ../../bin/oas/openapi.yaml -o ./src/clients/rest/generated -n hatchet.ts --modular --axios",
@@ -91,7 +92,7 @@
     "prepublish": "cp package.json dist/package.json; cp README.md dist/; cp -r scripts dist/",
     "publish:ci": "rm -rf ./dist && pnpm run dump-version && pnpm run tsc:build && pnpm run prepublish && cd dist && pnpm publish --access public --no-git-checks",
     "publish:ci:alpha": "rm -rf ./dist && pnpm run dump-version && pnpm run tsc:build && pnpm run prepublish && cd dist && pnpm publish --access public --no-git-checks --tag alpha",
-    "generate-docs": "typedoc",
+    "generate-docs": "ts-node docs/generate.ts",
     "exec": "npx dotenv -- ts-node -r tsconfig-paths/register --project tsconfig.json"
   }
 }

package/{clients → src/clients}/admin/admin-client.d.ts

@@ -1,13 +1,19 @@
 import { Channel, ClientFactory } from 'nice-grpc';
-import { CreateWorkflowVersionOpts, RateLimitDuration, WorkflowServiceClient } from '../../protoc/workflows';
-import { ClientConfig } from '../hatchet-client/client-config';
-import { Logger } from '../../util/logger';
-import WorkflowRunRef from '../../util/workflow-run-ref';
-import { AdminServiceClient, CreateWorkflowVersionRequest } from '../../protoc/v1/workflows';
-import { Priority, RunsClient } from '../../v1';
+import { CreateWorkflowVersionOpts, RateLimitDuration, WorkerLabelComparator, WorkflowServiceClient } from '../../../protoc/workflows';
+import { ClientConfig } from '../../../clients/hatchet-client/client-config';
+import { Logger } from '../../../util/logger';
+import WorkflowRunRef from '../../../util/workflow-run-ref';
+import { AdminServiceClient, CreateWorkflowVersionRequest } from '../../../protoc/v1/workflows';
+import { Priority, RunsClient } from '../../../v1';
 import { Api } from '../rest';
 import { WebhookWorkerCreateRequest, WorkflowRunStatus, WorkflowRunStatusList } from '../rest/generated/data-contracts';
 import { RunListenerClient } from '../listeners/run-listener/child-listener-client';
+type DesiredWorkerLabelOpt = {
+    value: string | number;
+    required?: boolean;
+    weight?: number;
+    comparator?: WorkerLabelComparator;
+};
 type WorkflowMetricsQuery = {
     workflowId?: string;
     workflowName?: string;
@@ -48,19 +54,19 @@ export declare class AdminClient {
     /**
      * @deprecated use putWorkflow instead
      */
-    put_workflow(opts: CreateWorkflowVersionOpts): Promise<import("../../protoc/workflows").WorkflowVersion>;
+    put_workflow(opts: CreateWorkflowVersionOpts): Promise<import("../../../protoc/workflows").WorkflowVersion>;
     /**
      * Creates a new workflow or updates an existing workflow. If the workflow already exists, Hatchet will automatically
      * determine if the workflow definition has changed and create a new version if necessary.
      * @param workflow a workflow definition to create
      */
-    putWorkflow(workflow: CreateWorkflowVersionOpts): Promise<import("../../protoc/workflows").WorkflowVersion>;
+    putWorkflow(workflow: CreateWorkflowVersionOpts): Promise<import("../../../protoc/workflows").WorkflowVersion>;
     /**
      * Creates a new workflow or updates an existing workflow. If the workflow already exists, Hatchet will automatically
      * determine if the workflow definition has changed and create a new version if necessary.
      * @param workflow a workflow definition to create
      */
-    putWorkflowV1(workflow: CreateWorkflowVersionRequest): Promise<import("../../protoc/v1/workflows").CreateWorkflowVersionResponse>;
+    putWorkflowV1(workflow: CreateWorkflowVersionRequest): Promise<import("../../../protoc/v1/workflows").CreateWorkflowVersionResponse>;
     /**
      * @deprecated use hatchet.ratelimits.upsert instead
      */
@@ -114,6 +120,7 @@ export declare class AdminClient {
         additionalMetadata?: Record<string, string> | undefined;
         desiredWorkerId?: string | undefined;
         priority?: Priority;
+        desiredWorkerLabels?: Record<string, DesiredWorkerLabelOpt>;
     }): WorkflowRunRef<P>;
     /**
      * Run multiple workflows runs with the given input and options. This will create new workflow runs and return their IDs.
@@ -142,6 +149,7 @@ export declare class AdminClient {
             additionalMetadata?: Record<string, string> | undefined;
             desiredWorkerId?: string | undefined;
             priority?: Priority;
+            desiredWorkerLabels?: Record<string, DesiredWorkerLabelOpt>;
         };
     }>): Promise<WorkflowRunRef<P>[]>;
     /**

package/{clients → src/clients}/admin/admin-client.js

@@ -24,12 +24,24 @@ var __importDefault = (this && this.__importDefault) || function (mod) {
 };
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.AdminClient = void 0;
-const workflows_1 = require("../../protoc/workflows");
-const hatchet_error_1 = __importDefault(require("../../util/errors/hatchet-error"));
-const retrier_1 = require("../../util/retrier");
-const workflow_run_ref_1 = __importDefault(require("../../util/workflow-run-ref"));
-const workflows_2 = require("../../protoc/v1/workflows");
-const apply_namespace_1 = require("../../util/apply-namespace");
+const workflows_1 = require("../../../protoc/workflows");
+const hatchet_error_1 = __importDefault(require("../../../util/errors/hatchet-error"));
+const retrier_1 = require("../../../util/retrier");
+const workflow_run_ref_1 = __importDefault(require("../../../util/workflow-run-ref"));
+const workflows_2 = require("../../../protoc/v1/workflows");
+const apply_namespace_1 = require("../../../util/apply-namespace");
+function convertDesiredWorkerLabels(labels) {
+    return Object.fromEntries(Object.entries(labels).map(([key, label]) => [
+        key,
+        {
+            strValue: typeof label.value === 'string' ? label.value : undefined,
+            intValue: typeof label.value === 'number' ? label.value : undefined,
+            required: label.required,
+            weight: label.weight,
+            comparator: label.comparator,
+        },
+    ]));
+}
 class AdminClient {
     constructor(config, channel, factory, api, tenantId, listenerClient, workflows) {
         this.config = config;
@@ -132,10 +144,12 @@ class AdminClient {
         try {
             const inputStr = JSON.stringify(input);
             const opts = options !== null && options !== void 0 ? options : {};
-            const { additionalMetadata, parentStepRunId, parentTaskRunExternalId } = opts, rest = __rest(opts, ["additionalMetadata", "parentStepRunId", "parentTaskRunExternalId"]);
+            const { additionalMetadata, parentStepRunId, parentTaskRunExternalId, desiredWorkerLabels } = opts, rest = __rest(opts, ["additionalMetadata", "parentStepRunId", "parentTaskRunExternalId", "desiredWorkerLabels"]);
             const resp = this.client.triggerWorkflow(Object.assign(Object.assign({ name: computedName, input: inputStr }, rest), { 
                 // API expects `parentTaskRunExternalId`; accept the old name as an alias.
-                parentTaskRunExternalId: parentTaskRunExternalId !== null && parentTaskRunExternalId !== void 0 ? parentTaskRunExternalId : parentStepRunId, additionalMetadata: additionalMetadata ? JSON.stringify(additionalMetadata) : undefined, priority: opts.priority }));
+                parentTaskRunExternalId: parentTaskRunExternalId !== null && parentTaskRunExternalId !== void 0 ? parentTaskRunExternalId : parentStepRunId, additionalMetadata: additionalMetadata ? JSON.stringify(additionalMetadata) : undefined, priority: opts.priority, desiredWorkerLabels: desiredWorkerLabels
+                    ? convertDesiredWorkerLabels(desiredWorkerLabels)
+                    : {} }));
             return new workflow_run_ref_1.default(resp, this.listenerClient, this.workflows, options === null || options === void 0 ? void 0 : options.parentId);
         }
         catch (e) {
@@ -154,8 +168,10 @@ class AdminClient {
             const computedName = (0, apply_namespace_1.applyNamespace)(workflowName, this.config.namespace);
             const inputStr = JSON.stringify(input);
             const opts = options !== null && options !== void 0 ? options : {};
-            const { additionalMetadata, parentStepRunId, parentTaskRunExternalId } = opts, rest = __rest(opts, ["additionalMetadata", "parentStepRunId", "parentTaskRunExternalId"]);
-            return Object.assign(Object.assign({ name: computedName, input: inputStr }, rest), { parentTaskRunExternalId: parentTaskRunExternalId !== null && parentTaskRunExternalId !== void 0 ? parentTaskRunExternalId : parentStepRunId, additionalMetadata: additionalMetadata ? JSON.stringify(additionalMetadata) : undefined });
+            const { additionalMetadata, parentStepRunId, parentTaskRunExternalId, desiredWorkerLabels } = opts, rest = __rest(opts, ["additionalMetadata", "parentStepRunId", "parentTaskRunExternalId", "desiredWorkerLabels"]);
+            return Object.assign(Object.assign({ name: computedName, input: inputStr }, rest), { parentTaskRunExternalId: parentTaskRunExternalId !== null && parentTaskRunExternalId !== void 0 ? parentTaskRunExternalId : parentStepRunId, additionalMetadata: additionalMetadata ? JSON.stringify(additionalMetadata) : undefined, desiredWorkerLabels: desiredWorkerLabels
+                    ? convertDesiredWorkerLabels(desiredWorkerLabels)
+                    : {} });
         });
         try {
             // Call the bulk trigger workflow method

package/{clients → src/clients}/dispatcher/action-listener.d.ts

@@ -1,6 +1,6 @@
-import { DispatcherClient as PbDispatcherClient, AssignedAction } from '../../protoc/dispatcher';
-import { ClientConfig } from '../hatchet-client/client-config';
-import { Logger } from '../../util/logger';
+import { DispatcherClient as PbDispatcherClient, AssignedAction } from '../../../protoc/dispatcher';
+import { ClientConfig } from '../../../clients/hatchet-client/client-config';
+import { Logger } from '../../../util/logger';
 import { DispatcherClient } from './dispatcher-client';
 import { Heartbeat } from './heartbeat/heartbeat-controller';
 declare enum ListenStrategy {
@@ -34,6 +34,6 @@ export declare class ActionListener {
     getListenStrategy(): Promise<ListenStrategy>;
     incrementRetries(): Promise<void>;
     getListenClient(): Promise<AsyncIterable<AssignedAction>>;
-    unregister(): Promise<import("../../protoc/dispatcher").WorkerUnsubscribeResponse>;
+    unregister(): Promise<import("../../../protoc/dispatcher").WorkerUnsubscribeResponse>;
 }
 export {};

package/{clients → src/clients}/dispatcher/action-listener.js

@@ -34,11 +34,11 @@ var __importDefault = (this && this.__importDefault) || function (mod) {
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.ActionListener = void 0;
 exports.createActionKey = createActionKey;
-const dispatcher_1 = require("../../protoc/dispatcher");
+const dispatcher_1 = require("../../../protoc/dispatcher");
 const nice_grpc_1 = require("nice-grpc");
 const abort_controller_x_1 = require("abort-controller-x");
-const sleep_1 = __importDefault(require("../../util/sleep"));
-const hatchet_error_1 = __importDefault(require("../../util/errors/hatchet-error"));
+const sleep_1 = __importDefault(require("../../../util/sleep"));
+const hatchet_error_1 = __importDefault(require("../../../util/errors/hatchet-error"));
 const heartbeat_controller_1 = require("./heartbeat/heartbeat-controller");
 const DEFAULT_ACTION_LISTENER_RETRY_INTERVAL = 5000; // milliseconds
 const DEFAULT_ACTION_LISTENER_RETRY_COUNT = 20;

package/{clients → src/clients}/dispatcher/dispatcher-client.d.ts

@@ -1,8 +1,8 @@
 import { Channel, ClientFactory } from 'nice-grpc';
-import { DispatcherClient as PbDispatcherClient, StepActionEvent, GroupKeyActionEvent, OverridesData, DeepPartial, WorkerLabels as PbWorkerAffinityConfig, RuntimeInfo } from '../../protoc/dispatcher';
-import { ClientConfig } from '../hatchet-client/client-config';
-import { Logger } from '../../util/logger';
-import { SlotConfig } from '../../v1/slot-types';
+import { DispatcherClient as PbDispatcherClient, StepActionEvent, GroupKeyActionEvent, OverridesData, DeepPartial, WorkerLabels as PbWorkerAffinityConfig, RuntimeInfo } from '../../../protoc/dispatcher';
+import { ClientConfig } from '../../../clients/hatchet-client/client-config';
+import { Logger } from '../../../util/logger';
+import { SlotConfig } from '../../../v1/slot-types';
 import { ActionListener } from './action-listener';
 export type WorkerLabels = Record<string, string | number | undefined>;
 interface GetActionListenerOptions {
@@ -16,7 +16,7 @@ interface GetActionListenerOptions {
     durableSlots?: number;
     /** @deprecated use slots */
     maxRuns?: number;
-    labels: Record<string, string | number | undefined>;
+    labels: WorkerLabels;
 }
 type StepActionEventInput = StepActionEvent & {
     /** @deprecated use taskId */
@@ -36,11 +36,11 @@ export declare class DispatcherClient {
      * Throws a gRPC error with code UNIMPLEMENTED on older engines.
      */
     getVersion(): Promise<string>;
-    sendStepActionEvent(in_: StepActionEventInput): Promise<import("../../protoc/dispatcher").ActionEventResponse>;
-    sendGroupKeyActionEvent(in_: GroupKeyActionEvent): Promise<import("../../protoc/dispatcher").ActionEventResponse>;
-    putOverridesData(in_: DeepPartial<OverridesData>): Promise<void | import("../../protoc/dispatcher").OverridesDataResponse>;
-    refreshTimeout(incrementTimeoutBy: string, taskRunExternalId: string): Promise<import("../../protoc/dispatcher").RefreshTimeoutResponse>;
-    upsertWorkerLabels(workerId: string, labels: WorkerLabels): Promise<import("../../protoc/dispatcher").UpsertWorkerLabelsResponse>;
+    sendStepActionEvent(in_: StepActionEventInput): Promise<import("../../../protoc/dispatcher").ActionEventResponse>;
+    sendGroupKeyActionEvent(in_: GroupKeyActionEvent): Promise<import("../../../protoc/dispatcher").ActionEventResponse>;
+    putOverridesData(in_: DeepPartial<OverridesData>): Promise<void | import("../../../protoc/dispatcher").OverridesDataResponse>;
+    refreshTimeout(incrementTimeoutBy: string, taskRunExternalId: string): Promise<import("../../../protoc/dispatcher").RefreshTimeoutResponse>;
+    upsertWorkerLabels(workerId: string, labels: WorkerLabels): Promise<import("../../../protoc/dispatcher").UpsertWorkerLabelsResponse>;
 }
 export declare function mapLabels(in_: WorkerLabels): Record<string, PbWorkerAffinityConfig>;
 export {};

package/{clients → src/clients}/dispatcher/dispatcher-client.js

@@ -25,11 +25,11 @@ var __importDefault = (this && this.__importDefault) || function (mod) {
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.DispatcherClient = void 0;
 exports.mapLabels = mapLabels;
-const dispatcher_1 = require("../../protoc/dispatcher");
-const hatchet_error_1 = __importDefault(require("../../util/errors/hatchet-error"));
-const retrier_1 = require("../../util/retrier");
-const version_1 = require("../../version");
-const slot_types_1 = require("../../v1/slot-types");
+const dispatcher_1 = require("../../../protoc/dispatcher");
+const hatchet_error_1 = __importDefault(require("../../../util/errors/hatchet-error"));
+const retrier_1 = require("../../../util/retrier");
+const version_1 = require("../../../version");
+const slot_types_1 = require("../../../v1/slot-types");
 const action_listener_1 = require("./action-listener");
 class DispatcherClient {
     constructor(config, channel, factory) {

package/{clients → src/clients}/dispatcher/heartbeat/heartbeat-controller.d.ts

@@ -1,5 +1,5 @@
-import { Logger } from '../../../util/logger';
-import { DispatcherClient as PbDispatcherClient } from '../../../protoc/dispatcher';
+import { Logger } from '../../../../util/logger';
+import { DispatcherClient as PbDispatcherClient } from '../../../../protoc/dispatcher';
 import { Worker } from 'worker_threads';
 import { ClientConfig } from '../../hatchet-client';
 import { DispatcherClient } from '../dispatcher-client';

package/{clients → src/clients}/dispatcher/heartbeat/heartbeat-controller.js

@@ -8,13 +8,24 @@ var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, ge
         step((generator = generator.apply(thisArg, _arguments || [])).next());
     });
 };
+var __rest = (this && this.__rest) || function (s, e) {
+    var t = {};
+    for (var p in s) if (Object.prototype.hasOwnProperty.call(s, p) && e.indexOf(p) < 0)
+        t[p] = s[p];
+    if (s != null && typeof Object.getOwnPropertySymbols === "function")
+        for (var i = 0, p = Object.getOwnPropertySymbols(s); i < p.length; i++) {
+            if (e.indexOf(p[i]) < 0 && Object.prototype.propertyIsEnumerable.call(s, p[i]))
+                t[p[i]] = s[p[i]];
+        }
+    return t;
+};
 var __importDefault = (this && this.__importDefault) || function (mod) {
     return (mod && mod.__esModule) ? mod : { "default": mod };
 };
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.Heartbeat = exports.STOP_HEARTBEAT = void 0;
 const path_1 = __importDefault(require("path"));
-const thread_helper_1 = require("../../../util/thread-helper");
+const thread_helper_1 = require("../../../../util/thread-helper");
 exports.STOP_HEARTBEAT = 'stop';
 class Heartbeat {
     constructor(client, workerId) {
@@ -26,9 +37,10 @@ class Heartbeat {
     start() {
         return __awaiter(this, void 0, void 0, function* () {
             if (!this.heartbeatWorker) {
+                const _a = this.config, { middleware: _m, logger: _l } = _a, clonableConfig = __rest(_a, ["middleware", "logger"]);
                 this.heartbeatWorker = (0, thread_helper_1.runThreaded)(path_1.default.join(__dirname, './heartbeat-worker'), {
                     workerData: {
-                        config: Object.assign(Object.assign({}, this.config), { logger: undefined }),
+                        config: clonableConfig,
                         workerId: this.workerId,
                     },
                 });

package/{clients → src/clients}/dispatcher/heartbeat/heartbeat-worker.js

@@ -10,10 +10,10 @@ var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, ge
 };
 Object.defineProperty(exports, "__esModule", { value: true });
 const worker_threads_1 = require("worker_threads");
-const hatchet_client_1 = require("../../hatchet-client");
-const config_loader_1 = require("../../../util/config-loader");
+const hatchet_client_1 = require("../../../../clients/hatchet-client");
+const config_loader_1 = require("../../../../util/config-loader");
 const nice_grpc_1 = require("nice-grpc");
-const grpc_helpers_1 = require("../../../util/grpc-helpers");
+const grpc_helpers_1 = require("../../../../util/grpc-helpers");
 const dispatcher_client_1 = require("../dispatcher-client");
 const heartbeat_controller_1 = require("./heartbeat-controller");
 const HEARTBEAT_INTERVAL = 4000;
@@ -55,7 +55,7 @@ class HeartbeatWorker {
                     const actualInterval = now - this.timeLastHeartbeat;
                     if (actualInterval > HEARTBEAT_INTERVAL * 1.2) {
                         const message = `Heartbeat interval delay (${actualInterval}ms >> ${HEARTBEAT_INTERVAL}ms)`;
-                        this.logger.warn(message);
+                        this.logger.debug(message);
                         postMessage({
                             type: 'warn',
                             message,
@@ -102,6 +102,7 @@ class HeartbeatWorker {
 }
 const heartbeat = new HeartbeatWorker(worker_threads_1.workerData.config, worker_threads_1.workerData.workerId);
 heartbeat.start();
-worker_threads_1.parentPort === null || worker_threads_1.parentPort === void 0 ? void 0 : worker_threads_1.parentPort.on(heartbeat_controller_1.STOP_HEARTBEAT, () => {
-    heartbeat.stop();
+worker_threads_1.parentPort === null || worker_threads_1.parentPort === void 0 ? void 0 : worker_threads_1.parentPort.on('message', (msg) => {
+    if (msg === heartbeat_controller_1.STOP_HEARTBEAT)
+        heartbeat.stop();
 });

package/{clients → src/clients}/event/event-client.d.ts

@@ -1,10 +1,9 @@
 import { Channel, ClientFactory } from 'nice-grpc';
-import { EventsServiceClient } from '../../protoc/events/events';
-import { ClientConfig } from '../hatchet-client/client-config';
-import { Logger } from '../../util/logger';
-import { retrier } from '../../util/retrier';
-import { HatchetClient } from '../../v1';
-import { LegacyHatchetClient } from '../hatchet-client';
+import { EventsServiceClient } from '../../../protoc/events/events';
+import { ClientConfig } from '../../../clients/hatchet-client/client-config';
+import { Logger } from '../../../util/logger';
+import { retrier } from '../../../util/retrier';
+import { HatchetClient } from '../../../v1';
 export declare enum LogLevel {
     INFO = "INFO",
     WARN = "WARN",
@@ -29,9 +28,9 @@ export declare class EventClient {
     api: HatchetClient['api'];
     tenantId: string;
     logger: Logger;
-    constructor(config: ClientConfig, channel: Channel, factory: ClientFactory, hatchetClient: LegacyHatchetClient);
-    push<T>(type: string, input: T, options?: PushEventOptions): Promise<import("../../protoc/events/events").Event>;
-    bulkPush<T>(type: string, inputs: EventWithMetadata<T>[], options?: PushEventOptions): Promise<import("../../protoc/events/events").Events>;
+    constructor(config: ClientConfig, channel: Channel, factory: ClientFactory, api: HatchetClient['api']);
+    push<T>(type: string, input: T, options?: PushEventOptions): Promise<import("../../../protoc/events/events").Event>;
+    bulkPush<T>(type: string, inputs: EventWithMetadata<T>[], options?: PushEventOptions): Promise<import("../../../protoc/events/events").Events>;
     putLog(taskRunExternalId: string, log: string, level?: LogLevel, taskRetryCount?: number, metadata?: Record<string, any>): Promise<void>;
     putStream(taskRunExternalId: string, data: string | Uint8Array, index: number | undefined): Promise<void>;
     list(opts?: Parameters<typeof this.api.v1EventList>[1]): Promise<import("../rest/generated/data-contracts").V1EventList>;

package/{clients → src/clients}/event/event-client.js

@@ -13,10 +13,10 @@ var __importDefault = (this && this.__importDefault) || function (mod) {
 };
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.EventClient = exports.LogLevel = void 0;
-const events_1 = require("../../protoc/events/events");
-const hatchet_error_1 = __importDefault(require("../../util/errors/hatchet-error"));
-const retrier_1 = require("../../util/retrier");
-const apply_namespace_1 = require("../../util/apply-namespace");
+const events_1 = require("../../../protoc/events/events");
+const hatchet_error_1 = __importDefault(require("../../../util/errors/hatchet-error"));
+const retrier_1 = require("../../../util/retrier");
+const apply_namespace_1 = require("../../../util/apply-namespace");
 // eslint-disable-next-line no-shadow
 var LogLevel;
 (function (LogLevel) {
@@ -26,12 +26,12 @@ var LogLevel;
     LogLevel["DEBUG"] = "DEBUG";
 })(LogLevel || (exports.LogLevel = LogLevel = {}));
 class EventClient {
-    constructor(config, channel, factory, hatchetClient) {
+    constructor(config, channel, factory, api) {
         this.config = config;
         this.client = factory.create(events_1.EventsServiceDefinition, channel);
         this.logger = config.logger(`Dispatcher`, config.log_level);
         this.retrier = retrier_1.retrier;
-        this.api = hatchetClient.api;
+        this.api = api;
         this.tenantId = config.tenant_id;
     }
     push(type, input, options = {}) {