hatchet-sdk 0.5.0 → 0.7.0

data/lib/hatchet/clients/rest/lib/hatchet-sdk-rest/models/v1_update_http_operator_request.rb

@@ -0,0 +1,252 @@
+=begin
+#Hatchet API
+
+#The Hatchet API
+
+The version of the OpenAPI document: 1.0.0
+
+Generated by: https://openapi-generator.tech
+Generator version: 7.14.0
+
+=end
+
+require 'date'
+require 'time'
+
+module HatchetSdkRest
+  # Fields to update on an HTTP operator. Omitted fields are left unchanged.
+  class V1UpdateHTTPOperatorRequest
+    # The HTTPS endpoint (https, port 443) that assigned tasks are delivered to.
+    attr_accessor :trigger_endpoint
+
+    # An optional HTTPS endpoint polled to verify the operator endpoint is reachable.
+    attr_accessor :healthcheck_endpoint
+
+    # The secret used to HMAC-sign delivered requests. Write-only: it is never returned in responses. Provide a new value to rotate it.
+    attr_accessor :signing_secret
+
+    # Optional override for the per-request timeout backstop, in seconds.
+    attr_accessor :request_timeout_seconds
+
+    # Attribute mapping from ruby-style variable name to JSON key.
+    def self.attribute_map
+      {
+        :'trigger_endpoint' => :'triggerEndpoint',
+        :'healthcheck_endpoint' => :'healthcheckEndpoint',
+        :'signing_secret' => :'signingSecret',
+        :'request_timeout_seconds' => :'requestTimeoutSeconds'
+      }
+    end
+
+    # Returns attribute mapping this model knows about
+    def self.acceptable_attribute_map
+      attribute_map
+    end
+
+    # Returns all the JSON keys this model knows about
+    def self.acceptable_attributes
+      acceptable_attribute_map.values
+    end
+
+    # Attribute type mapping.
+    def self.openapi_types
+      {
+        :'trigger_endpoint' => :'String',
+        :'healthcheck_endpoint' => :'String',
+        :'signing_secret' => :'String',
+        :'request_timeout_seconds' => :'Integer'
+      }
+    end
+
+    # List of attributes with nullable: true
+    def self.openapi_nullable
+      Set.new([
+      ])
+    end
+
+    # Initializes the object
+    # @param [Hash] attributes Model attributes in the form of hash
+    def initialize(attributes = {})
+      if (!attributes.is_a?(Hash))
+        fail ArgumentError, "The input argument (attributes) must be a hash in `HatchetSdkRest::V1UpdateHTTPOperatorRequest` initialize method"
+      end
+
+      # check to see if the attribute exists and convert string to symbol for hash key
+      acceptable_attribute_map = self.class.acceptable_attribute_map
+      attributes = attributes.each_with_object({}) { |(k, v), h|
+        if (!acceptable_attribute_map.key?(k.to_sym))
+          fail ArgumentError, "`#{k}` is not a valid attribute in `HatchetSdkRest::V1UpdateHTTPOperatorRequest`. Please check the name to make sure it's valid. List of attributes: " + acceptable_attribute_map.keys.inspect
+        end
+        h[k.to_sym] = v
+      }
+
+      if attributes.key?(:'trigger_endpoint')
+        self.trigger_endpoint = attributes[:'trigger_endpoint']
+      end
+
+      if attributes.key?(:'healthcheck_endpoint')
+        self.healthcheck_endpoint = attributes[:'healthcheck_endpoint']
+      end
+
+      if attributes.key?(:'signing_secret')
+        self.signing_secret = attributes[:'signing_secret']
+      end
+
+      if attributes.key?(:'request_timeout_seconds')
+        self.request_timeout_seconds = attributes[:'request_timeout_seconds']
+      end
+    end
+
+    # Show invalid properties with the reasons. Usually used together with valid?
+    # @return Array for valid properties with the reasons
+    def list_invalid_properties
+      warn '[DEPRECATED] the `list_invalid_properties` method is obsolete'
+      invalid_properties = Array.new
+      invalid_properties
+    end
+
+    # Check to see if the all the properties in the model are valid
+    # @return true if the model is valid
+    def valid?
+      warn '[DEPRECATED] the `valid?` method is obsolete'
+      true
+    end
+
+    # Checks equality by comparing each attribute.
+    # @param [Object] Object to be compared
+    def ==(o)
+      return true if self.equal?(o)
+      self.class == o.class &&
+          trigger_endpoint == o.trigger_endpoint &&
+          healthcheck_endpoint == o.healthcheck_endpoint &&
+          signing_secret == o.signing_secret &&
+          request_timeout_seconds == o.request_timeout_seconds
+    end
+
+    # @see the `==` method
+    # @param [Object] Object to be compared
+    def eql?(o)
+      self == o
+    end
+
+    # Calculates hash code according to all attributes.
+    # @return [Integer] Hash code
+    def hash
+      [trigger_endpoint, healthcheck_endpoint, signing_secret, request_timeout_seconds].hash
+    end
+
+    # Builds the object from hash
+    # @param [Hash] attributes Model attributes in the form of hash
+    # @return [Object] Returns the model itself
+    def self.build_from_hash(attributes)
+      return nil unless attributes.is_a?(Hash)
+      attributes = attributes.transform_keys(&:to_sym)
+      transformed_hash = {}
+      openapi_types.each_pair do |key, type|
+        if attributes.key?(attribute_map[key]) && attributes[attribute_map[key]].nil?
+          transformed_hash["#{key}"] = nil
+        elsif type =~ /\AArray<(.*)>/i
+          # check to ensure the input is an array given that the attribute
+          # is documented as an array but the input is not
+          if attributes[attribute_map[key]].is_a?(Array)
+            transformed_hash["#{key}"] = attributes[attribute_map[key]].map { |v| _deserialize($1, v) }
+          end
+        elsif !attributes[attribute_map[key]].nil?
+          transformed_hash["#{key}"] = _deserialize(type, attributes[attribute_map[key]])
+        end
+      end
+      new(transformed_hash)
+    end
+
+    # Deserializes the data based on type
+    # @param string type Data type
+    # @param string value Value to be deserialized
+    # @return [Object] Deserialized data
+    def self._deserialize(type, value)
+      case type.to_sym
+      when :Time
+        Time.parse(value)
+      when :Date
+        Date.parse(value)
+      when :String
+        value.to_s
+      when :Integer
+        value.to_i
+      when :Float
+        value.to_f
+      when :Boolean
+        if value.to_s =~ /\A(true|t|yes|y|1)\z/i
+          true
+        else
+          false
+        end
+      when :Object
+        # generic object (usually a Hash), return directly
+        value
+      when /\AArray<(?<inner_type>.+)>\z/
+        inner_type = Regexp.last_match[:inner_type]
+        value.map { |v| _deserialize(inner_type, v) }
+      when /\AHash<(?<k_type>.+?), (?<v_type>.+)>\z/
+        k_type = Regexp.last_match[:k_type]
+        v_type = Regexp.last_match[:v_type]
+        {}.tap do |hash|
+          value.each do |k, v|
+            hash[_deserialize(k_type, k)] = _deserialize(v_type, v)
+          end
+        end
+      else # model
+        # models (e.g. Pet) or oneOf
+        klass = HatchetSdkRest.const_get(type)
+        klass.respond_to?(:openapi_any_of) || klass.respond_to?(:openapi_one_of) ? klass.build(value) : klass.build_from_hash(value)
+      end
+    end
+
+    # Returns the string representation of the object
+    # @return [String] String presentation of the object
+    def to_s
+      to_hash.to_s
+    end
+
+    # to_body is an alias to to_hash (backward compatibility)
+    # @return [Hash] Returns the object in the form of hash
+    def to_body
+      to_hash
+    end
+
+    # Returns the object in the form of hash
+    # @return [Hash] Returns the object in the form of hash
+    def to_hash
+      hash = {}
+      self.class.attribute_map.each_pair do |attr, param|
+        value = self.send(attr)
+        if value.nil?
+          is_nullable = self.class.openapi_nullable.include?(attr)
+          next if !is_nullable || (is_nullable && !instance_variable_defined?(:"@#{attr}"))
+        end
+
+        hash[param] = _to_hash(value)
+      end
+      hash
+    end
+
+    # Outputs non-array value in the form of hash
+    # For object, use to_hash. Otherwise, just return the value
+    # @param [Object] value Any valid value
+    # @return [Hash] Returns the value in the form of hash
+    def _to_hash(value)
+      if value.is_a?(Array)
+        value.compact.map { |v| _to_hash(v) }
+      elsif value.is_a?(Hash)
+        {}.tap do |hash|
+          value.each { |k, v| hash[k] = _to_hash(v) }
+        end
+      elsif value.respond_to? :to_hash
+        value.to_hash
+      else
+        value
+      end
+    end
+
+  end
+
+end

data/lib/hatchet/clients/rest/lib/hatchet-sdk-rest/models/v1_workflow_run.rb

@@ -56,6 +56,9 @@ module HatchetSdkRest
 
     attr_accessor :parent_task_external_id
 
+    # True when input, output, and event payload fields were omitted because the caller cannot view payloads. Additional metadata is still included.
+    attr_accessor :payloads_restricted
+
     class EnumAttributeValidator
       attr_reader :datatype
       attr_reader :allowable_values
@@ -95,7 +98,8 @@ module HatchetSdkRest
         :'error_message' => :'errorMessage',
         :'workflow_version_id' => :'workflowVersionId',
         :'created_at' => :'createdAt',
-        :'parent_task_external_id' => :'parentTaskExternalId'
+        :'parent_task_external_id' => :'parentTaskExternalId',
+        :'payloads_restricted' => :'payloadsRestricted'
       }
     end
 
@@ -126,7 +130,8 @@ module HatchetSdkRest
         :'error_message' => :'String',
         :'workflow_version_id' => :'String',
         :'created_at' => :'Time',
-        :'parent_task_external_id' => :'String'
+        :'parent_task_external_id' => :'String',
+        :'payloads_restricted' => :'Boolean'
       }
     end
 
@@ -225,6 +230,10 @@ module HatchetSdkRest
       if attributes.key?(:'parent_task_external_id')
         self.parent_task_external_id = attributes[:'parent_task_external_id']
       end
+
+      if attributes.key?(:'payloads_restricted')
+        self.payloads_restricted = attributes[:'payloads_restricted']
+      end
     end
 
     # Show invalid properties with the reasons. Usually used together with valid?
@@ -412,7 +421,8 @@ module HatchetSdkRest
           error_message == o.error_message &&
           workflow_version_id == o.workflow_version_id &&
           created_at == o.created_at &&
-          parent_task_external_id == o.parent_task_external_id
+          parent_task_external_id == o.parent_task_external_id &&
+          payloads_restricted == o.payloads_restricted
     end
 
     # @see the `==` method
@@ -424,7 +434,7 @@ module HatchetSdkRest
     # Calculates hash code according to all attributes.
     # @return [Integer] Hash code
     def hash
-      [metadata, status, tenant_id, display_name, workflow_id, output, input, started_at, finished_at, duration, additional_metadata, error_message, workflow_version_id, created_at, parent_task_external_id].hash
+      [metadata, status, tenant_id, display_name, workflow_id, output, input, started_at, finished_at, duration, additional_metadata, error_message, workflow_version_id, created_at, parent_task_external_id, payloads_restricted].hash
     end
 
     # Builds the object from hash

data/lib/hatchet/clients/rest/lib/hatchet-sdk-rest/models/workflow.rb

@@ -29,6 +29,12 @@ module HatchetSdkRest
     # Whether the workflow is paused.
     attr_accessor :is_paused
 
+    # The behavior of cron runs triggered while the workflow is paused.
+    attr_accessor :paused_workflow_cron_run_queue_behavior
+
+    # The behavior of scheduled runs triggered while the workflow is paused.
+    attr_accessor :paused_workflow_scheduled_run_queue_behavior
+
     attr_accessor :versions
 
     # The tags of the workflow.
@@ -37,6 +43,28 @@ module HatchetSdkRest
     # The jobs of the workflow.
     attr_accessor :jobs
 
+    class EnumAttributeValidator
+      attr_reader :datatype
+      attr_reader :allowable_values
+
+      def initialize(datatype, allowable_values)
+        @allowable_values = allowable_values.map do |value|
+          case datatype.to_s
+          when /Integer/i
+            value.to_i
+          when /Float/i
+            value.to_f
+          else
+            value
+          end
+        end
+      end
+
+      def valid?(value)
+        !value || allowable_values.include?(value)
+      end
+    end
+
     # Attribute mapping from ruby-style variable name to JSON key.
     def self.attribute_map
       {
@@ -45,6 +73,8 @@ module HatchetSdkRest
         :'tenant_id' => :'tenantId',
         :'description' => :'description',
         :'is_paused' => :'isPaused',
+        :'paused_workflow_cron_run_queue_behavior' => :'pausedWorkflowCronRunQueueBehavior',
+        :'paused_workflow_scheduled_run_queue_behavior' => :'pausedWorkflowScheduledRunQueueBehavior',
         :'versions' => :'versions',
         :'tags' => :'tags',
         :'jobs' => :'jobs'
@@ -69,6 +99,8 @@ module HatchetSdkRest
         :'tenant_id' => :'String',
         :'description' => :'String',
         :'is_paused' => :'Boolean',
+        :'paused_workflow_cron_run_queue_behavior' => :'WorkflowPauseScheduledCronRunQueueBehavior',
+        :'paused_workflow_scheduled_run_queue_behavior' => :'WorkflowPauseScheduledCronRunQueueBehavior',
         :'versions' => :'Array<WorkflowVersionMeta>',
         :'tags' => :'Array<WorkflowTag>',
         :'jobs' => :'Array<Job>'
@@ -123,6 +155,14 @@ module HatchetSdkRest
         self.is_paused = attributes[:'is_paused']
       end
 
+      if attributes.key?(:'paused_workflow_cron_run_queue_behavior')
+        self.paused_workflow_cron_run_queue_behavior = attributes[:'paused_workflow_cron_run_queue_behavior']
+      end
+
+      if attributes.key?(:'paused_workflow_scheduled_run_queue_behavior')
+        self.paused_workflow_scheduled_run_queue_behavior = attributes[:'paused_workflow_scheduled_run_queue_behavior']
+      end
+
       if attributes.key?(:'versions')
         if (value = attributes[:'versions']).is_a?(Array)
           self.versions = value
@@ -212,6 +252,8 @@ module HatchetSdkRest
           tenant_id == o.tenant_id &&
           description == o.description &&
           is_paused == o.is_paused &&
+          paused_workflow_cron_run_queue_behavior == o.paused_workflow_cron_run_queue_behavior &&
+          paused_workflow_scheduled_run_queue_behavior == o.paused_workflow_scheduled_run_queue_behavior &&
           versions == o.versions &&
           tags == o.tags &&
           jobs == o.jobs
@@ -226,7 +268,7 @@ module HatchetSdkRest
     # Calculates hash code according to all attributes.
     # @return [Integer] Hash code
     def hash
-      [metadata, name, tenant_id, description, is_paused, versions, tags, jobs].hash
+      [metadata, name, tenant_id, description, is_paused, paused_workflow_cron_run_queue_behavior, paused_workflow_scheduled_run_queue_behavior, versions, tags, jobs].hash
     end
 
     # Builds the object from hash

data/lib/hatchet/clients/rest/lib/hatchet-sdk-rest/models/workflow_pause_scheduled_cron_run_queue_behavior.rb

@@ -0,0 +1,40 @@
+=begin
+#Hatchet API
+
+#The Hatchet API
+
+The version of the OpenAPI document: 1.0.0
+
+Generated by: https://openapi-generator.tech
+Generator version: 7.14.0
+
+=end
+
+require 'date'
+require 'time'
+
+module HatchetSdkRest
+  class WorkflowPauseScheduledCronRunQueueBehavior
+    DROP = "DROP".freeze
+    QUEUE = "QUEUE".freeze
+
+    def self.all_vars
+      @all_vars ||= [DROP, QUEUE].freeze
+    end
+
+    # Builds the enum from string
+    # @param [String] The enum value in the form of the string
+    # @return [String] The enum value
+    def self.build_from_hash(value)
+      new.build_from_hash(value)
+    end
+
+    # Builds the enum from string
+    # @param [String] The enum value in the form of the string
+    # @return [String] The enum value
+    def build_from_hash(value)
+      return value if WorkflowPauseScheduledCronRunQueueBehavior.all_vars.include?(value)
+      raise "Invalid ENUM value #{value} for class #WorkflowPauseScheduledCronRunQueueBehavior"
+    end
+  end
+end

data/lib/hatchet/clients/rest/lib/hatchet-sdk-rest/models/workflow_update_request.rb

@@ -15,13 +15,12 @@ require 'time'
 
 module HatchetSdkRest
   class WorkflowUpdateRequest
-    # Whether the workflow is paused.
-    attr_accessor :is_paused
+    attr_accessor :pause
 
     # Attribute mapping from ruby-style variable name to JSON key.
     def self.attribute_map
       {
-        :'is_paused' => :'isPaused'
+        :'pause' => :'pause'
       }
     end
 
@@ -38,7 +37,7 @@ module HatchetSdkRest
     # Attribute type mapping.
     def self.openapi_types
       {
-        :'is_paused' => :'Boolean'
+        :'pause' => :'PauseWorkflowRequest'
       }
     end
 
@@ -64,8 +63,8 @@ module HatchetSdkRest
         h[k.to_sym] = v
       }
 
-      if attributes.key?(:'is_paused')
-        self.is_paused = attributes[:'is_paused']
+      if attributes.key?(:'pause')
+        self.pause = attributes[:'pause']
       end
     end
 
@@ -89,7 +88,7 @@ module HatchetSdkRest
     def ==(o)
       return true if self.equal?(o)
       self.class == o.class &&
-          is_paused == o.is_paused
+          pause == o.pause
     end
 
     # @see the `==` method
@@ -101,7 +100,7 @@ module HatchetSdkRest
     # Calculates hash code according to all attributes.
     # @return [Integer] Hash code
     def hash
-      [is_paused].hash
+      [pause].hash
     end
 
     # Builds the object from hash

data/lib/hatchet/clients/rest/lib/hatchet-sdk-rest.rb

@@ -75,6 +75,9 @@ HatchetSdkRest.autoload :OtelSpanKind, 'hatchet-sdk-rest/models/otel_span_kind'
 HatchetSdkRest.autoload :OtelSpanList, 'hatchet-sdk-rest/models/otel_span_list'
 HatchetSdkRest.autoload :OtelStatusCode, 'hatchet-sdk-rest/models/otel_status_code'
 HatchetSdkRest.autoload :PaginationResponse, 'hatchet-sdk-rest/models/pagination_response'
+HatchetSdkRest.autoload :PauseWorkflowRequest, 'hatchet-sdk-rest/models/pause_workflow_request'
+HatchetSdkRest.autoload :PauseWorkflowRequestPause, 'hatchet-sdk-rest/models/pause_workflow_request_pause'
+HatchetSdkRest.autoload :PauseWorkflowRequestUnpause, 'hatchet-sdk-rest/models/pause_workflow_request_unpause'
 HatchetSdkRest.autoload :PullRequest, 'hatchet-sdk-rest/models/pull_request'
 HatchetSdkRest.autoload :PullRequestState, 'hatchet-sdk-rest/models/pull_request_state'
 HatchetSdkRest.autoload :QueueMetrics, 'hatchet-sdk-rest/models/queue_metrics'
@@ -158,6 +161,7 @@ HatchetSdkRest.autoload :V1CELDebugResponseStatus, 'hatchet-sdk-rest/models/v1_c
 HatchetSdkRest.autoload :V1CancelTaskRequest, 'hatchet-sdk-rest/models/v1_cancel_task_request'
 HatchetSdkRest.autoload :V1CancelledTasks, 'hatchet-sdk-rest/models/v1_cancelled_tasks'
 HatchetSdkRest.autoload :V1CreateFilterRequest, 'hatchet-sdk-rest/models/v1_create_filter_request'
+HatchetSdkRest.autoload :V1CreateHTTPOperatorRequest, 'hatchet-sdk-rest/models/v1_create_http_operator_request'
 HatchetSdkRest.autoload :V1CreateWebhookRequest, 'hatchet-sdk-rest/models/v1_create_webhook_request'
 HatchetSdkRest.autoload :V1CreateWebhookRequestAPIKey, 'hatchet-sdk-rest/models/v1_create_webhook_request_api_key'
 HatchetSdkRest.autoload :V1CreateWebhookRequestBase, 'hatchet-sdk-rest/models/v1_create_webhook_request_base'
@@ -174,6 +178,8 @@ HatchetSdkRest.autoload :V1EventTriggeredRun, 'hatchet-sdk-rest/models/v1_event_
 HatchetSdkRest.autoload :V1EventWorkflowRunSummary, 'hatchet-sdk-rest/models/v1_event_workflow_run_summary'
 HatchetSdkRest.autoload :V1Filter, 'hatchet-sdk-rest/models/v1_filter'
 HatchetSdkRest.autoload :V1FilterList, 'hatchet-sdk-rest/models/v1_filter_list'
+HatchetSdkRest.autoload :V1HTTPOperator, 'hatchet-sdk-rest/models/v1_http_operator'
+HatchetSdkRest.autoload :V1HTTPOperatorList, 'hatchet-sdk-rest/models/v1_http_operator_list'
 HatchetSdkRest.autoload :V1LogLine, 'hatchet-sdk-rest/models/v1_log_line'
 HatchetSdkRest.autoload :V1LogLineLevel, 'hatchet-sdk-rest/models/v1_log_line_level'
 HatchetSdkRest.autoload :V1LogLineList, 'hatchet-sdk-rest/models/v1_log_line_list'
@@ -200,6 +206,7 @@ HatchetSdkRest.autoload :V1TaskTiming, 'hatchet-sdk-rest/models/v1_task_timing'
 HatchetSdkRest.autoload :V1TaskTimingList, 'hatchet-sdk-rest/models/v1_task_timing_list'
 HatchetSdkRest.autoload :V1TriggerWorkflowRunRequest, 'hatchet-sdk-rest/models/v1_trigger_workflow_run_request'
 HatchetSdkRest.autoload :V1UpdateFilterRequest, 'hatchet-sdk-rest/models/v1_update_filter_request'
+HatchetSdkRest.autoload :V1UpdateHTTPOperatorRequest, 'hatchet-sdk-rest/models/v1_update_http_operator_request'
 HatchetSdkRest.autoload :V1UpdateWebhookRequest, 'hatchet-sdk-rest/models/v1_update_webhook_request'
 HatchetSdkRest.autoload :V1WaitItem, 'hatchet-sdk-rest/models/v1_wait_item'
 HatchetSdkRest.autoload :V1Webhook, 'hatchet-sdk-rest/models/v1_webhook'
@@ -238,6 +245,7 @@ HatchetSdkRest.autoload :WorkflowConcurrency, 'hatchet-sdk-rest/models/workflow_
 HatchetSdkRest.autoload :WorkflowKind, 'hatchet-sdk-rest/models/workflow_kind'
 HatchetSdkRest.autoload :WorkflowList, 'hatchet-sdk-rest/models/workflow_list'
 HatchetSdkRest.autoload :WorkflowMetrics, 'hatchet-sdk-rest/models/workflow_metrics'
+HatchetSdkRest.autoload :WorkflowPauseScheduledCronRunQueueBehavior, 'hatchet-sdk-rest/models/workflow_pause_scheduled_cron_run_queue_behavior'
 HatchetSdkRest.autoload :WorkflowRun, 'hatchet-sdk-rest/models/workflow_run'
 HatchetSdkRest.autoload :WorkflowRunList, 'hatchet-sdk-rest/models/workflow_run_list'
 HatchetSdkRest.autoload :WorkflowRunOrderByDirection, 'hatchet-sdk-rest/models/workflow_run_order_by_direction'
@@ -272,6 +280,7 @@ HatchetSdkRest.autoload :HealthcheckApi, 'hatchet-sdk-rest/api/healthcheck_api'
 HatchetSdkRest.autoload :LogApi, 'hatchet-sdk-rest/api/log_api'
 HatchetSdkRest.autoload :MetadataApi, 'hatchet-sdk-rest/api/metadata_api'
 HatchetSdkRest.autoload :ObservabilityApi, 'hatchet-sdk-rest/api/observability_api'
+HatchetSdkRest.autoload :OperatorApi, 'hatchet-sdk-rest/api/operator_api'
 HatchetSdkRest.autoload :RateLimitsApi, 'hatchet-sdk-rest/api/rate_limits_api'
 HatchetSdkRest.autoload :SNSApi, 'hatchet-sdk-rest/api/sns_api'
 HatchetSdkRest.autoload :SlackApi, 'hatchet-sdk-rest/api/slack_api'

data/lib/hatchet/concurrency.rb

@@ -7,6 +7,8 @@ module Hatchet
     CANCEL_NEWEST = :cancel_newest
     GROUP_ROUND_ROBIN = :group_round_robin
     QUEUE = :queue
+    CANCEL_QUEUED_EXCEPT_NEWEST = :cancel_queued_except_newest
+    CANCEL_QUEUED_EXCEPT_OLDEST = :cancel_queued_except_oldest
   end
 
   # Defines a concurrency expression for workflow or task-level concurrency control
@@ -30,7 +32,7 @@ module Hatchet
     # @return [Integer] Maximum concurrent runs for this key
     attr_reader :max_runs
 
-    # @return [Symbol] Strategy when limit is exceeded (:cancel_in_progress, :cancel_newest, :group_round_robin, :queue)
+    # @return [Symbol] Strategy when limit is exceeded (:cancel_in_progress, :cancel_newest, :group_round_robin, :queue, :cancel_queued_except_newest, :cancel_queued_except_oldest)
     attr_reader :limit_strategy
 
     # @param expression [String] CEL expression evaluated against input
@@ -59,6 +61,8 @@ module Hatchet
       group_round_robin: :GROUP_ROUND_ROBIN,
       queue: :QUEUE_NEWEST,
       drop_newest: :DROP_NEWEST,
+      cancel_queued_except_newest: :CANCEL_QUEUED_EXCEPT_NEWEST,
+      cancel_queued_except_oldest: :CANCEL_QUEUED_EXCEPT_OLDEST,
     }.freeze
 
     # Convert to a V1::Concurrency protobuf message

data/lib/hatchet/context.rb

@@ -1,5 +1,7 @@
 # frozen_string_literal: true
 
+require "json"
+
 module Hatchet
   # Context object passed to task execution blocks.
   #
@@ -128,10 +130,28 @@ module Hatchet
       @client&.logger&.info(msg) || puts(msg)
     end
 
-    # Cancel the current workflow run
+    # Cancel the current workflow run.
+    #
+    # Batch tasks share one context across every buffered member, so there is no single
+    # task run to cancel; instead this sends a batch CANCELLED event covering every
+    # member of the batch.
     def cancel
       @cancelled = true
       @exit_flag = true
+
+      if batch_action?
+        begin
+          @dispatcher_client&.send_batch_action_event(
+            action: @action,
+            event_type: :STEP_EVENT_TYPE_CANCELLED,
+            items: batch_member_ids.map { |id| { task_run_external_id: id } },
+          )
+        rescue StandardError
+          nil
+        end
+        return
+      end
+
       return unless @client && @workflow_run_id
 
       begin
@@ -198,11 +218,30 @@ module Hatchet
       task_run_errors.find { |e| e.respond_to?(:task_name) && e.task_name == key }
     end
 
-    # Access the worker context for worker-level operations
+    # Access the worker context for worker-level operations. The returned
+    # {WorkerContext} exposes +id+ (the worker id), +labels+ (the current
+    # worker labels), and +upsert_labels+ to add or update worker labels on
+    # the server at runtime.
     #
-    # @return [WorkerContext, nil]
+    # @return [WorkerContext, nil] The worker context, if one is attached
     def worker
       @worker_context
     end
+
+    private
+
+    def batch_action?
+      @action.respond_to?(:action_type) && @action.action_type == :START_BATCH
+    end
+
+    def batch_member_ids
+      raw = @action.respond_to?(:action_payload) ? @action.action_payload : nil
+      return [] if raw.nil? || raw.to_s.empty?
+
+      parsed = JSON.parse(raw)
+      parsed.is_a?(Hash) ? parsed.keys : []
+    rescue JSON::ParserError
+      []
+    end
   end
 end