fixing data store connection issue (#38635)

* fixing data store connection issue

* updating version file

Author: achauhan-scc

sdk/ml/azure-ai-ml/CHANGELOG.md

@@ -1,14 +1,9 @@
 # Release History
 
-## 1.22.2 (2024-11-18)
+## 1.22.4 (2024-11-21)
 
 ### Bugs Fixed
-  - **#3620407** - Fix Datastore credentials show up as NoneCredentials
-
-## 1.22.1 (2024-11-13)
-
-### Bugs Fixed
-  - **#38493** - Fix error NoneType object is not subscriptable
+  - **#38609** - Fix Datastore credentials for Account key base
 
 ## 1.22.3 (2024-11-20)
 

sdk/ml/azure-ai-ml/azure/ai/ml/_version.py

@@ -2,4 +2,4 @@
 # Copyright (c) Microsoft Corporation. All rights reserved.
 # ---------------------------------------------------------
 
-VERSION = "1.22.3"
+VERSION = "1.22.4"

sdk/ml/azure-ai-ml/azure/ai/ml/entities/_datastore/utils.py

@@ -30,7 +30,12 @@ def from_rest_datastore_credentials(
     config_class: Any = NoneCredentialConfiguration
 
     if isinstance(rest_credentials, (models.AccountKeyDatastoreCredentials, models2024.AccountKeyDatastoreCredentials)):
-        config_class = AccountKeyConfiguration
+        # we are no more using key for key base account.
+        # https://github.com/Azure/azure-sdk-for-python/pull/35716
+        if isinstance(rest_credentials.secrets, models2024.SasDatastoreSecrets):
+            config_class = SasTokenConfiguration
+        else:
+            config_class = AccountKeyConfiguration
     elif isinstance(rest_credentials, (models.SasDatastoreCredentials, models2024.SasDatastoreCredentials)):
         config_class = SasTokenConfiguration
     elif isinstance(