evolclaw 2.0.0

package/dist/core/message-stream.js

@@ -0,0 +1,59 @@
+/**
+ * MessageStream - Push-based async iterable for streaming user messages to the SDK.
+ * Based on HappyClaw's implementation.
+ */
+import { logger } from '../utils/logger.js';
+export class MessageStream {
+    queue = [];
+    waiting = null;
+    done = false;
+    push(text, images) {
+        let content;
+        if (images && images.length > 0) {
+            logger.debug('[MessageStream] Creating multimodal message with', images.length, 'images');
+            logger.debug('[MessageStream] Image sizes:', images.map(img => img.data.length).join(', '));
+            // 多模态消息：text + images
+            content = [
+                { type: 'text', text },
+                ...images.map((img) => ({
+                    type: 'image',
+                    source: {
+                        type: 'base64',
+                        media_type: img.mimeType || 'image/png',
+                        data: img.data,
+                    },
+                })),
+            ];
+        }
+        else {
+            // 纯文本消息
+            content = text;
+        }
+        const message = {
+            type: 'user',
+            message: { role: 'user', content },
+            parent_tool_use_id: null,
+            session_id: '',
+        };
+        logger.debug('[MessageStream] Pushing message, content type:', Array.isArray(content) ? 'array' : 'string');
+        this.queue.push(message);
+        this.waiting?.();
+    }
+    end() {
+        this.done = true;
+        this.waiting?.();
+    }
+    async *[Symbol.asyncIterator]() {
+        while (true) {
+            while (this.queue.length > 0) {
+                yield this.queue.shift();
+            }
+            if (this.done)
+                return;
+            await new Promise((r) => {
+                this.waiting = r;
+            });
+            this.waiting = null;
+        }
+    }
+}