vrr/qwen-code
2
0
Fork 0
mirror of https://github.com/QwenLM/qwen-code.git synced 2026-04-30 04:30:48 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 24

feat(channels): add streaming response hooks to ChannelBase

Browse source 
- Add onResponseChunk hook for progressive text display during streaming
- Add onResponseComplete hook for customizing response delivery
- Update mock plugin channel to support streaming chunks

This enables channels to display AI responses progressively as they stream,
improving user experience with real-time feedback.

Co-authored-by: Qwen-Coder <qwen-coder@alibabacloud.com>
This commit is contained in:
tanzhenxin 2026-03-27 14:08:09 +00:00
parent 0ca8cf86f6
commit f7979aa902
4 changed files with 117 additions and 17 deletions
Download patch file Download diff file Expand all files Collapse all files

44
packages/channels/plugin-example/src/mock-server.ts
View file

@ -53,9 +53,10 @@ export function createMockServer(
const pendingRequests = new Map<
string,
{
resolve: (text: string) => void;
resolve: (result: { text: string; chunks: string[] }) => void;
reject: (err: Error) => void;
timer: ReturnType<typeof setTimeout>;
chunks: string[];
}
>();
@ -73,12 +74,17 @@ export function createMockServer(
ws.on('message', (data) => {
try {
const msg = JSON.parse(data.toString());
if (msg.type === 'outbound' && msg.messageId) {
if (msg.type === 'chunk' && msg.messageId) {
const pending = pendingRequests.get(msg.messageId);
if (pending) {
pending.chunks.push(msg.text);
}
} else if (msg.type === 'outbound' && msg.messageId) {
const pending = pendingRequests.get(msg.messageId);
if (pending) {
clearTimeout(pending.timer);
pendingRequests.delete(msg.messageId);
pending.resolve(msg.text);
pending.resolve({ text: msg.text, chunks: pending.chunks });
}
}
} catch {
@ -138,18 +144,35 @@ export function createMockServer(
}),
);
const responsePromise = new Promise<string>((resolve, reject) => {
const responsePromise = new Promise<{
text: string;
chunks: string[];
}>((resolve, reject) => {
const timer = setTimeout(() => {
pendingRequests.delete(messageId);
reject(new Error('Timeout waiting for agent response'));
}, responseTimeoutMs);
pendingRequests.set(messageId, { resolve, reject, timer });
pendingRequests.set(messageId, {
resolve,
reject,
timer,
chunks: [],
});
});
responsePromise
.then((responseText) => {
.then((result) => {
res.writeHead(200, { 'Content-Type': 'application/json' });
res.end(JSON.stringify({ messageId, text: responseText }));
res.end(
JSON.stringify({
messageId,
text: result.text,
streaming: {
chunks: result.chunks.length,
bytes: result.chunks.reduce((n, c) => n + c.length, 0),
},
}),
);
})
.catch((err: Error) => {
res.writeHead(504, { 'Content-Type': 'application/json' });
@ -215,7 +238,12 @@ export function createMockServer(
pendingRequests.delete(messageId);
reject(new Error('Timeout waiting for agent response'));
}, responseTimeoutMs);
pendingRequests.set(messageId, { resolve, reject, timer });
pendingRequests.set(messageId, {
resolve: (result) => resolve(result.text),
reject,
timer,
chunks: [],
});
});
},