zimk/NekoAI
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat: 停止生成、代码块复制、消息复制、重新生成、会话导出 MD

Browse Source
This commit is contained in:
zimk
2026-03-17 00:41:46 +08:00
parent e1a3b629b4
commit a57bebb385
4 changed files with 274 additions and 34 deletions
Download Patch File Download Diff File Expand all files Collapse all files

153
public/app.js
View File

@@ -36,6 +36,8 @@ const mobileSidebarBackdropEl = document.getElementById('mobileSidebarBackdrop')
const fileInputEl = document.getElementById('fileInput');
const attachBtnEl = document.getElementById('attachBtn');
const attachmentListEl = document.getElementById('attachmentList');
const stopBtnEl = document.getElementById('stopBtn');
const exportBtnEl = document.getElementById('exportBtn');
boot();
@@ -130,6 +132,16 @@ function bindEvents() {
fileInputEl.value = '';
});
stopBtnEl.addEventListener('click', () => {
if (state.abortController) {
state.abortController.abort();
}
});
exportBtnEl.addEventListener('click', () => {
exportConversation();
});
messageInputEl.addEventListener('input', () => {
autoResizeTextarea();
});
@@ -434,14 +446,24 @@ function renderAttachments() {
async function sendChat(accessKey, model, conversation, assistantMessage, attachments) {
const payload = buildChatPayload(model, conversation, assistantMessage, attachments);
const response = await fetch(`${API_BASE_URL}/api/chat`, {
state.abortController = new AbortController();
const signal = state.abortController.signal;
let response;
try {
response = await fetch(`${API_BASE_URL}/api/chat`, {
method: 'POST',
headers: {
'Content-Type': 'application/json',
Authorization: `Bearer ${accessKey}`,
},
body: JSON.stringify(payload),
signal,
});
} catch (err) {
if (err.name === 'AbortError') return;
throw err;
}
if (!response.ok) {
const data = await response.json().catch(() => ({}));
@@ -460,6 +482,7 @@ async function sendChat(accessKey, model, conversation, assistantMessage, attach
const decoder = new TextDecoder();
let buffer = '';
try {
while (true) {
const { value, done } = await reader.read();
if (done) break;
@@ -496,6 +519,16 @@ async function sendChat(accessKey, model, conversation, assistantMessage, attach
}
}
}
} catch (err) {
if (err.name === 'AbortError') {
// 用户主动停止，保留已生成内容
updateThinkTiming(assistantMessage, true);
return;
}
throw err;
} finally {
state.abortController = null;
}
}
function buildChatPayload(model, conversation, assistantMessage, attachments) {
@@ -582,6 +615,73 @@ function buildUserMessage(message, attachments) {
return { role: 'user', content };
}
async function regenLastAssistant() {
if (state.sending) return;
const conversation = getActiveConversation();
if (!conversation) return;
// 找到最后一条 assistant 消息，删掉它，重新发
const lastIdx = conversation.messages.map((m) => m.role).lastIndexOf('assistant');
if (lastIdx === -1) return;
conversation.messages.splice(lastIdx, 1);
const accessKey = localStorage.getItem(ACCESS_KEY_STORAGE) || '';
const model = state.selectedModel;
if (!accessKey || !model) return;
const assistantMessage = { role: 'assistant', content: '', createdAt: Date.now() };
conversation.messages.push(assistantMessage);
conversation.updatedAt = Date.now();
state.sending = true;
updateSendingState();
renderMessages();
try {
await sendChat(accessKey, model, conversation, assistantMessage, []);
} catch (err) {
assistantMessage.content = `请求失败：${err.message}`;
} finally {
state.sending = false;
updateSendingState();
conversation.updatedAt = Date.now();
persistConversations();
renderConversationList();
renderMessages();
focusComposer();
}
}
function exportConversation() {
const conversation = getActiveConversation();
if (!conversation || !conversation.messages.length) {
alert('当前会话没有内容可以导出。');
return;
}
const lines = [`# ${conversation.title || '新会话'}`, ''];
conversation.messages.forEach((msg) => {
const role = msg.role === 'assistant' ? '**NekoAI**' : '**用户**';
const content = String(msg.content || '')
.replace(/<think>[\s\S]*?<\/think>/gi, '')
.replace(/<think>[\s\S]*$/gi, '')
.trim();
lines.push(`${role}\n\n${content}`, '');
if (msg.attachments?.length) {
msg.attachments.forEach((f) => lines.push(`> 附件：${f.name} (${formatBytes(f.size)})`, ''));
}
lines.push('---', '');
});
const blob = new Blob([lines.join('\n')], { type: 'text/markdown;charset=utf-8' });
const url = URL.createObjectURL(blob);
const a = document.createElement('a');
a.href = url;
a.download = `${(conversation.title || '会话').slice(0, 40)}.md`;
a.click();
URL.revokeObjectURL(url);
}
function extractAssistantText(data) {
const content = data?.choices?.[0]?.message?.content;
if (typeof content === 'string') return content;
@@ -781,9 +881,51 @@ function renderMessages() {
});
}
// 复制按钮（所有消息）
const copyBtn = document.createElement('button');
copyBtn.className = 'message-copy-btn';
copyBtn.type = 'button';
copyBtn.textContent = '复制';
copyBtn.addEventListener('click', () => {
const plainText = String(message.content || '')
.replace(/<think>[\s\S]*?<\/think>/gi, '')
.replace(/<think>[\s\S]*$/gi, '')
.trim();
navigator.clipboard.writeText(plainText).then(() => {
copyBtn.textContent = '已复制';
copyBtn.classList.add('copied');
setTimeout(() => { copyBtn.textContent = '复制'; copyBtn.classList.remove('copied'); }, 1500);
});
});
row.querySelector('.message-bubble').appendChild(copyBtn);
// 重新生成按钮（仅 assistant 最后一条）
const isLastAssistant = message.role === 'assistant' &&
conversation.messages[conversation.messages.length - 1] === message;
if (isLastAssistant && !state.sending) {
const regenBtn = document.createElement('button');
regenBtn.className = 'message-regen-btn';
regenBtn.type = 'button';
regenBtn.textContent = '重新生成';
regenBtn.addEventListener('click', () => regenLastAssistant());
row.querySelector('.message-bubble').appendChild(regenBtn);
}
messagesEl.appendChild(row);
});
// 代码块复制按钮事件委托
messagesEl.querySelectorAll('.copy-code-btn').forEach((btn) => {
btn.addEventListener('click', () => {
const code = btn.previousElementSibling?.querySelector('code')?.textContent || '';
navigator.clipboard.writeText(code).then(() => {
btn.textContent = '已复制';
btn.classList.add('copied');
setTimeout(() => { btn.textContent = '复制'; btn.classList.remove('copied'); }, 1500);
});
});
});
if (thinkingMsg) startLiveThinkTicker();
else stopLiveThinkTicker();
@@ -852,7 +994,7 @@ function renderMarkdown(input) {
renderer.code = ({ text, lang }) => {
const escaped = text.replace(/&/g,'&amp;').replace(/</g,'&lt;').replace(/>/g,'&gt;');
return `<pre class="md-pre"><code>${escaped}</code></pre>`;
return `<div class="md-pre-wrap"><pre class="md-pre"><code>${escaped}</code></pre><button class="copy-code-btn" type="button">复制</button></div>`;
};
renderer.image = ({ href, title, text }) => {
@@ -916,6 +1058,13 @@ function stopLiveThinkTicker() {
function updateSendingState() {
sendBtnEl.disabled = state.sending;
sendBtnEl.textContent = state.sending ? '发送中...' : '发送';
if (state.sending) {
stopBtnEl.classList.remove('hidden');
sendBtnEl.classList.add('hidden');
} else {
stopBtnEl.classList.add('hidden');
sendBtnEl.classList.remove('hidden');
}
}
function autoResizeTextarea() {

4
public/index.html
View File

@@ -40,6 +40,7 @@
<div class="sidebar-section-header">
<div class="sidebar-section-label">最近会话</div>
<button id="clearAllBtn" class="sidebar-text-btn" type="button">清空</button>
<button id="exportBtn" class="sidebar-text-btn" type="button">导出</button>
</div>
<div id="conversationList" class="conversation-list"></div>
</aside>
@@ -96,3 +97,6 @@
<script src="/app.js" defer></script>
</body>
</html>
>
</body>
</html>

86
public/styles.css
View File

@@ -497,6 +497,7 @@ body.theme-light .md-pre { background: #f3f4f6; border-color: rgba(15,23,42,.08)
align-items: center;
gap: 12px;
}
.attach-btn svg { display: block; }
.attach-btn {
width: 40px;
height: 40px;
@@ -706,3 +707,88 @@ body.theme-light .md-pre { background: #f3f4f6; border-color: rgba(15,23,42,.08)
place-items: center;
}
}
/* ── 新增样式 ── */
/* 停止按钮 */
.stop-btn {
padding: 10px 16px;
background: #ff4d4d;
color: #fff;
border: none;
font-weight: 600;
border-radius: 14px;
min-width: 72px;
font-size: 14px;
cursor: pointer;
transition: .18s ease;
margin-right: 8px;
}
.stop-btn:hover { background: #e03c3c; }
/* 代码块复制按钮 */
.md-pre-wrap {
position: relative;
margin: 0 0 12px;
}
.md-pre-wrap .md-pre {
margin: 0;
}
.copy-code-btn {
position: absolute;
top: 8px;
right: 8px;
padding: 4px 10px;
border-radius: 8px;
border: 1px solid rgba(255,255,255,.12);
background: rgba(255,255,255,.08);
color: var(--muted);
font-size: 12px;
cursor: pointer;
opacity: 0;
transition: opacity .15s ease, background .15s ease;
}
.md-pre-wrap:hover .copy-code-btn { opacity: 1; }
.copy-code-btn:hover { background: rgba(255,255,255,.16); color: var(--text); }
.copy-code-btn.copied { color: #4ade80; }
body.theme-light .copy-code-btn {
border-color: rgba(0,0,0,.1);
background: rgba(0,0,0,.04);
}
body.theme-light .copy-code-btn:hover { background: rgba(0,0,0,.08); }
/* 消息复制按钮 */
.message-copy-btn {
display: inline-block;
margin-top: 6px;
padding: 3px 10px;
border-radius: 8px;
border: 1px solid var(--border);
background: transparent;
color: var(--muted);
font-size: 12px;
cursor: pointer;
opacity: 0;
transition: opacity .15s ease, background .15s ease;
}
.message-bubble:hover .message-copy-btn { opacity: 1; }
.message-copy-btn:hover { background: var(--panel-hover); color: var(--text); }
.message-copy-btn.copied { color: #4ade80; }
/* 消息重新生成按钮 */
.message-regen-btn {
display: inline-block;
margin-top: 6px;
margin-left: 6px;
padding: 3px 10px;
border-radius: 8px;
border: 1px solid var(--border);
background: transparent;
color: var(--muted);
font-size: 12px;
cursor: pointer;
opacity: 0;
transition: opacity .15s ease, background .15s ease;
}
.message-bubble:hover .message-regen-btn { opacity: 1; }
.message-regen-btn:hover { background: var(--panel-hover); color: var(--text); }

1
wrangler.toml
View File

@@ -2,3 +2,4 @@ name = "nekoai-api"
main = "src/index.js"
compatibility_date = "2026-03-12"
workers_dev = true
pages_build_output_dir = "public"