@cubeai/dsh-feishu-channels 0.0.0-stage → 0.2.0

package/src/engine/reply-stream.js

@@ -0,0 +1,86 @@
+export function isAssistantTextDelta(chunk) {
+    if (!chunk || typeof chunk.text !== 'string' || chunk.text.length === 0)
+        return false;
+    return !chunk.type || chunk.type === 'text' || chunk.type === 'text-delta';
+}
+export class ReplyStreamHub {
+    streams = new Map();
+    texts = new Map();
+    tails = new Map();
+    delivered = new Set();
+    // 在增量排队时记录归属，开流尚未完成也能按会话取消。
+    owners = new Map();
+    // 回合纪元：reset 时自增，迟到的旧回合增量据此丢弃，不会重建流
+    generations = new Map();
+    onTextDelta(key, delta, start, sessionId) {
+        if (sessionId !== undefined) {
+            const owner = this.owners.get(key);
+            if (owner !== undefined && owner !== sessionId)
+                this.reset(key);
+            this.owners.set(key, sessionId);
+        }
+        const generation = this.generations.get(key) ?? 0;
+        const prev = this.tails.get(key) ?? Promise.resolve();
+        const next = prev.catch(() => undefined).then(async () => {
+            if (generation !== (this.generations.get(key) ?? 0))
+                return;
+            let stream = this.streams.get(key);
+            if (!stream) {
+                stream = await start();
+                // start 期间可能刚被 reset（模型切换等），旧回合的流不再入表
+                if (generation !== (this.generations.get(key) ?? 0))
+                    return;
+                if (!stream)
+                    return;
+                this.streams.set(key, stream);
+                this.texts.set(key, '');
+                this.delivered.delete(key);
+            }
+            const acc = (this.texts.get(key) ?? '') + delta;
+            this.texts.set(key, acc);
+            await stream.update(acc);
+        });
+        this.tails.set(key, next);
+        return next;
+    }
+    async take(key, sessionId) {
+        if (sessionId !== undefined && this.owners.has(key) && this.owners.get(key) !== sessionId)
+            return { text: '' };
+        const generation = this.generations.get(key) ?? 0;
+        await (this.tails.get(key) ?? Promise.resolve()).catch(() => undefined);
+        if (generation !== (this.generations.get(key) ?? 0))
+            return { text: '', invalidated: true };
+        const stream = this.streams.get(key);
+        const text = this.texts.get(key) ?? '';
+        this.streams.delete(key);
+        this.texts.delete(key);
+        this.tails.delete(key);
+        return { stream, text };
+    }
+    markDelivered(key, sessionId) {
+        if (sessionId !== undefined) {
+            const owner = this.owners.get(key);
+            if (owner !== undefined && owner !== sessionId)
+                return;
+            this.owners.set(key, sessionId);
+        }
+        this.delivered.add(key);
+    }
+    consumeDelivered(key, sessionId) {
+        if (sessionId !== undefined && this.owners.get(key) !== sessionId)
+            return false;
+        return this.delivered.delete(key);
+    }
+    reset(key, sessionId) {
+        // 清理其他历史登记不能取消当前会话的流；普通新回合仍可无条件重置。
+        if (sessionId !== undefined && this.owners.get(key) !== sessionId)
+            return;
+        // 新回合开始：清掉上一回合可能残留的流与累计文本，避免新内容拼进旧卡片
+        this.generations.set(key, (this.generations.get(key) ?? 0) + 1);
+        this.streams.delete(key);
+        this.texts.delete(key);
+        this.tails.delete(key);
+        this.delivered.delete(key);
+        this.owners.delete(key);
+    }
+}