badgr-cli 1.0.45 → 1.0.47

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "badgr-cli",
-  "version": "1.0.45",
+  "version": "1.0.47",
   "description": "Badgr — run or serve GPU workloads from one command",
   "type": "module",
   "bin": {
@@ -14,7 +14,9 @@
   "dependencies": {
     "@inquirer/prompts": "^8.5.2",
     "archiver": "^7.0.1",
-    "chalk": "^5.3.0"
+    "chalk": "^5.3.0",
+    "js-yaml": "^4.1.0",
+    "tar": "^7.4.3"
   },
   "devDependencies": {
     "vitest": "^4.1.8"

package/src/api.js

@@ -97,6 +97,36 @@ export async function callApi(path, { method = 'GET', apiKey, baseUrl, body, tim
   return res.json();
 }
 
+// ---- Uploads (POST /v1/uploads — generic blob store) ------------------------
+
+/**
+ * Upload a blob (project zip, dataset, input archive, ...) to the generic
+ * /v1/uploads store. Returns the parsed JSON response ({ upload_id,
+ * code_uri, expires_in }) — callers pick whichever field they need.
+ *
+ * Uses Node's built-in FormData/Blob/fetch (Node >=18) rather than a
+ * multipart HTTP library — no extra dependency, and fetch derives the
+ * multipart boundary/Content-Type from the FormData body automatically
+ * (setting Content-Type manually breaks the boundary).
+ */
+export async function uploadBlob(config, { data, filename, contentType }) {
+  const form = new FormData();
+  const blob = contentType ? new Blob([data], { type: contentType }) : new Blob([data]);
+  form.append('file', blob, filename);
+
+  const baseUrl = config.baseUrl.replace(/\/v1\/?$/, '');
+  const res = await fetch(`${baseUrl}/v1/uploads`, {
+    method: 'POST',
+    body: form,
+    headers: { Authorization: `Bearer ${config.apiKey}` },
+  });
+  if (!res.ok) {
+    const text = await res.text().catch(() => '');
+    throw new Error(`Upload failed: ${res.status} ${res.statusText}${text ? ` — ${text}` : ''}`);
+  }
+  return res.json();
+}
+
 // ---- Core: run & serve (POST /v1/run, POST /v1/serve) ----------------------
 
 export function runJob(config, body) {
@@ -168,6 +198,33 @@ export async function terminateDeployment(config, deploymentId) {
   throw lastErr;
 }
 
+export function restartDeployment(config, deploymentId) {
+  return callApi(`/deployments/${deploymentId}/restart`, {
+    method: 'POST',
+    apiKey: config.apiKey,
+    baseUrl: config.baseUrl,
+    timeoutMs: 30_000,
+  });
+}
+
+export function rerunDeployment(config, deploymentId) {
+  return callApi(`/deployments/${deploymentId}/rerun`, {
+    method: 'POST',
+    apiKey: config.apiKey,
+    baseUrl: config.baseUrl,
+    timeoutMs: 30_000,
+  });
+}
+
+export function heartbeatDeployment(config, deploymentId) {
+  return callApi(`/deployments/${deploymentId}/heartbeat`, {
+    method: 'POST',
+    apiKey: config.apiKey,
+    baseUrl: config.baseUrl,
+    timeoutMs: 10_000,
+  });
+}
+
 export function getDeploymentLogs(config, deploymentId) {
   return callApi(`/deployments/${deploymentId}/logs`, {
     apiKey: config.apiKey,

package/src/badgr.js

@@ -19,8 +19,12 @@ import { transcribeCommand } from './commands/transcribe.js';
 import { embedCommand }      from './commands/embed.js';
 import { templateCommand }   from './commands/template.js';
 import { workloadCommand }   from './commands/workload.js';
+import { batchCommand }      from './commands/batch.js';
 import { workspaceCommand }  from './commands/workspace.js';
 import { detectCommand }     from './commands/detect.js';
+import { restartCommand }    from './commands/restart.js';
+import { rerunCommand }      from './commands/rerun.js';
+import { heartbeatCommand }  from './commands/heartbeat.js';
 
 const HELP = `
 ${chalk.bold('badgr')} — run or serve GPU workloads from one command
@@ -34,6 +38,9 @@ ${chalk.bold('COMMANDS')}
   ${chalk.cyan('badgr status')}              Show what's running and what's billing
   ${chalk.cyan('badgr logs <id>')}           Stream logs for a running job or endpoint
   ${chalk.cyan('badgr down <id>')}           Stop a deployment and end billing
+  ${chalk.cyan('badgr restart <id>')}        Relaunch an endpoint with the same config and API key
+  ${chalk.cyan('badgr rerun <id>')}          Replay a past job or endpoint with its exact original spec
+  ${chalk.cyan('badgr heartbeat <id>')}      Reset an endpoint's idle-timeout clock
   ${chalk.cyan('badgr receipts')}            Show cost history
   ${chalk.cyan('badgr test')}               Run an end-to-end test (provision → run → teardown)
   ${chalk.cyan('badgr capacity')}            Check what GPU capacity is available right now
@@ -42,6 +49,11 @@ ${chalk.bold('COMMANDS')}
   ${chalk.cyan('badgr workload run <name>')} Rerun a saved workload
   ${chalk.cyan('badgr workspace list')}      List workspace trackers (job history + cost per named context)
   ${chalk.cyan('badgr workspace create')}    Create a workspace tracker (link jobs to a named storage path)
+  ${chalk.cyan('badgr batch run <workload.yml>')}   Run a generic containerized batch job with artifact capture
+  ${chalk.cyan('badgr batch status <run_id>')}      Show status, failure reason, cost, teardown
+  ${chalk.cyan('badgr batch artifacts <run_id>')}   Download and extract output artifacts
+  ${chalk.cyan('badgr batch receipt <run_id>')}     Show the full batch receipt
+  ${chalk.cyan('badgr batch compare <a> <b>')}      Compare success_metric between two runs
 
 ${chalk.bold('SHORTCUTS')}  ${chalk.dim('(wrappers around run / serve for common workloads)')}
   ${chalk.cyan('badgr comfyui run <workflow.json>')}   Launch ComfyUI, return endpoint URL
@@ -89,6 +101,11 @@ ${chalk.bold('EXAMPLES')}
   ${chalk.dim('# Generate embeddings:')}
   badgr embed BAAI/bge-large-en-v1.5 documents.txt --max-cost 2
 
+  ${chalk.dim('# Generic batch workload (CV/video/scientific batch, sim, physical-AI eval):')}
+  badgr batch run workload.yml
+  badgr batch artifacts dep-abc123
+  badgr batch compare dep-abc123 dep-def456
+
   ${chalk.dim('# Tier 2 — marketplace routing, lower-cost options:')}
   badgr serve meta-llama/Llama-3.1-8B-Instruct --tier 2 --max-cost 10
 
@@ -158,6 +175,9 @@ async function main() {
     case 'status':   return statusCommand(config, rest, chalk);
     case 'logs':     return logsCommand(config, rest, chalk);
     case 'down':     return downCommand(config, rest, chalk);
+    case 'restart':   return restartCommand(config, rest, chalk);
+    case 'rerun':     return rerunCommand(config, rest, chalk);
+    case 'heartbeat': return heartbeatCommand(config, rest, chalk);
     case 'receipts': return receiptsCommand(config, rest, chalk);
     case 'models':   return modelsCommand(config, chalk);
     case 'capacity': return capacityCommand(config, rest, chalk);
@@ -169,6 +189,7 @@ async function main() {
     case 'embed':     return embedCommand(config, rest, chalk);
     case 'template':  return templateCommand(config, rest, chalk);
     case 'workload':   return workloadCommand(config, rest, chalk);
+    case 'batch':      return batchCommand(config, rest, chalk);
     case 'workspace':  return workspaceCommand(config, rest, chalk);
     // legacy aliases kept for compatibility
     case 'up':        return upCommand(config, rest, chalk);