dao-ai 0.1.14__py3-none-any.whl → 0.1.16__py3-none-any.whl

dao_ai/cli.py

@@ -63,16 +63,26 @@ def detect_cloud_provider(profile: Optional[str] = None) -> Optional[str]:
         Cloud provider string ('azure', 'aws', 'gcp') or None if detection fails
     """
     try:
+        import os
         from databricks.sdk import WorkspaceClient
 
+        # Check for environment variables that might override profile
+        if profile and os.environ.get("DATABRICKS_HOST"):
+            logger.warning(
+                f"DATABRICKS_HOST environment variable is set, which may override --profile {profile}"
+            )
+
         # Create workspace client with optional profile
         if profile:
+            logger.debug(f"Creating WorkspaceClient with profile: {profile}")
             w = WorkspaceClient(profile=profile)
         else:
+            logger.debug("Creating WorkspaceClient with default/ambient credentials")
             w = WorkspaceClient()
 
         # Get the workspace URL from config
         host = w.config.host
+        logger.debug(f"WorkspaceClient host: {host}, profile used: {profile}")
         if not host:
             logger.warning("Could not determine workspace URL for cloud detection")
             return None
@@ -1143,7 +1153,7 @@ def run_databricks_command(
     app_config: AppConfig = AppConfig.from_file(config_path) if config_path else None
     normalized_name: str = normalize_name(app_config.app.name) if app_config else None
 
-    # Auto-detect cloud provider if not specified
+    # Auto-detect cloud provider if not specified (used for node_type selection)
     if not cloud:
         cloud = detect_cloud_provider(profile)
         if cloud:
@@ -1156,10 +1166,12 @@ def run_databricks_command(
     if config_path and app_config:
         generate_bundle_from_template(config_path, normalized_name)
 
-    # Use cloud as target (azure, aws, gcp) - can be overridden with explicit --target
+    # Use app-specific cloud target: {app_name}-{cloud}
+    # This ensures each app has unique deployment identity while supporting cloud-specific settings
+    # Can be overridden with explicit --target
     if not target:
-        target = cloud
-        logger.debug(f"Using cloud-based target: {target}")
+        target = f"{normalized_name}-{cloud}"
+        logger.info(f"Using app-specific cloud target: {target}")
 
     # Build databricks command
     # --profile is a global flag, --target is a subcommand flag for 'bundle'

dao_ai/config.py

@@ -373,11 +373,13 @@ class IsDatabricksResource(ABC, BaseModel):
         """
         from dao_ai.utils import normalize_host
 
+        logger.trace(f"workspace_client_from called", context=context, on_behalf_of_user=self.on_behalf_of_user)
+
         # Check if we have headers in context for OBO
         if context and context.headers and self.on_behalf_of_user:
             headers = context.headers
             # Try both lowercase and title-case header names (HTTP headers are case-insensitive)
-            forwarded_token = headers.get("x-forwarded-access-token") or headers.get(
+            forwarded_token: str = headers.get("x-forwarded-access-token") or headers.get(
                 "X-Forwarded-Access-Token"
             )
 
@@ -2088,9 +2090,10 @@ class McpFunctionModel(BaseFunctionModel, IsDatabricksResource):
         if self.sql:
             return f"{workspace_host}/api/2.0/mcp/sql"
 
-        # Databricks App
+        # Databricks App - MCP endpoint is at {app_url}/mcp
         if self.app:
-            return self.app.url
+            app_url = self.app.url.rstrip("/")
+            return f"{app_url}/mcp"
 
         # Vector Search
         if self.vector_search:

{dao_ai-0.1.14.dist-info → dao_ai-0.1.16.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: dao-ai
-Version: 0.1.14
+Version: 0.1.16
 Summary: DAO AI: A modular, multi-agent orchestration framework for complex AI workflows. Supports agent handoff, tool integration, and dynamic configuration via YAML.
 Project-URL: Homepage, https://github.com/natefleming/dao-ai
 Project-URL: Documentation, https://natefleming.github.io/dao-ai

{dao_ai-0.1.14.dist-info → dao_ai-0.1.16.dist-info}/RECORD

@@ -1,7 +1,7 @@
 dao_ai/__init__.py,sha256=18P98ExEgUaJ1Byw440Ct1ty59v6nxyWtc5S6Uq2m9Q,1062
 dao_ai/catalog.py,sha256=sPZpHTD3lPx4EZUtIWeQV7VQM89WJ6YH__wluk1v2lE,4947
-dao_ai/cli.py,sha256=1Ox8qjLKRlrKu2YXozm0lWoeZnDCouECeZSGVPkQgIQ,50923
-dao_ai/config.py,sha256=7MDuX7xGSyDuBpdFZbKNDUPuTiuVe9onnUEGFtDI0jc,130123
+dao_ai/cli.py,sha256=6qwlS07_Tei6iEPXIJ-19cQVnLXd7vJDpuY4Qu0k96E,51634
+dao_ai/config.py,sha256=9RcyCNiLg0YlARZd-ZJ9M4oirNX5emxaIS23VHca_Z4,130328
 dao_ai/graph.py,sha256=1-uQlo7iXZQTT3uU8aYu0N5rnhw5_g_2YLwVsAs6M-U,1119
 dao_ai/logging.py,sha256=lYy4BmucCHvwW7aI3YQkQXKJtMvtTnPDu9Hnd7_O4oc,1556
 dao_ai/messages.py,sha256=4ZBzO4iFdktGSLrmhHzFjzMIt2tpaL-aQLHOQJysGnY,6959
@@ -68,8 +68,8 @@ dao_ai/tools/sql.py,sha256=FG-Aa0FAUAnhCuZvao1J-y-cMM6bU5eCujNbsYn0xDw,7864
 dao_ai/tools/time.py,sha256=tufJniwivq29y0LIffbgeBTIDE6VgrLpmVf8Qr90qjw,9224
 dao_ai/tools/unity_catalog.py,sha256=oBlW6pH-Ne08g60QW9wVi_tyeVYDiecuNoxQbIIFmN8,16515
 dao_ai/tools/vector_search.py,sha256=LF_72vlEF6TwUjKVv6nkUetLK766l9Kl6DQQTc9ebJI,15888
-dao_ai-0.1.14.dist-info/METADATA,sha256=3cgCatKya02uIxRs9fP-P2R_GbV3DfrQ-_JsknH0kkg,16830
-dao_ai-0.1.14.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
-dao_ai-0.1.14.dist-info/entry_points.txt,sha256=Xa-UFyc6gWGwMqMJOt06ZOog2vAfygV_DSwg1AiP46g,43
-dao_ai-0.1.14.dist-info/licenses/LICENSE,sha256=YZt3W32LtPYruuvHE9lGk2bw6ZPMMJD8yLrjgHybyz4,1069
-dao_ai-0.1.14.dist-info/RECORD,,
+dao_ai-0.1.16.dist-info/METADATA,sha256=hKYbqBIQ8eNjVtpkhysLAJNS1ISg9Tuy1ek9SggxSC4,16830
+dao_ai-0.1.16.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+dao_ai-0.1.16.dist-info/entry_points.txt,sha256=Xa-UFyc6gWGwMqMJOt06ZOog2vAfygV_DSwg1AiP46g,43
+dao_ai-0.1.16.dist-info/licenses/LICENSE,sha256=YZt3W32LtPYruuvHE9lGk2bw6ZPMMJD8yLrjgHybyz4,1069
+dao_ai-0.1.16.dist-info/RECORD,,