durabletask 0.2.1__py3-none-any.whl → 0.4.0__py3-none-any.whl

durabletask/internal/orchestrator_service_pb2.pyi

@@ -2,6 +2,7 @@ from google.protobuf import timestamp_pb2 as _timestamp_pb2
 from google.protobuf import duration_pb2 as _duration_pb2
 from google.protobuf import wrappers_pb2 as _wrappers_pb2
 from google.protobuf import empty_pb2 as _empty_pb2
+from google.protobuf import struct_pb2 as _struct_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
@@ -21,11 +22,10 @@ class OrchestrationStatus(int, metaclass=_enum_type_wrapper.EnumTypeWrapper):
     ORCHESTRATION_STATUS_PENDING: _ClassVar[OrchestrationStatus]
     ORCHESTRATION_STATUS_SUSPENDED: _ClassVar[OrchestrationStatus]
 
-class CreateOrchestrationAction(int, metaclass=_enum_type_wrapper.EnumTypeWrapper):
+class WorkerCapability(int, metaclass=_enum_type_wrapper.EnumTypeWrapper):
     __slots__ = ()
-    ERROR: _ClassVar[CreateOrchestrationAction]
-    IGNORE: _ClassVar[CreateOrchestrationAction]
-    TERMINATE: _ClassVar[CreateOrchestrationAction]
+    WORKER_CAPABILITY_UNSPECIFIED: _ClassVar[WorkerCapability]
+    WORKER_CAPABILITY_HISTORY_STREAMING: _ClassVar[WorkerCapability]
 ORCHESTRATION_STATUS_RUNNING: OrchestrationStatus
 ORCHESTRATION_STATUS_COMPLETED: OrchestrationStatus
 ORCHESTRATION_STATUS_CONTINUED_AS_NEW: OrchestrationStatus
@@ -34,9 +34,8 @@ ORCHESTRATION_STATUS_CANCELED: OrchestrationStatus
 ORCHESTRATION_STATUS_TERMINATED: OrchestrationStatus
 ORCHESTRATION_STATUS_PENDING: OrchestrationStatus
 ORCHESTRATION_STATUS_SUSPENDED: OrchestrationStatus
-ERROR: CreateOrchestrationAction
-IGNORE: CreateOrchestrationAction
-TERMINATE: CreateOrchestrationAction
+WORKER_CAPABILITY_UNSPECIFIED: WorkerCapability
+WORKER_CAPABILITY_HISTORY_STREAMING: WorkerCapability
 
 class OrchestrationInstance(_message.Message):
     __slots__ = ("instanceId", "executionId")
@@ -113,7 +112,14 @@ class TraceContext(_message.Message):
     def __init__(self, traceParent: _Optional[str] = ..., spanID: _Optional[str] = ..., traceState: _Optional[_Union[_wrappers_pb2.StringValue, _Mapping]] = ...) -> None: ...
 
 class ExecutionStartedEvent(_message.Message):
-    __slots__ = ("name", "version", "input", "orchestrationInstance", "parentInstance", "scheduledStartTimestamp", "parentTraceContext", "orchestrationSpanID")
+    __slots__ = ("name", "version", "input", "orchestrationInstance", "parentInstance", "scheduledStartTimestamp", "parentTraceContext", "orchestrationSpanID", "tags")
+    class TagsEntry(_message.Message):
+        __slots__ = ("key", "value")
+        KEY_FIELD_NUMBER: _ClassVar[int]
+        VALUE_FIELD_NUMBER: _ClassVar[int]
+        key: str
+        value: str
+        def __init__(self, key: _Optional[str] = ..., value: _Optional[str] = ...) -> None: ...
     NAME_FIELD_NUMBER: _ClassVar[int]
     VERSION_FIELD_NUMBER: _ClassVar[int]
     INPUT_FIELD_NUMBER: _ClassVar[int]
@@ -122,6 +128,7 @@ class ExecutionStartedEvent(_message.Message):
     SCHEDULEDSTARTTIMESTAMP_FIELD_NUMBER: _ClassVar[int]
     PARENTTRACECONTEXT_FIELD_NUMBER: _ClassVar[int]
     ORCHESTRATIONSPANID_FIELD_NUMBER: _ClassVar[int]
+    TAGS_FIELD_NUMBER: _ClassVar[int]
     name: str
     version: _wrappers_pb2.StringValue
     input: _wrappers_pb2.StringValue
@@ -130,7 +137,8 @@ class ExecutionStartedEvent(_message.Message):
     scheduledStartTimestamp: _timestamp_pb2.Timestamp
     parentTraceContext: TraceContext
     orchestrationSpanID: _wrappers_pb2.StringValue
-    def __init__(self, name: _Optional[str] = ..., version: _Optional[_Union[_wrappers_pb2.StringValue, _Mapping]] = ..., input: _Optional[_Union[_wrappers_pb2.StringValue, _Mapping]] = ..., orchestrationInstance: _Optional[_Union[OrchestrationInstance, _Mapping]] = ..., parentInstance: _Optional[_Union[ParentInstanceInfo, _Mapping]] = ..., scheduledStartTimestamp: _Optional[_Union[_timestamp_pb2.Timestamp, _Mapping]] = ..., parentTraceContext: _Optional[_Union[TraceContext, _Mapping]] = ..., orchestrationSpanID: _Optional[_Union[_wrappers_pb2.StringValue, _Mapping]] = ...) -> None: ...
+    tags: _containers.ScalarMap[str, str]
+    def __init__(self, name: _Optional[str] = ..., version: _Optional[_Union[_wrappers_pb2.StringValue, _Mapping]] = ..., input: _Optional[_Union[_wrappers_pb2.StringValue, _Mapping]] = ..., orchestrationInstance: _Optional[_Union[OrchestrationInstance, _Mapping]] = ..., parentInstance: _Optional[_Union[ParentInstanceInfo, _Mapping]] = ..., scheduledStartTimestamp: _Optional[_Union[_timestamp_pb2.Timestamp, _Mapping]] = ..., parentTraceContext: _Optional[_Union[TraceContext, _Mapping]] = ..., orchestrationSpanID: _Optional[_Union[_wrappers_pb2.StringValue, _Mapping]] = ..., tags: _Optional[_Mapping[str, str]] = ...) -> None: ...
 
 class ExecutionCompletedEvent(_message.Message):
     __slots__ = ("orchestrationStatus", "result", "failureDetails")
@@ -151,16 +159,25 @@ class ExecutionTerminatedEvent(_message.Message):
     def __init__(self, input: _Optional[_Union[_wrappers_pb2.StringValue, _Mapping]] = ..., recurse: bool = ...) -> None: ...
 
 class TaskScheduledEvent(_message.Message):
-    __slots__ = ("name", "version", "input", "parentTraceContext")
+    __slots__ = ("name", "version", "input", "parentTraceContext", "tags")
+    class TagsEntry(_message.Message):
+        __slots__ = ("key", "value")
+        KEY_FIELD_NUMBER: _ClassVar[int]
+        VALUE_FIELD_NUMBER: _ClassVar[int]
+        key: str
+        value: str
+        def __init__(self, key: _Optional[str] = ..., value: _Optional[str] = ...) -> None: ...
     NAME_FIELD_NUMBER: _ClassVar[int]
     VERSION_FIELD_NUMBER: _ClassVar[int]
     INPUT_FIELD_NUMBER: _ClassVar[int]
     PARENTTRACECONTEXT_FIELD_NUMBER: _ClassVar[int]
+    TAGS_FIELD_NUMBER: _ClassVar[int]
     name: str
     version: _wrappers_pb2.StringValue
     input: _wrappers_pb2.StringValue
     parentTraceContext: TraceContext
-    def __init__(self, name: _Optional[str] = ..., version: _Optional[_Union[_wrappers_pb2.StringValue, _Mapping]] = ..., input: _Optional[_Union[_wrappers_pb2.StringValue, _Mapping]] = ..., parentTraceContext: _Optional[_Union[TraceContext, _Mapping]] = ...) -> None: ...
+    tags: _containers.ScalarMap[str, str]
+    def __init__(self, name: _Optional[str] = ..., version: _Optional[_Union[_wrappers_pb2.StringValue, _Mapping]] = ..., input: _Optional[_Union[_wrappers_pb2.StringValue, _Mapping]] = ..., parentTraceContext: _Optional[_Union[TraceContext, _Mapping]] = ..., tags: _Optional[_Mapping[str, str]] = ...) -> None: ...
 
 class TaskCompletedEvent(_message.Message):
     __slots__ = ("taskScheduledId", "result")
@@ -278,8 +295,84 @@ class ExecutionResumedEvent(_message.Message):
     input: _wrappers_pb2.StringValue
     def __init__(self, input: _Optional[_Union[_wrappers_pb2.StringValue, _Mapping]] = ...) -> None: ...
 
+class EntityOperationSignaledEvent(_message.Message):
+    __slots__ = ("requestId", "operation", "scheduledTime", "input", "targetInstanceId")
+    REQUESTID_FIELD_NUMBER: _ClassVar[int]
+    OPERATION_FIELD_NUMBER: _ClassVar[int]
+    SCHEDULEDTIME_FIELD_NUMBER: _ClassVar[int]
+    INPUT_FIELD_NUMBER: _ClassVar[int]
+    TARGETINSTANCEID_FIELD_NUMBER: _ClassVar[int]
+    requestId: str
+    operation: str
+    scheduledTime: _timestamp_pb2.Timestamp
+    input: _wrappers_pb2.StringValue
+    targetInstanceId: _wrappers_pb2.StringValue
+    def __init__(self, requestId: _Optional[str] = ..., operation: _Optional[str] = ..., scheduledTime: _Optional[_Union[_timestamp_pb2.Timestamp, _Mapping]] = ..., input: _Optional[_Union[_wrappers_pb2.StringValue, _Mapping]] = ..., targetInstanceId: _Optional[_Union[_wrappers_pb2.StringValue, _Mapping]] = ...) -> None: ...
+
+class EntityOperationCalledEvent(_message.Message):
+    __slots__ = ("requestId", "operation", "scheduledTime", "input", "parentInstanceId", "parentExecutionId", "targetInstanceId")
+    REQUESTID_FIELD_NUMBER: _ClassVar[int]
+    OPERATION_FIELD_NUMBER: _ClassVar[int]
+    SCHEDULEDTIME_FIELD_NUMBER: _ClassVar[int]
+    INPUT_FIELD_NUMBER: _ClassVar[int]
+    PARENTINSTANCEID_FIELD_NUMBER: _ClassVar[int]
+    PARENTEXECUTIONID_FIELD_NUMBER: _ClassVar[int]
+    TARGETINSTANCEID_FIELD_NUMBER: _ClassVar[int]
+    requestId: str
+    operation: str
+    scheduledTime: _timestamp_pb2.Timestamp
+    input: _wrappers_pb2.StringValue
+    parentInstanceId: _wrappers_pb2.StringValue
+    parentExecutionId: _wrappers_pb2.StringValue
+    targetInstanceId: _wrappers_pb2.StringValue
+    def __init__(self, requestId: _Optional[str] = ..., operation: _Optional[str] = ..., scheduledTime: _Optional[_Union[_timestamp_pb2.Timestamp, _Mapping]] = ..., input: _Optional[_Union[_wrappers_pb2.StringValue, _Mapping]] = ..., parentInstanceId: _Optional[_Union[_wrappers_pb2.StringValue, _Mapping]] = ..., parentExecutionId: _Optional[_Union[_wrappers_pb2.StringValue, _Mapping]] = ..., targetInstanceId: _Optional[_Union[_wrappers_pb2.StringValue, _Mapping]] = ...) -> None: ...
+
+class EntityLockRequestedEvent(_message.Message):
+    __slots__ = ("criticalSectionId", "lockSet", "position", "parentInstanceId")
+    CRITICALSECTIONID_FIELD_NUMBER: _ClassVar[int]
+    LOCKSET_FIELD_NUMBER: _ClassVar[int]
+    POSITION_FIELD_NUMBER: _ClassVar[int]
+    PARENTINSTANCEID_FIELD_NUMBER: _ClassVar[int]
+    criticalSectionId: str
+    lockSet: _containers.RepeatedScalarFieldContainer[str]
+    position: int
+    parentInstanceId: _wrappers_pb2.StringValue
+    def __init__(self, criticalSectionId: _Optional[str] = ..., lockSet: _Optional[_Iterable[str]] = ..., position: _Optional[int] = ..., parentInstanceId: _Optional[_Union[_wrappers_pb2.StringValue, _Mapping]] = ...) -> None: ...
+
+class EntityOperationCompletedEvent(_message.Message):
+    __slots__ = ("requestId", "output")
+    REQUESTID_FIELD_NUMBER: _ClassVar[int]
+    OUTPUT_FIELD_NUMBER: _ClassVar[int]
+    requestId: str
+    output: _wrappers_pb2.StringValue
+    def __init__(self, requestId: _Optional[str] = ..., output: _Optional[_Union[_wrappers_pb2.StringValue, _Mapping]] = ...) -> None: ...
+
+class EntityOperationFailedEvent(_message.Message):
+    __slots__ = ("requestId", "failureDetails")
+    REQUESTID_FIELD_NUMBER: _ClassVar[int]
+    FAILUREDETAILS_FIELD_NUMBER: _ClassVar[int]
+    requestId: str
+    failureDetails: TaskFailureDetails
+    def __init__(self, requestId: _Optional[str] = ..., failureDetails: _Optional[_Union[TaskFailureDetails, _Mapping]] = ...) -> None: ...
+
+class EntityUnlockSentEvent(_message.Message):
+    __slots__ = ("criticalSectionId", "parentInstanceId", "targetInstanceId")
+    CRITICALSECTIONID_FIELD_NUMBER: _ClassVar[int]
+    PARENTINSTANCEID_FIELD_NUMBER: _ClassVar[int]
+    TARGETINSTANCEID_FIELD_NUMBER: _ClassVar[int]
+    criticalSectionId: str
+    parentInstanceId: _wrappers_pb2.StringValue
+    targetInstanceId: _wrappers_pb2.StringValue
+    def __init__(self, criticalSectionId: _Optional[str] = ..., parentInstanceId: _Optional[_Union[_wrappers_pb2.StringValue, _Mapping]] = ..., targetInstanceId: _Optional[_Union[_wrappers_pb2.StringValue, _Mapping]] = ...) -> None: ...
+
+class EntityLockGrantedEvent(_message.Message):
+    __slots__ = ("criticalSectionId",)
+    CRITICALSECTIONID_FIELD_NUMBER: _ClassVar[int]
+    criticalSectionId: str
+    def __init__(self, criticalSectionId: _Optional[str] = ...) -> None: ...
+
 class HistoryEvent(_message.Message):
-    __slots__ = ("eventId", "timestamp", "executionStarted", "executionCompleted", "executionTerminated", "taskScheduled", "taskCompleted", "taskFailed", "subOrchestrationInstanceCreated", "subOrchestrationInstanceCompleted", "subOrchestrationInstanceFailed", "timerCreated", "timerFired", "orchestratorStarted", "orchestratorCompleted", "eventSent", "eventRaised", "genericEvent", "historyState", "continueAsNew", "executionSuspended", "executionResumed")
+    __slots__ = ("eventId", "timestamp", "executionStarted", "executionCompleted", "executionTerminated", "taskScheduled", "taskCompleted", "taskFailed", "subOrchestrationInstanceCreated", "subOrchestrationInstanceCompleted", "subOrchestrationInstanceFailed", "timerCreated", "timerFired", "orchestratorStarted", "orchestratorCompleted", "eventSent", "eventRaised", "genericEvent", "historyState", "continueAsNew", "executionSuspended", "executionResumed", "entityOperationSignaled", "entityOperationCalled", "entityOperationCompleted", "entityOperationFailed", "entityLockRequested", "entityLockGranted", "entityUnlockSent")
     EVENTID_FIELD_NUMBER: _ClassVar[int]
     TIMESTAMP_FIELD_NUMBER: _ClassVar[int]
     EXECUTIONSTARTED_FIELD_NUMBER: _ClassVar[int]
@@ -302,6 +395,13 @@ class HistoryEvent(_message.Message):
     CONTINUEASNEW_FIELD_NUMBER: _ClassVar[int]
     EXECUTIONSUSPENDED_FIELD_NUMBER: _ClassVar[int]
     EXECUTIONRESUMED_FIELD_NUMBER: _ClassVar[int]
+    ENTITYOPERATIONSIGNALED_FIELD_NUMBER: _ClassVar[int]
+    ENTITYOPERATIONCALLED_FIELD_NUMBER: _ClassVar[int]
+    ENTITYOPERATIONCOMPLETED_FIELD_NUMBER: _ClassVar[int]
+    ENTITYOPERATIONFAILED_FIELD_NUMBER: _ClassVar[int]
+    ENTITYLOCKREQUESTED_FIELD_NUMBER: _ClassVar[int]
+    ENTITYLOCKGRANTED_FIELD_NUMBER: _ClassVar[int]
+    ENTITYUNLOCKSENT_FIELD_NUMBER: _ClassVar[int]
     eventId: int
     timestamp: _timestamp_pb2.Timestamp
     executionStarted: ExecutionStartedEvent
@@ -324,17 +424,33 @@ class HistoryEvent(_message.Message):
     continueAsNew: ContinueAsNewEvent
     executionSuspended: ExecutionSuspendedEvent
     executionResumed: ExecutionResumedEvent
-    def __init__(self, eventId: _Optional[int] = ..., timestamp: _Optional[_Union[_timestamp_pb2.Timestamp, _Mapping]] = ..., executionStarted: _Optional[_Union[ExecutionStartedEvent, _Mapping]] = ..., executionCompleted: _Optional[_Union[ExecutionCompletedEvent, _Mapping]] = ..., executionTerminated: _Optional[_Union[ExecutionTerminatedEvent, _Mapping]] = ..., taskScheduled: _Optional[_Union[TaskScheduledEvent, _Mapping]] = ..., taskCompleted: _Optional[_Union[TaskCompletedEvent, _Mapping]] = ..., taskFailed: _Optional[_Union[TaskFailedEvent, _Mapping]] = ..., subOrchestrationInstanceCreated: _Optional[_Union[SubOrchestrationInstanceCreatedEvent, _Mapping]] = ..., subOrchestrationInstanceCompleted: _Optional[_Union[SubOrchestrationInstanceCompletedEvent, _Mapping]] = ..., subOrchestrationInstanceFailed: _Optional[_Union[SubOrchestrationInstanceFailedEvent, _Mapping]] = ..., timerCreated: _Optional[_Union[TimerCreatedEvent, _Mapping]] = ..., timerFired: _Optional[_Union[TimerFiredEvent, _Mapping]] = ..., orchestratorStarted: _Optional[_Union[OrchestratorStartedEvent, _Mapping]] = ..., orchestratorCompleted: _Optional[_Union[OrchestratorCompletedEvent, _Mapping]] = ..., eventSent: _Optional[_Union[EventSentEvent, _Mapping]] = ..., eventRaised: _Optional[_Union[EventRaisedEvent, _Mapping]] = ..., genericEvent: _Optional[_Union[GenericEvent, _Mapping]] = ..., historyState: _Optional[_Union[HistoryStateEvent, _Mapping]] = ..., continueAsNew: _Optional[_Union[ContinueAsNewEvent, _Mapping]] = ..., executionSuspended: _Optional[_Union[ExecutionSuspendedEvent, _Mapping]] = ..., executionResumed: _Optional[_Union[ExecutionResumedEvent, _Mapping]] = ...) -> None: ...
+    entityOperationSignaled: EntityOperationSignaledEvent
+    entityOperationCalled: EntityOperationCalledEvent
+    entityOperationCompleted: EntityOperationCompletedEvent
+    entityOperationFailed: EntityOperationFailedEvent
+    entityLockRequested: EntityLockRequestedEvent
+    entityLockGranted: EntityLockGrantedEvent
+    entityUnlockSent: EntityUnlockSentEvent
+    def __init__(self, eventId: _Optional[int] = ..., timestamp: _Optional[_Union[_timestamp_pb2.Timestamp, _Mapping]] = ..., executionStarted: _Optional[_Union[ExecutionStartedEvent, _Mapping]] = ..., executionCompleted: _Optional[_Union[ExecutionCompletedEvent, _Mapping]] = ..., executionTerminated: _Optional[_Union[ExecutionTerminatedEvent, _Mapping]] = ..., taskScheduled: _Optional[_Union[TaskScheduledEvent, _Mapping]] = ..., taskCompleted: _Optional[_Union[TaskCompletedEvent, _Mapping]] = ..., taskFailed: _Optional[_Union[TaskFailedEvent, _Mapping]] = ..., subOrchestrationInstanceCreated: _Optional[_Union[SubOrchestrationInstanceCreatedEvent, _Mapping]] = ..., subOrchestrationInstanceCompleted: _Optional[_Union[SubOrchestrationInstanceCompletedEvent, _Mapping]] = ..., subOrchestrationInstanceFailed: _Optional[_Union[SubOrchestrationInstanceFailedEvent, _Mapping]] = ..., timerCreated: _Optional[_Union[TimerCreatedEvent, _Mapping]] = ..., timerFired: _Optional[_Union[TimerFiredEvent, _Mapping]] = ..., orchestratorStarted: _Optional[_Union[OrchestratorStartedEvent, _Mapping]] = ..., orchestratorCompleted: _Optional[_Union[OrchestratorCompletedEvent, _Mapping]] = ..., eventSent: _Optional[_Union[EventSentEvent, _Mapping]] = ..., eventRaised: _Optional[_Union[EventRaisedEvent, _Mapping]] = ..., genericEvent: _Optional[_Union[GenericEvent, _Mapping]] = ..., historyState: _Optional[_Union[HistoryStateEvent, _Mapping]] = ..., continueAsNew: _Optional[_Union[ContinueAsNewEvent, _Mapping]] = ..., executionSuspended: _Optional[_Union[ExecutionSuspendedEvent, _Mapping]] = ..., executionResumed: _Optional[_Union[ExecutionResumedEvent, _Mapping]] = ..., entityOperationSignaled: _Optional[_Union[EntityOperationSignaledEvent, _Mapping]] = ..., entityOperationCalled: _Optional[_Union[EntityOperationCalledEvent, _Mapping]] = ..., entityOperationCompleted: _Optional[_Union[EntityOperationCompletedEvent, _Mapping]] = ..., entityOperationFailed: _Optional[_Union[EntityOperationFailedEvent, _Mapping]] = ..., entityLockRequested: _Optional[_Union[EntityLockRequestedEvent, _Mapping]] = ..., entityLockGranted: _Optional[_Union[EntityLockGrantedEvent, _Mapping]] = ..., entityUnlockSent: _Optional[_Union[EntityUnlockSentEvent, _Mapping]] = ...) -> None: ...
 
 class ScheduleTaskAction(_message.Message):
-    __slots__ = ("name", "version", "input")
+    __slots__ = ("name", "version", "input", "tags")
+    class TagsEntry(_message.Message):
+        __slots__ = ("key", "value")
+        KEY_FIELD_NUMBER: _ClassVar[int]
+        VALUE_FIELD_NUMBER: _ClassVar[int]
+        key: str
+        value: str
+        def __init__(self, key: _Optional[str] = ..., value: _Optional[str] = ...) -> None: ...
     NAME_FIELD_NUMBER: _ClassVar[int]
     VERSION_FIELD_NUMBER: _ClassVar[int]
     INPUT_FIELD_NUMBER: _ClassVar[int]
+    TAGS_FIELD_NUMBER: _ClassVar[int]
     name: str
     version: _wrappers_pb2.StringValue
     input: _wrappers_pb2.StringValue
-    def __init__(self, name: _Optional[str] = ..., version: _Optional[_Union[_wrappers_pb2.StringValue, _Mapping]] = ..., input: _Optional[_Union[_wrappers_pb2.StringValue, _Mapping]] = ...) -> None: ...
+    tags: _containers.ScalarMap[str, str]
+    def __init__(self, name: _Optional[str] = ..., version: _Optional[_Union[_wrappers_pb2.StringValue, _Mapping]] = ..., input: _Optional[_Union[_wrappers_pb2.StringValue, _Mapping]] = ..., tags: _Optional[_Mapping[str, str]] = ...) -> None: ...
 
 class CreateSubOrchestrationAction(_message.Message):
     __slots__ = ("instanceId", "name", "version", "input")
@@ -390,8 +506,20 @@ class TerminateOrchestrationAction(_message.Message):
     recurse: bool
     def __init__(self, instanceId: _Optional[str] = ..., reason: _Optional[_Union[_wrappers_pb2.StringValue, _Mapping]] = ..., recurse: bool = ...) -> None: ...
 
+class SendEntityMessageAction(_message.Message):
+    __slots__ = ("entityOperationSignaled", "entityOperationCalled", "entityLockRequested", "entityUnlockSent")
+    ENTITYOPERATIONSIGNALED_FIELD_NUMBER: _ClassVar[int]
+    ENTITYOPERATIONCALLED_FIELD_NUMBER: _ClassVar[int]
+    ENTITYLOCKREQUESTED_FIELD_NUMBER: _ClassVar[int]
+    ENTITYUNLOCKSENT_FIELD_NUMBER: _ClassVar[int]
+    entityOperationSignaled: EntityOperationSignaledEvent
+    entityOperationCalled: EntityOperationCalledEvent
+    entityLockRequested: EntityLockRequestedEvent
+    entityUnlockSent: EntityUnlockSentEvent
+    def __init__(self, entityOperationSignaled: _Optional[_Union[EntityOperationSignaledEvent, _Mapping]] = ..., entityOperationCalled: _Optional[_Union[EntityOperationCalledEvent, _Mapping]] = ..., entityLockRequested: _Optional[_Union[EntityLockRequestedEvent, _Mapping]] = ..., entityUnlockSent: _Optional[_Union[EntityUnlockSentEvent, _Mapping]] = ...) -> None: ...
+
 class OrchestratorAction(_message.Message):
-    __slots__ = ("id", "scheduleTask", "createSubOrchestration", "createTimer", "sendEvent", "completeOrchestration", "terminateOrchestration")
+    __slots__ = ("id", "scheduleTask", "createSubOrchestration", "createTimer", "sendEvent", "completeOrchestration", "terminateOrchestration", "sendEntityMessage")
     ID_FIELD_NUMBER: _ClassVar[int]
     SCHEDULETASK_FIELD_NUMBER: _ClassVar[int]
     CREATESUBORCHESTRATION_FIELD_NUMBER: _ClassVar[int]
@@ -399,6 +527,7 @@ class OrchestratorAction(_message.Message):
     SENDEVENT_FIELD_NUMBER: _ClassVar[int]
     COMPLETEORCHESTRATION_FIELD_NUMBER: _ClassVar[int]
     TERMINATEORCHESTRATION_FIELD_NUMBER: _ClassVar[int]
+    SENDENTITYMESSAGE_FIELD_NUMBER: _ClassVar[int]
     id: int
     scheduleTask: ScheduleTaskAction
     createSubOrchestration: CreateSubOrchestrationAction
@@ -406,36 +535,50 @@ class OrchestratorAction(_message.Message):
     sendEvent: SendEventAction
     completeOrchestration: CompleteOrchestrationAction
     terminateOrchestration: TerminateOrchestrationAction
-    def __init__(self, id: _Optional[int] = ..., scheduleTask: _Optional[_Union[ScheduleTaskAction, _Mapping]] = ..., createSubOrchestration: _Optional[_Union[CreateSubOrchestrationAction, _Mapping]] = ..., createTimer: _Optional[_Union[CreateTimerAction, _Mapping]] = ..., sendEvent: _Optional[_Union[SendEventAction, _Mapping]] = ..., completeOrchestration: _Optional[_Union[CompleteOrchestrationAction, _Mapping]] = ..., terminateOrchestration: _Optional[_Union[TerminateOrchestrationAction, _Mapping]] = ...) -> None: ...
+    sendEntityMessage: SendEntityMessageAction
+    def __init__(self, id: _Optional[int] = ..., scheduleTask: _Optional[_Union[ScheduleTaskAction, _Mapping]] = ..., createSubOrchestration: _Optional[_Union[CreateSubOrchestrationAction, _Mapping]] = ..., createTimer: _Optional[_Union[CreateTimerAction, _Mapping]] = ..., sendEvent: _Optional[_Union[SendEventAction, _Mapping]] = ..., completeOrchestration: _Optional[_Union[CompleteOrchestrationAction, _Mapping]] = ..., terminateOrchestration: _Optional[_Union[TerminateOrchestrationAction, _Mapping]] = ..., sendEntityMessage: _Optional[_Union[SendEntityMessageAction, _Mapping]] = ...) -> None: ...
 
 class OrchestratorRequest(_message.Message):
-    __slots__ = ("instanceId", "executionId", "pastEvents", "newEvents", "entityParameters")
+    __slots__ = ("instanceId", "executionId", "pastEvents", "newEvents", "entityParameters", "requiresHistoryStreaming", "properties")
+    class PropertiesEntry(_message.Message):
+        __slots__ = ("key", "value")
+        KEY_FIELD_NUMBER: _ClassVar[int]
+        VALUE_FIELD_NUMBER: _ClassVar[int]
+        key: str
+        value: _struct_pb2.Value
+        def __init__(self, key: _Optional[str] = ..., value: _Optional[_Union[_struct_pb2.Value, _Mapping]] = ...) -> None: ...
     INSTANCEID_FIELD_NUMBER: _ClassVar[int]
     EXECUTIONID_FIELD_NUMBER: _ClassVar[int]
     PASTEVENTS_FIELD_NUMBER: _ClassVar[int]
     NEWEVENTS_FIELD_NUMBER: _ClassVar[int]
     ENTITYPARAMETERS_FIELD_NUMBER: _ClassVar[int]
+    REQUIRESHISTORYSTREAMING_FIELD_NUMBER: _ClassVar[int]
+    PROPERTIES_FIELD_NUMBER: _ClassVar[int]
     instanceId: str
     executionId: _wrappers_pb2.StringValue
     pastEvents: _containers.RepeatedCompositeFieldContainer[HistoryEvent]
     newEvents: _containers.RepeatedCompositeFieldContainer[HistoryEvent]
     entityParameters: OrchestratorEntityParameters
-    def __init__(self, instanceId: _Optional[str] = ..., executionId: _Optional[_Union[_wrappers_pb2.StringValue, _Mapping]] = ..., pastEvents: _Optional[_Iterable[_Union[HistoryEvent, _Mapping]]] = ..., newEvents: _Optional[_Iterable[_Union[HistoryEvent, _Mapping]]] = ..., entityParameters: _Optional[_Union[OrchestratorEntityParameters, _Mapping]] = ...) -> None: ...
+    requiresHistoryStreaming: bool
+    properties: _containers.MessageMap[str, _struct_pb2.Value]
+    def __init__(self, instanceId: _Optional[str] = ..., executionId: _Optional[_Union[_wrappers_pb2.StringValue, _Mapping]] = ..., pastEvents: _Optional[_Iterable[_Union[HistoryEvent, _Mapping]]] = ..., newEvents: _Optional[_Iterable[_Union[HistoryEvent, _Mapping]]] = ..., entityParameters: _Optional[_Union[OrchestratorEntityParameters, _Mapping]] = ..., requiresHistoryStreaming: bool = ..., properties: _Optional[_Mapping[str, _struct_pb2.Value]] = ...) -> None: ...
 
 class OrchestratorResponse(_message.Message):
-    __slots__ = ("instanceId", "actions", "customStatus", "completionToken")
+    __slots__ = ("instanceId", "actions", "customStatus", "completionToken", "numEventsProcessed")
     INSTANCEID_FIELD_NUMBER: _ClassVar[int]
     ACTIONS_FIELD_NUMBER: _ClassVar[int]
     CUSTOMSTATUS_FIELD_NUMBER: _ClassVar[int]
     COMPLETIONTOKEN_FIELD_NUMBER: _ClassVar[int]
+    NUMEVENTSPROCESSED_FIELD_NUMBER: _ClassVar[int]
     instanceId: str
     actions: _containers.RepeatedCompositeFieldContainer[OrchestratorAction]
     customStatus: _wrappers_pb2.StringValue
     completionToken: str
-    def __init__(self, instanceId: _Optional[str] = ..., actions: _Optional[_Iterable[_Union[OrchestratorAction, _Mapping]]] = ..., customStatus: _Optional[_Union[_wrappers_pb2.StringValue, _Mapping]] = ..., completionToken: _Optional[str] = ...) -> None: ...
+    numEventsProcessed: _wrappers_pb2.Int32Value
+    def __init__(self, instanceId: _Optional[str] = ..., actions: _Optional[_Iterable[_Union[OrchestratorAction, _Mapping]]] = ..., customStatus: _Optional[_Union[_wrappers_pb2.StringValue, _Mapping]] = ..., completionToken: _Optional[str] = ..., numEventsProcessed: _Optional[_Union[_wrappers_pb2.Int32Value, _Mapping]] = ...) -> None: ...
 
 class CreateInstanceRequest(_message.Message):
-    __slots__ = ("instanceId", "name", "version", "input", "scheduledStartTimestamp", "orchestrationIdReusePolicy", "executionId", "tags")
+    __slots__ = ("instanceId", "name", "version", "input", "scheduledStartTimestamp", "orchestrationIdReusePolicy", "executionId", "tags", "parentTraceContext", "requestTime")
     class TagsEntry(_message.Message):
         __slots__ = ("key", "value")
         KEY_FIELD_NUMBER: _ClassVar[int]
@@ -451,6 +594,8 @@ class CreateInstanceRequest(_message.Message):
     ORCHESTRATIONIDREUSEPOLICY_FIELD_NUMBER: _ClassVar[int]
     EXECUTIONID_FIELD_NUMBER: _ClassVar[int]
     TAGS_FIELD_NUMBER: _ClassVar[int]
+    PARENTTRACECONTEXT_FIELD_NUMBER: _ClassVar[int]
+    REQUESTTIME_FIELD_NUMBER: _ClassVar[int]
     instanceId: str
     name: str
     version: _wrappers_pb2.StringValue
@@ -459,15 +604,15 @@ class CreateInstanceRequest(_message.Message):
     orchestrationIdReusePolicy: OrchestrationIdReusePolicy
     executionId: _wrappers_pb2.StringValue
     tags: _containers.ScalarMap[str, str]
-    def __init__(self, instanceId: _Optional[str] = ..., name: _Optional[str] = ..., version: _Optional[_Union[_wrappers_pb2.StringValue, _Mapping]] = ..., input: _Optional[_Union[_wrappers_pb2.StringValue, _Mapping]] = ..., scheduledStartTimestamp: _Optional[_Union[_timestamp_pb2.Timestamp, _Mapping]] = ..., orchestrationIdReusePolicy: _Optional[_Union[OrchestrationIdReusePolicy, _Mapping]] = ..., executionId: _Optional[_Union[_wrappers_pb2.StringValue, _Mapping]] = ..., tags: _Optional[_Mapping[str, str]] = ...) -> None: ...
+    parentTraceContext: TraceContext
+    requestTime: _timestamp_pb2.Timestamp
+    def __init__(self, instanceId: _Optional[str] = ..., name: _Optional[str] = ..., version: _Optional[_Union[_wrappers_pb2.StringValue, _Mapping]] = ..., input: _Optional[_Union[_wrappers_pb2.StringValue, _Mapping]] = ..., scheduledStartTimestamp: _Optional[_Union[_timestamp_pb2.Timestamp, _Mapping]] = ..., orchestrationIdReusePolicy: _Optional[_Union[OrchestrationIdReusePolicy, _Mapping]] = ..., executionId: _Optional[_Union[_wrappers_pb2.StringValue, _Mapping]] = ..., tags: _Optional[_Mapping[str, str]] = ..., parentTraceContext: _Optional[_Union[TraceContext, _Mapping]] = ..., requestTime: _Optional[_Union[_timestamp_pb2.Timestamp, _Mapping]] = ...) -> None: ...
 
 class OrchestrationIdReusePolicy(_message.Message):
-    __slots__ = ("operationStatus", "action")
-    OPERATIONSTATUS_FIELD_NUMBER: _ClassVar[int]
-    ACTION_FIELD_NUMBER: _ClassVar[int]
-    operationStatus: _containers.RepeatedScalarFieldContainer[OrchestrationStatus]
-    action: CreateOrchestrationAction
-    def __init__(self, operationStatus: _Optional[_Iterable[_Union[OrchestrationStatus, str]]] = ..., action: _Optional[_Union[CreateOrchestrationAction, str]] = ...) -> None: ...
+    __slots__ = ("replaceableStatus",)
+    REPLACEABLESTATUS_FIELD_NUMBER: _ClassVar[int]
+    replaceableStatus: _containers.RepeatedScalarFieldContainer[OrchestrationStatus]
+    def __init__(self, replaceableStatus: _Optional[_Iterable[_Union[OrchestrationStatus, str]]] = ...) -> None: ...
 
 class CreateInstanceResponse(_message.Message):
     __slots__ = ("instanceId",)
@@ -504,7 +649,14 @@ class RewindInstanceResponse(_message.Message):
     def __init__(self) -> None: ...
 
 class OrchestrationState(_message.Message):
-    __slots__ = ("instanceId", "name", "version", "orchestrationStatus", "scheduledStartTimestamp", "createdTimestamp", "lastUpdatedTimestamp", "input", "output", "customStatus", "failureDetails", "executionId", "completedTimestamp", "parentInstanceId")
+    __slots__ = ("instanceId", "name", "version", "orchestrationStatus", "scheduledStartTimestamp", "createdTimestamp", "lastUpdatedTimestamp", "input", "output", "customStatus", "failureDetails", "executionId", "completedTimestamp", "parentInstanceId", "tags")
+    class TagsEntry(_message.Message):
+        __slots__ = ("key", "value")
+        KEY_FIELD_NUMBER: _ClassVar[int]
+        VALUE_FIELD_NUMBER: _ClassVar[int]
+        key: str
+        value: str
+        def __init__(self, key: _Optional[str] = ..., value: _Optional[str] = ...) -> None: ...
     INSTANCEID_FIELD_NUMBER: _ClassVar[int]
     NAME_FIELD_NUMBER: _ClassVar[int]
     VERSION_FIELD_NUMBER: _ClassVar[int]
@@ -519,6 +671,7 @@ class OrchestrationState(_message.Message):
     EXECUTIONID_FIELD_NUMBER: _ClassVar[int]
     COMPLETEDTIMESTAMP_FIELD_NUMBER: _ClassVar[int]
     PARENTINSTANCEID_FIELD_NUMBER: _ClassVar[int]
+    TAGS_FIELD_NUMBER: _ClassVar[int]
     instanceId: str
     name: str
     version: _wrappers_pb2.StringValue
@@ -533,7 +686,8 @@ class OrchestrationState(_message.Message):
     executionId: _wrappers_pb2.StringValue
     completedTimestamp: _timestamp_pb2.Timestamp
     parentInstanceId: _wrappers_pb2.StringValue
-    def __init__(self, instanceId: _Optional[str] = ..., name: _Optional[str] = ..., version: _Optional[_Union[_wrappers_pb2.StringValue, _Mapping]] = ..., orchestrationStatus: _Optional[_Union[OrchestrationStatus, str]] = ..., scheduledStartTimestamp: _Optional[_Union[_timestamp_pb2.Timestamp, _Mapping]] = ..., createdTimestamp: _Optional[_Union[_timestamp_pb2.Timestamp, _Mapping]] = ..., lastUpdatedTimestamp: _Optional[_Union[_timestamp_pb2.Timestamp, _Mapping]] = ..., input: _Optional[_Union[_wrappers_pb2.StringValue, _Mapping]] = ..., output: _Optional[_Union[_wrappers_pb2.StringValue, _Mapping]] = ..., customStatus: _Optional[_Union[_wrappers_pb2.StringValue, _Mapping]] = ..., failureDetails: _Optional[_Union[TaskFailureDetails, _Mapping]] = ..., executionId: _Optional[_Union[_wrappers_pb2.StringValue, _Mapping]] = ..., completedTimestamp: _Optional[_Union[_timestamp_pb2.Timestamp, _Mapping]] = ..., parentInstanceId: _Optional[_Union[_wrappers_pb2.StringValue, _Mapping]] = ...) -> None: ...
+    tags: _containers.ScalarMap[str, str]
+    def __init__(self, instanceId: _Optional[str] = ..., name: _Optional[str] = ..., version: _Optional[_Union[_wrappers_pb2.StringValue, _Mapping]] = ..., orchestrationStatus: _Optional[_Union[OrchestrationStatus, str]] = ..., scheduledStartTimestamp: _Optional[_Union[_timestamp_pb2.Timestamp, _Mapping]] = ..., createdTimestamp: _Optional[_Union[_timestamp_pb2.Timestamp, _Mapping]] = ..., lastUpdatedTimestamp: _Optional[_Union[_timestamp_pb2.Timestamp, _Mapping]] = ..., input: _Optional[_Union[_wrappers_pb2.StringValue, _Mapping]] = ..., output: _Optional[_Union[_wrappers_pb2.StringValue, _Mapping]] = ..., customStatus: _Optional[_Union[_wrappers_pb2.StringValue, _Mapping]] = ..., failureDetails: _Optional[_Union[TaskFailureDetails, _Mapping]] = ..., executionId: _Optional[_Union[_wrappers_pb2.StringValue, _Mapping]] = ..., completedTimestamp: _Optional[_Union[_timestamp_pb2.Timestamp, _Mapping]] = ..., parentInstanceId: _Optional[_Union[_wrappers_pb2.StringValue, _Mapping]] = ..., tags: _Optional[_Mapping[str, str]] = ...) -> None: ...
 
 class RaiseEventRequest(_message.Message):
     __slots__ = ("instanceId", "name", "input")
@@ -642,10 +796,12 @@ class PurgeInstanceFilter(_message.Message):
     def __init__(self, createdTimeFrom: _Optional[_Union[_timestamp_pb2.Timestamp, _Mapping]] = ..., createdTimeTo: _Optional[_Union[_timestamp_pb2.Timestamp, _Mapping]] = ..., runtimeStatus: _Optional[_Iterable[_Union[OrchestrationStatus, str]]] = ...) -> None: ...
 
 class PurgeInstancesResponse(_message.Message):
-    __slots__ = ("deletedInstanceCount",)
+    __slots__ = ("deletedInstanceCount", "isComplete")
     DELETEDINSTANCECOUNT_FIELD_NUMBER: _ClassVar[int]
+    ISCOMPLETE_FIELD_NUMBER: _ClassVar[int]
     deletedInstanceCount: int
-    def __init__(self, deletedInstanceCount: _Optional[int] = ...) -> None: ...
+    isComplete: _wrappers_pb2.BoolValue
+    def __init__(self, deletedInstanceCount: _Optional[int] = ..., isComplete: _Optional[_Union[_wrappers_pb2.BoolValue, _Mapping]] = ...) -> None: ...
 
 class CreateTaskHubRequest(_message.Message):
     __slots__ = ("recreateIfExists",)
@@ -666,18 +822,22 @@ class DeleteTaskHubResponse(_message.Message):
     def __init__(self) -> None: ...
 
 class SignalEntityRequest(_message.Message):
-    __slots__ = ("instanceId", "name", "input", "requestId", "scheduledTime")
+    __slots__ = ("instanceId", "name", "input", "requestId", "scheduledTime", "parentTraceContext", "requestTime")
     INSTANCEID_FIELD_NUMBER: _ClassVar[int]
     NAME_FIELD_NUMBER: _ClassVar[int]
     INPUT_FIELD_NUMBER: _ClassVar[int]
     REQUESTID_FIELD_NUMBER: _ClassVar[int]
     SCHEDULEDTIME_FIELD_NUMBER: _ClassVar[int]
+    PARENTTRACECONTEXT_FIELD_NUMBER: _ClassVar[int]
+    REQUESTTIME_FIELD_NUMBER: _ClassVar[int]
     instanceId: str
     name: str
     input: _wrappers_pb2.StringValue
     requestId: str
     scheduledTime: _timestamp_pb2.Timestamp
-    def __init__(self, instanceId: _Optional[str] = ..., name: _Optional[str] = ..., input: _Optional[_Union[_wrappers_pb2.StringValue, _Mapping]] = ..., requestId: _Optional[str] = ..., scheduledTime: _Optional[_Union[_timestamp_pb2.Timestamp, _Mapping]] = ...) -> None: ...
+    parentTraceContext: TraceContext
+    requestTime: _timestamp_pb2.Timestamp
+    def __init__(self, instanceId: _Optional[str] = ..., name: _Optional[str] = ..., input: _Optional[_Union[_wrappers_pb2.StringValue, _Mapping]] = ..., requestId: _Optional[str] = ..., scheduledTime: _Optional[_Union[_timestamp_pb2.Timestamp, _Mapping]] = ..., parentTraceContext: _Optional[_Union[TraceContext, _Mapping]] = ..., requestTime: _Optional[_Union[_timestamp_pb2.Timestamp, _Mapping]] = ...) -> None: ...
 
 class SignalEntityResponse(_message.Message):
     __slots__ = ()
@@ -782,26 +942,44 @@ class EntityBatchRequest(_message.Message):
     def __init__(self, instanceId: _Optional[str] = ..., entityState: _Optional[_Union[_wrappers_pb2.StringValue, _Mapping]] = ..., operations: _Optional[_Iterable[_Union[OperationRequest, _Mapping]]] = ...) -> None: ...
 
 class EntityBatchResult(_message.Message):
-    __slots__ = ("results", "actions", "entityState", "failureDetails")
+    __slots__ = ("results", "actions", "entityState", "failureDetails", "completionToken", "operationInfos")
     RESULTS_FIELD_NUMBER: _ClassVar[int]
     ACTIONS_FIELD_NUMBER: _ClassVar[int]
     ENTITYSTATE_FIELD_NUMBER: _ClassVar[int]
     FAILUREDETAILS_FIELD_NUMBER: _ClassVar[int]
+    COMPLETIONTOKEN_FIELD_NUMBER: _ClassVar[int]
+    OPERATIONINFOS_FIELD_NUMBER: _ClassVar[int]
     results: _containers.RepeatedCompositeFieldContainer[OperationResult]
     actions: _containers.RepeatedCompositeFieldContainer[OperationAction]
     entityState: _wrappers_pb2.StringValue
     failureDetails: TaskFailureDetails
-    def __init__(self, results: _Optional[_Iterable[_Union[OperationResult, _Mapping]]] = ..., actions: _Optional[_Iterable[_Union[OperationAction, _Mapping]]] = ..., entityState: _Optional[_Union[_wrappers_pb2.StringValue, _Mapping]] = ..., failureDetails: _Optional[_Union[TaskFailureDetails, _Mapping]] = ...) -> None: ...
+    completionToken: str
+    operationInfos: _containers.RepeatedCompositeFieldContainer[OperationInfo]
+    def __init__(self, results: _Optional[_Iterable[_Union[OperationResult, _Mapping]]] = ..., actions: _Optional[_Iterable[_Union[OperationAction, _Mapping]]] = ..., entityState: _Optional[_Union[_wrappers_pb2.StringValue, _Mapping]] = ..., failureDetails: _Optional[_Union[TaskFailureDetails, _Mapping]] = ..., completionToken: _Optional[str] = ..., operationInfos: _Optional[_Iterable[_Union[OperationInfo, _Mapping]]] = ...) -> None: ...
+
+class EntityRequest(_message.Message):
+    __slots__ = ("instanceId", "executionId", "entityState", "operationRequests")
+    INSTANCEID_FIELD_NUMBER: _ClassVar[int]
+    EXECUTIONID_FIELD_NUMBER: _ClassVar[int]
+    ENTITYSTATE_FIELD_NUMBER: _ClassVar[int]
+    OPERATIONREQUESTS_FIELD_NUMBER: _ClassVar[int]
+    instanceId: str
+    executionId: str
+    entityState: _wrappers_pb2.StringValue
+    operationRequests: _containers.RepeatedCompositeFieldContainer[HistoryEvent]
+    def __init__(self, instanceId: _Optional[str] = ..., executionId: _Optional[str] = ..., entityState: _Optional[_Union[_wrappers_pb2.StringValue, _Mapping]] = ..., operationRequests: _Optional[_Iterable[_Union[HistoryEvent, _Mapping]]] = ...) -> None: ...
 
 class OperationRequest(_message.Message):
-    __slots__ = ("operation", "requestId", "input")
+    __slots__ = ("operation", "requestId", "input", "traceContext")
     OPERATION_FIELD_NUMBER: _ClassVar[int]
     REQUESTID_FIELD_NUMBER: _ClassVar[int]
     INPUT_FIELD_NUMBER: _ClassVar[int]
+    TRACECONTEXT_FIELD_NUMBER: _ClassVar[int]
     operation: str
     requestId: str
     input: _wrappers_pb2.StringValue
-    def __init__(self, operation: _Optional[str] = ..., requestId: _Optional[str] = ..., input: _Optional[_Union[_wrappers_pb2.StringValue, _Mapping]] = ...) -> None: ...
+    traceContext: TraceContext
+    def __init__(self, operation: _Optional[str] = ..., requestId: _Optional[str] = ..., input: _Optional[_Union[_wrappers_pb2.StringValue, _Mapping]] = ..., traceContext: _Optional[_Union[TraceContext, _Mapping]] = ...) -> None: ...
 
 class OperationResult(_message.Message):
     __slots__ = ("success", "failure")
@@ -811,17 +989,33 @@ class OperationResult(_message.Message):
     failure: OperationResultFailure
     def __init__(self, success: _Optional[_Union[OperationResultSuccess, _Mapping]] = ..., failure: _Optional[_Union[OperationResultFailure, _Mapping]] = ...) -> None: ...
 
+class OperationInfo(_message.Message):
+    __slots__ = ("requestId", "responseDestination")
+    REQUESTID_FIELD_NUMBER: _ClassVar[int]
+    RESPONSEDESTINATION_FIELD_NUMBER: _ClassVar[int]
+    requestId: str
+    responseDestination: OrchestrationInstance
+    def __init__(self, requestId: _Optional[str] = ..., responseDestination: _Optional[_Union[OrchestrationInstance, _Mapping]] = ...) -> None: ...
+
 class OperationResultSuccess(_message.Message):
-    __slots__ = ("result",)
+    __slots__ = ("result", "startTimeUtc", "endTimeUtc")
     RESULT_FIELD_NUMBER: _ClassVar[int]
+    STARTTIMEUTC_FIELD_NUMBER: _ClassVar[int]
+    ENDTIMEUTC_FIELD_NUMBER: _ClassVar[int]
     result: _wrappers_pb2.StringValue
-    def __init__(self, result: _Optional[_Union[_wrappers_pb2.StringValue, _Mapping]] = ...) -> None: ...
+    startTimeUtc: _timestamp_pb2.Timestamp
+    endTimeUtc: _timestamp_pb2.Timestamp
+    def __init__(self, result: _Optional[_Union[_wrappers_pb2.StringValue, _Mapping]] = ..., startTimeUtc: _Optional[_Union[_timestamp_pb2.Timestamp, _Mapping]] = ..., endTimeUtc: _Optional[_Union[_timestamp_pb2.Timestamp, _Mapping]] = ...) -> None: ...
 
 class OperationResultFailure(_message.Message):
-    __slots__ = ("failureDetails",)
+    __slots__ = ("failureDetails", "startTimeUtc", "endTimeUtc")
     FAILUREDETAILS_FIELD_NUMBER: _ClassVar[int]
+    STARTTIMEUTC_FIELD_NUMBER: _ClassVar[int]
+    ENDTIMEUTC_FIELD_NUMBER: _ClassVar[int]
     failureDetails: TaskFailureDetails
-    def __init__(self, failureDetails: _Optional[_Union[TaskFailureDetails, _Mapping]] = ...) -> None: ...
+    startTimeUtc: _timestamp_pb2.Timestamp
+    endTimeUtc: _timestamp_pb2.Timestamp
+    def __init__(self, failureDetails: _Optional[_Union[TaskFailureDetails, _Mapping]] = ..., startTimeUtc: _Optional[_Union[_timestamp_pb2.Timestamp, _Mapping]] = ..., endTimeUtc: _Optional[_Union[_timestamp_pb2.Timestamp, _Mapping]] = ...) -> None: ...
 
 class OperationAction(_message.Message):
     __slots__ = ("id", "sendSignal", "startNewOrchestration")
@@ -834,52 +1028,96 @@ class OperationAction(_message.Message):
     def __init__(self, id: _Optional[int] = ..., sendSignal: _Optional[_Union[SendSignalAction, _Mapping]] = ..., startNewOrchestration: _Optional[_Union[StartNewOrchestrationAction, _Mapping]] = ...) -> None: ...
 
 class SendSignalAction(_message.Message):
-    __slots__ = ("instanceId", "name", "input", "scheduledTime")
+    __slots__ = ("instanceId", "name", "input", "scheduledTime", "requestTime", "parentTraceContext")
     INSTANCEID_FIELD_NUMBER: _ClassVar[int]
     NAME_FIELD_NUMBER: _ClassVar[int]
     INPUT_FIELD_NUMBER: _ClassVar[int]
     SCHEDULEDTIME_FIELD_NUMBER: _ClassVar[int]
+    REQUESTTIME_FIELD_NUMBER: _ClassVar[int]
+    PARENTTRACECONTEXT_FIELD_NUMBER: _ClassVar[int]
     instanceId: str
     name: str
     input: _wrappers_pb2.StringValue
     scheduledTime: _timestamp_pb2.Timestamp
-    def __init__(self, instanceId: _Optional[str] = ..., name: _Optional[str] = ..., input: _Optional[_Union[_wrappers_pb2.StringValue, _Mapping]] = ..., scheduledTime: _Optional[_Union[_timestamp_pb2.Timestamp, _Mapping]] = ...) -> None: ...
+    requestTime: _timestamp_pb2.Timestamp
+    parentTraceContext: TraceContext
+    def __init__(self, instanceId: _Optional[str] = ..., name: _Optional[str] = ..., input: _Optional[_Union[_wrappers_pb2.StringValue, _Mapping]] = ..., scheduledTime: _Optional[_Union[_timestamp_pb2.Timestamp, _Mapping]] = ..., requestTime: _Optional[_Union[_timestamp_pb2.Timestamp, _Mapping]] = ..., parentTraceContext: _Optional[_Union[TraceContext, _Mapping]] = ...) -> None: ...
 
 class StartNewOrchestrationAction(_message.Message):
-    __slots__ = ("instanceId", "name", "version", "input", "scheduledTime")
+    __slots__ = ("instanceId", "name", "version", "input", "scheduledTime", "requestTime", "parentTraceContext")
     INSTANCEID_FIELD_NUMBER: _ClassVar[int]
     NAME_FIELD_NUMBER: _ClassVar[int]
     VERSION_FIELD_NUMBER: _ClassVar[int]
     INPUT_FIELD_NUMBER: _ClassVar[int]
     SCHEDULEDTIME_FIELD_NUMBER: _ClassVar[int]
+    REQUESTTIME_FIELD_NUMBER: _ClassVar[int]
+    PARENTTRACECONTEXT_FIELD_NUMBER: _ClassVar[int]
     instanceId: str
     name: str
     version: _wrappers_pb2.StringValue
     input: _wrappers_pb2.StringValue
     scheduledTime: _timestamp_pb2.Timestamp
-    def __init__(self, instanceId: _Optional[str] = ..., name: _Optional[str] = ..., version: _Optional[_Union[_wrappers_pb2.StringValue, _Mapping]] = ..., input: _Optional[_Union[_wrappers_pb2.StringValue, _Mapping]] = ..., scheduledTime: _Optional[_Union[_timestamp_pb2.Timestamp, _Mapping]] = ...) -> None: ...
+    requestTime: _timestamp_pb2.Timestamp
+    parentTraceContext: TraceContext
+    def __init__(self, instanceId: _Optional[str] = ..., name: _Optional[str] = ..., version: _Optional[_Union[_wrappers_pb2.StringValue, _Mapping]] = ..., input: _Optional[_Union[_wrappers_pb2.StringValue, _Mapping]] = ..., scheduledTime: _Optional[_Union[_timestamp_pb2.Timestamp, _Mapping]] = ..., requestTime: _Optional[_Union[_timestamp_pb2.Timestamp, _Mapping]] = ..., parentTraceContext: _Optional[_Union[TraceContext, _Mapping]] = ...) -> None: ...
+
+class AbandonActivityTaskRequest(_message.Message):
+    __slots__ = ("completionToken",)
+    COMPLETIONTOKEN_FIELD_NUMBER: _ClassVar[int]
+    completionToken: str
+    def __init__(self, completionToken: _Optional[str] = ...) -> None: ...
+
+class AbandonActivityTaskResponse(_message.Message):
+    __slots__ = ()
+    def __init__(self) -> None: ...
+
+class AbandonOrchestrationTaskRequest(_message.Message):
+    __slots__ = ("completionToken",)
+    COMPLETIONTOKEN_FIELD_NUMBER: _ClassVar[int]
+    completionToken: str
+    def __init__(self, completionToken: _Optional[str] = ...) -> None: ...
+
+class AbandonOrchestrationTaskResponse(_message.Message):
+    __slots__ = ()
+    def __init__(self) -> None: ...
+
+class AbandonEntityTaskRequest(_message.Message):
+    __slots__ = ("completionToken",)
+    COMPLETIONTOKEN_FIELD_NUMBER: _ClassVar[int]
+    completionToken: str
+    def __init__(self, completionToken: _Optional[str] = ...) -> None: ...
+
+class AbandonEntityTaskResponse(_message.Message):
+    __slots__ = ()
+    def __init__(self) -> None: ...
 
 class GetWorkItemsRequest(_message.Message):
-    __slots__ = ("maxConcurrentOrchestrationWorkItems", "maxConcurrentActivityWorkItems")
+    __slots__ = ("maxConcurrentOrchestrationWorkItems", "maxConcurrentActivityWorkItems", "maxConcurrentEntityWorkItems", "capabilities")
     MAXCONCURRENTORCHESTRATIONWORKITEMS_FIELD_NUMBER: _ClassVar[int]
     MAXCONCURRENTACTIVITYWORKITEMS_FIELD_NUMBER: _ClassVar[int]
+    MAXCONCURRENTENTITYWORKITEMS_FIELD_NUMBER: _ClassVar[int]
+    CAPABILITIES_FIELD_NUMBER: _ClassVar[int]
     maxConcurrentOrchestrationWorkItems: int
     maxConcurrentActivityWorkItems: int
-    def __init__(self, maxConcurrentOrchestrationWorkItems: _Optional[int] = ..., maxConcurrentActivityWorkItems: _Optional[int] = ...) -> None: ...
+    maxConcurrentEntityWorkItems: int
+    capabilities: _containers.RepeatedScalarFieldContainer[WorkerCapability]
+    def __init__(self, maxConcurrentOrchestrationWorkItems: _Optional[int] = ..., maxConcurrentActivityWorkItems: _Optional[int] = ..., maxConcurrentEntityWorkItems: _Optional[int] = ..., capabilities: _Optional[_Iterable[_Union[WorkerCapability, str]]] = ...) -> None: ...
 
 class WorkItem(_message.Message):
-    __slots__ = ("orchestratorRequest", "activityRequest", "entityRequest", "healthPing", "completionToken")
+    __slots__ = ("orchestratorRequest", "activityRequest", "entityRequest", "healthPing", "entityRequestV2", "completionToken")
     ORCHESTRATORREQUEST_FIELD_NUMBER: _ClassVar[int]
     ACTIVITYREQUEST_FIELD_NUMBER: _ClassVar[int]
     ENTITYREQUEST_FIELD_NUMBER: _ClassVar[int]
     HEALTHPING_FIELD_NUMBER: _ClassVar[int]
+    ENTITYREQUESTV2_FIELD_NUMBER: _ClassVar[int]
     COMPLETIONTOKEN_FIELD_NUMBER: _ClassVar[int]
     orchestratorRequest: OrchestratorRequest
     activityRequest: ActivityRequest
     entityRequest: EntityBatchRequest
     healthPing: HealthPing
+    entityRequestV2: EntityRequest
     completionToken: str
-    def __init__(self, orchestratorRequest: _Optional[_Union[OrchestratorRequest, _Mapping]] = ..., activityRequest: _Optional[_Union[ActivityRequest, _Mapping]] = ..., entityRequest: _Optional[_Union[EntityBatchRequest, _Mapping]] = ..., healthPing: _Optional[_Union[HealthPing, _Mapping]] = ..., completionToken: _Optional[str] = ...) -> None: ...
+    def __init__(self, orchestratorRequest: _Optional[_Union[OrchestratorRequest, _Mapping]] = ..., activityRequest: _Optional[_Union[ActivityRequest, _Mapping]] = ..., entityRequest: _Optional[_Union[EntityBatchRequest, _Mapping]] = ..., healthPing: _Optional[_Union[HealthPing, _Mapping]] = ..., entityRequestV2: _Optional[_Union[EntityRequest, _Mapping]] = ..., completionToken: _Optional[str] = ...) -> None: ...
 
 class CompleteTaskResponse(_message.Message):
     __slots__ = ()
@@ -888,3 +1126,19 @@ class CompleteTaskResponse(_message.Message):
 class HealthPing(_message.Message):
     __slots__ = ()
     def __init__(self) -> None: ...
+
+class StreamInstanceHistoryRequest(_message.Message):
+    __slots__ = ("instanceId", "executionId", "forWorkItemProcessing")
+    INSTANCEID_FIELD_NUMBER: _ClassVar[int]
+    EXECUTIONID_FIELD_NUMBER: _ClassVar[int]
+    FORWORKITEMPROCESSING_FIELD_NUMBER: _ClassVar[int]
+    instanceId: str
+    executionId: _wrappers_pb2.StringValue
+    forWorkItemProcessing: bool
+    def __init__(self, instanceId: _Optional[str] = ..., executionId: _Optional[_Union[_wrappers_pb2.StringValue, _Mapping]] = ..., forWorkItemProcessing: bool = ...) -> None: ...
+
+class HistoryChunk(_message.Message):
+    __slots__ = ("events",)
+    EVENTS_FIELD_NUMBER: _ClassVar[int]
+    events: _containers.RepeatedCompositeFieldContainer[HistoryEvent]
+    def __init__(self, events: _Optional[_Iterable[_Union[HistoryEvent, _Mapping]]] = ...) -> None: ...