@barivia/barsom-mcp 0.15.0 → 0.15.4

package/dist/train_submit_message.js

@@ -0,0 +1,52 @@
+import { getVizPort } from "./shared.js";
+export function buildTrainSubmitExtras(opts) {
+    const { newJobId, totalRows, hasTransforms, prepareJobId, variantPrefix, paramSummary, impute, resultsHint } = opts;
+    const monitorUrl = getVizPort() > 0
+        ? `http://localhost:${getVizPort()}/viz/training-monitor?mode=standalone&job_id=${encodeURIComponent(newJobId)}`
+        : null;
+    let msg = `Job submitted (${variantPrefix}, ${paramSummary}). `;
+    if (prepareJobId) {
+        msg += `Preprocessing job prepare_training_matrix (${prepareJobId}) runs on worker-io first; train job ${newJobId} starts after it completes. Poll jobs(action=status, job_id="${prepareJobId}") until completed, then poll the train job. `;
+    }
+    msg += `Recommended: training_monitor(job_id="${newJobId}")`;
+    if (monitorUrl)
+        msg += ` or open ${monitorUrl}`;
+    msg += ` while polling jobs(action=status, job_id="${newJobId}") every 60–120s for large jobs. `;
+    msg += `When status is completed, check jobs(action=status) for finalize_job_id — if present, poll that finalize_training job before results(action=get). `;
+    msg += `Then use results(action=get, job_id="${newJobId}")`;
+    if (resultsHint) {
+        msg += ` to ${resultsHint}`;
+    }
+    else if (impute) {
+        msg += ` for the map and imputed.csv`;
+    }
+    else {
+        msg += ` to view the map and metrics`;
+    }
+    msg += `.`;
+    if (totalRows >= 1_000_000 || hasTransforms) {
+        msg += ` Large-table note: progress may stay low during staging preprocess before epochs start.`;
+        if (hasTransforms && totalRows >= 1_000_000) {
+            if (prepareJobId) {
+                msg += ` Transforms run in prepare_training_matrix on worker-io (full dataset, no CSV re-download on GPU).`;
+            }
+            else {
+                msg += ` Transforms require prepare_training_matrix on worker-io for staged datasets; the API enqueues it automatically when needed.`;
+            }
+        }
+    }
+    const pollSteps = prepareJobId
+        ? [
+            `1) jobs(action=status, job_id="${prepareJobId}") until prepare completes`,
+            `2) jobs(action=status, job_id="${newJobId}") every 60–120s until completed`,
+            `3) if finalize_job_id is set, poll it until completed`,
+            `4) results(action=download, job_id="${newJobId}", folder=".", include_json=true)`,
+        ]
+        : [
+            `1) training_monitor(job_id="${newJobId}")`,
+            `2) jobs(action=status, job_id="${newJobId}") every 60–120s until completed`,
+            `3) if finalize_job_id is set, poll it until completed`,
+            `4) results(action=download, job_id="${newJobId}", folder=".", include_json=true)`,
+        ];
+    return { message: msg, suggested_next_step: pollSteps.join(" → ") };
+}