updated as per PR's copilot autofixes

Author: rupali-atlan

application_sdk/clients/azure/azure.py

@@ -249,4 +249,26 @@ def __enter__(self):
 
     def __exit__(self, exc_type, exc_val, exc_tb):
         """Context manager exit."""
-        asyncio.create_task(self.close()) 
+        # Note: This is a synchronous context manager.
+        # For proper async cleanup, use the async context manager instead.
+        # This method is kept for backward compatibility but doesn't guarantee cleanup.
+        logger.warning(
+            "Using synchronous context manager. For proper async cleanup, "
+            "use 'async with AzureClient() as client:' instead."
+        )
+        # Schedule cleanup but don't wait for it
+        try:
+            loop = asyncio.get_event_loop()
+            if loop.is_running():
+                loop.create_task(self.close())
+        except RuntimeError:
+            # No event loop running, can't schedule async cleanup
+            logger.warning("No event loop running, async cleanup not possible")
+
+    async def __aenter__(self):
+        """Async context manager entry."""
+        return self
+
+    async def __aexit__(self, exc_type, exc_val, exc_tb):
+        """Async context manager exit."""
+        await self.close() 

application_sdk/clients/azure/azure_utils.py

@@ -38,9 +38,9 @@ def parse_azure_url(url: str) -> Dict[str, str]:
         hostname = parsed.netloc.lower()
         service_type = None
 
-        if ".blob.core.windows.net" in hostname:
+        if hostname.endswith(".blob.core.windows.net"):
             service_type = "blob"
-        elif ".dfs.core.windows.net" in hostname:
+        elif hostname.endswith(".dfs.core.windows.net"):
             service_type = "datalake"
         else:
             service_type = "unknown"
@@ -298,4 +298,4 @@ def format_azure_error_message(error: Exception, context: Optional[str] = None)
 
     except Exception as e:
         logger.error(f"Failed to format Azure error message: {str(e)}")
-        return f"Azure Error: {str(error)}" 
+        return f"Azure Error: {str(error)}"