ensuring cosmos client has valid url

microsoft · Oct 3, 2024 · 26dc092 · 26dc092
1 parent f4a9ae7
commit 26dc092
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/pctasks/core/pctasks/core/cosmos/settings.py b/pctasks/core/pctasks/core/cosmos/settings.py
@@ -124,6 +124,7 @@ def get_client(self) -> CosmosClient:
         else:
             # If the connection string is not set, the credetials are
             # automatically picked up from the environment/managed identity
+            assert self.url
             credential = DefaultAzureCredential()
             return CosmosClient(
                 self.url, credential=credential, connection_verify=connection_verify
@@ -145,6 +146,7 @@ def get_async_client(self) -> AsyncCosmosClient:
         else:
             # If the connection string is not set, the credetials are
             # automatically picked up from the environment/managed identity
+            assert self.url
             credential = azure.identity.aio.DefaultAzureCredential()
             return AsyncCosmosClient(
                 self.url, credential=credential, connection_verify=connection_verify