Add Azure Managed on-demand sandbox preview APIs

Makefile

@@ -9,8 +9,12 @@ install:
 
 gen-proto:
 	curl -o durabletask/internal/orchestrator_service.proto https://raw.githubusercontent.com/microsoft/durabletask-protobuf/refs/heads/main/protos/orchestrator_service.proto
-	curl -H "Accept: application/vnd.github.v3+json" "https://api.github.com/repos/microsoft/durabletask-protobuf/commits?path=protos/orchestrator_service.proto&sha=main&per_page=1" | jq -r '.[0].sha' >> durabletask/internal/PROTO_SOURCE_COMMIT_HASH
+	curl -H "Accept: application/vnd.github.v3+json" "https://api.github.com/repos/microsoft/durabletask-protobuf/commits?path=protos/orchestrator_service.proto&sha=main&per_page=1" | jq -r '.[0].sha' > durabletask/internal/PROTO_SOURCE_COMMIT_HASH
 	python3 -m grpc_tools.protoc --proto_path=.  --python_out=. --pyi_out=. --grpc_python_out=. ./durabletask/internal/orchestrator_service.proto
 	rm durabletask/internal/*.proto
+	ON_DEMAND_SANDBOX_PROTO_SOURCE_COMMIT_HASH=$$(cat durabletask-azuremanaged/durabletask/azuremanaged/internal/ON_DEMAND_SANDBOX_PROTO_SOURCE_COMMIT_HASH); \
+	curl -o durabletask-azuremanaged/durabletask/azuremanaged/internal/on_demand_sandbox_activities_service.proto https://raw.githubusercontent.com/microsoft/durabletask-protobuf/$${ON_DEMAND_SANDBOX_PROTO_SOURCE_COMMIT_HASH}/protos/serverless_activities_service.proto
+	python3 -m grpc_tools.protoc --proto_path=. --python_out=. --pyi_out=. --grpc_python_out=. ./durabletask-azuremanaged/durabletask/azuremanaged/internal/on_demand_sandbox_activities_service.proto
+	rm durabletask-azuremanaged/durabletask/azuremanaged/internal/*.proto
 
 .PHONY: init test gen-proto install

durabletask-azuremanaged/CHANGELOG.md

@@ -5,6 +5,12 @@ All notable changes to this project will be documented in this file.
 The format is based on [Keep a Changelog](https://keepachangelog.com/en/1.0.0/), and this project
 adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0.html).
 
+## Unreleased
+
+- Added preview `durabletask.azuremanaged.preview.on_demand_sandbox` APIs for
+  declaring Durable Task Scheduler on-demand sandbox activities, registering
+  worker profiles, and running sandbox activity workers.
+
 ## v1.5.0
 
 - Updates base dependency to durabletask v1.5.0

durabletask-azuremanaged/durabletask/azuremanaged/internal/ON_DEMAND_SANDBOX_PROTO_SOURCE_COMMIT_HASH

@@ -0,0 +1 @@
+3c481ecf6e4888766dd1317456693ace92ca3669

durabletask-azuremanaged/durabletask/azuremanaged/internal/on_demand_sandbox_activities_service_pb2.pyi

@@ -0,0 +1,111 @@
+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 SubstrateKind(int, metaclass=_enum_type_wrapper.EnumTypeWrapper):
+    __slots__ = ()
+    SUBSTRATE_KIND_UNSPECIFIED: _ClassVar[SubstrateKind]
+    SUBSTRATE_KIND_ACA_SESSION_POOL: _ClassVar[SubstrateKind]
+    SUBSTRATE_KIND_SANDBOX: _ClassVar[SubstrateKind]
+SUBSTRATE_KIND_UNSPECIFIED: SubstrateKind
+SUBSTRATE_KIND_ACA_SESSION_POOL: SubstrateKind
+SUBSTRATE_KIND_SANDBOX: SubstrateKind
+
+class OnDemandSandboxActivityWorkerMessage(_message.Message):
+    __slots__ = ("start", "heartbeat")
+    START_FIELD_NUMBER: _ClassVar[int]
+    HEARTBEAT_FIELD_NUMBER: _ClassVar[int]
+    start: OnDemandSandboxActivityWorkerStart
+    heartbeat: OnDemandSandboxActivityWorkerHeartbeat
+    def __init__(self, start: _Optional[_Union[OnDemandSandboxActivityWorkerStart, _Mapping]] = ..., heartbeat: _Optional[_Union[OnDemandSandboxActivityWorkerHeartbeat, _Mapping]] = ...) -> None: ...
+
+class OnDemandSandboxActivityWorkerStart(_message.Message):
+    __slots__ = ("task_hub", "max_activities_count", "substrate", "dts_sandbox_identifier", "worker_profile_id", "activity_names")
+    TASK_HUB_FIELD_NUMBER: _ClassVar[int]
+    MAX_ACTIVITIES_COUNT_FIELD_NUMBER: _ClassVar[int]
+    SUBSTRATE_FIELD_NUMBER: _ClassVar[int]
+    DTS_SANDBOX_IDENTIFIER_FIELD_NUMBER: _ClassVar[int]
+    WORKER_PROFILE_ID_FIELD_NUMBER: _ClassVar[int]
+    ACTIVITY_NAMES_FIELD_NUMBER: _ClassVar[int]
+    task_hub: str
+    max_activities_count: int
+    substrate: SubstrateKind
+    dts_sandbox_identifier: str
+    worker_profile_id: str
+    activity_names: _containers.RepeatedScalarFieldContainer[str]
+    def __init__(self, task_hub: _Optional[str] = ..., max_activities_count: _Optional[int] = ..., substrate: _Optional[_Union[SubstrateKind, str]] = ..., dts_sandbox_identifier: _Optional[str] = ..., worker_profile_id: _Optional[str] = ..., activity_names: _Optional[_Iterable[str]] = ...) -> None: ...
+
+class OnDemandSandboxActivityWorkerHeartbeat(_message.Message):
+    __slots__ = ("active_activities_count",)
+    ACTIVE_ACTIVITIES_COUNT_FIELD_NUMBER: _ClassVar[int]
+    active_activities_count: int
+    def __init__(self, active_activities_count: _Optional[int] = ...) -> None: ...
+
+class OnDemandSandboxActivityWorkerSessionResult(_message.Message):
+    __slots__ = ("message",)
+    MESSAGE_FIELD_NUMBER: _ClassVar[int]
+    message: str
+    def __init__(self, message: _Optional[str] = ...) -> None: ...
+
+class OnDemandSandboxActivityDeclaration(_message.Message):
+    __slots__ = ("worker_profile_id", "activity_names", "image", "environment_variables", "max_concurrent_activities", "resources", "entrypoint", "cmd", "scheduler_managed_identity_client_id")
+    class EnvironmentVariablesEntry(_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: ...
+    WORKER_PROFILE_ID_FIELD_NUMBER: _ClassVar[int]
+    ACTIVITY_NAMES_FIELD_NUMBER: _ClassVar[int]
+    IMAGE_FIELD_NUMBER: _ClassVar[int]
+    ENVIRONMENT_VARIABLES_FIELD_NUMBER: _ClassVar[int]
+    MAX_CONCURRENT_ACTIVITIES_FIELD_NUMBER: _ClassVar[int]
+    RESOURCES_FIELD_NUMBER: _ClassVar[int]
+    ENTRYPOINT_FIELD_NUMBER: _ClassVar[int]
+    CMD_FIELD_NUMBER: _ClassVar[int]
+    SCHEDULER_MANAGED_IDENTITY_CLIENT_ID_FIELD_NUMBER: _ClassVar[int]
+    worker_profile_id: str
+    activity_names: _containers.RepeatedScalarFieldContainer[str]
+    image: OnDemandSandboxActivityImage
+    environment_variables: _containers.ScalarMap[str, str]
+    max_concurrent_activities: int
+    resources: OnDemandSandboxActivityResources
+    entrypoint: _containers.RepeatedScalarFieldContainer[str]
+    cmd: _containers.RepeatedScalarFieldContainer[str]
+    scheduler_managed_identity_client_id: str
+    def __init__(self, worker_profile_id: _Optional[str] = ..., activity_names: _Optional[_Iterable[str]] = ..., image: _Optional[_Union[OnDemandSandboxActivityImage, _Mapping]] = ..., environment_variables: _Optional[_Mapping[str, str]] = ..., max_concurrent_activities: _Optional[int] = ..., resources: _Optional[_Union[OnDemandSandboxActivityResources, _Mapping]] = ..., entrypoint: _Optional[_Iterable[str]] = ..., cmd: _Optional[_Iterable[str]] = ..., scheduler_managed_identity_client_id: _Optional[str] = ...) -> None: ...
+
+class OnDemandSandboxActivityImage(_message.Message):
+    __slots__ = ("image_ref", "managed_identity_client_id")
+    IMAGE_REF_FIELD_NUMBER: _ClassVar[int]
+    MANAGED_IDENTITY_CLIENT_ID_FIELD_NUMBER: _ClassVar[int]
+    image_ref: str
+    managed_identity_client_id: str
+    def __init__(self, image_ref: _Optional[str] = ..., managed_identity_client_id: _Optional[str] = ...) -> None: ...
+
+class OnDemandSandboxActivityResources(_message.Message):
+    __slots__ = ("cpu", "memory")
+    CPU_FIELD_NUMBER: _ClassVar[int]
+    MEMORY_FIELD_NUMBER: _ClassVar[int]
+    cpu: str
+    memory: str
+    def __init__(self, cpu: _Optional[str] = ..., memory: _Optional[str] = ...) -> None: ...
+
+class OnDemandSandboxActivityDeclarationResult(_message.Message):
+    __slots__ = ()
+    def __init__(self) -> None: ...
+
+class RemoveOnDemandSandboxActivityDeclarationRequest(_message.Message):
+    __slots__ = ("worker_profile_id",)
+    WORKER_PROFILE_ID_FIELD_NUMBER: _ClassVar[int]
+    worker_profile_id: str
+    def __init__(self, worker_profile_id: _Optional[str] = ...) -> None: ...
+
+class RemoveOnDemandSandboxActivityDeclarationResult(_message.Message):
+    __slots__ = ()
+    def __init__(self) -> None: ...