genesis-flow 1.0.5__py3-none-any.whl → 1.0.6__py3-none-any.whl

{genesis_flow-1.0.5.dist-info → genesis_flow-1.0.6.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: genesis-flow
-Version: 1.0.5
+Version: 1.0.6
 Summary: Genesis-Flow: MLflow v3.1.4 compatible fork for Genesis platform
 Maintainer-email: Databricks <mlflow-oss-maintainers@googlegroups.com>
 License: Copyright 2018 Databricks, Inc.  All rights reserved.

{genesis_flow-1.0.5.dist-info → genesis_flow-1.0.6.dist-info}/RECORD

@@ -1,4 +1,4 @@
-genesis_flow-1.0.5.dist-info/licenses/LICENSE.txt,sha256=Y5U1Xebzka__NZlqMPtBsYm0mRpMtUmTrONatpoL-ig,11382
+genesis_flow-1.0.6.dist-info/licenses/LICENSE.txt,sha256=Y5U1Xebzka__NZlqMPtBsYm0mRpMtUmTrONatpoL-ig,11382
 mlflow/__init__.py,sha256=-_r__N5Afed81pLVtr2wKbHQIA0aj9u9n_7kWGxLWi4,11194
 mlflow/__main__.py,sha256=_PcdoxKehR_a2MI6GqBfzYzRCXZhVyDCSdbxDWVlWd4,39
 mlflow/cli.py,sha256=f1ObrWZ03HgRiRoVEE1Gffe-dGcSY7CxJyEFgb5VUMM,26137
@@ -26,7 +26,7 @@ mlflow/azure/client.py,sha256=dUaMFj04OLrlyvYSRTHXF4CaLHCGUdSzhjM5u8ucv8g,11509
 mlflow/azure/config.py,sha256=0DwiEYN3TefmdaGO9KogNNflX_-0wOeHwKt2hMxKEdU,4491
 mlflow/azure/connection_factory.py,sha256=1hcwPmc5xpP_MS1t3CIlzYfJ_mzEn2GBxnz5TJ1sHOY,12391
 mlflow/azure/exceptions.py,sha256=3jQo_kimK4g94LmSCtzfRQqEOl6bDBtwAQ6utXbgw8k,494
-mlflow/azure/stores.py,sha256=MnZaPU2eOAerzGfcwSoyuMCkd9ad1cOmsb6fclgxF4Q,12890
+mlflow/azure/stores.py,sha256=EwcAnb7RolKU-cI11Ak1Rdsv6a-TOQJbAOOaaoIcKuE,12876
 mlflow/azure/utils.py,sha256=uU4OWU9ffz3KhLf0IgGqabZTTiAskLrNGVH5bIcwKUU,5563
 mlflow/bedrock/__init__.py,sha256=LNOczTpqmgbdek-anWMNisPGDcnn-VbWQm5VVxQ7FdA,1846
 mlflow/bedrock/_autolog.py,sha256=7c2mj9pK1aj-IKCySl74g6gGxze2Bldo8bHSIZAQj5M,7800
@@ -641,8 +641,8 @@ mlflow/utils/autologging_utils/metrics_queue.py,sha256=bwpMX7Go6xFxrpYROi6rDBdVt
 mlflow/utils/autologging_utils/safety.py,sha256=IwbTbusyE87Hc4qkhhvMikoaZqX9kpr972FWS2B8goc,51465
 mlflow/utils/autologging_utils/versioning.py,sha256=2hSN4KXFWEJCcopDdLG6BiPeqSoqjETeNMuUBsCCwlI,3762
 mlflow/utils/import_hooks/__init__.py,sha256=werje98Woelkbwrhtlb8wmRdt3RtiL--LqGru7Xh3YU,13589
-genesis_flow-1.0.5.dist-info/METADATA,sha256=c2PrO48hnz6iVieC5srClmgBx4MN_61P3W_NvJvFsQE,33401
-genesis_flow-1.0.5.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-genesis_flow-1.0.5.dist-info/entry_points.txt,sha256=fwlE6mYWea9uCNIsmXdT5qMIOZ8rYbTvO6XcZR8fBbE,495
-genesis_flow-1.0.5.dist-info/top_level.txt,sha256=wm8UqYyUHI21EvrTDHb3eYICy0dOVDLBhAL-jp5zbuI,7
-genesis_flow-1.0.5.dist-info/RECORD,,
+genesis_flow-1.0.6.dist-info/METADATA,sha256=_ofi_wrmOHLBu-0gjHwRW97dMrrVpnksdsLhUr1WrlU,33401
+genesis_flow-1.0.6.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+genesis_flow-1.0.6.dist-info/entry_points.txt,sha256=YZZiTHSYQpv8vou6gdqWI17piK9Pm0P3BmZ-xepUtnw,449
+genesis_flow-1.0.6.dist-info/top_level.txt,sha256=wm8UqYyUHI21EvrTDHb3eYICy0dOVDLBhAL-jp5zbuI,7
+genesis_flow-1.0.6.dist-info/RECORD,,

{genesis_flow-1.0.5.dist-info → genesis_flow-1.0.6.dist-info}/entry_points.txt

@@ -16,4 +16,3 @@ openai = mlflow.deployments.openai
 
 [mlflow.tracking_store]
 azure-postgres = mlflow.azure.stores:create_store
-postgresql = mlflow.azure.stores:create_store

mlflow/azure/stores.py

@@ -172,22 +172,21 @@ def create_store(store_uri: str, artifact_uri: Optional[str] = None):
         "auth_method=managed_identity" in store_uri
     )
     
-    # If auth is not enabled via config and not explicitly requested in URI, skip Azure auth
+    # Check if we need to enable auth based on explicit request in URI
+    if explicit_azure_request and not config.auth_enabled:
+        # URI explicitly requests Azure auth, so enable it
+        config.auth_enabled = True
+        if not config.auth_method or config.auth_method == AuthMethod.SQL_AUTH:
+            config.auth_method = AuthMethod.MANAGED_IDENTITY
+        logger.debug("Enabled Azure auth due to explicit request in URI")
+    
+    # Log the decision for debugging
     if not config.auth_enabled and not explicit_azure_request:
-        # Log for debugging
         logger.debug(
             "Azure auth not enabled: config.auth_enabled=%s, explicit_request=%s",
             config.auth_enabled,
             explicit_azure_request
         )
-        return None
-    
-    # If we reach here, Azure auth is either enabled or explicitly requested
-    # Make sure config reflects this
-    if explicit_azure_request and not config.auth_enabled:
-        config.auth_enabled = True
-        if not config.auth_method or config.auth_method == AuthMethod.SQL_AUTH:
-            config.auth_method = AuthMethod.MANAGED_IDENTITY
     
     if config.should_use_azure_auth:
         logger.info("Creating Azure-enabled tracking store via plugin: auth_method=%s", config.auth_method.value)