@llmindset/hf-mcp 0.2.32 → 0.2.34

package/src/jobs/formatters.ts

@@ -0,0 +1,149 @@
+import type { JobInfo, ScheduledJobInfo } from './types.js';
+
+/**
+ * Truncate a string to a maximum length with ellipsis
+ */
+function truncate(str: string, maxLength: number): string {
+	if (str.length <= maxLength) {
+		return str;
+	}
+	return str.substring(0, maxLength - 3) + '...';
+}
+
+/**
+ * Format a date string to a readable format
+ */
+function formatDate(dateStr: string | undefined): string {
+	if (!dateStr) {
+		return 'N/A';
+	}
+	try {
+		const date = new Date(dateStr);
+		return date.toISOString().replace('T', ' ').substring(0, 19);
+	} catch {
+		return dateStr;
+	}
+}
+
+/**
+ * Format command array as a single string
+ */
+function formatCommand(command?: string[]): string {
+	if (!command || command.length === 0) {
+		return 'N/A';
+	}
+	return command.join(' ');
+}
+
+/**
+ * Get image/space identifier from job
+ */
+function getImageOrSpace(job: JobInfo | { dockerImage?: string; spaceId?: string }): string {
+	if (job.spaceId) {
+		return job.spaceId;
+	}
+	if (job.dockerImage) {
+		return job.dockerImage;
+	}
+	return 'N/A';
+}
+
+/**
+ * Format jobs as a markdown table
+ */
+export function formatJobsTable(jobs: JobInfo[]): string {
+	if (jobs.length === 0) {
+		return 'No jobs found.';
+	}
+
+	// Calculate dynamic ID column width - never truncate IDs!
+	const longestIdLength = Math.max(...jobs.map((job) => job.id.length));
+	const idColumnWidth = Math.max(longestIdLength, 'JOB ID'.length);
+
+	// Define column widths
+	const colWidths = {
+		id: idColumnWidth,
+		image: 20,
+		command: 30,
+		created: 19,
+		status: 12,
+	};
+
+	// Build header
+	const header = `| ${'JOB ID'.padEnd(colWidths.id)} | ${'IMAGE/SPACE'.padEnd(colWidths.image)} | ${'COMMAND'.padEnd(colWidths.command)} | ${'CREATED'.padEnd(colWidths.created)} | ${'STATUS'.padEnd(colWidths.status)} |`;
+	const separator = `|${'-'.repeat(colWidths.id + 2)}|${'-'.repeat(colWidths.image + 2)}|${'-'.repeat(colWidths.command + 2)}|${'-'.repeat(colWidths.created + 2)}|${'-'.repeat(colWidths.status + 2)}|`;
+
+	// Build rows
+	const rows = jobs.map((job) => {
+		const id = job.id; // Never truncate IDs!
+		const image = truncate(getImageOrSpace(job), colWidths.image);
+		const command = truncate(formatCommand(job.command), colWidths.command);
+		const created = truncate(formatDate(job.createdAt), colWidths.created);
+		const status = truncate(job.status.stage, colWidths.status);
+
+		return `| ${id.padEnd(colWidths.id)} | ${image.padEnd(colWidths.image)} | ${command.padEnd(colWidths.command)} | ${created.padEnd(colWidths.created)} | ${status.padEnd(colWidths.status)} |`;
+	});
+
+	return [header, separator, ...rows].join('\n');
+}
+
+/**
+ * Format scheduled jobs as a markdown table
+ */
+export function formatScheduledJobsTable(jobs: ScheduledJobInfo[]): string {
+	if (jobs.length === 0) {
+		return 'No scheduled jobs found.';
+	}
+
+	// Calculate dynamic ID column width - never truncate IDs!
+	const longestIdLength = Math.max(...jobs.map((job) => job.id.length));
+	const idColumnWidth = Math.max(longestIdLength, 'ID'.length);
+
+	// Define column widths
+	const colWidths = {
+		id: idColumnWidth,
+		schedule: 12,
+		image: 18,
+		command: 25,
+		lastRun: 19,
+		nextRun: 19,
+		suspend: 9,
+	};
+
+	// Build header
+	const header = `| ${'ID'.padEnd(colWidths.id)} | ${'SCHEDULE'.padEnd(colWidths.schedule)} | ${'IMAGE/SPACE'.padEnd(colWidths.image)} | ${'COMMAND'.padEnd(colWidths.command)} | ${'LAST RUN'.padEnd(colWidths.lastRun)} | ${'NEXT RUN'.padEnd(colWidths.nextRun)} | ${'SUSPENDED'.padEnd(colWidths.suspend)} |`;
+	const separator = `|${'-'.repeat(colWidths.id + 2)}|${'-'.repeat(colWidths.schedule + 2)}|${'-'.repeat(colWidths.image + 2)}|${'-'.repeat(colWidths.command + 2)}|${'-'.repeat(colWidths.lastRun + 2)}|${'-'.repeat(colWidths.nextRun + 2)}|${'-'.repeat(colWidths.suspend + 2)}|`;
+
+	// Build rows
+	const rows = jobs.map((job) => {
+		const id = job.id; // Never truncate IDs!
+		const schedule = truncate(job.schedule, colWidths.schedule);
+		const image = truncate(getImageOrSpace(job.jobSpec), colWidths.image);
+		const command = truncate(formatCommand(job.jobSpec.command), colWidths.command);
+		const lastRun = truncate(formatDate(job.lastRun), colWidths.lastRun);
+		const nextRun = truncate(formatDate(job.nextRun), colWidths.nextRun);
+		const suspend = job.suspend ? 'Yes' : 'No';
+
+		return `| ${id.padEnd(colWidths.id)} | ${schedule.padEnd(colWidths.schedule)} | ${image.padEnd(colWidths.image)} | ${command.padEnd(colWidths.command)} | ${lastRun.padEnd(colWidths.lastRun)} | ${nextRun.padEnd(colWidths.nextRun)} | ${suspend.padEnd(colWidths.suspend)} |`;
+	});
+
+	return [header, separator, ...rows].join('\n');
+}
+
+/**
+ * Format job details as JSON in a markdown code block
+ */
+export function formatJobDetails(jobs: JobInfo | JobInfo[]): string {
+	const jobArray = Array.isArray(jobs) ? jobs : [jobs];
+	const json = JSON.stringify(jobArray, null, 2);
+	return `\`\`\`json\n${json}\n\`\`\``;
+}
+
+/**
+ * Format scheduled job details as JSON in a markdown code block
+ */
+export function formatScheduledJobDetails(jobs: ScheduledJobInfo | ScheduledJobInfo[]): string {
+	const jobArray = Array.isArray(jobs) ? jobs : [jobs];
+	const json = JSON.stringify(jobArray, null, 2);
+	return `\`\`\`json\n${json}\n\`\`\``;
+}

package/src/jobs/sse-handler.ts

@@ -0,0 +1,144 @@
+import type { LogEvent } from './types.js';
+
+/**
+ * Options for fetching logs via SSE
+ */
+export interface SseLogOptions {
+	/** Maximum time to collect logs in milliseconds (default: 10000 = 10s) */
+	maxDuration?: number;
+	/** Maximum number of lines to return (default: 20) */
+	maxLines?: number;
+	/** HF API token for authentication */
+	token?: string;
+}
+
+/**
+ * Result from fetching logs
+ */
+export interface SseLogResult {
+	/** Log lines collected */
+	logs: string[];
+	/** Whether the job finished during collection */
+	finished: boolean;
+	/** Whether collection was truncated due to timeout */
+	truncated: boolean;
+}
+
+/**
+ * Fetch logs from a job via Server-Sent Events (SSE)
+ * Collects logs for a maximum duration and returns the last N lines
+ *
+ * @param url - The SSE endpoint URL for job logs
+ * @param options - Options for log collection
+ * @returns Log result with collected lines and status
+ */
+export async function fetchJobLogs(url: string, options: SseLogOptions = {}): Promise<SseLogResult> {
+	const { maxDuration = 10000, maxLines = 20, token } = options;
+
+	const logLines: string[] = [];
+	let finished = false;
+	let truncated = false;
+
+	// Create abort controller for timeout
+	const controller = new AbortController();
+	const timeoutId = setTimeout(() => {
+		controller.abort();
+		truncated = true;
+	}, maxDuration);
+
+	try {
+		const headers: Record<string, string> = {
+			Accept: 'text/event-stream',
+		};
+		if (token) {
+			headers['Authorization'] = `Bearer ${token}`;
+		}
+
+		const response = await fetch(url, {
+			headers,
+			signal: controller.signal,
+		});
+
+		if (!response.ok) {
+			throw new Error(`Failed to fetch logs: ${response.status} ${response.statusText}`);
+		}
+
+		if (!response.body) {
+			throw new Error('Response body is null');
+		}
+
+		// Process the SSE stream
+		const reader = response.body.getReader();
+		const decoder = new TextDecoder();
+		let buffer = '';
+
+		while (true) {
+			const { done, value } = await reader.read();
+
+			if (done) {
+				break;
+			}
+
+			// Decode chunk and add to buffer
+			buffer += decoder.decode(value, { stream: true });
+
+			// Process complete lines
+			const lines = buffer.split('\n');
+			buffer = lines.pop() || ''; // Keep incomplete line in buffer
+
+			for (const line of lines) {
+				// SSE format: "data: {json}"
+				if (line.startsWith('data: ')) {
+					try {
+						const jsonStr = line.substring(6); // Remove "data: " prefix
+						const event = JSON.parse(jsonStr) as LogEvent;
+
+						// Filter out system messages
+						if (event.data.startsWith('===== Job started')) {
+							continue;
+						}
+
+						// Check for job finished message
+						if (event.data.startsWith('===== Job finished')) {
+							finished = true;
+							// Extract status from message if present
+							// e.g., "===== Job finished: status=COMPLETED ====="
+							logLines.push(event.data);
+							break;
+						}
+
+						// Add log line
+						logLines.push(event.data);
+					} catch {
+						// Ignore malformed JSON
+						continue;
+					}
+				}
+			}
+
+			// Break if job finished
+			if (finished) {
+				break;
+			}
+		}
+
+		// Close the reader
+		await reader.cancel();
+	} catch (error) {
+		// If aborted due to timeout, that's expected
+		if ((error as Error).name !== 'AbortError') {
+			throw error;
+		}
+	} finally {
+		clearTimeout(timeoutId);
+	}
+
+	// Return last N lines
+	const lastLines = logLines.slice(-maxLines);
+
+	return {
+		logs: lastLines,
+		finished,
+		truncated: truncated && !finished,
+	};
+}