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

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: genesis-flow
-Version: 1.0.4
+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.4.dist-info → genesis_flow-1.0.6.dist-info}/RECORD

@@ -1,4 +1,4 @@
-genesis_flow-1.0.4.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=dlmdDVZRINgJw2wxzLhEsnAp1W76bR3NhQ0MftH9T74,12732
+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.4.dist-info/METADATA,sha256=0ZYCp0XfX9xral_05JfSsI8gh0jXigpbeHEANiP2EIM,33401
-genesis_flow-1.0.4.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-genesis_flow-1.0.4.dist-info/entry_points.txt,sha256=fwlE6mYWea9uCNIsmXdT5qMIOZ8rYbTvO6XcZR8fBbE,495
-genesis_flow-1.0.4.dist-info/top_level.txt,sha256=wm8UqYyUHI21EvrTDHb3eYICy0dOVDLBhAL-jp5zbuI,7
-genesis_flow-1.0.4.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.4.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

@@ -8,7 +8,7 @@ from typing import Optional, Any, Dict
 
 from sqlalchemy import Engine
 
-from mlflow.azure.config import AzureAuthConfig
+from mlflow.azure.config import AzureAuthConfig, AuthMethod
 from mlflow.azure.connection_factory import ConnectionFactory
 from mlflow.azure.exceptions import ConnectionError, ConfigurationError
 
@@ -162,25 +162,31 @@ def create_store(store_uri: str, artifact_uri: Optional[str] = None):
     # Check if Azure authentication should be used
     config = AzureAuthConfig()
     
-    # Auto-detect Azure PostgreSQL and enable auth if credentials are available
-    is_azure_postgres = (
+    # IMPORTANT: Do NOT auto-detect Azure PostgreSQL based on hostname
+    # Only use Azure auth if explicitly enabled through configuration
+    # The MLFLOW_AZURE_AUTH_ENABLED flag should be the single source of truth
+    
+    # Check if URI explicitly requests Azure auth (via scheme or parameter)
+    explicit_azure_request = (
         store_uri.startswith("azure-postgres://") or
-        ".postgres.database.azure.com" in store_uri or
         "auth_method=managed_identity" in store_uri
     )
     
-    # For Azure PostgreSQL, enable auth if we have any Azure credentials
-    if is_azure_postgres and not config.auth_enabled:
-        # Enable auth automatically if we have Azure credentials
-        if config.client_id or config.client_secret or os.getenv("AZURE_CLIENT_ID"):
-            config.auth_enabled = True
-            if not config.auth_method or config.auth_method == config.auth_method.SQL_AUTH:
-                # Default to service principal if we have credentials, managed identity otherwise  
-                config.auth_method = (
-                    config.auth_method.SERVICE_PRINCIPAL 
-                    if config.client_secret or os.getenv("AZURE_CLIENT_SECRET") 
-                    else config.auth_method.MANAGED_IDENTITY
-                )
+    # 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:
+        logger.debug(
+            "Azure auth not enabled: config.auth_enabled=%s, explicit_request=%s",
+            config.auth_enabled,
+            explicit_azure_request
+        )
     
     if config.should_use_azure_auth:
         logger.info("Creating Azure-enabled tracking store via plugin: auth_method=%s", config.auth_method.value)