add files

R/asa_api_helpers.R

@@ -131,6 +131,42 @@ asa_api_apply_env_defaults <- function() {
   }
 }
 
+asa_api_get_asa_namespace_function <- function(name, optional = FALSE) {
+  if (!requireNamespace("asa", quietly = TRUE)) {
+    stop("Package `asa` is not installed in this environment.", call. = FALSE)
+  }
+
+  exports <- tryCatch(getNamespaceExports("asa"), error = function(e) character(0))
+  if (name %in% exports) {
+    return(getExportedValue("asa", name))
+  }
+
+  asa_ns <- asNamespace("asa")
+  if (exists(name, envir = asa_ns, inherits = FALSE)) {
+    return(get(name, envir = asa_ns, inherits = FALSE))
+  }
+
+  if (isTRUE(optional)) {
+    return(NULL)
+  }
+
+  stop(
+    sprintf(
+      "Package `asa` does not provide `%s`. Upgrade `asa` or disable direct-provider mode.",
+      name
+    ),
+    call. = FALSE
+  )
+}
+
+asa_api_get_run_direct_task <- function(optional = FALSE) {
+  asa_api_get_asa_namespace_function("run_direct_task", optional = optional)
+}
+
+asa_api_has_run_direct_task <- function() {
+  !is.null(asa_api_get_run_direct_task(optional = TRUE))
+}
+
 asa_api_parse_proxy_url <- function(proxy_url) {
   proxy_url <- trimws(asa_api_scalar_chr(proxy_url, default = ""))
   if (!nzchar(proxy_url)) {
@@ -383,12 +419,12 @@ asa_api_build_run_args <- function(payload) {
   asa_api_filter_formals(asa::run_task, run)
 }
 
-asa_api_build_direct_args <- function(payload) {
+asa_api_build_direct_args <- function(payload, run_direct_task_fun = asa_api_get_run_direct_task()) {
   direct <- asa_api_build_run_args(payload)
   direct$config <- NULL
   direct$agent <- NULL
 
-  asa_api_filter_formals(asa::run_direct_task, direct)
+  asa_api_filter_formals(run_direct_task_fun, direct)
 }
 
 asa_api_build_batch_args <- function(payload) {
@@ -544,14 +580,15 @@ asa_api_run_single_via_asa <- function(payload) {
 
 asa_api_run_single_via_direct <- function(payload) {
   asa_api_apply_env_defaults()
+  run_direct_task_fun <- asa_api_get_run_direct_task()
   prompt <- asa_api_require_prompt(payload)
   config <- asa_api_build_config(payload)
-  direct_args <- asa_api_build_direct_args(payload)
+  direct_args <- asa_api_build_direct_args(payload, run_direct_task_fun = run_direct_task_fun)
 
   args <- c(list(prompt = prompt, config = config), direct_args)
-  args <- asa_api_filter_formals(asa::run_direct_task, args)
+  args <- asa_api_filter_formals(run_direct_task_fun, args)
 
-  result <- do.call(asa::run_direct_task, args)
+  result <- do.call(run_direct_task_fun, args)
   include_raw_output <- asa_api_to_bool(payload$include_raw_output, default = FALSE)
   include_trace_json <- asa_api_to_bool(payload$include_trace_json, default = FALSE)
 
@@ -613,6 +650,7 @@ asa_api_run_batch <- function(payload) {
 
 asa_api_health_payload <- function(boot_error = NULL) {
   asa_installed <- requireNamespace("asa", quietly = TRUE)
+  direct_provider_available <- asa_installed && isTRUE(asa_api_has_run_direct_task())
   has_boot_error <- is.character(boot_error) && nzchar(trimws(boot_error))
   tor_health <- asa_api_tor_health()
   healthy <- asa_installed && !has_boot_error && (!isTRUE(tor_health$tor_enabled) || isTRUE(tor_health$tor_ready))
@@ -622,6 +660,7 @@ asa_api_health_payload <- function(boot_error = NULL) {
     service = "asa-api",
     time_utc = format(Sys.time(), tz = "UTC", usetz = TRUE),
     asa_installed = asa_installed,
+    direct_provider_available = direct_provider_available,
     boot_error = if (has_boot_error) boot_error else NULL,
     defaults = list(
       backend = getOption("asa.default_backend", NULL),

R/plumber.R

@@ -45,6 +45,9 @@ asa_api_error_payload <- function(res, status, message) {
 }
 
 asa_api_error_status <- function(message) {
+  if (grepl("direct-provider mode|does not provide `run_direct_task`", message, ignore.case = TRUE)) {
+    return(501L)
+  }
   if (grepl("invalid json|required|must be|non-empty|unauthorized|password", message, ignore.case = TRUE)) {
     return(400L)
   }

Tests/api_contract_smoke.R

@@ -105,6 +105,17 @@ assert_true(
   "Batch-compatible shared options should still flow into run_task_batch."
 )
 
+assert_true(
+  identical(asa_api_has_run_direct_task(), !is.null(asa_api_get_run_direct_task(optional = TRUE))),
+  "Direct-provider capability checks should agree on run_direct_task availability."
+)
+
+health_payload <- asa_api_health_payload()
+assert_true(
+  identical(isTRUE(health_payload$direct_provider_available), isTRUE(asa_api_has_run_direct_task())),
+  "Health payload should report direct-provider availability from the same capability check."
+)
+
 {
   original_asa <- asa_api_run_single_via_asa
   original_direct <- asa_api_run_single_via_direct

www/index.html

@@ -324,6 +324,7 @@
         <input id="use-direct-provider" type="checkbox" />
         <span>Use direct provider call (skip ASA)</span>
       </label>
+      <p id="direct-provider-note" class="hint" hidden></p>
 
       <div class="actions">
         <button id="submit-btn" type="button">Run Query</button>
@@ -358,6 +359,7 @@
     const promptInput = document.getElementById("prompt");
     const outputFormatInput = document.getElementById("output-format");
     const useDirectProviderInput = document.getElementById("use-direct-provider");
+    const directProviderNoteEl = document.getElementById("direct-provider-note");
     const outputPre = document.getElementById("output");
     const statusEl = document.getElementById("status");
     const modeBadgeEl = document.getElementById("mode-badge");
@@ -376,6 +378,42 @@
       modeBadgeEl.textContent = label || "Mode: not run yet";
     }
 
+    function setDirectProviderAvailability(available, note) {
+      if (available) {
+        useDirectProviderInput.disabled = false;
+        directProviderNoteEl.hidden = true;
+        directProviderNoteEl.textContent = "";
+        return;
+      }
+
+      useDirectProviderInput.checked = false;
+      useDirectProviderInput.disabled = true;
+      directProviderNoteEl.hidden = false;
+      directProviderNoteEl.textContent = note || "Direct provider mode requires an asa build that provides run_direct_task.";
+    }
+
+    async function loadCapabilities() {
+      try {
+        const response = await fetch("/healthz", {
+          headers: {
+            "Accept": "application/json"
+          }
+        });
+        if (!response.ok) {
+          return;
+        }
+
+        const data = await response.json();
+        if (data && data.direct_provider_available === false) {
+          setDirectProviderAvailability(false);
+        } else if (data && data.direct_provider_available === true) {
+          setDirectProviderAvailability(true);
+        }
+      } catch (error) {
+        // Keep the GUI usable even if health probing fails.
+      }
+    }
+
     async function runQuery() {
       const password = passwordInput.value || "";
       const prompt = promptInput.value || "";
@@ -443,6 +481,8 @@
         runQuery();
       }
     });
+
+    loadCapabilities();
   </script>
 </body>
 </html>