fix: updated as per comments

Author: rupali-atlan

application_sdk/clients/azure/__init__.py

@@ -1,44 +0,0 @@
-"""
-Azure client module for the application-sdk framework.
-
-This module provides a unified interface for connecting to and interacting
-with Azure Storage services including Blob Storage and Data Lake Storage Gen2.
-
-The module includes:
-- AzureClient: Main client for unified Azure service access
-- AzureAuthProvider: Authentication provider for different Azure auth methods
-- AzureStorageClient: Storage service client for Blob and Data Lake Storage
-- Utilities: Common Azure helper functions
-"""
-
-from .azure import AzureClient
-from .azure_auth import AzureAuthProvider
-from .azure_services import AzureStorageClient
-from .azure_utils import (
-    build_azure_connection_string,
-    extract_azure_resource_info,
-    format_azure_error_message,
-    get_azure_service_endpoint,
-    parse_azure_url,
-    validate_azure_credentials,
-    validate_azure_permissions,
-)
-
-__all__ = [
-    # Main client
-    "AzureClient",
-    # Authentication
-    "AzureAuthProvider",
-    # Service-specific clients
-    "AzureStorageClient",
-    # Utilities
-    "parse_azure_url",
-    "validate_azure_credentials",
-    "build_azure_connection_string",
-    "extract_azure_resource_info",
-    "validate_azure_permissions",
-    "get_azure_service_endpoint",
-    "format_azure_error_message",
-]
-
-__version__ = "0.1.0"

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.clients.azure.azure_services import AzureStorageClient
 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
@@ -28,7 +27,7 @@ class AzureClient(ClientInterface):
     Main Azure client for the application-sdk framework.
 
     This client provides a unified interface for connecting to and interacting
-    with Azure Storage services. It supports Service Principal authentication
+    with Azure services. It supports Service Principal authentication
     and provides service-specific subclients for different Azure services.
 
     Attributes:
@@ -136,38 +135,13 @@ async def close(self) -> None:
         except Exception as e:
             logger.error(f"Error closing Azure client: {str(e)}")
 
-    async def get_storage_client(self) -> AzureStorageClient:
-        """
-        Get Azure Storage service client.
-
-        Returns:
-            AzureStorageClient: Configured Azure Storage client.
-
-        Raises:
-            ClientError: If client is not loaded or storage client creation fails.
-        """
-        if not self._connection_health:
-            raise ClientError(f"{ClientError.CLIENT_AUTH_ERROR}: Client not loaded")
-
-        if "storage" not in self._services:
-            try:
-                self._services["storage"] = AzureStorageClient(
-                    credential=self.credential, **self._kwargs
-                )
-                await self._services["storage"].load(self.resolved_credentials)
-            except Exception as e:
-                logger.error(f"Failed to create storage client: {str(e)}")
-                raise ClientError(f"{ClientError.CLIENT_AUTH_ERROR}: {str(e)}")
-
-        return self._services["storage"]
-
     async def get_service_client(self, service_type: str) -> Any:
         """
         Get a service client by type.
 
         Args:
             service_type (str): Type of service client to retrieve.
-                Supported values: 'storage'.
+                Currently no services are supported.
 
         Returns:
             Any: The requested service client.
@@ -176,14 +150,7 @@ async def get_service_client(self, service_type: str) -> Any:
             ValueError: If service_type is not supported.
             ClientError: If client creation fails.
         """
-        service_mapping = {
-            "storage": self.get_storage_client,
-        }
-
-        if service_type not in service_mapping:
-            raise ValueError(f"Unsupported service type: {service_type}")
-
-        return await service_mapping[service_type]()
+        raise ValueError(f"Unsupported service type: {service_type}")
 
     async def health_check(self) -> Dict[str, Any]:
         """