Support DTS

Author: andystaples

durabletask-azuremanaged/durabletask/azuremanaged/client.py

@@ -13,6 +13,7 @@
     DTSDefaultClientInterceptorImpl,
 )
 from durabletask.client import AsyncTaskHubGrpcClient, TaskHubGrpcClient
+from durabletask.payload.store import PayloadStore
 
 
 # Client class used for Durable Task Scheduler (DTS)
@@ -23,6 +24,7 @@ def __init__(self, *,
                  token_credential: Optional[TokenCredential],
                  secure_channel: bool = True,
                  default_version: Optional[str] = None,
+                 payload_store: Optional[PayloadStore] = None,
                  log_handler: Optional[logging.Handler] = None,
                  log_formatter: Optional[logging.Formatter] = None):
 
@@ -40,7 +42,8 @@ def __init__(self, *,
             log_handler=log_handler,
             log_formatter=log_formatter,
             interceptors=interceptors,
-            default_version=default_version)
+            default_version=default_version,
+            payload_store=payload_store)
 
 
 # Async client class used for Durable Task Scheduler (DTS)
@@ -60,6 +63,8 @@ class AsyncDurableTaskSchedulerClient(AsyncTaskHubGrpcClient):
         secure_channel (bool, optional): Whether to use a secure gRPC channel (TLS).
             Defaults to True.
         default_version (Optional[str], optional): Default version string for orchestrations.
+        payload_store (Optional[PayloadStore], optional): A payload store for
+            externalizing large payloads. If None, payloads are sent inline.
         log_handler (Optional[logging.Handler], optional): Custom logging handler for client logs.
         log_formatter (Optional[logging.Formatter], optional): Custom log formatter for client logs.
 
@@ -85,6 +90,7 @@ def __init__(self, *,
                  token_credential: Optional[AsyncTokenCredential],
                  secure_channel: bool = True,
                  default_version: Optional[str] = None,
+                 payload_store: Optional[PayloadStore] = None,
                  log_handler: Optional[logging.Handler] = None,
                  log_formatter: Optional[logging.Formatter] = None):
 
@@ -102,4 +108,5 @@ def __init__(self, *,
             log_handler=log_handler,
             log_formatter=log_formatter,
             interceptors=interceptors,
-            default_version=default_version)
+            default_version=default_version,
+            payload_store=payload_store)

durabletask-azuremanaged/durabletask/azuremanaged/worker.py

@@ -9,6 +9,7 @@
 
 from durabletask.azuremanaged.internal.durabletask_grpc_interceptor import \
     DTSDefaultClientInterceptorImpl
+from durabletask.payload.store import PayloadStore
 from durabletask.worker import ConcurrencyOptions, TaskHubGrpcWorker
 
 
@@ -30,6 +31,8 @@ class DurableTaskSchedulerWorker(TaskHubGrpcWorker):
         concurrency_options (Optional[ConcurrencyOptions], optional): Configuration
             for controlling worker concurrency limits. If None, default concurrency
             settings will be used.
+        payload_store (Optional[PayloadStore], optional): A payload store for
+            externalizing large payloads. If None, payloads are sent inline.
         log_handler (Optional[logging.Handler], optional): Custom logging handler for worker logs.
         log_formatter (Optional[logging.Formatter], optional): Custom log formatter for worker logs.
 
@@ -63,6 +66,7 @@ def __init__(self, *,
                  token_credential: Optional[TokenCredential],
                  secure_channel: bool = True,
                  concurrency_options: Optional[ConcurrencyOptions] = None,
+                 payload_store: Optional[PayloadStore] = None,
                  log_handler: Optional[logging.Handler] = None,
                  log_formatter: Optional[logging.Formatter] = None):
 
@@ -80,4 +84,5 @@ def __init__(self, *,
             log_handler=log_handler,
             log_formatter=log_formatter,
             interceptors=interceptors,
-            concurrency_options=concurrency_options)
+            concurrency_options=concurrency_options,
+            payload_store=payload_store)

durabletask-azuremanaged/pyproject.toml

@@ -30,6 +30,11 @@ dependencies = [
     "azure-identity>=1.19.0"
 ]
 
+[project.optional-dependencies]
+azure-blob-payloads = [
+    "durabletask[azure-blob-payloads]>=1.3.0"
+]
+
 [project.urls]
 repository = "https://github.com/microsoft/durabletask-python"
 changelog = "https://github.com/microsoft/durabletask-python/blob/main/CHANGELOG.md"