@barivia/barsom-mcp 0.22.2 → 0.22.4

package/dist/training_monitor_curve.js

@@ -83,3 +83,35 @@ export function lastEpochTeFromCurves(data) {
     }
     return null;
 }
+export function isTerminalJobStatus(status) {
+    return status === "completed" || status === "failed" || status === "cancelled";
+}
+/** True when compute + finalize (if any) are done — safe to open results_explorer. */
+export function isFullyComplete(data) {
+    const status = String(data.status ?? "");
+    if (status !== "completed")
+        return isTerminalJobStatus(status);
+    const finalizeId = data.finalize_job_id != null && String(data.finalize_job_id) !== ""
+        ? String(data.finalize_job_id)
+        : null;
+    if (!finalizeId)
+        return true;
+    return String(data.finalize_status ?? "") === "completed";
+}
+/** Keep polling while finalize is still running after parent compute completed. */
+export function shouldKeepPollingJob(data) {
+    const status = String(data.status ?? "");
+    if (!isTerminalJobStatus(status))
+        return true;
+    if (status !== "completed")
+        return false;
+    if (data.finalize_failed === true)
+        return false;
+    const finalizeId = data.finalize_job_id != null && String(data.finalize_job_id) !== ""
+        ? String(data.finalize_job_id)
+        : null;
+    const finalizeStatus = data.finalize_status != null && String(data.finalize_status) !== ""
+        ? String(data.finalize_status)
+        : null;
+    return finalizeId != null && finalizeStatus !== "completed";
+}

package/dist/ui-delivery.js

@@ -0,0 +1,185 @@
+/**
+ * Client-capability adaptation for MCP App tools (training_monitor, results_explorer).
+ * Respects BARIVIA_UI_DELIVERY override and emits a structured block agents can rely on.
+ */
+import { apiRawCall, getClientSupportsMcpApps, getVizPort, mimeForFilename, resetInlineAttachBudget, } from "./shared.js";
+import { buildVizStandaloneUrl } from "./viz_links.js";
+export function parseUiDeliveryOverride() {
+    const raw = (process.env.BARIVIA_UI_DELIVERY ?? "auto").trim().toLowerCase();
+    if (raw === "apps" || raw === "localhost" || raw === "inline" || raw === "auto") {
+        return raw;
+    }
+    return "auto";
+}
+function vizPortPinned() {
+    return Boolean(process.env.BARIVIA_VIZ_PORT?.trim());
+}
+function viewForTool(tool) {
+    return tool === "training_monitor" ? "training-monitor" : "results-explorer";
+}
+/** Prefer viz_links (single URL builder) so train/monitor/explorer stay consistent. */
+export function buildStandaloneVizUrl(view, jobId) {
+    return buildVizStandaloneUrl(viewForTool(view), jobId);
+}
+export function buildHealthUrl(jobId) {
+    const port = getVizPort();
+    if (!port)
+        return undefined;
+    const base = `http://localhost:${port}/api/health`;
+    return jobId ? `${base}?job_id=${encodeURIComponent(jobId)}` : base;
+}
+function hintForTier(tier, tool, opts) {
+    const label = tool === "training_monitor" ? "training monitor" : "results explorer";
+    switch (tier) {
+        case "embedded":
+            return `The ${label} MCP App panel is embedded in this client. Also mention the standalone URL if the user may open a browser tab.`;
+        case "localhost":
+            return opts.standaloneUrl
+                ? `AGENT: Post a clickable markdown link to the user (e.g. "[Open ${label}](${opts.standaloneUrl})") — do not bury it in tool output or use broken workspace file paths. Re-post after the job completes if status was running.`
+                : `AGENT: Viz server unavailable — use jobs(status) and results(get) for progress and figures.`;
+        case "inline_image":
+            return `AGENT: Inline raster figure(s) are attached below. Summarize metrics in your reply; offer results_explorer or results(get) for additional figures.`;
+        case "text_only":
+            return `AGENT: No embedded panel or localhost viz — use jobs(status) for training progress and results(get) or results_explorer for figures.`;
+    }
+}
+export function resolveUiDelivery(options) {
+    const override = parseUiDeliveryOverride();
+    const mcpApps = getClientSupportsMcpApps();
+    const port = getVizPort() || null;
+    const standaloneUrl = buildStandaloneVizUrl(options.tool, options.jobId);
+    let delivery;
+    if (override === "apps") {
+        delivery = mcpApps ? "embedded" : port ? "localhost" : "text_only";
+    }
+    else if (override === "localhost") {
+        delivery = port ? "localhost" : options.inlineImageAttached ? "inline_image" : "text_only";
+    }
+    else if (override === "inline") {
+        delivery = options.inlineImageAttached ? "inline_image" : "text_only";
+    }
+    else {
+        // auto
+        if (mcpApps) {
+            delivery = "embedded";
+        }
+        else if (port) {
+            delivery = "localhost";
+        }
+        else if (options.inlineImageAttached) {
+            delivery = "inline_image";
+        }
+        else {
+            delivery = "text_only";
+        }
+    }
+    const urls = [];
+    if (standaloneUrl && (delivery === "embedded" || delivery === "localhost")) {
+        urls.push(standaloneUrl);
+    }
+    const healthUrl = buildHealthUrl(options.jobId);
+    if (healthUrl && delivery === "localhost") {
+        urls.push(healthUrl);
+    }
+    const crossLink = options.tool === "training_monitor" && options.jobStatus === "completed"
+        ? buildStandaloneVizUrl("results_explorer", options.jobId)
+        : options.tool === "results_explorer"
+            ? buildStandaloneVizUrl("training_monitor", options.jobId)
+            : undefined;
+    if (crossLink && !urls.includes(crossLink)) {
+        urls.push(crossLink);
+    }
+    return {
+        delivery,
+        urls: urls.length > 0 ? urls : undefined,
+        hint_for_agent: hintForTier(delivery, options.tool, { standaloneUrl, jobStatus: options.jobStatus }),
+        mcp_apps_supported: mcpApps,
+        viz_port: port,
+        port_pinned: vizPortPinned(),
+        override,
+    };
+}
+/**
+ * Prepend prominent standalone URL (hosts may advertise Apps but not mount them),
+ * then structured ui_delivery JSON + agent hint.
+ */
+export function appendUiDeliveryContent(content, block, options) {
+    const primary = options?.primaryUrl ?? block.urls?.[0];
+    const label = options?.linkLabel ?? "Open interactive view";
+    const prefix = [];
+    if (primary) {
+        const embedNote = block.delivery === "embedded"
+            ? "Embedded panel if your host supports MCP Apps — otherwise open the standalone URL below (expected on some Cursor builds)."
+            : "Open the standalone URL below when no inline panel appears.";
+        prefix.push({
+            type: "text",
+            text: `${embedNote}\n` +
+                `Standalone URL (copy if Open is blocked):\n${primary}\n` +
+                `[${label}](${primary})\n` +
+                `${block.hint_for_agent}`,
+        });
+    }
+    prefix.push({
+        type: "text",
+        text: "```json\n" + JSON.stringify({ ui_delivery: block }, null, 2) + "\n```",
+    });
+    if (primary && !block.port_pinned && block.delivery !== "text_only") {
+        prefix.push({
+            type: "text",
+            text: "Tip: set BARIVIA_VIZ_PORT in MCP env for a session-stable localhost port; re-run this tool if the page stops updating.",
+        });
+    }
+    content.unshift(...prefix);
+}
+export async function tryAttachLearningCurve(content, jobId) {
+    for (const ext of ["png", "svg"]) {
+        try {
+            const filename = `learning_curve.${ext}`;
+            const { data: buf } = await apiRawCall(`/v1/results/${jobId}/image/${filename}`);
+            content.push({
+                type: "image",
+                data: buf.toString("base64"),
+                mimeType: mimeForFilename(filename),
+                annotations: { audience: ["user"], priority: 0.75 },
+            });
+            return true;
+        }
+        catch {
+            continue;
+        }
+    }
+    return false;
+}
+export function getMcpClientDiagnostics() {
+    const override = parseUiDeliveryOverride();
+    const mcpApps = getClientSupportsMcpApps();
+    const port = getVizPort() || null;
+    let activeTier = "text_only";
+    if (override === "apps") {
+        activeTier = mcpApps ? "embedded" : port ? "localhost" : "text_only";
+    }
+    else if (override === "localhost") {
+        activeTier = port ? "localhost" : "text_only";
+    }
+    else if (override === "inline") {
+        activeTier = "inline_image";
+    }
+    else if (mcpApps) {
+        activeTier = "embedded";
+    }
+    else if (port) {
+        activeTier = "localhost";
+    }
+    return {
+        mcp_apps_supported: mcpApps,
+        viz_port: port,
+        viz_port_pinned: vizPortPinned(),
+        ui_delivery_override: override,
+        active_delivery_tier: activeTier,
+        health_url: buildHealthUrl(),
+    };
+}
+/** Reset inline budget at tool entry when attaching fallback images. */
+export function beginInlineFallback() {
+    resetInlineAttachBudget();
+}