585 lines
22 KiB
JavaScript
585 lines
22 KiB
JavaScript
import { getProcesses, stopProcess as apiStop, deleteProcess as apiDelete } from '../services/apiClient.js';
|
||
import { showNotification } from '../ui/notify.js';
|
||
import { showConfirmDialog } from '../ui/dialog.js';
|
||
import { el, clearChildren } from '../utils/dom.js';
|
||
import { showLogs, refreshLog, clearLogTimerFor } from './logs.js';
|
||
|
||
// 计算训练进度信息
|
||
function calculateRemainingTime(current, total, logText) {
|
||
// 尝试从日志中提取时间信息
|
||
const timePatterns = [
|
||
/remaining[\s:=]\s*(\d+)[\s:]?(\d+)?[\s:]?(\d+)?/i, // remaining: 1:30:45 or remaining: 90
|
||
/ETA[\s:=]\s*(\d+):(\d+):(\d+)/i, // ETA: 1:30:45
|
||
/预计剩余[\s:=]\s*(\d+)[\s小时]*[\s:]?(\d+)?[\s分钟]*/i, // 预计剩余: 1小时30分钟
|
||
/剩余时间[\s:=]\s*(\d+)[\s小时]*[\s:]?(\d+)?[\s分钟]*/i, // 剩余时间: 1小时30分钟
|
||
/time left[\s:=]\s*(\d+)[\s:]?(\d+)?[\s:]?(\d+)?/i, // time left: 1:30:45
|
||
/还需[\s:=]\s*(\d+)[\s小时]*[\s:]?(\d+)?[\s分钟]*/i // 还需: 1小时30分钟
|
||
];
|
||
|
||
for (const pattern of timePatterns) {
|
||
const match = logText.match(pattern);
|
||
if (match) {
|
||
const hours = parseInt(match[1]) || 0;
|
||
const minutes = parseInt(match[2]) || 0;
|
||
const seconds = parseInt(match[3]) || 0;
|
||
|
||
if (hours > 0 || minutes > 0 || seconds > 0) {
|
||
const parts = [];
|
||
if (hours > 0) parts.push(`${hours}小时`);
|
||
if (minutes > 0) parts.push(`${minutes}分钟`);
|
||
if (seconds > 0 && hours === 0 && minutes === 0) parts.push(`${seconds}秒`);
|
||
|
||
return parts.join('');
|
||
}
|
||
}
|
||
}
|
||
|
||
// 如果没有找到时间信息，根据进度估算
|
||
if (current > 0 && current < total) {
|
||
const remainingEpochs = total - current;
|
||
// 假设每个epoch大约需要一定时间，这里使用简单的线性估算
|
||
// 实际应用中可以根据历史数据更准确地估算
|
||
return `约${remainingEpochs}个epoch`;
|
||
}
|
||
|
||
return '计算中...';
|
||
}
|
||
|
||
function calculateProgress(process) {
|
||
const defaultProgress = {
|
||
percentage: 0,
|
||
current: 0,
|
||
total: 0,
|
||
remaining: '计算中...',
|
||
loss: null,
|
||
epoch: null,
|
||
lr: null
|
||
};
|
||
|
||
// 如果进程不在运行，返回默认进度
|
||
if (!process.running) return defaultProgress;
|
||
|
||
// 从进程数据中提取进度信息
|
||
if (process.progress) {
|
||
return {
|
||
percentage: process.progress.percentage || 0,
|
||
current: process.progress.current_epoch || 0,
|
||
total: process.progress.total_epochs || 0,
|
||
remaining: process.progress.remaining_time || '计算中...',
|
||
loss: process.progress.current_loss || null,
|
||
epoch: process.progress.current_epoch ? `${process.progress.current_epoch}/${process.progress.total_epochs}` : null,
|
||
lr: process.progress.current_lr || null
|
||
};
|
||
}
|
||
|
||
// 尝试从日志中提取进度信息（增强版本）
|
||
if (process.logs) {
|
||
const logText = process.logs.slice(-2000); // 取最近2000字符以获取更多上下文
|
||
|
||
// 提取epoch信息 - 支持多种格式
|
||
const epochPatterns = [
|
||
/epoch\s+(\d+)\s*\/\s*(\d+)/i, // epoch 3/10
|
||
/Epoch\s+(\d+)\s*of\s*(\d+)/i, // Epoch 3 of 10
|
||
/\[(\d+)\/(\d+)\]/i, // [3/10]
|
||
/epoch\s*[:：]\s*(\d+)\s*\/\s*(\d+)/i, // epoch: 3/10
|
||
/第\s*(\d+)\s*轮\s*\/\s*共\s*(\d+)\s*轮/i // 第3轮/共10轮
|
||
];
|
||
|
||
let current = 0;
|
||
let total = 0;
|
||
let percentage = 0;
|
||
|
||
for (const pattern of epochPatterns) {
|
||
const match = logText.match(pattern);
|
||
if (match) {
|
||
current = parseInt(match[1]);
|
||
total = parseInt(match[2]);
|
||
percentage = total > 0 ? Math.round((current / total) * 100) : 0;
|
||
break;
|
||
}
|
||
}
|
||
|
||
// 提取loss信息 - 支持多种格式
|
||
const lossPatterns = [
|
||
/loss[\s:=]\s*([\d.]+(?:e[+-]?\d+)?)/i, // loss: 4.32 or loss = 4.32
|
||
/training_loss[\s:=]\s*([\d.]+(?:e[+-]?\d+)?)/i, // training_loss: 4.32
|
||
/train_loss[\s:=]\s*([\d.]+(?:e[+-]?\d+)?)/i, // train_loss: 4.32
|
||
/Loss[\s:=]\s*([\d.]+(?:e[+-]?\d+)?)/i, // Loss: 4.32
|
||
/训练损失[\s:=]\s*([\d.]+(?:e[+-]?\d+)?)/i, // 训练损失: 4.32
|
||
/损失[\s:=]\s*([\d.]+(?:e[+-]?\d+)?)/i, // 损失: 4.32
|
||
/\s+([\d.]+(?:e[+-]?\d+)?)\s*loss/i, // 4.32 loss
|
||
/\s+([\d.]+(?:e[+-]?\d+)?)\s*训练损失/i, // 4.32 训练损失
|
||
/(?:loss|损失|training_loss|train_loss)\s*=\s*([\d.]+(?:e[+-]?\d+)?)/i // loss = 4.32
|
||
];
|
||
|
||
let currentLoss = null;
|
||
for (const pattern of lossPatterns) {
|
||
const matches = [...logText.matchAll(pattern)];
|
||
if (matches.length > 0) {
|
||
// 取最后一个匹配的loss值
|
||
const lastMatch = matches[matches.length - 1];
|
||
const lossValue = parseFloat(lastMatch[1]);
|
||
if (!isNaN(lossValue) && lossValue > 0 && lossValue < 100) { // 合理的loss范围
|
||
currentLoss = lossValue.toFixed(4);
|
||
break;
|
||
}
|
||
}
|
||
}
|
||
|
||
// 提取学习率信息
|
||
const lrPatterns = [
|
||
/lr[\s:=]\s*([\d.e+-]+)/i, // lr: 1e-4
|
||
/learning_rate[\s:=]\s*([\d.e+-]+)/i, // learning_rate: 1e-4
|
||
/LR[\s:=]\s*([\d.e+-]+)/i, // LR: 1e-4
|
||
/学习率[\s:=]\s*([\d.e+-]+)/i // 学习率: 1e-4
|
||
];
|
||
|
||
let currentLr = null;
|
||
for (const pattern of lrPatterns) {
|
||
const matches = [...logText.matchAll(pattern)];
|
||
if (matches.length > 0) {
|
||
const lastMatch = matches[matches.length - 1];
|
||
const lrValue = parseFloat(lastMatch[1]);
|
||
if (!isNaN(lrValue) && lrValue > 0 && lrValue < 1) { // 合理的lr范围
|
||
currentLr = lrValue.toExponential(2);
|
||
break;
|
||
}
|
||
}
|
||
}
|
||
|
||
// 如果找到了有效的epoch信息，返回进度
|
||
if (total > 0) {
|
||
return {
|
||
percentage,
|
||
current,
|
||
total,
|
||
remaining: calculateRemainingTime(current, total, logText),
|
||
loss: currentLoss,
|
||
epoch: `${current}/${total}`,
|
||
lr: currentLr
|
||
};
|
||
}
|
||
}
|
||
|
||
return defaultProgress;
|
||
}
|
||
|
||
let processPollingTimer = null;
|
||
|
||
export function startProcessPolling() {
|
||
if (processPollingTimer) clearInterval(processPollingTimer);
|
||
// 使用2秒间隔进行实时更新，确保进度信息及时刷新
|
||
processPollingTimer = setInterval(() => {
|
||
const tab = document.querySelector('.tab.active');
|
||
if (tab && tab.textContent.includes('进程')) {
|
||
checkProcessStatusChanges();
|
||
}
|
||
}, 2000);
|
||
}
|
||
|
||
export function stopProcessPolling() {
|
||
if (processPollingTimer) {
|
||
clearInterval(processPollingTimer);
|
||
processPollingTimer = null;
|
||
}
|
||
}
|
||
|
||
export function checkProcessStatusChanges() {
|
||
return getProcesses()
|
||
.then((data) => {
|
||
let updatedCount = 0;
|
||
data.forEach((p) => {
|
||
const item = document.querySelector(`[data-process-id="${p.id}"]`);
|
||
if (!item) return;
|
||
const cur = item.dataset.processStatus;
|
||
const next = p.status;
|
||
|
||
// 如果状态发生变化，更新整个项目
|
||
if (cur !== next) {
|
||
updateProcessItem(item, p);
|
||
if (next === '出错') showNotification(`进程 ${p.train_type} 已出错`, 'error');
|
||
updatedCount++;
|
||
}
|
||
// 如果进程正在运行，即使状态没变也要更新进度信息
|
||
else if (p.running) {
|
||
updateProcessProgress(item, p);
|
||
updatedCount++;
|
||
}
|
||
});
|
||
|
||
// 调试用：在控制台显示更新信息（生产环境中可以移除）
|
||
if (updatedCount > 0) {
|
||
console.log(`[${new Date().toLocaleTimeString()}] 更新了 ${updatedCount} 个进程的进度信息`);
|
||
}
|
||
})
|
||
.catch(() => {
|
||
showNotification('连接服务器失败，请刷新页面重试', 'error');
|
||
});
|
||
}
|
||
|
||
export function loadProcesses() {
|
||
return getProcesses().then((data) => {
|
||
const list = document.getElementById('process-list');
|
||
clearChildren(list);
|
||
if (data.length === 0) {
|
||
list.innerHTML = '<p>暂无训练进程</p>';
|
||
return;
|
||
}
|
||
data.sort((a, b) => new Date(b.start_time) - new Date(a.start_time));
|
||
const groups = {};
|
||
data.forEach((p) => {
|
||
if (!groups[p.train_type]) groups[p.train_type] = [];
|
||
groups[p.train_type].push(p);
|
||
});
|
||
const order = ['pretrain', 'sft', 'lora', 'dpo'];
|
||
const types = [...order.filter((t) => groups[t]), ...Object.keys(groups).filter((t) => !order.includes(t))];
|
||
types.forEach((t) => {
|
||
const g = createTypeGroupWithToggle(t, groups[t]);
|
||
list.appendChild(g);
|
||
});
|
||
});
|
||
}
|
||
|
||
function createTypeGroupWithToggle(trainType, processes) {
|
||
const group = el('div', { class: 'process-type-group' });
|
||
const header = el('div', { class: 'process-type-header' });
|
||
header.dataset.expanded = 'true';
|
||
const title = el('h3', { class: 'process-type-title', text: getTrainTypeDisplayName(trainType) });
|
||
const toggle = el('button', { class: 'toggle-btn' });
|
||
toggle.innerHTML = '▼';
|
||
toggle.onclick = (e) => {
|
||
e.stopPropagation();
|
||
toggleGroup(header);
|
||
};
|
||
header.appendChild(title);
|
||
header.appendChild(toggle);
|
||
header.onclick = () => toggleGroup(header);
|
||
const content = el('div', { class: 'process-type-content' });
|
||
processes.forEach((p) => addProcessItemToGroup(content, p));
|
||
group.appendChild(header);
|
||
group.appendChild(content);
|
||
return group;
|
||
}
|
||
|
||
function toggleGroup(header) {
|
||
const expanded = header.dataset.expanded === 'true';
|
||
const content = header.nextElementSibling;
|
||
const toggle = header.querySelector('.toggle-btn');
|
||
if (expanded) {
|
||
header.dataset.expanded = 'false';
|
||
content.style.maxHeight = '0';
|
||
content.style.overflow = 'hidden';
|
||
toggle.innerHTML = '▶';
|
||
} else {
|
||
content.style.overflow = 'hidden';
|
||
content.style.maxHeight = 'none';
|
||
const h = content.scrollHeight;
|
||
content.style.maxHeight = '0';
|
||
content.offsetHeight;
|
||
header.dataset.expanded = 'true';
|
||
content.style.maxHeight = h + 'px';
|
||
setTimeout(() => {
|
||
content.style.maxHeight = 'none';
|
||
content.style.overflow = 'visible';
|
||
}, 300);
|
||
toggle.innerHTML = '▼';
|
||
}
|
||
}
|
||
|
||
function getTrainTypeDisplayName(trainType) {
|
||
const names = {
|
||
pretrain: '预训练 (Pretrain)',
|
||
sft: '全参数监督微调 (SFT - Full)',
|
||
lora: 'LoRA监督微调 (SFT - Lora)',
|
||
dpo: '直接偏好优化 (RL - DPO)',
|
||
ppo: 'PPO',
|
||
grpo: 'GRPO',
|
||
spo: 'SPO',
|
||
};
|
||
return names[trainType] || trainType;
|
||
}
|
||
|
||
export function addProcessItemToGroup(parent, process) {
|
||
const item = el('div', { class: 'process-item' });
|
||
let statusClass = 'status-completed';
|
||
if (process.status === '运行中') statusClass = 'status-running';
|
||
else if (process.status === '手动停止') statusClass = 'status-manual-stop';
|
||
else if (process.status === '出错') statusClass = 'status-error';
|
||
item.dataset.processId = process.id;
|
||
item.dataset.processStatus = process.status;
|
||
item.dataset.trainMonitor = process.train_monitor || 'none';
|
||
item.dataset.swanlabUrl = process.swanlab_url || '';
|
||
const showDelete = !process.running;
|
||
const showSwanlab = process.train_monitor !== 'none';
|
||
const swanBtn = showSwanlab ? `<button class="btn-swanlab" data-swan="${process.id}">SwanLab</button>` : '';
|
||
|
||
// 计算进度信息
|
||
const progressInfo = calculateProgress(process);
|
||
const progressBar = process.running ? `
|
||
<div class="progress-container">
|
||
<div class="progress-bar">
|
||
<div class="progress-fill" style="width: ${progressInfo.percentage}%"></div>
|
||
</div>
|
||
<div class="progress-info">
|
||
<span>进度: ${progressInfo.current}/${progressInfo.total}</span>
|
||
<span>剩余时间: ${progressInfo.remaining}</span>
|
||
</div>
|
||
<div class="progress-metrics">
|
||
${progressInfo.loss ? `<div class="metric-item"><span class="metric-label">Loss:</span><span class="metric-value">${progressInfo.loss}</span></div>` : ''}
|
||
${progressInfo.epoch ? `<div class="metric-item"><span class="metric-label">Epoch:</span><span class="metric-value">${progressInfo.epoch}</span></div>` : ''}
|
||
${progressInfo.lr ? `<div class="metric-item"><span class="metric-label">LR:</span><span class="metric-value">${progressInfo.lr}</span></div>` : ''}
|
||
</div>
|
||
</div>
|
||
` : '';
|
||
|
||
item.innerHTML = `
|
||
<div class="process-info">
|
||
<div><strong>${process.start_time}</strong></div>
|
||
<div><span class="process-status ${statusClass}">${process.status}</span></div>
|
||
</div>
|
||
${progressBar}
|
||
<div>
|
||
<button class="btn-logs" data-show="${process.id}">查看日志</button>
|
||
<button class="btn-logs" data-refresh="${process.id}">刷新日志</button>
|
||
${swanBtn}
|
||
${process.running ? `<button class="btn-stop" data-stop="${process.id}">停止训练</button>` : ''}
|
||
${showDelete ? `<button class="btn-delete" data-del="${process.id}">删除</button>` : ''}
|
||
</div>
|
||
<div id="logs-${process.id}" class="logs-container hidden"></div>
|
||
`;
|
||
parent.appendChild(item);
|
||
bindItemButtons(item, process);
|
||
}
|
||
|
||
function bindItemButtons(item, process) {
|
||
const showBtn = item.querySelector('[data-show]');
|
||
if (showBtn) showBtn.addEventListener('click', () => showLogs(process.id));
|
||
const refreshBtn = item.querySelector('[data-refresh]');
|
||
if (refreshBtn) refreshBtn.addEventListener('click', () => refreshLog(process.id));
|
||
const swanBtn = item.querySelector('[data-swan]');
|
||
if (swanBtn) swanBtn.addEventListener('click', () => checkAndOpenSwanlab(process.id));
|
||
const stopBtn = item.querySelector('[data-stop]');
|
||
if (stopBtn) stopBtn.addEventListener('click', () => stopProcess(process.id));
|
||
const delBtn = item.querySelector('[data-del]');
|
||
if (delBtn) delBtn.addEventListener('click', () => deleteProcess(process.id));
|
||
}
|
||
|
||
export function updateProcessProgress(item, process) {
|
||
// 只更新进度信息，不更新整个项目
|
||
const progressInfo = calculateProgress(process);
|
||
|
||
// 更新进度条
|
||
const progressFill = item.querySelector('.progress-fill');
|
||
const progressText = item.querySelector('.progress-info span:first-child');
|
||
const remainingText = item.querySelector('.progress-info span:last-child');
|
||
const metricsContainer = item.querySelector('.progress-metrics');
|
||
|
||
if (progressFill) {
|
||
progressFill.style.width = `${progressInfo.percentage}%`;
|
||
}
|
||
|
||
if (progressText) {
|
||
progressText.textContent = `进度: ${progressInfo.current}/${progressInfo.total}`;
|
||
}
|
||
|
||
if (remainingText) {
|
||
remainingText.textContent = `剩余时间: ${progressInfo.remaining}`;
|
||
}
|
||
|
||
if (metricsContainer) {
|
||
// 更新指标 - 只更新有变化的值来减少DOM操作
|
||
const lossItem = metricsContainer.querySelector('.metric-item:nth-child(1) .metric-value');
|
||
const epochItem = metricsContainer.querySelector('.metric-item:nth-child(2) .metric-value');
|
||
const lrItem = metricsContainer.querySelector('.metric-item:nth-child(3) .metric-value');
|
||
|
||
if (progressInfo.loss && lossItem) {
|
||
lossItem.textContent = progressInfo.loss;
|
||
}
|
||
if (progressInfo.epoch && epochItem) {
|
||
epochItem.textContent = progressInfo.epoch;
|
||
}
|
||
if (progressInfo.lr && lrItem) {
|
||
lrItem.textContent = progressInfo.lr;
|
||
}
|
||
}
|
||
}
|
||
|
||
export function updateProcessItem(item, process) {
|
||
item.dataset.processStatus = process.status;
|
||
item.dataset.trainMonitor = process.train_monitor || 'none';
|
||
if (process.swanlab_url) item.dataset.swanlabUrl = process.swanlab_url;
|
||
const statusEl = item.querySelector('.process-status');
|
||
if (statusEl) {
|
||
statusEl.classList.remove('status-running', 'status-manual-stop', 'status-error', 'status-completed');
|
||
let cls = 'status-completed';
|
||
if (process.status === '运行中') cls = 'status-running';
|
||
else if (process.status === '手动停止') cls = 'status-manual-stop';
|
||
else if (process.status === '出错') cls = 'status-error';
|
||
statusEl.classList.add(cls);
|
||
statusEl.textContent = process.status;
|
||
}
|
||
const btnContainer = item.querySelector('div:nth-child(2)');
|
||
const existingSwan = item.querySelector('.btn-swanlab');
|
||
const showSwan = process.train_monitor !== 'none';
|
||
if (showSwan && !existingSwan && btnContainer) {
|
||
const b = el('button', { class: 'btn-swanlab' });
|
||
b.textContent = 'SwanLab';
|
||
b.onclick = () => checkAndOpenSwanlab(process.id);
|
||
const stop = btnContainer.querySelector('.btn-stop');
|
||
if (stop) btnContainer.insertBefore(b, stop);
|
||
else btnContainer.appendChild(b);
|
||
} else if (!showSwan && existingSwan) existingSwan.remove();
|
||
const stopBtn = item.querySelector('.btn-stop');
|
||
if (stopBtn) {
|
||
if (!process.running) stopBtn.remove();
|
||
} else if (process.running && btnContainer) {
|
||
const n = el('button', { class: 'btn-stop' });
|
||
n.textContent = '停止训练';
|
||
n.onclick = () => stopProcess(process.id);
|
||
btnContainer.appendChild(n);
|
||
}
|
||
const delBtn = item.querySelector('.btn-delete');
|
||
if (!process.running) {
|
||
if (!delBtn) {
|
||
const c = item.querySelector('div:last-child');
|
||
if (c) {
|
||
const d = el('button', { class: 'btn-delete' });
|
||
d.textContent = '删除';
|
||
d.onclick = () => deleteProcess(process.id);
|
||
c.appendChild(d);
|
||
}
|
||
}
|
||
} else if (delBtn) delBtn.remove();
|
||
if (!process.running) clearLogTimerFor(process.id);
|
||
}
|
||
|
||
export function deleteProcess(processId) {
|
||
showConfirmDialog('确定要删除这个训练进程吗？此操作不可恢复。', () => {
|
||
apiDelete(processId)
|
||
.then(() => {
|
||
const item = document.querySelector(`[data-process-id="${processId}"]`);
|
||
if (item && item.parentNode) {
|
||
item.style.transition = 'opacity 0.3s, transform 0.3s';
|
||
item.style.opacity = '0';
|
||
item.style.transform = 'translateX(-20px)';
|
||
setTimeout(() => {
|
||
const content = item.closest('.process-type-content');
|
||
const group = content ? content.closest('.process-type-group') : null;
|
||
item.parentNode.removeChild(item);
|
||
if (content) {
|
||
const remain = content.querySelectorAll('.process-item');
|
||
if (remain.length === 0 && group) {
|
||
setTimeout(() => {
|
||
group.style.transition = 'opacity 0.3s, transform 0.3s';
|
||
group.style.opacity = '0';
|
||
group.style.transform = 'translateY(-10px)';
|
||
setTimeout(() => {
|
||
if (group.parentNode) group.parentNode.removeChild(group);
|
||
const left = document.querySelectorAll('.process-item');
|
||
if (left.length === 0) {
|
||
const list = document.getElementById('process-list');
|
||
list.innerHTML = '<p>暂无训练进程</p>';
|
||
}
|
||
}, 300);
|
||
}, 100);
|
||
} else {
|
||
const header = content.previousElementSibling;
|
||
if (header && header.dataset.expanded === 'true') content.style.maxHeight = content.scrollHeight + 'px';
|
||
const left = document.querySelectorAll('.process-item');
|
||
if (left.length === 0) {
|
||
const list = document.getElementById('process-list');
|
||
list.innerHTML = '<p>暂无训练进程</p>';
|
||
}
|
||
}
|
||
}
|
||
}, 300);
|
||
}
|
||
clearLogTimerFor(processId);
|
||
showNotification('训练进程已删除', 'success');
|
||
})
|
||
.catch(() => {
|
||
showNotification('删除进程失败，请刷新页面重试', 'error');
|
||
});
|
||
});
|
||
}
|
||
|
||
export function stopProcess(processId) {
|
||
showConfirmDialog('确定要停止这个训练进程吗？', () => {
|
||
apiStop(processId)
|
||
.then(() => {
|
||
const item = document.querySelector(`[data-process-id="${processId}"]`);
|
||
if (item) {
|
||
item.dataset.processStatus = '手动停止';
|
||
const statusEl = item.querySelector('.process-status');
|
||
if (statusEl) {
|
||
statusEl.classList.remove('status-running', 'status-error', 'status-completed');
|
||
statusEl.classList.add('status-manual-stop');
|
||
statusEl.textContent = '手动停止';
|
||
}
|
||
const stopBtn = item.querySelector('.btn-stop');
|
||
if (stopBtn) stopBtn.remove();
|
||
clearLogTimerFor(processId);
|
||
}
|
||
showNotification('训练进程已停止', 'info');
|
||
getProcesses()
|
||
.then((data) => {
|
||
const updated = data.find((p) => p.id === processId);
|
||
if (updated && item) updateProcessItem(item, updated);
|
||
})
|
||
.catch(() => {});
|
||
})
|
||
.catch(() => {
|
||
showNotification('停止进程失败', 'error');
|
||
});
|
||
}, () => {
|
||
showNotification('已取消停止操作', 'info');
|
||
});
|
||
}
|
||
|
||
export function checkAndOpenSwanlab(processId) {
|
||
const item = document.querySelector(`[data-process-id="${processId}"]`);
|
||
const monitor = item ? item.dataset.trainMonitor : 'none';
|
||
if (monitor === 'none') {
|
||
showNotification('此训练未启用监控功能', 'info');
|
||
return;
|
||
}
|
||
let url = item ? item.dataset.swanlabUrl : '';
|
||
if (!url || url.trim() === '') {
|
||
getProcesses()
|
||
.then((data) => {
|
||
const p = data.find((x) => x.id === processId);
|
||
if (p && p.swanlab_url) {
|
||
url = p.swanlab_url;
|
||
if (item) item.dataset.swanlabUrl = url;
|
||
openSwanlab(url);
|
||
} else {
|
||
showNotification('SwanLab链接尚未生成，请稍后再试', 'info');
|
||
}
|
||
})
|
||
.catch(() => {
|
||
showNotification('获取SwanLab链接失败，请稍后再试', 'error');
|
||
});
|
||
} else openSwanlab(url);
|
||
}
|
||
|
||
function openSwanlab(url) {
|
||
if (!isValidUrl(url)) {
|
||
showNotification('SwanLab链接无效或尚未生成', 'info');
|
||
return;
|
||
}
|
||
const w = window.open(url, '_blank');
|
||
if (w) showNotification('正在打开SwanLab页面', 'info');
|
||
else showNotification('无法打开新窗口，请检查浏览器设置', 'error');
|
||
}
|
||
|
||
function isValidUrl(url) {
|
||
try {
|
||
new URL(url);
|
||
return true;
|
||
} catch {
|
||
const u = String(url).toLowerCase();
|
||
return u.startsWith('http://') || u.startsWith('https://');
|
||
}
|
||
}
|
||
|