Making token credential optional (#45)

Signed-off-by: Ryan Lettieri <[email protected]>

Author: RyanLettieri

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:

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)
 

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: