updated azure.py

Author: rupali-atlan

application_sdk/clients/azure/azure.py

@@ -15,7 +15,6 @@
 
 from application_sdk.clients import ClientInterface
 from application_sdk.clients.azure.azure_auth import AzureAuthProvider
-from application_sdk.common.credential_utils import resolve_credentials
 from application_sdk.common.error_codes import ClientError
 from application_sdk.observability.logger_adaptor import get_logger
 
@@ -82,8 +81,22 @@ async def load(self, credentials: Optional[Dict[str, Any]] = None) -> None:
         try:
             logger.info("Loading Azure client...")
 
-            # Resolve credentials using framework's credential resolution
-            self.resolved_credentials = await resolve_credentials(self.credentials)
+            # Handle credential resolution
+            if "credential_guid" in self.credentials:
+                # If we have a credential_guid, use the async get_credentials function
+                from application_sdk.common.credential_utils import get_credentials
+                self.resolved_credentials = await get_credentials(self.credentials["credential_guid"])
+            else:
+                # If credentials are already resolved (direct format), use them as-is
+                # For direct credentials, we need to check if they need resolution
+                if "secret-path" in self.credentials or "credentialSource" in self.credentials:
+                    # Credentials need resolution - this is a complex case
+                    # For now, assume credentials are already resolved if no credential_guid
+                    logger.warning("Credentials appear to need resolution but no credential_guid provided. Using as-is.")
+                    self.resolved_credentials = self.credentials
+                else:
+                    # Credentials are already in the correct format
+                    self.resolved_credentials = self.credentials
 
             # Create Azure credential using Service Principal authentication
             self.credential = await self.auth_provider.create_credential(

application_sdk/common/credential_utils.py

@@ -59,6 +59,7 @@ def _get_credentials_sync(credential_guid: str) -> Dict[str, Any]:
             f"Failed to resolve credentials: {str(e)}",
         )
 
+
 def resolve_credentials(
     credential_config: Dict[str, Any], secret_data: Dict[str, Any]
 ) -> Dict[str, Any]:
@@ -79,6 +80,7 @@ def resolve_credentials(
                 if value in secret_data:
                     credentials["extra"][key] = secret_data[value]
                 elif value in secret_data.get("extra", {}):
+                    
                     credentials["extra"][key] = secret_data["extra"][value]
 
     return credentials

application_sdk/transformers/common/utils.py

@@ -139,7 +139,6 @@ def flatten_yaml_columns(
     flat_columns: List[Dict[str, Any]] = []
     for key, value in nested.items():
         new_key = f"{parent_key}{sep}{key}" if parent_key else key
-        
         if isinstance(value, dict) and any(isinstance(v, dict) for v in value.values()):
             # If value is a dict and has nested dicts, recurse
             flat_columns.extend(flatten_yaml_columns(value, new_key, sep=sep))
@@ -150,5 +149,4 @@ def flatten_yaml_columns(
             )
             col_def["name"] = new_key
             flat_columns.append(col_def)
-    
-    return flat_columns
+    return flat_columns