sdks: add idempotency_key to trigger workflow requests

Author: colonelpanic8

sdks/python/hatchet_sdk/contracts/v1/workflows_pb2.py

@@ -119,16 +119,18 @@ class ReplayTasksResponse(_message.Message):
     def __init__(self, replayed_tasks: _Optional[_Iterable[str]] = ...) -> None: ...
 
 class TriggerWorkflowRunRequest(_message.Message):
-    __slots__ = ("workflow_name", "input", "additional_metadata", "priority")
+    __slots__ = ("workflow_name", "input", "additional_metadata", "priority", "idempotency_key")
     WORKFLOW_NAME_FIELD_NUMBER: _ClassVar[int]
     INPUT_FIELD_NUMBER: _ClassVar[int]
     ADDITIONAL_METADATA_FIELD_NUMBER: _ClassVar[int]
     PRIORITY_FIELD_NUMBER: _ClassVar[int]
+    IDEMPOTENCY_KEY_FIELD_NUMBER: _ClassVar[int]
     workflow_name: str
     input: bytes
     additional_metadata: bytes
     priority: int
-    def __init__(self, workflow_name: _Optional[str] = ..., input: _Optional[bytes] = ..., additional_metadata: _Optional[bytes] = ..., priority: _Optional[int] = ...) -> None: ...
+    idempotency_key: str
+    def __init__(self, workflow_name: _Optional[str] = ..., input: _Optional[bytes] = ..., additional_metadata: _Optional[bytes] = ..., priority: _Optional[int] = ..., idempotency_key: _Optional[str] = ...) -> None: ...
 
 class TriggerWorkflowRunResponse(_message.Message):
     __slots__ = ("external_id",)

sdks/python/hatchet_sdk/contracts/v1/workflows_pb2.pyi

@@ -3,7 +3,7 @@
 import grpc
 import warnings
 
-from hatchet_sdk.contracts.v1 import workflows_pb2 as v1_dot_workflows__pb2
+from v1 import workflows_pb2 as v1_dot_workflows__pb2
 
 GRPC_GENERATED_VERSION = '1.76.0'
 GRPC_VERSION = grpc.__version__
@@ -29,7 +29,7 @@ class AdminServiceStub(object):
     """AdminService represents a set of RPCs for admin management of tasks, workflows, etc.
     """
 
-    def __init__(self, channel: grpc.Channel | grpc.aio.Channel) -> None:
+    def __init__(self, channel):
         """Constructor.
 
         Args:

sdks/python/hatchet_sdk/contracts/v1/workflows_pb2_grpc.py

@@ -143,7 +143,7 @@ class DesiredWorkerLabels(_message.Message):
     required: bool
     comparator: WorkerLabelComparator
     weight: int
-    def __init__(self, str_value: _Optional[str] = ..., int_value: _Optional[int] = ..., required: bool = ..., comparator: _Optional[_Union[WorkerLabelComparator, str]] = ..., weight: _Optional[int] = ...) -> None: ...
+    def __init__(self, str_value: _Optional[str] = ..., int_value: _Optional[int] = ..., required: _Optional[bool] = ..., comparator: _Optional[_Union[WorkerLabelComparator, str]] = ..., weight: _Optional[int] = ...) -> None: ...
 
 class CreateWorkflowStepOpts(_message.Message):
     __slots__ = ("readable_id", "action", "timeout", "inputs", "parents", "user_data", "retries", "rate_limits", "worker_labels", "backoff_factor", "backoff_max_seconds")
@@ -275,7 +275,7 @@ class BulkTriggerWorkflowResponse(_message.Message):
     def __init__(self, workflow_run_ids: _Optional[_Iterable[str]] = ...) -> None: ...
 
 class TriggerWorkflowRequest(_message.Message):
-    __slots__ = ("name", "input", "parent_id", "parent_task_run_external_id", "child_index", "child_key", "additional_metadata", "desired_worker_id", "priority")
+    __slots__ = ("name", "input", "parent_id", "parent_task_run_external_id", "child_index", "child_key", "additional_metadata", "desired_worker_id", "priority", "idempotency_key")
     NAME_FIELD_NUMBER: _ClassVar[int]
     INPUT_FIELD_NUMBER: _ClassVar[int]
     PARENT_ID_FIELD_NUMBER: _ClassVar[int]
@@ -285,6 +285,7 @@ class TriggerWorkflowRequest(_message.Message):
     ADDITIONAL_METADATA_FIELD_NUMBER: _ClassVar[int]
     DESIRED_WORKER_ID_FIELD_NUMBER: _ClassVar[int]
     PRIORITY_FIELD_NUMBER: _ClassVar[int]
+    IDEMPOTENCY_KEY_FIELD_NUMBER: _ClassVar[int]
     name: str
     input: str
     parent_id: str
@@ -294,7 +295,8 @@ class TriggerWorkflowRequest(_message.Message):
     additional_metadata: str
     desired_worker_id: str
     priority: int
-    def __init__(self, name: _Optional[str] = ..., input: _Optional[str] = ..., parent_id: _Optional[str] = ..., parent_task_run_external_id: _Optional[str] = ..., child_index: _Optional[int] = ..., child_key: _Optional[str] = ..., additional_metadata: _Optional[str] = ..., desired_worker_id: _Optional[str] = ..., priority: _Optional[int] = ...) -> None: ...
+    idempotency_key: str
+    def __init__(self, name: _Optional[str] = ..., input: _Optional[str] = ..., parent_id: _Optional[str] = ..., parent_task_run_external_id: _Optional[str] = ..., child_index: _Optional[int] = ..., child_key: _Optional[str] = ..., additional_metadata: _Optional[str] = ..., desired_worker_id: _Optional[str] = ..., priority: _Optional[int] = ..., idempotency_key: _Optional[str] = ...) -> None: ...
 
 class TriggerWorkflowResponse(_message.Message):
     __slots__ = ("workflow_run_id",)

sdks/python/hatchet_sdk/contracts/workflows_pb2.py

@@ -1,3 +1,7 @@
+#!/usr/bin/env bash
+
+set -euo pipefail
+
 # Directory to write generated code to (.js and .d.ts files)
 
 OUT_DIR="./src/protoc"
@@ -8,12 +12,22 @@ else
     IN_DIR="../../api-contracts"
 fi
 
-# Generate code
-./node_modules/.bin/grpc_tools_node_protoc \
+# Generate code.
+# Prefer system `protoc` (available in Nix environments) over `grpc-tools`' bundled protoc.
+PROTOC_BIN="${PROTOC_BIN:-}"
+if [ -z "$PROTOC_BIN" ]; then
+  if command -v protoc >/dev/null 2>&1; then
+    PROTOC_BIN="protoc"
+  else
+    PROTOC_BIN="./node_modules/.bin/grpc_tools_node_protoc"
+  fi
+fi
+
+"$PROTOC_BIN" \
   --plugin=protoc-gen-ts_proto=./node_modules/.bin/protoc-gen-ts_proto \
-  --ts_proto_out=$OUT_DIR \
+  --ts_proto_out="$OUT_DIR" \
   --ts_proto_opt=outputServices=nice-grpc,outputServices=generic-definitions,useExactTypes=false \
-  --proto_path=$IN_DIR \
-  $IN_DIR/**/*.proto
+  --proto_path="$IN_DIR" \
+  $(find "$IN_DIR" -type f -name '*.proto' -print)
 
 pnpm lint:fix

sdks/python/hatchet_sdk/contracts/workflows_pb2.pyi

@@ -52,6 +52,7 @@ export type WorkflowRun<T = object> = {
     parentStepRunId?: string | undefined;
     childIndex?: number | undefined;
     childKey?: string | undefined;
+    idempotencyKey?: string | undefined;
     additionalMetadata?: Record<string, string> | undefined;
   };
 };
@@ -171,6 +172,7 @@ export class AdminClient {
       parentStepRunId?: string | undefined;
       childIndex?: number | undefined;
       childKey?: string | undefined;
+      idempotencyKey?: string | undefined;
       additionalMetadata?: Record<string, string> | undefined;
     }
   ) {
@@ -203,6 +205,7 @@ export class AdminClient {
       parentStepRunId?: string | undefined;
       childIndex?: number | undefined;
       childKey?: string | undefined;
+      idempotencyKey?: string | undefined;
       additionalMetadata?: Record<string, string> | undefined;
       desiredWorkerId?: string | undefined;
       priority?: Priority;
@@ -256,6 +259,7 @@ export class AdminClient {
         parentStepRunId?: string | undefined;
         childIndex?: number | undefined;
         childKey?: string | undefined;
+        idempotencyKey?: string | undefined;
         additionalMetadata?: Record<string, string> | undefined;
         desiredWorkerId?: string | undefined;
         priority?: Priority;