sdks: add idempotency_key to trigger workflow requests

Author: colonelpanic8

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

@@ -107,7 +107,7 @@ class ReplayTasksResponse(_message.Message):
     def __init__(self, replayed_tasks: _Optional[_Iterable[str]] = ...) -> None: ...
 
 class TriggerWorkflowRunRequest(_message.Message):
-    __slots__ = ("workflow_name", "input", "additional_metadata", "priority", "desired_worker_labels")
+    __slots__ = ("workflow_name", "input", "additional_metadata", "priority", "desired_worker_labels", "idempotency_key")
     class DesiredWorkerLabelsEntry(_message.Message):
         __slots__ = ("key", "value")
         KEY_FIELD_NUMBER: _ClassVar[int]
@@ -120,12 +120,14 @@ class TriggerWorkflowRunRequest(_message.Message):
     ADDITIONAL_METADATA_FIELD_NUMBER: _ClassVar[int]
     PRIORITY_FIELD_NUMBER: _ClassVar[int]
     DESIRED_WORKER_LABELS_FIELD_NUMBER: _ClassVar[int]
+    IDEMPOTENCY_KEY_FIELD_NUMBER: _ClassVar[int]
     workflow_name: str
     input: bytes
     additional_metadata: bytes
     priority: int
     desired_worker_labels: _containers.MessageMap[str, _trigger_pb2.DesiredWorkerLabels]
-    def __init__(self, workflow_name: _Optional[str] = ..., input: _Optional[bytes] = ..., additional_metadata: _Optional[bytes] = ..., priority: _Optional[int] = ..., desired_worker_labels: _Optional[_Mapping[str, _trigger_pb2.DesiredWorkerLabels]] = ...) -> None: ...
+    idempotency_key: str
+    def __init__(self, workflow_name: _Optional[str] = ..., input: _Optional[bytes] = ..., additional_metadata: _Optional[bytes] = ..., priority: _Optional[int] = ..., desired_worker_labels: _Optional[_Mapping[str, _trigger_pb2.DesiredWorkerLabels]] = ..., idempotency_key: _Optional[str] = ...) -> None: ...
 
 class TriggerWorkflowRunResponse(_message.Message):
     __slots__ = ("external_id",)

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

@@ -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/typescript/generate-protoc.sh

@@ -77,6 +77,7 @@ export type WorkflowRun<T = object> = {
     parentStepRunId?: string | undefined;
     childIndex?: number | undefined;
     childKey?: string | undefined;
+    idempotencyKey?: string | undefined;
     additionalMetadata?: Record<string, string> | undefined;
   };
 };
@@ -196,6 +197,7 @@ export class AdminClient {
       parentStepRunId?: string | undefined;
       childIndex?: number | undefined;
       childKey?: string | undefined;
+      idempotencyKey?: string | undefined;
       additionalMetadata?: Record<string, string> | undefined;
     }
   ) {
@@ -228,6 +230,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;
@@ -291,6 +294,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;