From 7d1cb4ef646cad687afb89de25240afd69ecb0c7 Mon Sep 17 00:00:00 2001 From: Ryan Lettieri Date: Mon, 10 Mar 2025 13:37:07 -0600 Subject: [PATCH] Making token credential optional Signed-off-by: Ryan Lettieri --- durabletask-azuremanaged/durabletask/azuremanaged/client.py | 3 ++- .../azuremanaged/internal/durabletask_grpc_interceptor.py | 4 +++- durabletask-azuremanaged/durabletask/azuremanaged/worker.py | 3 ++- 3 files changed, 7 insertions(+), 3 deletions(-) diff --git a/durabletask-azuremanaged/durabletask/azuremanaged/client.py b/durabletask-azuremanaged/durabletask/azuremanaged/client.py index f641eae..1d8cecd 100644 --- a/durabletask-azuremanaged/durabletask/azuremanaged/client.py +++ b/durabletask-azuremanaged/durabletask/azuremanaged/client.py @@ -2,6 +2,7 @@ # Licensed under the MIT License. from azure.core.credentials import TokenCredential +from typing import Optional from durabletask.azuremanaged.internal.durabletask_grpc_interceptor import \ DTSDefaultClientInterceptorImpl @@ -13,7 +14,7 @@ class DurableTaskSchedulerClient(TaskHubGrpcClient): def __init__(self, *, host_address: str, taskhub: str, - token_credential: TokenCredential, + token_credential: Optional[TokenCredential], secure_channel: bool = True): if not taskhub: diff --git a/durabletask-azuremanaged/durabletask/azuremanaged/internal/durabletask_grpc_interceptor.py b/durabletask-azuremanaged/durabletask/azuremanaged/internal/durabletask_grpc_interceptor.py index a23cac9..077905e 100644 --- a/durabletask-azuremanaged/durabletask/azuremanaged/internal/durabletask_grpc_interceptor.py +++ b/durabletask-azuremanaged/durabletask/azuremanaged/internal/durabletask_grpc_interceptor.py @@ -2,6 +2,8 @@ # Licensed under the MIT License. import grpc +from typing import Optional + from azure.core.credentials import TokenCredential from durabletask.azuremanaged.internal.access_token_manager import \ @@ -15,7 +17,7 @@ class DTSDefaultClientInterceptorImpl (DefaultClientInterceptorImpl): StreamUnaryClientInterceptor and StreamStreamClientInterceptor from grpc to add an interceptor to add additional headers to all calls as needed.""" - def __init__(self, token_credential: TokenCredential, taskhub_name: str): + def __init__(self, token_credential: Optional[TokenCredential], taskhub_name: str): self._metadata = [("taskhub", taskhub_name)] super().__init__(self._metadata) diff --git a/durabletask-azuremanaged/durabletask/azuremanaged/worker.py b/durabletask-azuremanaged/durabletask/azuremanaged/worker.py index d10c2f7..8bdff3d 100644 --- a/durabletask-azuremanaged/durabletask/azuremanaged/worker.py +++ b/durabletask-azuremanaged/durabletask/azuremanaged/worker.py @@ -2,6 +2,7 @@ # Licensed under the MIT License. from azure.core.credentials import TokenCredential +from typing import Optional from durabletask.azuremanaged.internal.durabletask_grpc_interceptor import \ DTSDefaultClientInterceptorImpl @@ -13,7 +14,7 @@ class DurableTaskSchedulerWorker(TaskHubGrpcWorker): def __init__(self, *, host_address: str, taskhub: str, - token_credential: TokenCredential, + token_credential: Optional[TokenCredential], secure_channel: bool = True): if not taskhub: