hatchet-sdk 1.4.1__tar.gz → 1.6.1__tar.gz

{hatchet_sdk-1.4.1 → hatchet_sdk-1.6.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: hatchet-sdk
-Version: 1.4.1
+Version: 1.6.1
 Summary: 
 License: MIT
 Author: Alexander Belanger

{hatchet_sdk-1.4.1 → hatchet_sdk-1.6.1}/hatchet_sdk/clients/admin.py

@@ -42,6 +42,7 @@ class ScheduleTriggerWorkflowOptions(BaseModel):
     child_key: str | None = None
     namespace: str | None = None
     additional_metadata: JSONSerializableMapping = Field(default_factory=dict)
+    priority: int | None = None
 
 
 class TriggerWorkflowOptions(ScheduleTriggerWorkflowOptions):
@@ -126,12 +127,18 @@ class AdminClient:
         except json.JSONDecodeError as e:
             raise ValueError(f"Error encoding payload: {e}")
 
-        _options = self.TriggerWorkflowRequest.model_validate(
-            options.model_dump()
-        ).model_dump()
+        _options = self.TriggerWorkflowRequest.model_validate(options.model_dump())
 
         return v0_workflow_protos.TriggerWorkflowRequest(
-            name=workflow_name, input=payload_data, **_options
+            name=workflow_name,
+            input=payload_data,
+            parent_id=_options.parent_id,
+            parent_step_run_id=_options.parent_step_run_id,
+            child_index=_options.child_index,
+            child_key=_options.child_key,
+            additional_metadata=_options.additional_metadata,
+            desired_worker_id=_options.desired_worker_id,
+            priority=_options.priority,
         )
 
     def _prepare_put_workflow_request(
@@ -178,6 +185,7 @@ class AdminClient:
             child_index=options.child_index,
             child_key=options.child_key,
             additional_metadata=json.dumps(options.additional_metadata),
+            priority=options.priority,
         )
 
     @tenacity_retry
@@ -315,6 +323,7 @@ class AdminClient:
             child_index=child_index,
             additional_metadata=options.additional_metadata,
             desired_worker_id=desired_worker_id,
+            priority=options.priority,
         )
 
         namespace = options.namespace or self.namespace

{hatchet_sdk-1.4.1 → hatchet_sdk-1.6.1}/hatchet_sdk/clients/dispatcher/action_listener.py

@@ -7,7 +7,6 @@ from typing import Any, AsyncGenerator, cast
 
 import grpc
 import grpc.aio
-from grpc._cython import cygrpc  # type: ignore[attr-defined]
 from pydantic import BaseModel, ConfigDict, Field, field_validator, model_validator
 
 from hatchet_sdk.clients.event_ts import ThreadSafeEvent, read_with_interrupt
@@ -105,6 +104,8 @@ class Action(BaseModel):
     child_workflow_key: str | None = None
     parent_workflow_run_id: str | None = None
 
+    priority: int | None = None
+
     def _dump_payload_to_str(self) -> str:
         try:
             return json.dumps(self.action_payload.model_dump(), default=str)
@@ -265,7 +266,6 @@ class ActionListener:
                     await self.interrupt.wait()
 
                     if not t.done():
-                        # print a warning
                         logger.warning(
                             "Interrupted read_with_interrupt task of action listener"
                         )
@@ -275,9 +275,10 @@ class ActionListener:
 
                         break
 
-                    assigned_action, _ = t.result()
+                    assigned_action, _, is_eof = t.result()
 
-                    if assigned_action is cygrpc.EOF:
+                    if is_eof:
+                        logger.debug("Handling EOF in Action Listener")
                         self.retries = self.retries + 1
                         break
 
@@ -320,6 +321,7 @@ class ActionListener:
                         child_workflow_index=assigned_action.child_workflow_index,
                         child_workflow_key=assigned_action.child_workflow_key,
                         parent_workflow_run_id=assigned_action.parent_workflow_run_id,
+                        priority=assigned_action.priority,
                     )
 
                     yield action

{hatchet_sdk-1.4.1 → hatchet_sdk-1.6.1}/hatchet_sdk/clients/event_ts.py

@@ -4,6 +4,8 @@ from typing import Callable, TypeVar, cast, overload
 import grpc.aio
 from grpc._cython import cygrpc  # type: ignore[attr-defined]
 
+from hatchet_sdk.logger import logger
+
 
 class ThreadSafeEvent(asyncio.Event):
     """
@@ -32,7 +34,7 @@ async def read_with_interrupt(
     listener: grpc.aio.UnaryStreamCall[TRequest, TResponse],
     interrupt: ThreadSafeEvent,
     key_generator: Callable[[TResponse], str],
-) -> tuple[TResponse, str]: ...
+) -> tuple[TResponse, str, bool]: ...
 
 
 @overload
@@ -40,22 +42,23 @@ async def read_with_interrupt(
     listener: grpc.aio.UnaryStreamCall[TRequest, TResponse],
     interrupt: ThreadSafeEvent,
     key_generator: None = None,
-) -> tuple[TResponse, None]: ...
+) -> tuple[TResponse, None, bool]: ...
 
 
 async def read_with_interrupt(
     listener: grpc.aio.UnaryStreamCall[TRequest, TResponse],
     interrupt: ThreadSafeEvent,
     key_generator: Callable[[TResponse], str] | None = None,
-) -> tuple[TResponse, str | None]:
+) -> tuple[TResponse, str | None, bool]:
     try:
         result = cast(TResponse, await listener.read())
 
         if result is cygrpc.EOF:
-            raise ValueError("Unexpected EOF")
+            logger.warning("Received EOF from engine")
+            return cast(TResponse, None), None, True
 
         key = key_generator(result) if key_generator else None
 
-        return result, key
+        return result, key, False
     finally:
         interrupt.set()

{hatchet_sdk-1.4.1 → hatchet_sdk-1.6.1}/hatchet_sdk/clients/listeners/pooled_listener.py

@@ -5,7 +5,6 @@ from typing import Generic, Literal, TypeVar
 
 import grpc
 import grpc.aio
-from grpc._cython import cygrpc  # type: ignore[attr-defined]
 
 from hatchet_sdk.clients.event_ts import ThreadSafeEvent, read_with_interrupt
 from hatchet_sdk.config import ClientConfig
@@ -131,9 +130,12 @@ class PooledListener(Generic[R, T, L], ABC):
                                 await asyncio.sleep(DEFAULT_LISTENER_RETRY_INTERVAL)
                                 break
 
-                            event, key = t.result()
+                            event, key, is_eof = t.result()
 
-                            if event is cygrpc.EOF:
+                            if is_eof:
+                                logger.debug(
+                                    f"Handling EOF in Pooled Listener {self.__class__.__name__}"
+                                )
                                 break
 
                             subscriptions = self.to_subscriptions.get(key, [])

{hatchet_sdk-1.4.1/hatchet_sdk/v0 → hatchet_sdk-1.6.1/hatchet_sdk}/clients/rest/models/create_cron_workflow_trigger_request.py

@@ -20,7 +20,7 @@ import re  # noqa: F401
 from typing import Any, ClassVar, Dict, List, Optional, Set
 
 from pydantic import BaseModel, ConfigDict, Field, StrictStr
-from typing_extensions import Self
+from typing_extensions import Annotated, Self
 
 
 class CreateCronWorkflowTriggerRequest(BaseModel):
@@ -32,11 +32,13 @@ class CreateCronWorkflowTriggerRequest(BaseModel):
     additional_metadata: Dict[str, Any] = Field(alias="additionalMetadata")
     cron_name: StrictStr = Field(alias="cronName")
     cron_expression: StrictStr = Field(alias="cronExpression")
+    priority: Optional[Annotated[int, Field(le=3, strict=True, ge=1)]] = None
     __properties: ClassVar[List[str]] = [
         "input",
         "additionalMetadata",
         "cronName",
         "cronExpression",
+        "priority",
     ]
 
     model_config = ConfigDict(
@@ -93,6 +95,7 @@ class CreateCronWorkflowTriggerRequest(BaseModel):
                 "additionalMetadata": obj.get("additionalMetadata"),
                 "cronName": obj.get("cronName"),
                 "cronExpression": obj.get("cronExpression"),
+                "priority": obj.get("priority"),
             }
         )
         return _obj

{hatchet_sdk-1.4.1 → hatchet_sdk-1.6.1}/hatchet_sdk/clients/rest/models/cron_workflows.py

@@ -20,7 +20,7 @@ import re  # noqa: F401
 from typing import Any, ClassVar, Dict, List, Optional, Set
 
 from pydantic import BaseModel, ConfigDict, Field, StrictBool, StrictStr
-from typing_extensions import Self
+from typing_extensions import Annotated, Self
 
 from hatchet_sdk.clients.rest.models.api_resource_meta import APIResourceMeta
 from hatchet_sdk.clients.rest.models.cron_workflows_method import CronWorkflowsMethod
@@ -44,6 +44,7 @@ class CronWorkflows(BaseModel):
     )
     enabled: StrictBool
     method: CronWorkflowsMethod
+    priority: Optional[Annotated[int, Field(le=3, strict=True, ge=1)]] = None
     __properties: ClassVar[List[str]] = [
         "metadata",
         "tenantId",
@@ -56,6 +57,7 @@ class CronWorkflows(BaseModel):
         "additionalMetadata",
         "enabled",
         "method",
+        "priority",
     ]
 
     model_config = ConfigDict(
@@ -126,6 +128,7 @@ class CronWorkflows(BaseModel):
                 "additionalMetadata": obj.get("additionalMetadata"),
                 "enabled": obj.get("enabled"),
                 "method": obj.get("method"),
+                "priority": obj.get("priority"),
             }
         )
         return _obj

{hatchet_sdk-1.4.1/hatchet_sdk/v0 → hatchet_sdk-1.6.1/hatchet_sdk}/clients/rest/models/schedule_workflow_run_request.py

@@ -21,7 +21,7 @@ from datetime import datetime
 from typing import Any, ClassVar, Dict, List, Optional, Set
 
 from pydantic import BaseModel, ConfigDict, Field
-from typing_extensions import Self
+from typing_extensions import Annotated, Self
 
 
 class ScheduleWorkflowRunRequest(BaseModel):
@@ -32,7 +32,13 @@ class ScheduleWorkflowRunRequest(BaseModel):
     input: Dict[str, Any]
     additional_metadata: Dict[str, Any] = Field(alias="additionalMetadata")
     trigger_at: datetime = Field(alias="triggerAt")
-    __properties: ClassVar[List[str]] = ["input", "additionalMetadata", "triggerAt"]
+    priority: Optional[Annotated[int, Field(le=3, strict=True, ge=1)]] = None
+    __properties: ClassVar[List[str]] = [
+        "input",
+        "additionalMetadata",
+        "triggerAt",
+        "priority",
+    ]
 
     model_config = ConfigDict(
         populate_by_name=True,
@@ -87,6 +93,7 @@ class ScheduleWorkflowRunRequest(BaseModel):
                 "input": obj.get("input"),
                 "additionalMetadata": obj.get("additionalMetadata"),
                 "triggerAt": obj.get("triggerAt"),
+                "priority": obj.get("priority"),
             }
         )
         return _obj

{hatchet_sdk-1.4.1 → hatchet_sdk-1.6.1}/hatchet_sdk/clients/rest/models/scheduled_workflows.py

@@ -58,6 +58,7 @@ class ScheduledWorkflows(BaseModel):
         Annotated[str, Field(min_length=36, strict=True, max_length=36)]
     ] = Field(default=None, alias="workflowRunId")
     method: ScheduledWorkflowsMethod
+    priority: Optional[Annotated[int, Field(le=3, strict=True, ge=1)]] = None
     __properties: ClassVar[List[str]] = [
         "metadata",
         "tenantId",
@@ -72,6 +73,7 @@ class ScheduledWorkflows(BaseModel):
         "workflowRunStatus",
         "workflowRunId",
         "method",
+        "priority",
     ]
 
     model_config = ConfigDict(
@@ -144,6 +146,7 @@ class ScheduledWorkflows(BaseModel):
                 "workflowRunStatus": obj.get("workflowRunStatus"),
                 "workflowRunId": obj.get("workflowRunId"),
                 "method": obj.get("method"),
+                "priority": obj.get("priority"),
             }
         )
         return _obj

{hatchet_sdk-1.4.1 → hatchet_sdk-1.6.1}/hatchet_sdk/clients/rest/models/v1_trigger_workflow_run_request.py

@@ -19,7 +19,7 @@ import pprint
 import re  # noqa: F401
 from typing import Any, ClassVar, Dict, List, Optional, Set
 
-from pydantic import BaseModel, ConfigDict, Field, StrictStr
+from pydantic import BaseModel, ConfigDict, Field, StrictInt, StrictStr
 from typing_extensions import Self
 
 
@@ -35,7 +35,15 @@ class V1TriggerWorkflowRunRequest(BaseModel):
     additional_metadata: Optional[Dict[str, Any]] = Field(
         default=None, alias="additionalMetadata"
     )
-    __properties: ClassVar[List[str]] = ["workflowName", "input", "additionalMetadata"]
+    priority: Optional[StrictInt] = Field(
+        default=None, description="The priority of the workflow run."
+    )
+    __properties: ClassVar[List[str]] = [
+        "workflowName",
+        "input",
+        "additionalMetadata",
+        "priority",
+    ]
 
     model_config = ConfigDict(
         populate_by_name=True,
@@ -90,6 +98,7 @@ class V1TriggerWorkflowRunRequest(BaseModel):
                 "workflowName": obj.get("workflowName"),
                 "input": obj.get("input"),
                 "additionalMetadata": obj.get("additionalMetadata"),
+                "priority": obj.get("priority"),
             }
         )
         return _obj

{hatchet_sdk-1.4.1/hatchet_sdk/v0 → hatchet_sdk-1.6.1/hatchet_sdk}/clients/rest/models/workflow_runs_metrics.py

@@ -22,13 +22,17 @@ from typing import Any, ClassVar, Dict, List, Optional, Set
 from pydantic import BaseModel, ConfigDict
 from typing_extensions import Self
 
+from hatchet_sdk.clients.rest.models.workflow_runs_metrics_counts import (
+    WorkflowRunsMetricsCounts,
+)
+
 
 class WorkflowRunsMetrics(BaseModel):
     """
     WorkflowRunsMetrics
     """  # noqa: E501
 
-    counts: Optional[Dict[str, Any]] = None
+    counts: Optional[WorkflowRunsMetricsCounts] = None
     __properties: ClassVar[List[str]] = ["counts"]
 
     model_config = ConfigDict(

{hatchet_sdk-1.4.1 → hatchet_sdk-1.6.1}/hatchet_sdk/context/context.py

@@ -228,6 +228,10 @@ class Context:
     def parent_workflow_run_id(self) -> str | None:
         return self.action.parent_workflow_run_id
 
+    @property
+    def priority(self) -> int | None:
+        return self.action.priority
+
     @property
     def task_run_errors(self) -> dict[str, str]:
         errors = self.data.step_run_errors

hatchet_sdk-1.6.1/hatchet_sdk/contracts/dispatcher_pb2.py

@@ -0,0 +1,102 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: dispatcher.proto
+# Protobuf Python Version: 5.26.1
+"""Generated protocol buffer code."""
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import descriptor_pool as _descriptor_pool
+from google.protobuf import symbol_database as _symbol_database
+from google.protobuf.internal import builder as _builder
+# @@protoc_insertion_point(imports)
+
+_sym_db = _symbol_database.Default()
+
+
+from google.protobuf import timestamp_pb2 as google_dot_protobuf_dot_timestamp__pb2
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x10\x64ispatcher.proto\x1a\x1fgoogle/protobuf/timestamp.proto\"V\n\x0cWorkerLabels\x12\x15\n\x08strValue\x18\x01 \x01(\tH\x00\x88\x01\x01\x12\x15\n\x08intValue\x18\x02 \x01(\x05H\x01\x88\x01\x01\x42\x0b\n\t_strValueB\x0b\n\t_intValue\"\xc8\x01\n\x0bRuntimeInfo\x12\x17\n\nsdkVersion\x18\x01 \x01(\tH\x00\x88\x01\x01\x12\x1c\n\x08language\x18\x02 \x01(\x0e\x32\x05.SDKSH\x01\x88\x01\x01\x12\x1c\n\x0flanguageVersion\x18\x03 \x01(\tH\x02\x88\x01\x01\x12\x0f\n\x02os\x18\x04 \x01(\tH\x03\x88\x01\x01\x12\x12\n\x05\x65xtra\x18\x05 \x01(\tH\x04\x88\x01\x01\x42\r\n\x0b_sdkVersionB\x0b\n\t_languageB\x12\n\x10_languageVersionB\x05\n\x03_osB\x08\n\x06_extra\"\xc0\x02\n\x15WorkerRegisterRequest\x12\x12\n\nworkerName\x18\x01 \x01(\t\x12\x0f\n\x07\x61\x63tions\x18\x02 \x03(\t\x12\x10\n\x08services\x18\x03 \x03(\t\x12\x14\n\x07maxRuns\x18\x04 \x01(\x05H\x00\x88\x01\x01\x12\x32\n\x06labels\x18\x05 \x03(\x0b\x32\".WorkerRegisterRequest.LabelsEntry\x12\x16\n\twebhookId\x18\x06 \x01(\tH\x01\x88\x01\x01\x12&\n\x0bruntimeInfo\x18\x07 \x01(\x0b\x32\x0c.RuntimeInfoH\x02\x88\x01\x01\x1a<\n\x0bLabelsEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\x1c\n\x05value\x18\x02 \x01(\x0b\x32\r.WorkerLabels:\x02\x38\x01\x42\n\n\x08_maxRunsB\x0c\n\n_webhookIdB\x0e\n\x0c_runtimeInfo\"P\n\x16WorkerRegisterResponse\x12\x10\n\x08tenantId\x18\x01 \x01(\t\x12\x10\n\x08workerId\x18\x02 \x01(\t\x12\x12\n\nworkerName\x18\x03 \x01(\t\"\xa3\x01\n\x19UpsertWorkerLabelsRequest\x12\x10\n\x08workerId\x18\x01 \x01(\t\x12\x36\n\x06labels\x18\x02 \x03(\x0b\x32&.UpsertWorkerLabelsRequest.LabelsEntry\x1a<\n\x0bLabelsEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\x1c\n\x05value\x18\x02 \x01(\x0b\x32\r.WorkerLabels:\x02\x38\x01\"@\n\x1aUpsertWorkerLabelsResponse\x12\x10\n\x08tenantId\x18\x01 \x01(\t\x12\x10\n\x08workerId\x18\x02 \x01(\t\"\x98\x04\n\x0e\x41ssignedAction\x12\x10\n\x08tenantId\x18\x01 \x01(\t\x12\x15\n\rworkflowRunId\x18\x02 \x01(\t\x12\x18\n\x10getGroupKeyRunId\x18\x03 \x01(\t\x12\r\n\x05jobId\x18\x04 \x01(\t\x12\x0f\n\x07jobName\x18\x05 \x01(\t\x12\x10\n\x08jobRunId\x18\x06 \x01(\t\x12\x0e\n\x06stepId\x18\x07 \x01(\t\x12\x11\n\tstepRunId\x18\x08 \x01(\t\x12\x10\n\x08\x61\x63tionId\x18\t \x01(\t\x12\x1f\n\nactionType\x18\n \x01(\x0e\x32\x0b.ActionType\x12\x15\n\ractionPayload\x18\x0b \x01(\t\x12\x10\n\x08stepName\x18\x0c \x01(\t\x12\x12\n\nretryCount\x18\r \x01(\x05\x12 \n\x13\x61\x64\x64itional_metadata\x18\x0e \x01(\tH\x00\x88\x01\x01\x12!\n\x14\x63hild_workflow_index\x18\x0f \x01(\x05H\x01\x88\x01\x01\x12\x1f\n\x12\x63hild_workflow_key\x18\x10 \x01(\tH\x02\x88\x01\x01\x12#\n\x16parent_workflow_run_id\x18\x11 \x01(\tH\x03\x88\x01\x01\x12\x10\n\x08priority\x18\x12 \x01(\x05\x42\x16\n\x14_additional_metadataB\x17\n\x15_child_workflow_indexB\x15\n\x13_child_workflow_keyB\x19\n\x17_parent_workflow_run_id\"\'\n\x13WorkerListenRequest\x12\x10\n\x08workerId\x18\x01 \x01(\t\",\n\x18WorkerUnsubscribeRequest\x12\x10\n\x08workerId\x18\x01 \x01(\t\"?\n\x19WorkerUnsubscribeResponse\x12\x10\n\x08tenantId\x18\x01 \x01(\t\x12\x10\n\x08workerId\x18\x02 \x01(\t\"\xe1\x01\n\x13GroupKeyActionEvent\x12\x10\n\x08workerId\x18\x01 \x01(\t\x12\x15\n\rworkflowRunId\x18\x02 \x01(\t\x12\x18\n\x10getGroupKeyRunId\x18\x03 \x01(\t\x12\x10\n\x08\x61\x63tionId\x18\x04 \x01(\t\x12\x32\n\x0e\x65ventTimestamp\x18\x05 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12+\n\teventType\x18\x06 \x01(\x0e\x32\x18.GroupKeyActionEventType\x12\x14\n\x0c\x65ventPayload\x18\x07 \x01(\t\"\xc4\x02\n\x0fStepActionEvent\x12\x10\n\x08workerId\x18\x01 \x01(\t\x12\r\n\x05jobId\x18\x02 \x01(\t\x12\x10\n\x08jobRunId\x18\x03 \x01(\t\x12\x0e\n\x06stepId\x18\x04 \x01(\t\x12\x11\n\tstepRunId\x18\x05 \x01(\t\x12\x10\n\x08\x61\x63tionId\x18\x06 \x01(\t\x12\x32\n\x0e\x65ventTimestamp\x18\x07 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12\'\n\teventType\x18\x08 \x01(\x0e\x32\x14.StepActionEventType\x12\x14\n\x0c\x65ventPayload\x18\t \x01(\t\x12\x17\n\nretryCount\x18\n \x01(\x05H\x00\x88\x01\x01\x12\x1b\n\x0eshouldNotRetry\x18\x0b \x01(\x08H\x01\x88\x01\x01\x42\r\n\x0b_retryCountB\x11\n\x0f_shouldNotRetry\"9\n\x13\x41\x63tionEventResponse\x12\x10\n\x08tenantId\x18\x01 \x01(\t\x12\x10\n\x08workerId\x18\x02 \x01(\t\"\xc0\x01\n SubscribeToWorkflowEventsRequest\x12\x1a\n\rworkflowRunId\x18\x01 \x01(\tH\x00\x88\x01\x01\x12\x1e\n\x11\x61\x64\x64itionalMetaKey\x18\x02 \x01(\tH\x01\x88\x01\x01\x12 \n\x13\x61\x64\x64itionalMetaValue\x18\x03 \x01(\tH\x02\x88\x01\x01\x42\x10\n\x0e_workflowRunIdB\x14\n\x12_additionalMetaKeyB\x16\n\x14_additionalMetaValue\"7\n\x1eSubscribeToWorkflowRunsRequest\x12\x15\n\rworkflowRunId\x18\x01 \x01(\t\"\xb2\x02\n\rWorkflowEvent\x12\x15\n\rworkflowRunId\x18\x01 \x01(\t\x12#\n\x0cresourceType\x18\x02 \x01(\x0e\x32\r.ResourceType\x12%\n\teventType\x18\x03 \x01(\x0e\x32\x12.ResourceEventType\x12\x12\n\nresourceId\x18\x04 \x01(\t\x12\x32\n\x0e\x65ventTimestamp\x18\x05 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12\x14\n\x0c\x65ventPayload\x18\x06 \x01(\t\x12\x0e\n\x06hangup\x18\x07 \x01(\x08\x12\x18\n\x0bstepRetries\x18\x08 \x01(\x05H\x00\x88\x01\x01\x12\x17\n\nretryCount\x18\t \x01(\x05H\x01\x88\x01\x01\x42\x0e\n\x0c_stepRetriesB\r\n\x0b_retryCount\"\xa8\x01\n\x10WorkflowRunEvent\x12\x15\n\rworkflowRunId\x18\x01 \x01(\t\x12(\n\teventType\x18\x02 \x01(\x0e\x32\x15.WorkflowRunEventType\x12\x32\n\x0e\x65ventTimestamp\x18\x03 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12\x1f\n\x07results\x18\x04 \x03(\x0b\x32\x0e.StepRunResult\"\x8a\x01\n\rStepRunResult\x12\x11\n\tstepRunId\x18\x01 \x01(\t\x12\x16\n\x0estepReadableId\x18\x02 \x01(\t\x12\x10\n\x08jobRunId\x18\x03 \x01(\t\x12\x12\n\x05\x65rror\x18\x04 \x01(\tH\x00\x88\x01\x01\x12\x13\n\x06output\x18\x05 \x01(\tH\x01\x88\x01\x01\x42\x08\n\x06_errorB\t\n\x07_output\"W\n\rOverridesData\x12\x11\n\tstepRunId\x18\x01 \x01(\t\x12\x0c\n\x04path\x18\x02 \x01(\t\x12\r\n\x05value\x18\x03 \x01(\t\x12\x16\n\x0e\x63\x61llerFilename\x18\x04 \x01(\t\"\x17\n\x15OverridesDataResponse\"U\n\x10HeartbeatRequest\x12\x10\n\x08workerId\x18\x01 \x01(\t\x12/\n\x0bheartbeatAt\x18\x02 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\"\x13\n\x11HeartbeatResponse\"F\n\x15RefreshTimeoutRequest\x12\x11\n\tstepRunId\x18\x01 \x01(\t\x12\x1a\n\x12incrementTimeoutBy\x18\x02 \x01(\t\"G\n\x16RefreshTimeoutResponse\x12-\n\ttimeoutAt\x18\x01 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\"\'\n\x12ReleaseSlotRequest\x12\x11\n\tstepRunId\x18\x01 \x01(\t\"\x15\n\x13ReleaseSlotResponse*7\n\x04SDKS\x12\x0b\n\x07UNKNOWN\x10\x00\x12\x06\n\x02GO\x10\x01\x12\n\n\x06PYTHON\x10\x02\x12\x0e\n\nTYPESCRIPT\x10\x03*N\n\nActionType\x12\x12\n\x0eSTART_STEP_RUN\x10\x00\x12\x13\n\x0f\x43\x41NCEL_STEP_RUN\x10\x01\x12\x17\n\x13START_GET_GROUP_KEY\x10\x02*\xa2\x01\n\x17GroupKeyActionEventType\x12 \n\x1cGROUP_KEY_EVENT_TYPE_UNKNOWN\x10\x00\x12 \n\x1cGROUP_KEY_EVENT_TYPE_STARTED\x10\x01\x12\"\n\x1eGROUP_KEY_EVENT_TYPE_COMPLETED\x10\x02\x12\x1f\n\x1bGROUP_KEY_EVENT_TYPE_FAILED\x10\x03*\xac\x01\n\x13StepActionEventType\x12\x1b\n\x17STEP_EVENT_TYPE_UNKNOWN\x10\x00\x12\x1b\n\x17STEP_EVENT_TYPE_STARTED\x10\x01\x12\x1d\n\x19STEP_EVENT_TYPE_COMPLETED\x10\x02\x12\x1a\n\x16STEP_EVENT_TYPE_FAILED\x10\x03\x12 \n\x1cSTEP_EVENT_TYPE_ACKNOWLEDGED\x10\x04*e\n\x0cResourceType\x12\x19\n\x15RESOURCE_TYPE_UNKNOWN\x10\x00\x12\x1a\n\x16RESOURCE_TYPE_STEP_RUN\x10\x01\x12\x1e\n\x1aRESOURCE_TYPE_WORKFLOW_RUN\x10\x02*\xfe\x01\n\x11ResourceEventType\x12\x1f\n\x1bRESOURCE_EVENT_TYPE_UNKNOWN\x10\x00\x12\x1f\n\x1bRESOURCE_EVENT_TYPE_STARTED\x10\x01\x12!\n\x1dRESOURCE_EVENT_TYPE_COMPLETED\x10\x02\x12\x1e\n\x1aRESOURCE_EVENT_TYPE_FAILED\x10\x03\x12!\n\x1dRESOURCE_EVENT_TYPE_CANCELLED\x10\x04\x12!\n\x1dRESOURCE_EVENT_TYPE_TIMED_OUT\x10\x05\x12\x1e\n\x1aRESOURCE_EVENT_TYPE_STREAM\x10\x06*<\n\x14WorkflowRunEventType\x12$\n WORKFLOW_RUN_EVENT_TYPE_FINISHED\x10\x00\x32\xf8\x06\n\nDispatcher\x12=\n\x08Register\x12\x16.WorkerRegisterRequest\x1a\x17.WorkerRegisterResponse\"\x00\x12\x33\n\x06Listen\x12\x14.WorkerListenRequest\x1a\x0f.AssignedAction\"\x00\x30\x01\x12\x35\n\x08ListenV2\x12\x14.WorkerListenRequest\x1a\x0f.AssignedAction\"\x00\x30\x01\x12\x34\n\tHeartbeat\x12\x11.HeartbeatRequest\x1a\x12.HeartbeatResponse\"\x00\x12R\n\x19SubscribeToWorkflowEvents\x12!.SubscribeToWorkflowEventsRequest\x1a\x0e.WorkflowEvent\"\x00\x30\x01\x12S\n\x17SubscribeToWorkflowRuns\x12\x1f.SubscribeToWorkflowRunsRequest\x1a\x11.WorkflowRunEvent\"\x00(\x01\x30\x01\x12?\n\x13SendStepActionEvent\x12\x10.StepActionEvent\x1a\x14.ActionEventResponse\"\x00\x12G\n\x17SendGroupKeyActionEvent\x12\x14.GroupKeyActionEvent\x1a\x14.ActionEventResponse\"\x00\x12<\n\x10PutOverridesData\x12\x0e.OverridesData\x1a\x16.OverridesDataResponse\"\x00\x12\x46\n\x0bUnsubscribe\x12\x19.WorkerUnsubscribeRequest\x1a\x1a.WorkerUnsubscribeResponse\"\x00\x12\x43\n\x0eRefreshTimeout\x12\x16.RefreshTimeoutRequest\x1a\x17.RefreshTimeoutResponse\"\x00\x12:\n\x0bReleaseSlot\x12\x13.ReleaseSlotRequest\x1a\x14.ReleaseSlotResponse\"\x00\x12O\n\x12UpsertWorkerLabels\x12\x1a.UpsertWorkerLabelsRequest\x1a\x1b.UpsertWorkerLabelsResponse\"\x00\x42GZEgithub.com/hatchet-dev/hatchet/internal/services/dispatcher/contractsb\x06proto3')
+
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'dispatcher_pb2', _globals)
+if not _descriptor._USE_C_DESCRIPTORS:
+  _globals['DESCRIPTOR']._loaded_options = None
+  _globals['DESCRIPTOR']._serialized_options = b'ZEgithub.com/hatchet-dev/hatchet/internal/services/dispatcher/contracts'
+  _globals['_WORKERREGISTERREQUEST_LABELSENTRY']._loaded_options = None
+  _globals['_WORKERREGISTERREQUEST_LABELSENTRY']._serialized_options = b'8\001'
+  _globals['_UPSERTWORKERLABELSREQUEST_LABELSENTRY']._loaded_options = None
+  _globals['_UPSERTWORKERLABELSREQUEST_LABELSENTRY']._serialized_options = b'8\001'
+  _globals['_SDKS']._serialized_start=3590
+  _globals['_SDKS']._serialized_end=3645
+  _globals['_ACTIONTYPE']._serialized_start=3647
+  _globals['_ACTIONTYPE']._serialized_end=3725
+  _globals['_GROUPKEYACTIONEVENTTYPE']._serialized_start=3728
+  _globals['_GROUPKEYACTIONEVENTTYPE']._serialized_end=3890
+  _globals['_STEPACTIONEVENTTYPE']._serialized_start=3893
+  _globals['_STEPACTIONEVENTTYPE']._serialized_end=4065
+  _globals['_RESOURCETYPE']._serialized_start=4067
+  _globals['_RESOURCETYPE']._serialized_end=4168
+  _globals['_RESOURCEEVENTTYPE']._serialized_start=4171
+  _globals['_RESOURCEEVENTTYPE']._serialized_end=4425
+  _globals['_WORKFLOWRUNEVENTTYPE']._serialized_start=4427
+  _globals['_WORKFLOWRUNEVENTTYPE']._serialized_end=4487
+  _globals['_WORKERLABELS']._serialized_start=53
+  _globals['_WORKERLABELS']._serialized_end=139
+  _globals['_RUNTIMEINFO']._serialized_start=142
+  _globals['_RUNTIMEINFO']._serialized_end=342
+  _globals['_WORKERREGISTERREQUEST']._serialized_start=345
+  _globals['_WORKERREGISTERREQUEST']._serialized_end=665
+  _globals['_WORKERREGISTERREQUEST_LABELSENTRY']._serialized_start=563
+  _globals['_WORKERREGISTERREQUEST_LABELSENTRY']._serialized_end=623
+  _globals['_WORKERREGISTERRESPONSE']._serialized_start=667
+  _globals['_WORKERREGISTERRESPONSE']._serialized_end=747
+  _globals['_UPSERTWORKERLABELSREQUEST']._serialized_start=750
+  _globals['_UPSERTWORKERLABELSREQUEST']._serialized_end=913
+  _globals['_UPSERTWORKERLABELSREQUEST_LABELSENTRY']._serialized_start=563
+  _globals['_UPSERTWORKERLABELSREQUEST_LABELSENTRY']._serialized_end=623
+  _globals['_UPSERTWORKERLABELSRESPONSE']._serialized_start=915
+  _globals['_UPSERTWORKERLABELSRESPONSE']._serialized_end=979
+  _globals['_ASSIGNEDACTION']._serialized_start=982
+  _globals['_ASSIGNEDACTION']._serialized_end=1518
+  _globals['_WORKERLISTENREQUEST']._serialized_start=1520
+  _globals['_WORKERLISTENREQUEST']._serialized_end=1559
+  _globals['_WORKERUNSUBSCRIBEREQUEST']._serialized_start=1561
+  _globals['_WORKERUNSUBSCRIBEREQUEST']._serialized_end=1605
+  _globals['_WORKERUNSUBSCRIBERESPONSE']._serialized_start=1607
+  _globals['_WORKERUNSUBSCRIBERESPONSE']._serialized_end=1670
+  _globals['_GROUPKEYACTIONEVENT']._serialized_start=1673
+  _globals['_GROUPKEYACTIONEVENT']._serialized_end=1898
+  _globals['_STEPACTIONEVENT']._serialized_start=1901
+  _globals['_STEPACTIONEVENT']._serialized_end=2225
+  _globals['_ACTIONEVENTRESPONSE']._serialized_start=2227
+  _globals['_ACTIONEVENTRESPONSE']._serialized_end=2284
+  _globals['_SUBSCRIBETOWORKFLOWEVENTSREQUEST']._serialized_start=2287
+  _globals['_SUBSCRIBETOWORKFLOWEVENTSREQUEST']._serialized_end=2479
+  _globals['_SUBSCRIBETOWORKFLOWRUNSREQUEST']._serialized_start=2481
+  _globals['_SUBSCRIBETOWORKFLOWRUNSREQUEST']._serialized_end=2536
+  _globals['_WORKFLOWEVENT']._serialized_start=2539
+  _globals['_WORKFLOWEVENT']._serialized_end=2845
+  _globals['_WORKFLOWRUNEVENT']._serialized_start=2848
+  _globals['_WORKFLOWRUNEVENT']._serialized_end=3016
+  _globals['_STEPRUNRESULT']._serialized_start=3019
+  _globals['_STEPRUNRESULT']._serialized_end=3157
+  _globals['_OVERRIDESDATA']._serialized_start=3159
+  _globals['_OVERRIDESDATA']._serialized_end=3246
+  _globals['_OVERRIDESDATARESPONSE']._serialized_start=3248
+  _globals['_OVERRIDESDATARESPONSE']._serialized_end=3271
+  _globals['_HEARTBEATREQUEST']._serialized_start=3273
+  _globals['_HEARTBEATREQUEST']._serialized_end=3358
+  _globals['_HEARTBEATRESPONSE']._serialized_start=3360
+  _globals['_HEARTBEATRESPONSE']._serialized_end=3379
+  _globals['_REFRESHTIMEOUTREQUEST']._serialized_start=3381
+  _globals['_REFRESHTIMEOUTREQUEST']._serialized_end=3451
+  _globals['_REFRESHTIMEOUTRESPONSE']._serialized_start=3453
+  _globals['_REFRESHTIMEOUTRESPONSE']._serialized_end=3524
+  _globals['_RELEASESLOTREQUEST']._serialized_start=3526
+  _globals['_RELEASESLOTREQUEST']._serialized_end=3565
+  _globals['_RELEASESLOTRESPONSE']._serialized_start=3567
+  _globals['_RELEASESLOTRESPONSE']._serialized_end=3588
+  _globals['_DISPATCHER']._serialized_start=4490
+  _globals['_DISPATCHER']._serialized_end=5378
+# @@protoc_insertion_point(module_scope)

{hatchet_sdk-1.4.1 → hatchet_sdk-1.6.1}/hatchet_sdk/contracts/dispatcher_pb2.pyi

@@ -163,7 +163,7 @@ class UpsertWorkerLabelsResponse(_message.Message):
     def __init__(self, tenantId: _Optional[str] = ..., workerId: _Optional[str] = ...) -> None: ...
 
 class AssignedAction(_message.Message):
-    __slots__ = ("tenantId", "workflowRunId", "getGroupKeyRunId", "jobId", "jobName", "jobRunId", "stepId", "stepRunId", "actionId", "actionType", "actionPayload", "stepName", "retryCount", "additional_metadata", "child_workflow_index", "child_workflow_key", "parent_workflow_run_id")
+    __slots__ = ("tenantId", "workflowRunId", "getGroupKeyRunId", "jobId", "jobName", "jobRunId", "stepId", "stepRunId", "actionId", "actionType", "actionPayload", "stepName", "retryCount", "additional_metadata", "child_workflow_index", "child_workflow_key", "parent_workflow_run_id", "priority")
     TENANTID_FIELD_NUMBER: _ClassVar[int]
     WORKFLOWRUNID_FIELD_NUMBER: _ClassVar[int]
     GETGROUPKEYRUNID_FIELD_NUMBER: _ClassVar[int]
@@ -181,6 +181,7 @@ class AssignedAction(_message.Message):
     CHILD_WORKFLOW_INDEX_FIELD_NUMBER: _ClassVar[int]
     CHILD_WORKFLOW_KEY_FIELD_NUMBER: _ClassVar[int]
     PARENT_WORKFLOW_RUN_ID_FIELD_NUMBER: _ClassVar[int]
+    PRIORITY_FIELD_NUMBER: _ClassVar[int]
     tenantId: str
     workflowRunId: str
     getGroupKeyRunId: str
@@ -198,7 +199,8 @@ class AssignedAction(_message.Message):
     child_workflow_index: int
     child_workflow_key: str
     parent_workflow_run_id: str
-    def __init__(self, tenantId: _Optional[str] = ..., workflowRunId: _Optional[str] = ..., getGroupKeyRunId: _Optional[str] = ..., jobId: _Optional[str] = ..., jobName: _Optional[str] = ..., jobRunId: _Optional[str] = ..., stepId: _Optional[str] = ..., stepRunId: _Optional[str] = ..., actionId: _Optional[str] = ..., actionType: _Optional[_Union[ActionType, str]] = ..., actionPayload: _Optional[str] = ..., stepName: _Optional[str] = ..., retryCount: _Optional[int] = ..., additional_metadata: _Optional[str] = ..., child_workflow_index: _Optional[int] = ..., child_workflow_key: _Optional[str] = ..., parent_workflow_run_id: _Optional[str] = ...) -> None: ...
+    priority: int
+    def __init__(self, tenantId: _Optional[str] = ..., workflowRunId: _Optional[str] = ..., getGroupKeyRunId: _Optional[str] = ..., jobId: _Optional[str] = ..., jobName: _Optional[str] = ..., jobRunId: _Optional[str] = ..., stepId: _Optional[str] = ..., stepRunId: _Optional[str] = ..., actionId: _Optional[str] = ..., actionType: _Optional[_Union[ActionType, str]] = ..., actionPayload: _Optional[str] = ..., stepName: _Optional[str] = ..., retryCount: _Optional[int] = ..., additional_metadata: _Optional[str] = ..., child_workflow_index: _Optional[int] = ..., child_workflow_key: _Optional[str] = ..., parent_workflow_run_id: _Optional[str] = ..., priority: _Optional[int] = ...) -> None: ...
 
 class WorkerListenRequest(_message.Message):
     __slots__ = ("workerId",)

hatchet_sdk-1.6.1/hatchet_sdk/contracts/v1/workflows_pb2.py

@@ -0,0 +1,67 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: v1/workflows.proto
+# Protobuf Python Version: 5.26.1
+"""Generated protocol buffer code."""
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import descriptor_pool as _descriptor_pool
+from google.protobuf import symbol_database as _symbol_database
+from google.protobuf.internal import builder as _builder
+# @@protoc_insertion_point(imports)
+
+_sym_db = _symbol_database.Default()
+
+
+from google.protobuf import timestamp_pb2 as google_dot_protobuf_dot_timestamp__pb2
+from hatchet_sdk.contracts.v1.shared import condition_pb2 as v1_dot_shared_dot_condition__pb2
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x12v1/workflows.proto\x12\x02v1\x1a\x1fgoogle/protobuf/timestamp.proto\x1a\x19v1/shared/condition.proto\"Z\n\x12\x43\x61ncelTasksRequest\x12\x13\n\x0b\x65xternalIds\x18\x01 \x03(\t\x12$\n\x06\x66ilter\x18\x02 \x01(\x0b\x32\x0f.v1.TasksFilterH\x00\x88\x01\x01\x42\t\n\x07_filter\"Z\n\x12ReplayTasksRequest\x12\x13\n\x0b\x65xternalIds\x18\x01 \x03(\t\x12$\n\x06\x66ilter\x18\x02 \x01(\x0b\x32\x0f.v1.TasksFilterH\x00\x88\x01\x01\x42\t\n\x07_filter\"\xb7\x01\n\x0bTasksFilter\x12\x10\n\x08statuses\x18\x01 \x03(\t\x12)\n\x05since\x18\x02 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12.\n\x05until\x18\x03 \x01(\x0b\x32\x1a.google.protobuf.TimestampH\x00\x88\x01\x01\x12\x14\n\x0cworkflow_ids\x18\x04 \x03(\t\x12\x1b\n\x13\x61\x64\x64itional_metadata\x18\x05 \x03(\tB\x08\n\x06_until\".\n\x13\x43\x61ncelTasksResponse\x12\x17\n\x0f\x63\x61ncelled_tasks\x18\x01 \x03(\t\"-\n\x13ReplayTasksResponse\x12\x16\n\x0ereplayed_tasks\x18\x01 \x03(\t\"\x82\x01\n\x19TriggerWorkflowRunRequest\x12\x15\n\rworkflow_name\x18\x01 \x01(\t\x12\r\n\x05input\x18\x02 \x01(\x0c\x12\x1b\n\x13\x61\x64\x64itional_metadata\x18\x03 \x01(\x0c\x12\x15\n\x08priority\x18\x04 \x01(\x05H\x00\x88\x01\x01\x42\x0b\n\t_priority\"1\n\x1aTriggerWorkflowRunResponse\x12\x13\n\x0b\x65xternal_id\x18\x01 \x01(\t\"\xca\x03\n\x1c\x43reateWorkflowVersionRequest\x12\x0c\n\x04name\x18\x01 \x01(\t\x12\x13\n\x0b\x64\x65scription\x18\x02 \x01(\t\x12\x0f\n\x07version\x18\x03 \x01(\t\x12\x16\n\x0e\x65vent_triggers\x18\x04 \x03(\t\x12\x15\n\rcron_triggers\x18\x05 \x03(\t\x12!\n\x05tasks\x18\x06 \x03(\x0b\x32\x12.v1.CreateTaskOpts\x12$\n\x0b\x63oncurrency\x18\x07 \x01(\x0b\x32\x0f.v1.Concurrency\x12\x17\n\ncron_input\x18\x08 \x01(\tH\x00\x88\x01\x01\x12\x30\n\x0fon_failure_task\x18\t \x01(\x0b\x32\x12.v1.CreateTaskOptsH\x01\x88\x01\x01\x12\'\n\x06sticky\x18\n \x01(\x0e\x32\x12.v1.StickyStrategyH\x02\x88\x01\x01\x12\x1d\n\x10\x64\x65\x66\x61ult_priority\x18\x0b \x01(\x05H\x03\x88\x01\x01\x12(\n\x0f\x63oncurrency_arr\x18\x0c \x03(\x0b\x32\x0f.v1.ConcurrencyB\r\n\x0b_cron_inputB\x12\n\x10_on_failure_taskB\t\n\x07_stickyB\x13\n\x11_default_priority\"\x93\x01\n\x0b\x43oncurrency\x12\x12\n\nexpression\x18\x01 \x01(\t\x12\x15\n\x08max_runs\x18\x02 \x01(\x05H\x00\x88\x01\x01\x12\x39\n\x0elimit_strategy\x18\x03 \x01(\x0e\x32\x1c.v1.ConcurrencyLimitStrategyH\x01\x88\x01\x01\x42\x0b\n\t_max_runsB\x11\n\x0f_limit_strategy\"\xe4\x01\n\x13\x44\x65siredWorkerLabels\x12\x15\n\x08strValue\x18\x01 \x01(\tH\x00\x88\x01\x01\x12\x15\n\x08intValue\x18\x02 \x01(\x05H\x01\x88\x01\x01\x12\x15\n\x08required\x18\x03 \x01(\x08H\x02\x88\x01\x01\x12\x32\n\ncomparator\x18\x04 \x01(\x0e\x32\x19.v1.WorkerLabelComparatorH\x03\x88\x01\x01\x12\x13\n\x06weight\x18\x05 \x01(\x05H\x04\x88\x01\x01\x42\x0b\n\t_strValueB\x0b\n\t_intValueB\x0b\n\t_requiredB\r\n\x0b_comparatorB\t\n\x07_weight\"\xb1\x04\n\x0e\x43reateTaskOpts\x12\x13\n\x0breadable_id\x18\x01 \x01(\t\x12\x0e\n\x06\x61\x63tion\x18\x02 \x01(\t\x12\x0f\n\x07timeout\x18\x03 \x01(\t\x12\x0e\n\x06inputs\x18\x04 \x01(\t\x12\x0f\n\x07parents\x18\x05 \x03(\t\x12\x0f\n\x07retries\x18\x06 \x01(\x05\x12,\n\x0brate_limits\x18\x07 \x03(\x0b\x32\x17.v1.CreateTaskRateLimit\x12;\n\rworker_labels\x18\x08 \x03(\x0b\x32$.v1.CreateTaskOpts.WorkerLabelsEntry\x12\x1b\n\x0e\x62\x61\x63koff_factor\x18\t \x01(\x02H\x00\x88\x01\x01\x12 \n\x13\x62\x61\x63koff_max_seconds\x18\n \x01(\x05H\x01\x88\x01\x01\x12$\n\x0b\x63oncurrency\x18\x0b \x03(\x0b\x32\x0f.v1.Concurrency\x12+\n\nconditions\x18\x0c \x01(\x0b\x32\x12.v1.TaskConditionsH\x02\x88\x01\x01\x12\x1d\n\x10schedule_timeout\x18\r \x01(\tH\x03\x88\x01\x01\x1aL\n\x11WorkerLabelsEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12&\n\x05value\x18\x02 \x01(\x0b\x32\x17.v1.DesiredWorkerLabels:\x02\x38\x01\x42\x11\n\x0f_backoff_factorB\x16\n\x14_backoff_max_secondsB\r\n\x0b_conditionsB\x13\n\x11_schedule_timeout\"\xfd\x01\n\x13\x43reateTaskRateLimit\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\x12\n\x05units\x18\x02 \x01(\x05H\x00\x88\x01\x01\x12\x15\n\x08key_expr\x18\x03 \x01(\tH\x01\x88\x01\x01\x12\x17\n\nunits_expr\x18\x04 \x01(\tH\x02\x88\x01\x01\x12\x1e\n\x11limit_values_expr\x18\x05 \x01(\tH\x03\x88\x01\x01\x12,\n\x08\x64uration\x18\x06 \x01(\x0e\x32\x15.v1.RateLimitDurationH\x04\x88\x01\x01\x42\x08\n\x06_unitsB\x0b\n\t_key_exprB\r\n\x0b_units_exprB\x14\n\x12_limit_values_exprB\x0b\n\t_duration\"@\n\x1d\x43reateWorkflowVersionResponse\x12\n\n\x02id\x18\x01 \x01(\t\x12\x13\n\x0bworkflow_id\x18\x02 \x01(\t*$\n\x0eStickyStrategy\x12\x08\n\x04SOFT\x10\x00\x12\x08\n\x04HARD\x10\x01*]\n\x11RateLimitDuration\x12\n\n\x06SECOND\x10\x00\x12\n\n\x06MINUTE\x10\x01\x12\x08\n\x04HOUR\x10\x02\x12\x07\n\x03\x44\x41Y\x10\x03\x12\x08\n\x04WEEK\x10\x04\x12\t\n\x05MONTH\x10\x05\x12\x08\n\x04YEAR\x10\x06*\x7f\n\x18\x43oncurrencyLimitStrategy\x12\x16\n\x12\x43\x41NCEL_IN_PROGRESS\x10\x00\x12\x0f\n\x0b\x44ROP_NEWEST\x10\x01\x12\x10\n\x0cQUEUE_NEWEST\x10\x02\x12\x15\n\x11GROUP_ROUND_ROBIN\x10\x03\x12\x11\n\rCANCEL_NEWEST\x10\x04*\x85\x01\n\x15WorkerLabelComparator\x12\t\n\x05\x45QUAL\x10\x00\x12\r\n\tNOT_EQUAL\x10\x01\x12\x10\n\x0cGREATER_THAN\x10\x02\x12\x19\n\x15GREATER_THAN_OR_EQUAL\x10\x03\x12\r\n\tLESS_THAN\x10\x04\x12\x16\n\x12LESS_THAN_OR_EQUAL\x10\x05\x32\xb7\x02\n\x0c\x41\x64minService\x12R\n\x0bPutWorkflow\x12 .v1.CreateWorkflowVersionRequest\x1a!.v1.CreateWorkflowVersionResponse\x12>\n\x0b\x43\x61ncelTasks\x12\x16.v1.CancelTasksRequest\x1a\x17.v1.CancelTasksResponse\x12>\n\x0bReplayTasks\x12\x16.v1.ReplayTasksRequest\x1a\x17.v1.ReplayTasksResponse\x12S\n\x12TriggerWorkflowRun\x12\x1d.v1.TriggerWorkflowRunRequest\x1a\x1e.v1.TriggerWorkflowRunResponseBBZ@github.com/hatchet-dev/hatchet/internal/services/shared/proto/v1b\x06proto3')
+
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'v1.workflows_pb2', _globals)
+if not _descriptor._USE_C_DESCRIPTORS:
+  _globals['DESCRIPTOR']._loaded_options = None
+  _globals['DESCRIPTOR']._serialized_options = b'Z@github.com/hatchet-dev/hatchet/internal/services/shared/proto/v1'
+  _globals['_CREATETASKOPTS_WORKERLABELSENTRY']._loaded_options = None
+  _globals['_CREATETASKOPTS_WORKERLABELSENTRY']._serialized_options = b'8\001'
+  _globals['_STICKYSTRATEGY']._serialized_start=2463
+  _globals['_STICKYSTRATEGY']._serialized_end=2499
+  _globals['_RATELIMITDURATION']._serialized_start=2501
+  _globals['_RATELIMITDURATION']._serialized_end=2594
+  _globals['_CONCURRENCYLIMITSTRATEGY']._serialized_start=2596
+  _globals['_CONCURRENCYLIMITSTRATEGY']._serialized_end=2723
+  _globals['_WORKERLABELCOMPARATOR']._serialized_start=2726
+  _globals['_WORKERLABELCOMPARATOR']._serialized_end=2859
+  _globals['_CANCELTASKSREQUEST']._serialized_start=86
+  _globals['_CANCELTASKSREQUEST']._serialized_end=176
+  _globals['_REPLAYTASKSREQUEST']._serialized_start=178
+  _globals['_REPLAYTASKSREQUEST']._serialized_end=268
+  _globals['_TASKSFILTER']._serialized_start=271
+  _globals['_TASKSFILTER']._serialized_end=454
+  _globals['_CANCELTASKSRESPONSE']._serialized_start=456
+  _globals['_CANCELTASKSRESPONSE']._serialized_end=502
+  _globals['_REPLAYTASKSRESPONSE']._serialized_start=504
+  _globals['_REPLAYTASKSRESPONSE']._serialized_end=549
+  _globals['_TRIGGERWORKFLOWRUNREQUEST']._serialized_start=552
+  _globals['_TRIGGERWORKFLOWRUNREQUEST']._serialized_end=682
+  _globals['_TRIGGERWORKFLOWRUNRESPONSE']._serialized_start=684
+  _globals['_TRIGGERWORKFLOWRUNRESPONSE']._serialized_end=733
+  _globals['_CREATEWORKFLOWVERSIONREQUEST']._serialized_start=736
+  _globals['_CREATEWORKFLOWVERSIONREQUEST']._serialized_end=1194
+  _globals['_CONCURRENCY']._serialized_start=1197
+  _globals['_CONCURRENCY']._serialized_end=1344
+  _globals['_DESIREDWORKERLABELS']._serialized_start=1347
+  _globals['_DESIREDWORKERLABELS']._serialized_end=1575
+  _globals['_CREATETASKOPTS']._serialized_start=1578
+  _globals['_CREATETASKOPTS']._serialized_end=2139
+  _globals['_CREATETASKOPTS_WORKERLABELSENTRY']._serialized_start=1984
+  _globals['_CREATETASKOPTS_WORKERLABELSENTRY']._serialized_end=2060
+  _globals['_CREATETASKRATELIMIT']._serialized_start=2142
+  _globals['_CREATETASKRATELIMIT']._serialized_end=2395
+  _globals['_CREATEWORKFLOWVERSIONRESPONSE']._serialized_start=2397
+  _globals['_CREATEWORKFLOWVERSIONRESPONSE']._serialized_end=2461
+  _globals['_ADMINSERVICE']._serialized_start=2862
+  _globals['_ADMINSERVICE']._serialized_end=3173
+# @@protoc_insertion_point(module_scope)

{hatchet_sdk-1.4.1 → hatchet_sdk-1.6.1}/hatchet_sdk/contracts/v1/workflows_pb2.pyi

@@ -103,14 +103,16 @@ class ReplayTasksResponse(_message.Message):
     def __init__(self, replayed_tasks: _Optional[_Iterable[str]] = ...) -> None: ...
 
 class TriggerWorkflowRunRequest(_message.Message):
-    __slots__ = ("workflow_name", "input", "additional_metadata")
+    __slots__ = ("workflow_name", "input", "additional_metadata", "priority")
     WORKFLOW_NAME_FIELD_NUMBER: _ClassVar[int]
     INPUT_FIELD_NUMBER: _ClassVar[int]
     ADDITIONAL_METADATA_FIELD_NUMBER: _ClassVar[int]
+    PRIORITY_FIELD_NUMBER: _ClassVar[int]
     workflow_name: str
     input: bytes
     additional_metadata: bytes
-    def __init__(self, workflow_name: _Optional[str] = ..., input: _Optional[bytes] = ..., additional_metadata: _Optional[bytes] = ...) -> None: ...
+    priority: int
+    def __init__(self, workflow_name: _Optional[str] = ..., input: _Optional[bytes] = ..., additional_metadata: _Optional[bytes] = ..., priority: _Optional[int] = ...) -> None: ...
 
 class TriggerWorkflowRunResponse(_message.Message):
     __slots__ = ("external_id",)
@@ -119,7 +121,7 @@ class TriggerWorkflowRunResponse(_message.Message):
     def __init__(self, external_id: _Optional[str] = ...) -> None: ...
 
 class CreateWorkflowVersionRequest(_message.Message):
-    __slots__ = ("name", "description", "version", "event_triggers", "cron_triggers", "tasks", "concurrency", "cron_input", "on_failure_task", "sticky")
+    __slots__ = ("name", "description", "version", "event_triggers", "cron_triggers", "tasks", "concurrency", "cron_input", "on_failure_task", "sticky", "default_priority", "concurrency_arr")
     NAME_FIELD_NUMBER: _ClassVar[int]
     DESCRIPTION_FIELD_NUMBER: _ClassVar[int]
     VERSION_FIELD_NUMBER: _ClassVar[int]
@@ -130,6 +132,8 @@ class CreateWorkflowVersionRequest(_message.Message):
     CRON_INPUT_FIELD_NUMBER: _ClassVar[int]
     ON_FAILURE_TASK_FIELD_NUMBER: _ClassVar[int]
     STICKY_FIELD_NUMBER: _ClassVar[int]
+    DEFAULT_PRIORITY_FIELD_NUMBER: _ClassVar[int]
+    CONCURRENCY_ARR_FIELD_NUMBER: _ClassVar[int]
     name: str
     description: str
     version: str
@@ -140,7 +144,9 @@ class CreateWorkflowVersionRequest(_message.Message):
     cron_input: str
     on_failure_task: CreateTaskOpts
     sticky: StickyStrategy
-    def __init__(self, name: _Optional[str] = ..., description: _Optional[str] = ..., version: _Optional[str] = ..., event_triggers: _Optional[_Iterable[str]] = ..., cron_triggers: _Optional[_Iterable[str]] = ..., tasks: _Optional[_Iterable[_Union[CreateTaskOpts, _Mapping]]] = ..., concurrency: _Optional[_Union[Concurrency, _Mapping]] = ..., cron_input: _Optional[str] = ..., on_failure_task: _Optional[_Union[CreateTaskOpts, _Mapping]] = ..., sticky: _Optional[_Union[StickyStrategy, str]] = ...) -> None: ...
+    default_priority: int
+    concurrency_arr: _containers.RepeatedCompositeFieldContainer[Concurrency]
+    def __init__(self, name: _Optional[str] = ..., description: _Optional[str] = ..., version: _Optional[str] = ..., event_triggers: _Optional[_Iterable[str]] = ..., cron_triggers: _Optional[_Iterable[str]] = ..., tasks: _Optional[_Iterable[_Union[CreateTaskOpts, _Mapping]]] = ..., concurrency: _Optional[_Union[Concurrency, _Mapping]] = ..., cron_input: _Optional[str] = ..., on_failure_task: _Optional[_Union[CreateTaskOpts, _Mapping]] = ..., sticky: _Optional[_Union[StickyStrategy, str]] = ..., default_priority: _Optional[int] = ..., concurrency_arr: _Optional[_Iterable[_Union[Concurrency, _Mapping]]] = ...) -> None: ...
 
 class Concurrency(_message.Message):
     __slots__ = ("expression", "max_runs", "limit_strategy")