dao-ai 0.0.34__py3-none-any.whl → 0.0.35__py3-none-any.whl

dao_ai/config.py

@@ -1192,21 +1192,26 @@ class McpFunctionModel(BaseFunctionModel, HasFullName):
         Otherwise, falls back to the default Databricks host.
 
         Returns:
-            str: The workspace host URL without trailing slash
+            str: The workspace host URL with https:// scheme and without trailing slash
         """
-        from dao_ai.utils import get_default_databricks_host
+        from dao_ai.utils import get_default_databricks_host, normalize_host
 
         # Try to get workspace_host from config
         workspace_host: str | None = (
-            value_of(self.workspace_host) if self.workspace_host else None
+            normalize_host(value_of(self.workspace_host))
+            if self.workspace_host
+            else None
         )
 
         # If no workspace_host in config, get it from workspace client
         if not workspace_host:
             # Use connection's workspace client if available
             if self.connection:
-                workspace_host = self.connection.workspace_client.config.host
+                workspace_host = normalize_host(
+                    self.connection.workspace_client.config.host
+                )
             else:
+                # get_default_databricks_host already normalizes the host
                 workspace_host = get_default_databricks_host()
 
         if not workspace_host:

dao_ai/providers/databricks.py

@@ -73,6 +73,7 @@ from dao_ai.utils import (
     get_installed_packages,
     is_installed,
     is_lib_provided,
+    normalize_host,
     normalize_name,
 )
 from dao_ai.vector_search import endpoint_exists, index_exists
@@ -94,15 +95,18 @@ def _workspace_client(
     Create a WorkspaceClient instance with the provided parameters.
     If no parameters are provided, it will use the default configuration.
     """
-    if client_id and client_secret and workspace_host:
+    # Normalize the workspace host to ensure it has https:// scheme
+    normalized_host = normalize_host(workspace_host)
+
+    if client_id and client_secret and normalized_host:
         return WorkspaceClient(
-            host=workspace_host,
+            host=normalized_host,
             client_id=client_id,
             client_secret=client_secret,
             auth_type="oauth-m2m",
         )
     elif pat:
-        return WorkspaceClient(host=workspace_host, token=pat, auth_type="pat")
+        return WorkspaceClient(host=normalized_host, token=pat, auth_type="pat")
     else:
         return WorkspaceClient()
 
@@ -117,15 +121,18 @@ def _vector_search_client(
     Create a VectorSearchClient instance with the provided parameters.
     If no parameters are provided, it will use the default configuration.
     """
-    if client_id and client_secret and workspace_host:
+    # Normalize the workspace host to ensure it has https:// scheme
+    normalized_host = normalize_host(workspace_host)
+
+    if client_id and client_secret and normalized_host:
         return VectorSearchClient(
-            workspace_url=workspace_host,
+            workspace_url=normalized_host,
             service_principal_client_id=client_id,
             service_principal_client_secret=client_secret,
         )
-    elif pat and workspace_host:
+    elif pat and normalized_host:
         return VectorSearchClient(
-            workspace_url=workspace_host,
+            workspace_url=normalized_host,
             personal_access_token=pat,
         )
     else:

dao_ai/tools/unity_catalog.py

@@ -15,6 +15,7 @@ from dao_ai.config import (
     value_of,
 )
 from dao_ai.tools.human_in_the_loop import as_human_in_the_loop
+from dao_ai.utils import normalize_host
 
 
 def create_uc_tools(
@@ -299,9 +300,11 @@ def with_partial_args(
             if "client_secret" not in resolved_args:
                 resolved_args["client_secret"] = value_of(sp.client_secret)
 
-    # Normalize host/workspace_host - accept either key
+    # Normalize host/workspace_host - accept either key, ensure https:// scheme
     if "workspace_host" in resolved_args and "host" not in resolved_args:
-        resolved_args["host"] = resolved_args.pop("workspace_host")
+        resolved_args["host"] = normalize_host(resolved_args.pop("workspace_host"))
+    elif "host" in resolved_args:
+        resolved_args["host"] = normalize_host(resolved_args["host"])
 
     # Default host from WorkspaceClient if not provided
     if "host" not in resolved_args:

dao_ai/tools/vector_search.py

@@ -20,6 +20,7 @@ from dao_ai.config import (
     RetrieverModel,
     VectorStoreModel,
 )
+from dao_ai.utils import normalize_host
 
 
 def create_vector_search_tool(
@@ -108,8 +109,9 @@ def create_vector_search_tool(
     # The workspace_client parameter in DatabricksVectorSearch is only used to detect
     # model serving mode - it doesn't pass credentials to VectorSearchClient.
     client_args: dict[str, Any] = {}
-    if os.environ.get("DATABRICKS_HOST"):
-        client_args["workspace_url"] = os.environ.get("DATABRICKS_HOST")
+    databricks_host = normalize_host(os.environ.get("DATABRICKS_HOST"))
+    if databricks_host:
+        client_args["workspace_url"] = databricks_host
     if os.environ.get("DATABRICKS_TOKEN"):
         client_args["personal_access_token"] = os.environ.get("DATABRICKS_TOKEN")
     if os.environ.get("DATABRICKS_CLIENT_ID"):

dao_ai/utils.py

@@ -38,6 +38,30 @@ def normalize_name(name: str) -> str:
     return normalized.strip("_")
 
 
+def normalize_host(host: str | None) -> str | None:
+    """Ensure host URL has https:// scheme.
+
+    The DATABRICKS_HOST environment variable should always include the https://
+    scheme, but some environments (e.g., Databricks Apps infrastructure) may
+    provide the host without it. This function normalizes the host to ensure
+    it has the proper scheme.
+
+    Args:
+        host: The host URL, with or without scheme
+
+    Returns:
+        The host URL with https:// scheme, or None if host is None/empty
+    """
+    if not host:
+        return None
+    host = host.strip()
+    if not host:
+        return None
+    if not host.startswith("http://") and not host.startswith("https://"):
+        return f"https://{host}"
+    return host
+
+
 def get_default_databricks_host() -> str | None:
     """Get the default Databricks workspace host.
 
@@ -46,19 +70,19 @@ def get_default_databricks_host() -> str | None:
     2. WorkspaceClient ambient authentication (e.g., from ~/.databrickscfg)
 
     Returns:
-        The Databricks workspace host URL, or None if not available.
+        The Databricks workspace host URL (with https:// scheme), or None if not available.
     """
     # Try environment variable first
     host: str | None = os.environ.get("DATABRICKS_HOST")
     if host:
-        return host
+        return normalize_host(host)
 
     # Fall back to WorkspaceClient
     try:
         from databricks.sdk import WorkspaceClient
 
         w: WorkspaceClient = WorkspaceClient()
-        return w.config.host
+        return normalize_host(w.config.host)
     except Exception:
         logger.debug("Could not get default Databricks host from WorkspaceClient")
         return None

{dao_ai-0.0.34.dist-info → dao_ai-0.0.35.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: dao-ai
-Version: 0.0.34
+Version: 0.0.35
 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.0.34.dist-info → dao_ai-0.0.35.dist-info}/RECORD

@@ -3,7 +3,7 @@ dao_ai/agent_as_code.py,sha256=sviZQV7ZPxE5zkZ9jAbfegI681nra5i8yYxw05e3X7U,552
 dao_ai/catalog.py,sha256=sPZpHTD3lPx4EZUtIWeQV7VQM89WJ6YH__wluk1v2lE,4947
 dao_ai/chat_models.py,sha256=uhwwOTeLyHWqoTTgHrs4n5iSyTwe4EQcLKnh3jRxPWI,8626
 dao_ai/cli.py,sha256=gq-nsapWxDA1M6Jua3vajBvIwf0Oa6YLcB58lEtMKUo,22503
-dao_ai/config.py,sha256=Jzb0ePrt2TM2WuXI_LtmTafbseKBlJ8J8J2ExyBowbM,79491
+dao_ai/config.py,sha256=57oObqVD7OHYJ1ZWHFzSgYa1EOh6tGiJ_nuKl5_-iCc,79700
 dao_ai/graph.py,sha256=9kjJx0oFZKq5J9-Kpri4-0VCJILHYdYyhqQnj0_noxQ,8913
 dao_ai/guardrails.py,sha256=4TKArDONRy8RwHzOT1plZ1rhy3x9GF_aeGpPCRl6wYA,4016
 dao_ai/messages.py,sha256=xl_3-WcFqZKCFCiov8sZOPljTdM3gX3fCHhxq-xFg2U,7005
@@ -12,7 +12,7 @@ dao_ai/nodes.py,sha256=iQ_5vL6mt1UcRnhwgz-l1D8Ww4CMQrSMVnP_Lu7fFjU,8781
 dao_ai/prompts.py,sha256=iA2Iaky7yzjwWT5cxg0cUIgwo1z1UVQua__8WPnvV6g,1633
 dao_ai/state.py,sha256=_lF9krAYYjvFDMUwZzVKOn0ZnXKcOrbjWKdre0C5B54,1137
 dao_ai/types.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-dao_ai/utils.py,sha256=oIPmz02kZ3LMntbqxUajFXh4nswOhbvEjOTi4e5_cvI,8500
+dao_ai/utils.py,sha256=6gmcu2cu4WE2wfrVj45auPxISJ2y0lmyFZisbilB4kQ,9297
 dao_ai/vector_search.py,sha256=jlaFS_iizJ55wblgzZmswMM3UOL-qOp2BGJc0JqXYSg,2839
 dao_ai/hooks/__init__.py,sha256=LlHGIuiZt6vGW8K5AQo1XJEkBP5vDVtMhq0IdjcLrD4,417
 dao_ai/hooks/core.py,sha256=ZShHctUSoauhBgdf1cecy9-D7J6-sGn-pKjuRMumW5U,6663
@@ -22,7 +22,7 @@ dao_ai/memory/core.py,sha256=DnEjQO3S7hXr3CDDd7C2eE7fQUmcCS_8q9BXEgjPH3U,4271
 dao_ai/memory/postgres.py,sha256=vvI3osjx1EoU5GBA6SCUstTBKillcmLl12hVgDMjfJY,15346
 dao_ai/providers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 dao_ai/providers/base.py,sha256=-fjKypCOk28h6vioPfMj9YZSw_3Kcbi2nMuAyY7vX9k,1383
-dao_ai/providers/databricks.py,sha256=WEigNPGRTlIPVjwp97My8o1zOHn5ftuMsMrpqrBeaLg,66012
+dao_ai/providers/databricks.py,sha256=t9a3QBQ0v_14PbUtYRj2My423OskDj3VQ9b19vuAeco,66283
 dao_ai/tools/__init__.py,sha256=G5-5Yi6zpQOH53b5IzLdtsC6g0Ep6leI5GxgxOmgw7Q,1203
 dao_ai/tools/agent.py,sha256=WbQnyziiT12TLMrA7xK0VuOU029tdmUBXbUl-R1VZ0Q,1886
 dao_ai/tools/core.py,sha256=kN77fWOzVY7qOs4NiW72cUxCsSTC0DnPp73s6VJEZOQ,1991
@@ -32,10 +32,10 @@ dao_ai/tools/mcp.py,sha256=5aQoRtx2z4xm6zgRslc78rSfEQe-mfhqov2NsiybYfc,8416
 dao_ai/tools/python.py,sha256=XcQiTMshZyLUTVR5peB3vqsoUoAAy8gol9_pcrhddfI,1831
 dao_ai/tools/slack.py,sha256=SCvyVcD9Pv_XXPXePE_fSU1Pd8VLTEkKDLvoGTZWy2Y,4775
 dao_ai/tools/time.py,sha256=Y-23qdnNHzwjvnfkWvYsE7PoWS1hfeKy44tA7sCnNac,8759
-dao_ai/tools/unity_catalog.py,sha256=K9t8M4spsbxbecWmV5yEZy16s_AG7AfaoxT-7IDW43I,14438
-dao_ai/tools/vector_search.py,sha256=3cdiUaFpox25GSRNec7FKceY3DuLp7dLVH8FRA0BgeY,12624
-dao_ai-0.0.34.dist-info/METADATA,sha256=vq51NEV-pg7WTOD5z56jyOrC5_6Q-nUIL51RI5lL-Hg,42761
-dao_ai-0.0.34.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
-dao_ai-0.0.34.dist-info/entry_points.txt,sha256=Xa-UFyc6gWGwMqMJOt06ZOog2vAfygV_DSwg1AiP46g,43
-dao_ai-0.0.34.dist-info/licenses/LICENSE,sha256=YZt3W32LtPYruuvHE9lGk2bw6ZPMMJD8yLrjgHybyz4,1069
-dao_ai-0.0.34.dist-info/RECORD,,
+dao_ai/tools/unity_catalog.py,sha256=eB-7lrRqtf47rQab3WPdJiuJGDQzW3mLJUc9-oCm3j8,14622
+dao_ai/tools/vector_search.py,sha256=VnDNhVbDKiG6COhX9Eh44X79yfbi3AEC2eGBvPC6gv0,12700
+dao_ai-0.0.35.dist-info/METADATA,sha256=MPVim3KsOASWHmuVjKjRQIvl_CW1LldpQVysKNjudwM,42761
+dao_ai-0.0.35.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+dao_ai-0.0.35.dist-info/entry_points.txt,sha256=Xa-UFyc6gWGwMqMJOt06ZOog2vAfygV_DSwg1AiP46g,43
+dao_ai-0.0.35.dist-info/licenses/LICENSE,sha256=YZt3W32LtPYruuvHE9lGk2bw6ZPMMJD8yLrjgHybyz4,1069
+dao_ai-0.0.35.dist-info/RECORD,,