abbyswag/qwen-code
1
0
Fork 0
mirror of https://github.com/QwenLM/qwen-code.git synced 2025-12-21 17:27:54 +00:00
Code Issues Packages Projects Releases Wiki Activity

style(vscode-ide-companion): adjust chat session initialization logic and optimize tool invocation component style

Browse Source
This commit is contained in:
yiliang114
2025-12-06 22:44:31 +08:00
parent ed44520e51 57a684ad97
commit 67eee14ca9
15 changed files with 373 additions and 387 deletions
Download Patch File Download Diff File Expand all files Collapse all files

44
packages/vscode-ide-companion/src/webview/hooks/message/useMessageHandling.ts
View File

@@ -70,26 +70,32 @@ export const useMessageHandling = () => {
/**
* Add stream chunk
*/
const appendStreamChunk = useCallback((chunk: string) => {
setMessages((prev) => {
let idx = streamingMessageIndexRef.current;
const next = prev.slice();
const appendStreamChunk = useCallback(
(chunk: string) => {
// Ignore late chunks after user cancelled streaming (until next streamStart)
if (!isStreaming) return;
// If there is no active placeholder (e.g., after a tool call), start a new one
if (idx === null) {
idx = next.length;
streamingMessageIndexRef.current = idx;
next.push({ role: 'assistant', content: '', timestamp: Date.now() });
}
setMessages((prev) => {
let idx = streamingMessageIndexRef.current;
const next = prev.slice();
if (idx < 0 || idx >= next.length) {
return prev;
}
const target = next[idx];
next[idx] = { ...target, content: (target.content || '') + chunk };
return next;
});
}, []);
// If there is no active placeholder (e.g., after a tool call), start a new one
if (idx === null) {
idx = next.length;
streamingMessageIndexRef.current = idx;
next.push({ role: 'assistant', content: '', timestamp: Date.now() });
}
if (idx < 0 || idx >= next.length) {
return prev;
}
const target = next[idx];
next[idx] = { ...target, content: (target.content || '') + chunk };
return next;
});
},
[isStreaming],
);
/**
* Break current assistant stream segment (e.g., when a tool call starts/updates)
@@ -150,6 +156,8 @@ export const useMessageHandling = () => {
endStreaming,
// Thought handling
appendThinkingChunk: (chunk: string) => {
// Ignore late thoughts after user cancelled streaming
if (!isStreaming) return;
setMessages((prev) => {
let idx = thinkingMessageIndexRef.current;
const next = prev.slice();