`;
if (role === 'assistant') {
- html += `
${marked.parse(content)}
`;
+ html += `
+
+
${marked.parse(content)}
+
+
`;
} else {
html += `
${escapeHtml(content)}
`;
}
@@ -287,10 +332,19 @@
html += `
`;
html += ``;
if (role === 'assistant') {
- html += ``;
+ html += ``;
}
html += `
`;
+ // assistant消息的版本切换控件（初始隐藏，有多个版本时显示）
+ if (role === 'assistant') {
+ html += `
+
+ 1/1
+
+
`;
+ }
+
// Agent信息
if (role === 'assistant' && agentName) {
html += `
${agentName}
`;
@@ -302,6 +356,153 @@
container.scrollTop = container.scrollHeight;
}
+ // 添加新版本到现有消息
+ function addResponseVersion(messageId, content, thinking) {
+ const versions = messageVersions[messageId];
+ if (!versions) return;
+
+ // 添加新版本
+ versions.push({ content, thinking });
+ const newVersionIndex = versions.length - 1;
+
+ // 更新容器
+ const container = document.getElementById(`${messageId}_container`);
+ if (container) {
+ // 添加思考块（如果有）
+ const messageBody = container.closest('.message-body');
+ if (thinking && messageBody) {
+ // 先移除旧的思考块
+ const oldThinking = messageBody.querySelector('.thinking-block');
+ if (oldThinking) oldThinking.remove();
+
+ // 添加新的思考块
+ const thinkingHtml = `
+
+
${escapeHtml(thinking)}
+
`;
+ messageBody.insertAdjacentHTML('afterbegin', thinkingHtml);
+ }
+
+ // 隐藏所有旧版本，添加新版本
+ container.querySelectorAll('.response-version').forEach(v => v.classList.remove('active'));
+ const newVersionHtml = `
+
${marked.parse(content)}
+
`;
+ container.insertAdjacentHTML('beforeend', newVersionHtml);
+ }
+
+ // 更新复制源
+ const messageDiv = document.getElementById(messageId);
+ if (messageDiv) {
+ const copySource = messageDiv.querySelector('.copy-source');
+ if (copySource) copySource.value = content;
+ }
+
+ // 显示版本切换控件
+ showVersionControls(messageId);
+ }
+
+ // 显示loading动画
+ function showLoadingIndicator(messageId) {
+ const container = document.getElementById(`${messageId}_container`);
+ if (container) {
+ // 隐藏所有版本
+ container.querySelectorAll('.response-version').forEach(v => v.classList.remove('active'));
+ // 显示loading
+ const loadingHtml = `
`;
+ container.insertAdjacentHTML('beforeend', loadingHtml);
+ }
+ }
+
+ // 隐藏loading动画
+ function hideLoadingIndicator(messageId) {
+ const loading = document.getElementById(`${messageId}_loading`);
+ if (loading) loading.remove();
+ }
+
+ // 显示版本切换控件
+ function showVersionControls(messageId) {
+ const controls = document.getElementById(`${messageId}_version_controls`);
+ const versions = messageVersions[messageId];
+ if (controls && versions && versions.length > 1) {
+ controls.style.display = 'flex';
+ updateVersionIndicator(messageId);
+ }
+ }
+
+ // 更新版本指示器
+ function updateVersionIndicator(messageId) {
+ const container = document.getElementById(`${messageId}_container`);
+ const indicator = document.getElementById(`${messageId}_version_indicator`);
+ const versions = messageVersions[messageId];
+ if (!container || !indicator || !versions) return;
+
+ // 找到当前激活的版本
+ const activeVersion = container.querySelector('.response-version.active');
+ const currentIndex = activeVersion ? parseInt(activeVersion.dataset.version) : 0;
+
+ indicator.textContent = `${currentIndex + 1}/${versions.length}`;
+
+ // 更新按钮状态
+ const prevBtn = indicator.parentElement.querySelector('[data-dir="prev"]');
+ const nextBtn = indicator.parentElement.querySelector('[data-dir="next"]');
+ if (prevBtn) prevBtn.disabled = currentIndex === 0;
+ if (nextBtn) nextBtn.disabled = currentIndex === versions.length - 1;
+ }
+
+ // 切换版本
+ function switchVersion(messageId, direction) {
+ const container = document.getElementById(`${messageId}_container`);
+ const versions = messageVersions[messageId];
+ if (!container || !versions) return;
+
+ // 找到当前激活的版本
+ const activeVersion = container.querySelector('.response-version.active');
+ const currentIndex = activeVersion ? parseInt(activeVersion.dataset.version) : 0;
+ const newIndex = currentIndex + direction;
+
+ if (newIndex < 0 || newIndex >= versions.length) return;
+
+ // 切换显示
+ container.querySelectorAll('.response-version').forEach(v => {
+ v.classList.remove('active');
+ if (parseInt(v.dataset.version) === newIndex) v.classList.add('active');
+ });
+
+ // 更新复制源
+ const messageDiv = document.getElementById(messageId);
+ if (messageDiv) {
+ const copySource = messageDiv.querySelector('.copy-source');
+ if (copySource) copySource.value = versions[newIndex].content;
+ }
+
+ // 更新思考块
+ const messageBody = container.closest('.message-body');
+ if (messageBody && versions[newIndex].thinking) {
+ // 移除旧的思考块
+ const oldThinking = messageBody.querySelector('.thinking-block');
+ if (oldThinking) oldThinking.remove();
+
+ // 添加新版本的思考块
+ const thinkingHtml = `
+
+
${escapeHtml(versions[newIndex].thinking)}
+
`;
+ messageBody.insertAdjacentHTML('afterbegin', thinkingHtml);
+ }
+
+ updateVersionIndicator(messageId);
+ }
+
function toggleThinking(header) {
const block = header.parentElement;
const content = block.querySelector('.thinking-content');
@@ -357,19 +558,18 @@
document.body.removeChild(textarea);
}
- function regenerateMessage() {
+ function regenerateMessage(messageId) {
if (!lastUserMessage) {
alert('没有可重新生成的消息');
return;
}
// 设置重新生成标志，避免再次显示用户消息
isRegenerating = true;
- // 移除最后一条助手消息
- const container = document.getElementById('messagesContainer');
- const messages = container.querySelectorAll('.message.assistant');
- if (messages.length > 0) {
- messages[messages.length - 1].remove();
- }
+ regeneratingMessageId = messageId;
+
+ // 显示loading动画
+ showLoadingIndicator(messageId);
+
// 重新发送最后一条用户消息
document.getElementById('sendBtn').disabled = true;
if (ws?.readyState === WebSocket.OPEN) {