Bare ucode: dedicated message when the feature is disabled

On bare `ucode`, a disabled feature previously showed the generic
"no managed config found; using your local settings" line, because the
launch read collapses FEATURE_DISABLED to an empty result. Re-read on
the no-config path to tell the two apart and, when the feature is off,
point the developer at a specific agent (`ucode codex` / `ucode claude`).

Also promote is_feature_disabled to a public helper (now shared by cli,
managed_wizard, and managed_config).

Co-authored-by: Isaac

Author: lilly-luo

src/ucode/cli.py

@@ -44,6 +44,7 @@
     ensure_ai_gateway_v2,
     ensure_databricks_auth,
     ensure_pat_bearer,
+    fetch_managed_coding_agent_configs,
     find_profile_name_for_host,
     get_databricks_profiles,
     get_databricks_token,
@@ -63,6 +64,7 @@
 )
 from ucode.managed_config import (
     get_model_recommendation,
+    is_feature_disabled,
     load_managed_state,
     managed_agent_config_enabled,
     refresh_managed_config,
@@ -1853,6 +1855,14 @@ def _launch_managed_default(
                 "dry-run. Run `ucode` without --dry-run to pull your workspace's config first."
             )
             return
+        token = get_databricks_token(current, state.get("profile"))
+        _, reason = fetch_managed_coding_agent_configs(current, token)
+        if reason is not None and is_feature_disabled(reason):
+            print_warning(
+                "Managed coding agent config is not enabled for your workspace. Please launch "
+                "ucode with a specific agent, ex. `ucode codex` or `ucode claude`."
+            )
+            return
         _print_no_managed_config_guidance(current, state.get("profile"))
         return
     # The budget tier can move the org to a cheaper agent, so it outranks the config's

src/ucode/managed_config.py

@@ -276,9 +276,7 @@ def get_model_recommendation(workspace: str, token: str) -> tuple[dict | None, s
     """
     payload, reason = fetch_model_recommendation(workspace, token)
     if reason is not None:
-        # A disabled feature means the workspace behaves as though no config existed: fall back to
-        # the default model silently instead of warning about a budget we couldn't read.
-        if _is_feature_disabled(reason):
+        if is_feature_disabled(reason):
             return None, None
         return None, reason
     agent = AGENT_ENUM_TO_TOOL.get(_str(payload.get("recommended_agent")) or "")
@@ -314,11 +312,7 @@ def get_managed_config(workspace: str, token: str) -> tuple[dict | None, str | N
     """
     configs, reason = fetch_managed_coding_agent_configs(workspace, token)
     if reason is not None:
-        # A NOT_FOUND means the admin hasn't defined a config for this workspace — not a failure.
-        # A FEATURE_DISABLED means the feature is switched off server-side; either way the workspace
-        # has no config in effect, so both collapse to the authoritative (None, None) that clears a
-        # previously cached config rather than a warning that would reapply it.
-        if _is_not_found(reason) or _is_feature_disabled(reason):
+        if _is_not_found(reason) or is_feature_disabled(reason):
             return None, None
         return None, reason
     if not configs:
@@ -335,7 +329,7 @@ def _is_not_found(reason: str) -> bool:
     return "http 404" in lowered or "not_found" in lowered
 
 
-def _is_feature_disabled(reason: str) -> bool:
+def is_feature_disabled(reason: str) -> bool:
     """True when a read failure means the coding-agent config feature is switched off server-side.
 
     The gateway gates the feature behind a SAFE flag; when it is off, reads fail with a

src/ucode/managed_wizard.py

@@ -40,8 +40,8 @@
     update_coding_agent_config,
 )
 from ucode.managed_config import (
-    _is_feature_disabled,
     get_managed_config,
+    is_feature_disabled,
     load_managed_state,
     managed_state_workspace,
     save_managed_state,
@@ -874,7 +874,7 @@ def _require_feature_enabled(workspace: str, token: str) -> None:
     """
     with spinner("Checking whether managed coding-agent config is enabled..."):
         _, reason = fetch_managed_coding_agent_configs(workspace, token)
-    if reason is not None and _is_feature_disabled(reason):
+    if reason is not None and is_feature_disabled(reason):
         raise RuntimeError(
             f"Managed coding-agent config is not enabled for {workspace}. Reach out to your "
             "account admin to enable the Enhanced Unity AI Gateway Preview."

tests/test_managed_config.py

@@ -186,8 +186,6 @@ def test_not_found_is_treated_as_no_config(self, monkeypatch, not_found_reason):
         ],
     )
     def test_feature_disabled_is_treated_as_no_config(self, monkeypatch, disabled_reason):
-        # A FEATURE_DISABLED means the feature is switched off server-side — the workspace behaves as
-        # though no config existed, so it collapses to (None, None) and clears any cached config.
         monkeypatch.setattr(
             mc_mod,
             "fetch_managed_coding_agent_configs",
@@ -512,7 +510,6 @@ def test_failed_read_surfaces_the_reason(self, monkeypatch):
         assert mc_mod.get_model_recommendation("https://w", "tok") == (None, "HTTP 500")
 
     def test_feature_disabled_is_no_recommendation(self, monkeypatch):
-        # A disabled feature reads as no recommendation, not a failure to warn the developer about.
         self._stub(
             monkeypatch, {}, reason='HTTP 400 Bad Request: {"error_code":"FEATURE_DISABLED"}'
         )

tests/test_managed_wizard.py

@@ -227,7 +227,6 @@ def test_feature_enabled_passes(self):
             wizard._require_feature_enabled(WORKSPACE, "token")  # must not raise
 
     def test_transient_read_failure_is_allowed_through(self):
-        # A non-FEATURE_DISABLED read failure must not block setup — the API still gates writes.
         with patch.object(
             wizard,
             "fetch_managed_coding_agent_configs",