cadence-python-client 0.1.0__py3-none-any.whl

cadence/activity.py

@@ -0,0 +1,255 @@
+import inspect
+from abc import ABC, abstractmethod
+from contextlib import contextmanager
+from contextvars import ContextVar
+from dataclasses import dataclass
+from datetime import timedelta, datetime
+from enum import Enum
+from functools import update_wrapper
+from inspect import signature, Parameter
+from typing import (
+    Iterator,
+    TypedDict,
+    Unpack,
+    Callable,
+    Type,
+    ParamSpec,
+    TypeVar,
+    Generic,
+    get_type_hints,
+    Any,
+    overload,
+    Tuple,
+    Sequence,
+)
+
+from cadence import Client
+from cadence.workflow import WorkflowContext, ActivityOptions, execute_activity
+
+
+@dataclass(frozen=True)
+class ActivityInfo:
+    task_token: bytes
+    workflow_type: str
+    workflow_domain: str
+    workflow_id: str
+    workflow_run_id: str
+    activity_id: str
+    activity_type: str
+    task_list: str
+    heartbeat_timeout: timedelta
+    scheduled_timestamp: datetime
+    started_timestamp: datetime
+    start_to_close_timeout: timedelta
+    attempt: int
+
+
+def client() -> Client:
+    return ActivityContext.get().client()
+
+
+def in_activity() -> bool:
+    return ActivityContext.is_set()
+
+
+def info() -> ActivityInfo:
+    return ActivityContext.get().info()
+
+
+class ActivityContext(ABC):
+    _var: ContextVar["ActivityContext"] = ContextVar("activity")
+
+    @abstractmethod
+    def info(self) -> ActivityInfo: ...
+
+    @abstractmethod
+    def client(self) -> Client: ...
+
+    @contextmanager
+    def _activate(self) -> Iterator[None]:
+        token = ActivityContext._var.set(self)
+        yield None
+        ActivityContext._var.reset(token)
+
+    @staticmethod
+    def is_set() -> bool:
+        return ActivityContext._var.get(None) is not None
+
+    @staticmethod
+    def get() -> "ActivityContext":
+        return ActivityContext._var.get()
+
+
+@dataclass(frozen=True)
+class ActivityParameter:
+    name: str
+    type_hint: Type | None
+    has_default: bool
+    default_value: Any
+
+
+class ExecutionStrategy(Enum):
+    ASYNC = "async"
+    THREAD_POOL = "thread_pool"
+
+
+class ActivityDefinitionOptions(TypedDict, total=False):
+    name: str
+
+
+P = ParamSpec("P")
+T = TypeVar("T")
+
+
+class ActivityDefinition(Generic[P, T]):
+    def __init__(
+        self,
+        wrapped: Callable[P, T],
+        name: str,
+        strategy: ExecutionStrategy,
+        params: list[ActivityParameter],
+        result_type: Type[T],
+    ):
+        self._wrapped = wrapped
+        self._name = name
+        self._strategy = strategy
+        self._params = params
+        self._result_type = result_type
+        self._execution_options = ActivityOptions()
+        update_wrapper(self, wrapped)
+
+    def __call__(self, *args: P.args, **kwargs: P.kwargs) -> T:
+        if WorkflowContext.is_set():
+            # If the original function is async then this is fine
+            # If it's not async then this is invalid typing, but still allowed
+            # Users can use execute as a guaranteed type safe option if the function is sync
+            return self.execute(*args, **kwargs)  # type: ignore
+        return self._wrapped(*args, **kwargs)
+
+    def with_options(
+        self, **kwargs: Unpack[ActivityOptions]
+    ) -> "ActivityDefinition[P, T]":
+        res = ActivityDefinition(
+            self._wrapped, self._name, self.strategy, self.params, self.result_type
+        )
+        new_opts = self._execution_options.copy()
+        new_opts.update(kwargs)
+        res._execution_options = new_opts
+        return res
+
+    async def execute(self, *args: P.args, **kwargs: P.kwargs) -> T:
+        return await execute_activity(
+            self.name,
+            self.result_type,
+            *_to_parameters(self.params, args, kwargs),
+            **self._execution_options,
+        )
+
+    @property
+    def name(self) -> str:
+        return self._name
+
+    @property
+    def strategy(self) -> ExecutionStrategy:
+        return self._strategy
+
+    @property
+    def params(self) -> list[ActivityParameter]:
+        return self._params
+
+    @property
+    def result_type(self) -> Type[T]:
+        return self._result_type
+
+    @staticmethod
+    def wrap(
+        fn: Callable[P, T], opts: ActivityDefinitionOptions
+    ) -> "ActivityDefinition[P, T]":
+        name = fn.__qualname__
+        if "name" in opts and opts["name"]:
+            name = opts["name"]
+
+        strategy = ExecutionStrategy.THREAD_POOL
+        if inspect.iscoroutinefunction(fn) or inspect.iscoroutinefunction(fn.__call__):  # type: ignore
+            strategy = ExecutionStrategy.ASYNC
+
+        params, result_type = _get_signature(fn)
+        return ActivityDefinition(fn, name, strategy, params, result_type)
+
+
+ActivityDecorator = Callable[[Callable[P, T]], ActivityDefinition[P, T]]
+
+
+@overload
+def defn(fn: Callable[P, T]) -> ActivityDefinition[P, T]: ...
+
+
+@overload
+def defn(**kwargs: Unpack[ActivityDefinitionOptions]) -> ActivityDecorator: ...
+
+
+def defn(
+    fn: Callable[P, T] | None = None, **kwargs: Unpack[ActivityDefinitionOptions]
+) -> ActivityDecorator | ActivityDefinition[P, T]:
+    options = ActivityDefinitionOptions(**kwargs)
+
+    def decorator(inner_fn: Callable[P, T]) -> ActivityDefinition[P, T]:
+        return ActivityDefinition.wrap(inner_fn, options)
+
+    if fn is not None:
+        return decorator(fn)
+
+    return decorator
+
+
+def _get_signature(fn: Callable[P, T]) -> Tuple[list[ActivityParameter], Type[T]]:
+    sig = signature(fn)
+    args = sig.parameters
+    hints = get_type_hints(fn)
+    params = []
+    for name, param in args.items():
+        # "unbound functions" aren't a thing in the Python spec. We don't have a way to determine whether the function
+        # is part of a class or is standalone.
+        # Filter out the self parameter and hope they followed the convention.
+        if param.name == "self":
+            continue
+        default = None
+        has_default = False
+        if param.default != Parameter.empty:
+            default = param.default
+            has_default = param.default is not None
+        if param.kind in (Parameter.POSITIONAL_ONLY, Parameter.POSITIONAL_OR_KEYWORD):
+            type_hint = hints.get(name, None)
+            params.append(ActivityParameter(name, type_hint, has_default, default))
+        else:
+            raise ValueError(
+                f"Parameters must be positional. {name} is {param.kind}, and not valid"
+            )
+
+    # Treat unspecified return type
+    return_type = hints.get("return", dict)
+
+    return params, return_type
+
+
+def _to_parameters(
+    params: list[ActivityParameter], args: Sequence[Any], kwargs: dict[str, Any]
+) -> list[Any]:
+    result: list[Any] = []
+    for value, param_spec in zip(args, params):
+        result.append(value)
+
+    i = len(result)
+    while i < len(params):
+        param = params[i]
+        if param.name not in kwargs and not param.has_default:
+            raise ValueError(f"Missing parameter: {param.name}")
+
+        value = kwargs.pop(param.name, param.default_value)
+        result.append(value)
+        i = i + 1
+
+    if len(kwargs) > 0:
+        raise ValueError(f"Unexpected keyword arguments: {kwargs}")
+
+    return result

cadence/api/v1/__init__.py

@@ -0,0 +1,92 @@
+# Auto-generated __init__.py file
+# Import all generated protobuf modules
+from . import common_pb2
+from . import common_pb2_grpc
+from . import decision_pb2
+from . import decision_pb2_grpc
+from . import domain_pb2
+from . import domain_pb2_grpc
+from . import error_pb2
+from . import error_pb2_grpc
+from . import history_pb2
+from . import history_pb2_grpc
+from . import query_pb2
+from . import query_pb2_grpc
+from . import service_domain_pb2
+from . import service_domain_pb2_grpc
+from . import service_meta_pb2
+from . import service_meta_pb2_grpc
+from . import service_visibility_pb2
+from . import service_visibility_pb2_grpc
+from . import service_worker_pb2
+from . import service_worker_pb2_grpc
+from . import service_workflow_pb2
+from . import service_workflow_pb2_grpc
+from . import tasklist_pb2
+from . import tasklist_pb2_grpc
+from . import visibility_pb2
+from . import visibility_pb2_grpc
+from . import workflow_pb2
+from . import workflow_pb2_grpc
+
+# Create cleaner aliases for easier imports
+common = common_pb2
+common_grpc = common_pb2_grpc
+decision = decision_pb2
+decision_grpc = decision_pb2_grpc
+domain = domain_pb2
+domain_grpc = domain_pb2_grpc
+error = error_pb2
+error_grpc = error_pb2_grpc
+history = history_pb2
+history_grpc = history_pb2_grpc
+query = query_pb2
+query_grpc = query_pb2_grpc
+service_domain = service_domain_pb2
+service_domain_grpc = service_domain_pb2_grpc
+service_meta = service_meta_pb2
+service_meta_grpc = service_meta_pb2_grpc
+service_visibility = service_visibility_pb2
+service_visibility_grpc = service_visibility_pb2_grpc
+service_worker = service_worker_pb2
+service_worker_grpc = service_worker_pb2_grpc
+service_workflow = service_workflow_pb2
+service_workflow_grpc = service_workflow_pb2_grpc
+tasklist = tasklist_pb2
+tasklist_grpc = tasklist_pb2_grpc
+visibility = visibility_pb2
+visibility_grpc = visibility_pb2_grpc
+workflow = workflow_pb2
+workflow_grpc = workflow_pb2_grpc
+
+# Only expose clean module names (no _pb2)
+__all__ = [
+    'common',
+    'common_grpc',
+    'decision',
+    'decision_grpc',
+    'domain',
+    'domain_grpc',
+    'error',
+    'error_grpc',
+    'history',
+    'history_grpc',
+    'query',
+    'query_grpc',
+    'service_domain',
+    'service_domain_grpc',
+    'service_meta',
+    'service_meta_grpc',
+    'service_visibility',
+    'service_visibility_grpc',
+    'service_worker',
+    'service_worker_grpc',
+    'service_workflow',
+    'service_workflow_grpc',
+    'tasklist',
+    'tasklist_grpc',
+    'visibility',
+    'visibility_grpc',
+    'workflow',
+    'workflow_grpc',
+]

cadence/api/v1/common_pb2.py

@@ -0,0 +1,90 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# NO CHECKED-IN PROTOBUF GENCODE
+# source: cadence/api/v1/common.proto
+# Protobuf Python Version: 5.29.0
+"""Generated protocol buffer code."""
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import descriptor_pool as _descriptor_pool
+from google.protobuf import runtime_version as _runtime_version
+from google.protobuf import symbol_database as _symbol_database
+from google.protobuf.internal import builder as _builder
+_runtime_version.ValidateProtobufRuntimeVersion(
+    _runtime_version.Domain.PUBLIC,
+    5,
+    29,
+    0,
+    '',
+    'cadence/api/v1/common.proto'
+)
+# @@protoc_insertion_point(imports)
+
+_sym_db = _symbol_database.Default()
+
+
+from google.protobuf import duration_pb2 as google_dot_protobuf_dot_duration__pb2
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x1b\x63\x61\x64\x65nce/api/v1/common.proto\x12\x13uber.cadence.api.v1\x1a\x1egoogle/protobuf/duration.proto\"8\n\x11WorkflowExecution\x12\x13\n\x0bworkflow_id\x18\x01 \x01(\t\x12\x0e\n\x06run_id\x18\x02 \x01(\t\"\x1c\n\x0cWorkflowType\x12\x0c\n\x04name\x18\x01 \x01(\t\"\x1c\n\x0c\x41\x63tivityType\x12\x0c\n\x04name\x18\x01 \x01(\t\"\x17\n\x07Payload\x12\x0c\n\x04\x64\x61ta\x18\x01 \x01(\x0c\"*\n\x07\x46\x61ilure\x12\x0e\n\x06reason\x18\x01 \x01(\t\x12\x0f\n\x07\x64\x65tails\x18\x02 \x01(\x0c\"\x8a\x01\n\x04Memo\x12\x35\n\x06\x66ields\x18\x01 \x03(\x0b\x32%.uber.cadence.api.v1.Memo.FieldsEntry\x1aK\n\x0b\x46ieldsEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12+\n\x05value\x18\x02 \x01(\x0b\x32\x1c.uber.cadence.api.v1.Payload:\x02\x38\x01\"\x8e\x01\n\x06Header\x12\x37\n\x06\x66ields\x18\x01 \x03(\x0b\x32\'.uber.cadence.api.v1.Header.FieldsEntry\x1aK\n\x0b\x46ieldsEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12+\n\x05value\x18\x02 \x01(\x0b\x32\x1c.uber.cadence.api.v1.Payload:\x02\x38\x01\"\xb8\x01\n\x10SearchAttributes\x12P\n\x0eindexed_fields\x18\x01 \x03(\x0b\x32\x38.uber.cadence.api.v1.SearchAttributes.IndexedFieldsEntry\x1aR\n\x12IndexedFieldsEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12+\n\x05value\x18\x02 \x01(\x0b\x32\x1c.uber.cadence.api.v1.Payload:\x02\x38\x01\"R\n\x08\x44\x61taBlob\x12\x38\n\rencoding_type\x18\x01 \x01(\x0e\x32!.uber.cadence.api.v1.EncodingType\x12\x0c\n\x04\x64\x61ta\x18\x02 \x01(\x0c\":\n\x11WorkerVersionInfo\x12\x0c\n\x04impl\x18\x01 \x01(\t\x12\x17\n\x0f\x66\x65\x61ture_version\x18\x02 \x01(\t\";\n\x17SupportedClientVersions\x12\x0e\n\x06go_sdk\x18\x01 \x01(\t\x12\x10\n\x08java_sdk\x18\x02 \x01(\t\"\x8b\x02\n\x0bRetryPolicy\x12\x33\n\x10initial_interval\x18\x01 \x01(\x0b\x32\x19.google.protobuf.Duration\x12\x1b\n\x13\x62\x61\x63koff_coefficient\x18\x02 \x01(\x01\x12\x33\n\x10maximum_interval\x18\x03 \x01(\x0b\x32\x19.google.protobuf.Duration\x12\x18\n\x10maximum_attempts\x18\x04 \x01(\x05\x12#\n\x1bnon_retryable_error_reasons\x18\x05 \x03(\t\x12\x36\n\x13\x65xpiration_interval\x18\x06 \x01(\x0b\x32\x19.google.protobuf.Duration\"`\n\x17IsolationGroupPartition\x12\x0c\n\x04name\x18\x01 \x01(\t\x12\x37\n\x05state\x18\x02 \x01(\x0e\x32(.uber.cadence.api.v1.IsolationGroupState\"e\n\x1bIsolationGroupConfiguration\x12\x46\n\x10isolation_groups\x18\x01 \x03(\x0b\x32,.uber.cadence.api.v1.IsolationGroupPartition\"\x95\x01\n\x1a\x41syncWorkflowConfiguration\x12\x0f\n\x07\x65nabled\x18\x01 \x01(\x08\x12\x1d\n\x15predefined_queue_name\x18\x02 \x01(\t\x12\x12\n\nqueue_type\x18\x03 \x01(\t\x12\x33\n\x0cqueue_config\x18\x04 \x01(\x0b\x32\x1d.uber.cadence.api.v1.DataBlob\"\xc6\x02\n\x1c\x41\x63tiveClusterSelectionPolicy\x12\x45\n\x08strategy\x18\x01 \x01(\x0e\x32\x33.uber.cadence.api.v1.ActiveClusterSelectionStrategy\x12\x63\n#active_cluster_sticky_region_config\x18\x02 \x01(\x0b\x32\x34.uber.cadence.api.v1.ActiveClusterStickyRegionConfigH\x00\x12g\n%active_cluster_external_entity_config\x18\x03 \x01(\x0b\x32\x36.uber.cadence.api.v1.ActiveClusterExternalEntityConfigH\x00\x42\x11\n\x0fstrategy_config\"8\n\x1f\x41\x63tiveClusterStickyRegionConfig\x12\x15\n\rsticky_region\x18\x01 \x01(\t\"^\n!ActiveClusterExternalEntityConfig\x12\x1c\n\x14\x65xternal_entity_type\x18\x01 \x01(\t\x12\x1b\n\x13\x65xternal_entity_key\x18\x02 \x01(\t*w\n\x0c\x45ncodingType\x12\x19\n\x15\x45NCODING_TYPE_INVALID\x10\x00\x12\x1a\n\x16\x45NCODING_TYPE_THRIFTRW\x10\x01\x12\x16\n\x12\x45NCODING_TYPE_JSON\x10\x02\x12\x18\n\x14\x45NCODING_TYPE_PROTO3\x10\x03*~\n\x13IsolationGroupState\x12!\n\x1dISOLATION_GROUP_STATE_INVALID\x10\x00\x12!\n\x1dISOLATION_GROUP_STATE_HEALTHY\x10\x01\x12!\n\x1dISOLATION_GROUP_STATE_DRAINED\x10\x02*\xbb\x01\n\x1e\x41\x63tiveClusterSelectionStrategy\x12-\n)ACTIVE_CLUSTER_SELECTION_STRATEGY_INVALID\x10\x00\x12\x33\n/ACTIVE_CLUSTER_SELECTION_STRATEGY_REGION_STICKY\x10\x01\x12\x35\n1ACTIVE_CLUSTER_SELECTION_STRATEGY_EXTERNAL_ENTITY\x10\x02\x42[\n\x17\x63om.uber.cadence.api.v1B\x0b\x43ommonProtoP\x01Z1github.com/uber/cadence-idl/go/proto/api/v1;apiv1b\x06proto3')
+
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'cadence.api.v1.common_pb2', _globals)
+if not _descriptor._USE_C_DESCRIPTORS:
+  _globals['DESCRIPTOR']._loaded_options = None
+  _globals['DESCRIPTOR']._serialized_options = b'\n\027com.uber.cadence.api.v1B\013CommonProtoP\001Z1github.com/uber/cadence-idl/go/proto/api/v1;apiv1'
+  _globals['_MEMO_FIELDSENTRY']._loaded_options = None
+  _globals['_MEMO_FIELDSENTRY']._serialized_options = b'8\001'
+  _globals['_HEADER_FIELDSENTRY']._loaded_options = None
+  _globals['_HEADER_FIELDSENTRY']._serialized_options = b'8\001'
+  _globals['_SEARCHATTRIBUTES_INDEXEDFIELDSENTRY']._loaded_options = None
+  _globals['_SEARCHATTRIBUTES_INDEXEDFIELDSENTRY']._serialized_options = b'8\001'
+  _globals['_ENCODINGTYPE']._serialized_start=2055
+  _globals['_ENCODINGTYPE']._serialized_end=2174
+  _globals['_ISOLATIONGROUPSTATE']._serialized_start=2176
+  _globals['_ISOLATIONGROUPSTATE']._serialized_end=2302
+  _globals['_ACTIVECLUSTERSELECTIONSTRATEGY']._serialized_start=2305
+  _globals['_ACTIVECLUSTERSELECTIONSTRATEGY']._serialized_end=2492
+  _globals['_WORKFLOWEXECUTION']._serialized_start=84
+  _globals['_WORKFLOWEXECUTION']._serialized_end=140
+  _globals['_WORKFLOWTYPE']._serialized_start=142
+  _globals['_WORKFLOWTYPE']._serialized_end=170
+  _globals['_ACTIVITYTYPE']._serialized_start=172
+  _globals['_ACTIVITYTYPE']._serialized_end=200
+  _globals['_PAYLOAD']._serialized_start=202
+  _globals['_PAYLOAD']._serialized_end=225
+  _globals['_FAILURE']._serialized_start=227
+  _globals['_FAILURE']._serialized_end=269
+  _globals['_MEMO']._serialized_start=272
+  _globals['_MEMO']._serialized_end=410
+  _globals['_MEMO_FIELDSENTRY']._serialized_start=335
+  _globals['_MEMO_FIELDSENTRY']._serialized_end=410
+  _globals['_HEADER']._serialized_start=413
+  _globals['_HEADER']._serialized_end=555
+  _globals['_HEADER_FIELDSENTRY']._serialized_start=335
+  _globals['_HEADER_FIELDSENTRY']._serialized_end=410
+  _globals['_SEARCHATTRIBUTES']._serialized_start=558
+  _globals['_SEARCHATTRIBUTES']._serialized_end=742
+  _globals['_SEARCHATTRIBUTES_INDEXEDFIELDSENTRY']._serialized_start=660
+  _globals['_SEARCHATTRIBUTES_INDEXEDFIELDSENTRY']._serialized_end=742
+  _globals['_DATABLOB']._serialized_start=744
+  _globals['_DATABLOB']._serialized_end=826
+  _globals['_WORKERVERSIONINFO']._serialized_start=828
+  _globals['_WORKERVERSIONINFO']._serialized_end=886
+  _globals['_SUPPORTEDCLIENTVERSIONS']._serialized_start=888
+  _globals['_SUPPORTEDCLIENTVERSIONS']._serialized_end=947
+  _globals['_RETRYPOLICY']._serialized_start=950
+  _globals['_RETRYPOLICY']._serialized_end=1217
+  _globals['_ISOLATIONGROUPPARTITION']._serialized_start=1219
+  _globals['_ISOLATIONGROUPPARTITION']._serialized_end=1315
+  _globals['_ISOLATIONGROUPCONFIGURATION']._serialized_start=1317
+  _globals['_ISOLATIONGROUPCONFIGURATION']._serialized_end=1418
+  _globals['_ASYNCWORKFLOWCONFIGURATION']._serialized_start=1421
+  _globals['_ASYNCWORKFLOWCONFIGURATION']._serialized_end=1570
+  _globals['_ACTIVECLUSTERSELECTIONPOLICY']._serialized_start=1573
+  _globals['_ACTIVECLUSTERSELECTIONPOLICY']._serialized_end=1899
+  _globals['_ACTIVECLUSTERSTICKYREGIONCONFIG']._serialized_start=1901
+  _globals['_ACTIVECLUSTERSTICKYREGIONCONFIG']._serialized_end=1957
+  _globals['_ACTIVECLUSTEREXTERNALENTITYCONFIG']._serialized_start=1959
+  _globals['_ACTIVECLUSTEREXTERNALENTITYCONFIG']._serialized_end=2053
+# @@protoc_insertion_point(module_scope)

cadence/api/v1/common_pb2.pyi

@@ -0,0 +1,200 @@
+from google.protobuf import duration_pb2 as _duration_pb2
+from google.protobuf.internal import containers as _containers
+from google.protobuf.internal import enum_type_wrapper as _enum_type_wrapper
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import message as _message
+from typing import ClassVar as _ClassVar, Iterable as _Iterable, Mapping as _Mapping, Optional as _Optional, Union as _Union
+
+DESCRIPTOR: _descriptor.FileDescriptor
+
+class EncodingType(int, metaclass=_enum_type_wrapper.EnumTypeWrapper):
+    __slots__ = ()
+    ENCODING_TYPE_INVALID: _ClassVar[EncodingType]
+    ENCODING_TYPE_THRIFTRW: _ClassVar[EncodingType]
+    ENCODING_TYPE_JSON: _ClassVar[EncodingType]
+    ENCODING_TYPE_PROTO3: _ClassVar[EncodingType]
+
+class IsolationGroupState(int, metaclass=_enum_type_wrapper.EnumTypeWrapper):
+    __slots__ = ()
+    ISOLATION_GROUP_STATE_INVALID: _ClassVar[IsolationGroupState]
+    ISOLATION_GROUP_STATE_HEALTHY: _ClassVar[IsolationGroupState]
+    ISOLATION_GROUP_STATE_DRAINED: _ClassVar[IsolationGroupState]
+
+class ActiveClusterSelectionStrategy(int, metaclass=_enum_type_wrapper.EnumTypeWrapper):
+    __slots__ = ()
+    ACTIVE_CLUSTER_SELECTION_STRATEGY_INVALID: _ClassVar[ActiveClusterSelectionStrategy]
+    ACTIVE_CLUSTER_SELECTION_STRATEGY_REGION_STICKY: _ClassVar[ActiveClusterSelectionStrategy]
+    ACTIVE_CLUSTER_SELECTION_STRATEGY_EXTERNAL_ENTITY: _ClassVar[ActiveClusterSelectionStrategy]
+ENCODING_TYPE_INVALID: EncodingType
+ENCODING_TYPE_THRIFTRW: EncodingType
+ENCODING_TYPE_JSON: EncodingType
+ENCODING_TYPE_PROTO3: EncodingType
+ISOLATION_GROUP_STATE_INVALID: IsolationGroupState
+ISOLATION_GROUP_STATE_HEALTHY: IsolationGroupState
+ISOLATION_GROUP_STATE_DRAINED: IsolationGroupState
+ACTIVE_CLUSTER_SELECTION_STRATEGY_INVALID: ActiveClusterSelectionStrategy
+ACTIVE_CLUSTER_SELECTION_STRATEGY_REGION_STICKY: ActiveClusterSelectionStrategy
+ACTIVE_CLUSTER_SELECTION_STRATEGY_EXTERNAL_ENTITY: ActiveClusterSelectionStrategy
+
+class WorkflowExecution(_message.Message):
+    __slots__ = ("workflow_id", "run_id")
+    WORKFLOW_ID_FIELD_NUMBER: _ClassVar[int]
+    RUN_ID_FIELD_NUMBER: _ClassVar[int]
+    workflow_id: str
+    run_id: str
+    def __init__(self, workflow_id: _Optional[str] = ..., run_id: _Optional[str] = ...) -> None: ...
+
+class WorkflowType(_message.Message):
+    __slots__ = ("name",)
+    NAME_FIELD_NUMBER: _ClassVar[int]
+    name: str
+    def __init__(self, name: _Optional[str] = ...) -> None: ...
+
+class ActivityType(_message.Message):
+    __slots__ = ("name",)
+    NAME_FIELD_NUMBER: _ClassVar[int]
+    name: str
+    def __init__(self, name: _Optional[str] = ...) -> None: ...
+
+class Payload(_message.Message):
+    __slots__ = ("data",)
+    DATA_FIELD_NUMBER: _ClassVar[int]
+    data: bytes
+    def __init__(self, data: _Optional[bytes] = ...) -> None: ...
+
+class Failure(_message.Message):
+    __slots__ = ("reason", "details")
+    REASON_FIELD_NUMBER: _ClassVar[int]
+    DETAILS_FIELD_NUMBER: _ClassVar[int]
+    reason: str
+    details: bytes
+    def __init__(self, reason: _Optional[str] = ..., details: _Optional[bytes] = ...) -> None: ...
+
+class Memo(_message.Message):
+    __slots__ = ("fields",)
+    class FieldsEntry(_message.Message):
+        __slots__ = ("key", "value")
+        KEY_FIELD_NUMBER: _ClassVar[int]
+        VALUE_FIELD_NUMBER: _ClassVar[int]
+        key: str
+        value: Payload
+        def __init__(self, key: _Optional[str] = ..., value: _Optional[_Union[Payload, _Mapping]] = ...) -> None: ...
+    FIELDS_FIELD_NUMBER: _ClassVar[int]
+    fields: _containers.MessageMap[str, Payload]
+    def __init__(self, fields: _Optional[_Mapping[str, Payload]] = ...) -> None: ...
+
+class Header(_message.Message):
+    __slots__ = ("fields",)
+    class FieldsEntry(_message.Message):
+        __slots__ = ("key", "value")
+        KEY_FIELD_NUMBER: _ClassVar[int]
+        VALUE_FIELD_NUMBER: _ClassVar[int]
+        key: str
+        value: Payload
+        def __init__(self, key: _Optional[str] = ..., value: _Optional[_Union[Payload, _Mapping]] = ...) -> None: ...
+    FIELDS_FIELD_NUMBER: _ClassVar[int]
+    fields: _containers.MessageMap[str, Payload]
+    def __init__(self, fields: _Optional[_Mapping[str, Payload]] = ...) -> None: ...
+
+class SearchAttributes(_message.Message):
+    __slots__ = ("indexed_fields",)
+    class IndexedFieldsEntry(_message.Message):
+        __slots__ = ("key", "value")
+        KEY_FIELD_NUMBER: _ClassVar[int]
+        VALUE_FIELD_NUMBER: _ClassVar[int]
+        key: str
+        value: Payload
+        def __init__(self, key: _Optional[str] = ..., value: _Optional[_Union[Payload, _Mapping]] = ...) -> None: ...
+    INDEXED_FIELDS_FIELD_NUMBER: _ClassVar[int]
+    indexed_fields: _containers.MessageMap[str, Payload]
+    def __init__(self, indexed_fields: _Optional[_Mapping[str, Payload]] = ...) -> None: ...
+
+class DataBlob(_message.Message):
+    __slots__ = ("encoding_type", "data")
+    ENCODING_TYPE_FIELD_NUMBER: _ClassVar[int]
+    DATA_FIELD_NUMBER: _ClassVar[int]
+    encoding_type: EncodingType
+    data: bytes
+    def __init__(self, encoding_type: _Optional[_Union[EncodingType, str]] = ..., data: _Optional[bytes] = ...) -> None: ...
+
+class WorkerVersionInfo(_message.Message):
+    __slots__ = ("impl", "feature_version")
+    IMPL_FIELD_NUMBER: _ClassVar[int]
+    FEATURE_VERSION_FIELD_NUMBER: _ClassVar[int]
+    impl: str
+    feature_version: str
+    def __init__(self, impl: _Optional[str] = ..., feature_version: _Optional[str] = ...) -> None: ...
+
+class SupportedClientVersions(_message.Message):
+    __slots__ = ("go_sdk", "java_sdk")
+    GO_SDK_FIELD_NUMBER: _ClassVar[int]
+    JAVA_SDK_FIELD_NUMBER: _ClassVar[int]
+    go_sdk: str
+    java_sdk: str
+    def __init__(self, go_sdk: _Optional[str] = ..., java_sdk: _Optional[str] = ...) -> None: ...
+
+class RetryPolicy(_message.Message):
+    __slots__ = ("initial_interval", "backoff_coefficient", "maximum_interval", "maximum_attempts", "non_retryable_error_reasons", "expiration_interval")
+    INITIAL_INTERVAL_FIELD_NUMBER: _ClassVar[int]
+    BACKOFF_COEFFICIENT_FIELD_NUMBER: _ClassVar[int]
+    MAXIMUM_INTERVAL_FIELD_NUMBER: _ClassVar[int]
+    MAXIMUM_ATTEMPTS_FIELD_NUMBER: _ClassVar[int]
+    NON_RETRYABLE_ERROR_REASONS_FIELD_NUMBER: _ClassVar[int]
+    EXPIRATION_INTERVAL_FIELD_NUMBER: _ClassVar[int]
+    initial_interval: _duration_pb2.Duration
+    backoff_coefficient: float
+    maximum_interval: _duration_pb2.Duration
+    maximum_attempts: int
+    non_retryable_error_reasons: _containers.RepeatedScalarFieldContainer[str]
+    expiration_interval: _duration_pb2.Duration
+    def __init__(self, initial_interval: _Optional[_Union[_duration_pb2.Duration, _Mapping]] = ..., backoff_coefficient: _Optional[float] = ..., maximum_interval: _Optional[_Union[_duration_pb2.Duration, _Mapping]] = ..., maximum_attempts: _Optional[int] = ..., non_retryable_error_reasons: _Optional[_Iterable[str]] = ..., expiration_interval: _Optional[_Union[_duration_pb2.Duration, _Mapping]] = ...) -> None: ...
+
+class IsolationGroupPartition(_message.Message):
+    __slots__ = ("name", "state")
+    NAME_FIELD_NUMBER: _ClassVar[int]
+    STATE_FIELD_NUMBER: _ClassVar[int]
+    name: str
+    state: IsolationGroupState
+    def __init__(self, name: _Optional[str] = ..., state: _Optional[_Union[IsolationGroupState, str]] = ...) -> None: ...
+
+class IsolationGroupConfiguration(_message.Message):
+    __slots__ = ("isolation_groups",)
+    ISOLATION_GROUPS_FIELD_NUMBER: _ClassVar[int]
+    isolation_groups: _containers.RepeatedCompositeFieldContainer[IsolationGroupPartition]
+    def __init__(self, isolation_groups: _Optional[_Iterable[_Union[IsolationGroupPartition, _Mapping]]] = ...) -> None: ...
+
+class AsyncWorkflowConfiguration(_message.Message):
+    __slots__ = ("enabled", "predefined_queue_name", "queue_type", "queue_config")
+    ENABLED_FIELD_NUMBER: _ClassVar[int]
+    PREDEFINED_QUEUE_NAME_FIELD_NUMBER: _ClassVar[int]
+    QUEUE_TYPE_FIELD_NUMBER: _ClassVar[int]
+    QUEUE_CONFIG_FIELD_NUMBER: _ClassVar[int]
+    enabled: bool
+    predefined_queue_name: str
+    queue_type: str
+    queue_config: DataBlob
+    def __init__(self, enabled: bool = ..., predefined_queue_name: _Optional[str] = ..., queue_type: _Optional[str] = ..., queue_config: _Optional[_Union[DataBlob, _Mapping]] = ...) -> None: ...
+
+class ActiveClusterSelectionPolicy(_message.Message):
+    __slots__ = ("strategy", "active_cluster_sticky_region_config", "active_cluster_external_entity_config")
+    STRATEGY_FIELD_NUMBER: _ClassVar[int]
+    ACTIVE_CLUSTER_STICKY_REGION_CONFIG_FIELD_NUMBER: _ClassVar[int]
+    ACTIVE_CLUSTER_EXTERNAL_ENTITY_CONFIG_FIELD_NUMBER: _ClassVar[int]
+    strategy: ActiveClusterSelectionStrategy
+    active_cluster_sticky_region_config: ActiveClusterStickyRegionConfig
+    active_cluster_external_entity_config: ActiveClusterExternalEntityConfig
+    def __init__(self, strategy: _Optional[_Union[ActiveClusterSelectionStrategy, str]] = ..., active_cluster_sticky_region_config: _Optional[_Union[ActiveClusterStickyRegionConfig, _Mapping]] = ..., active_cluster_external_entity_config: _Optional[_Union[ActiveClusterExternalEntityConfig, _Mapping]] = ...) -> None: ...
+
+class ActiveClusterStickyRegionConfig(_message.Message):
+    __slots__ = ("sticky_region",)
+    STICKY_REGION_FIELD_NUMBER: _ClassVar[int]
+    sticky_region: str
+    def __init__(self, sticky_region: _Optional[str] = ...) -> None: ...
+
+class ActiveClusterExternalEntityConfig(_message.Message):
+    __slots__ = ("external_entity_type", "external_entity_key")
+    EXTERNAL_ENTITY_TYPE_FIELD_NUMBER: _ClassVar[int]
+    EXTERNAL_ENTITY_KEY_FIELD_NUMBER: _ClassVar[int]
+    external_entity_type: str
+    external_entity_key: str
+    def __init__(self, external_entity_type: _Optional[str] = ..., external_entity_key: _Optional[str] = ...) -> None: ...

cadence/api/v1/common_pb2_grpc.py

@@ -0,0 +1,24 @@
+# Generated by the gRPC Python protocol compiler plugin. DO NOT EDIT!
+"""Client and server classes corresponding to protobuf-defined services."""
+import grpc
+import warnings
+
+
+GRPC_GENERATED_VERSION = '1.71.2'
+GRPC_VERSION = grpc.__version__
+_version_not_supported = False
+
+try:
+    from grpc._utilities import first_version_is_lower
+    _version_not_supported = first_version_is_lower(GRPC_VERSION, GRPC_GENERATED_VERSION)
+except ImportError:
+    _version_not_supported = True
+
+if _version_not_supported:
+    raise RuntimeError(
+        f'The grpc package installed is at version {GRPC_VERSION},'
+        + f' but the generated code in cadence/api/v1/common_pb2_grpc.py depends on'
+        + f' grpcio>={GRPC_GENERATED_VERSION}.'
+        + f' Please upgrade your grpc module to grpcio>={GRPC_GENERATED_VERSION}'
+        + f' or downgrade your generated code using grpcio-tools<={GRPC_VERSION}.'
+    )