@scenemesh/entity-engine-aimodule 1.0.0 → 1.0.1

package/dist/utils/persistence.js

@@ -0,0 +1,286 @@
+/**
+ * 🎯 Entity Engine AIUI - Message Persistence Utils
+ *
+ * 消息持久化工具，支持消息验证、ID生成和存储管理
+ */
+/**
+ * 创建ID生成器
+ */
+export function createIdGenerator(options = {}) {
+    const { prefix = '', size = 12 } = options;
+    return () => {
+        const chars = 'ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789';
+        let result = '';
+        for (let i = 0; i < size; i++) {
+            result += chars.charAt(Math.floor(Math.random() * chars.length));
+        }
+        return prefix ? `${prefix}-${result}` : result;
+    };
+}
+/**
+ * 生成唯一ID
+ */
+export function generateId() {
+    return createIdGenerator()();
+}
+/**
+ * 验证UI消息
+ */
+export async function validateUIMessages({ messages, tools, dataPartsSchema, metadataSchema, }) {
+    const validatedMessages = [];
+    for (const message of messages) {
+        try {
+            const validatedMessage = await validateSingleMessage(message, {
+                tools,
+                dataPartsSchema,
+                metadataSchema,
+            });
+            validatedMessages.push(validatedMessage);
+        }
+        catch (error) {
+            // 如果验证失败，记录错误但继续处理
+            // 可以选择跳过无效消息或使用默认值
+            validatedMessages.push(sanitizeMessage(message));
+        }
+    }
+    return validatedMessages;
+}
+/**
+ * 验证单个消息
+ */
+async function validateSingleMessage(message, options) {
+    const { tools, dataPartsSchema, metadataSchema } = options;
+    // 验证消息结构
+    if (!message.id || !message.role || !Array.isArray(message.parts)) {
+        throw new Error('Invalid message structure');
+    }
+    // 验证消息部分
+    const validatedParts = [];
+    for (const part of message.parts) {
+        const validatedPart = await validateMessagePart(part, tools);
+        validatedParts.push(validatedPart);
+    }
+    // 验证元数据
+    let validatedMetadata = message.metadata;
+    if (metadataSchema && message.metadata) {
+        try {
+            validatedMetadata = metadataSchema.parse(message.metadata);
+        }
+        catch (error) {
+            validatedMetadata = undefined;
+        }
+    }
+    return {
+        ...message,
+        parts: validatedParts,
+        metadata: validatedMetadata,
+    };
+}
+/**
+ * 验证消息部分
+ */
+async function validateMessagePart(part, tools) {
+    // 验证文本部分
+    if (part.type === 'text') {
+        if (typeof part.text !== 'string') {
+            throw new Error('Text part must have a string text property');
+        }
+        return part;
+    }
+    // 验证文件部分
+    if (part.type === 'file') {
+        if (!part.filename || !part.url) {
+            throw new Error('File part must have filename and url properties');
+        }
+        return part;
+    }
+    // 验证工具调用部分
+    if (part.type.startsWith('tool-') || part.type === 'dynamic-tool') {
+        const toolName = part.type === 'dynamic-tool' ? part.toolName : part.type.replace('tool-', '');
+        // 如果提供了工具定义，验证工具是否存在
+        if (tools && toolName && !tools[toolName]) {
+            console.warn(`Tool ${toolName} not found in provided tools`);
+        }
+        return part;
+    }
+    // 验证数据部分
+    if (part.type.startsWith('data-')) {
+        if (!part.data) {
+            throw new Error('Data part must have a data property');
+        }
+        return part;
+    }
+    return part;
+}
+/**
+ * 清理消息（移除无效部分）
+ */
+function sanitizeMessage(message) {
+    const sanitizedParts = message.parts.filter(part => 
+    // 保留有效的部分类型
+    part && typeof part === 'object' && part.type);
+    return {
+        ...message,
+        parts: sanitizedParts.length > 0 ? sanitizedParts : [
+            { type: 'text', text: '[Message content could not be validated]' }
+        ],
+    };
+}
+// ================================
+// 🎯 类型验证错误
+// ================================
+/**
+ * 类型验证错误类
+ */
+export class TypeValidationError extends Error {
+    constructor(message, details) {
+        super(message);
+        this.details = details;
+        this.name = 'TypeValidationError';
+    }
+}
+/**
+ * 内存消息存储（用于开发和测试）
+ */
+export class InMemoryMessageStore {
+    constructor() {
+        this.chats = new Map();
+        this.metadata = new Map();
+    }
+    async createChat() {
+        const id = generateId();
+        this.chats.set(id, []);
+        this.metadata.set(id, { updatedAt: new Date() });
+        return id;
+    }
+    async loadChat(id) {
+        const messages = this.chats.get(id);
+        if (!messages) {
+            throw new Error(`Chat ${id} not found`);
+        }
+        return [...messages];
+    }
+    async saveChat({ chatId, messages }) {
+        this.chats.set(chatId, [...messages]);
+        // 更新元数据
+        const existing = this.metadata.get(chatId) || {};
+        this.metadata.set(chatId, {
+            ...existing,
+            updatedAt: new Date(),
+            title: existing.title || this.extractTitle(messages),
+        });
+    }
+    async deleteChat(id) {
+        this.chats.delete(id);
+        this.metadata.delete(id);
+    }
+    async listChats() {
+        return Array.from(this.metadata.entries()).map(([id, meta]) => ({
+            id,
+            ...meta,
+        }));
+    }
+    extractTitle(messages) {
+        // 提取第一个用户消息作为标题
+        const firstUserMessage = messages.find(m => m.role === 'user');
+        if (firstUserMessage) {
+            const textPart = firstUserMessage.parts.find(p => p.type === 'text');
+            if (textPart?.text) {
+                return textPart.text.slice(0, 50) + (textPart.text.length > 50 ? '...' : '');
+            }
+        }
+        return 'Untitled Chat';
+    }
+}
+/**
+ * 本地存储消息存储
+ */
+export class LocalStorageMessageStore {
+    constructor() {
+        this.keyPrefix = 'entity-engine-aiui-chat-';
+    }
+    async createChat() {
+        const id = generateId();
+        const key = this.keyPrefix + id;
+        const chatData = {
+            id,
+            messages: [],
+            createdAt: new Date().toISOString(),
+            updatedAt: new Date().toISOString(),
+        };
+        localStorage.setItem(key, JSON.stringify(chatData));
+        return id;
+    }
+    async loadChat(id) {
+        const key = this.keyPrefix + id;
+        const data = localStorage.getItem(key);
+        if (!data) {
+            throw new Error(`Chat ${id} not found`);
+        }
+        try {
+            const chatData = JSON.parse(data);
+            return chatData.messages || [];
+        }
+        catch (error) {
+            throw new Error(`Failed to parse chat data: ${error}`);
+        }
+    }
+    async saveChat({ chatId, messages }) {
+        const key = this.keyPrefix + chatId;
+        const existing = localStorage.getItem(key);
+        const chatData = {
+            id: chatId,
+            messages,
+            createdAt: new Date().toISOString(),
+            updatedAt: new Date().toISOString(),
+        };
+        if (existing) {
+            try {
+                const parsedExisting = JSON.parse(existing);
+                chatData.createdAt = parsedExisting.createdAt || chatData.createdAt;
+            }
+            catch {
+                // Ignore validation errors
+            }
+        }
+        localStorage.setItem(key, JSON.stringify(chatData));
+    }
+    async deleteChat(id) {
+        const key = this.keyPrefix + id;
+        localStorage.removeItem(key);
+    }
+    async listChats() {
+        const chats = [];
+        for (let i = 0; i < localStorage.length; i++) {
+            const key = localStorage.key(i);
+            if (key?.startsWith(this.keyPrefix)) {
+                try {
+                    const data = localStorage.getItem(key);
+                    if (data) {
+                        const chatData = JSON.parse(data);
+                        chats.push({
+                            id: chatData.id,
+                            title: this.extractTitle(chatData.messages || []),
+                            updatedAt: new Date(chatData.updatedAt || chatData.createdAt),
+                        });
+                    }
+                }
+                catch {
+                    // Ignore part validation errors
+                }
+            }
+        }
+        return chats.sort((a, b) => b.updatedAt.getTime() - a.updatedAt.getTime());
+    }
+    extractTitle(messages) {
+        const firstUserMessage = messages.find(m => m.role === 'user');
+        if (firstUserMessage) {
+            const textPart = firstUserMessage.parts.find(p => p.type === 'text');
+            if (textPart?.text) {
+                return textPart.text.slice(0, 50) + (textPart.text.length > 50 ? '...' : '');
+            }
+        }
+        return 'Untitled Chat';
+    }
+}
+//# sourceMappingURL=persistence.js.map

package/dist/utils/persistence.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"persistence.js","sourceRoot":"","sources":["../../src/utils/persistence.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAiBH;;GAEG;AACH,MAAM,UAAU,iBAAiB,CAAC,UAA8B,EAAE;IAChE,MAAM,EAAE,MAAM,GAAG,EAAE,EAAE,IAAI,GAAG,EAAE,EAAE,GAAG,OAAO,CAAC;IAE3C,OAAO,GAAW,EAAE;QAClB,MAAM,KAAK,GAAG,gEAAgE,CAAC;QAC/E,IAAI,MAAM,GAAG,EAAE,CAAC;QAEhB,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,IAAI,EAAE,CAAC,EAAE,EAAE,CAAC;YAC9B,MAAM,IAAI,KAAK,CAAC,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,MAAM,EAAE,GAAG,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC;QACnE,CAAC;QAED,OAAO,MAAM,CAAC,CAAC,CAAC,GAAG,MAAM,IAAI,MAAM,EAAE,CAAC,CAAC,CAAC,MAAM,CAAC;IACjD,CAAC,CAAC;AACJ,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,UAAU;IACxB,OAAO,iBAAiB,EAAE,EAAE,CAAC;AAC/B,CAAC;AAgBD;;GAEG;AACH,MAAM,CAAC,KAAK,UAAU,kBAAkB,CAAgC,EACtE,QAAQ,EACR,KAAK,EACL,eAAe,EACf,cAAc,GACe;IAC7B,MAAM,iBAAiB,GAAgB,EAAE,CAAC;IAE1C,KAAK,MAAM,OAAO,IAAI,QAAQ,EAAE,CAAC;QAC/B,IAAI,CAAC;YACH,MAAM,gBAAgB,GAAG,MAAM,qBAAqB,CAAC,OAAO,EAAE;gBAC5D,KAAK;gBACL,eAAe;gBACf,cAAc;aACf,CAAC,CAAC;YACH,iBAAiB,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC;QAC3C,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,mBAAmB;YACnB,mBAAmB;YACnB,iBAAiB,CAAC,IAAI,CAAC,eAAe,CAAC,OAAO,CAAC,CAAC,CAAC;QACnD,CAAC;IACH,CAAC;IAED,OAAO,iBAAiB,CAAC;AAC3B,CAAC;AAED;;GAEG;AACH,KAAK,UAAU,qBAAqB,CAClC,OAAkB,EAClB,OAAuD;IAEvD,MAAM,EAAE,KAAK,EAAE,eAAe,EAAE,cAAc,EAAE,GAAG,OAAO,CAAC;IAE3D,SAAS;IACT,IAAI,CAAC,OAAO,CAAC,EAAE,IAAI,CAAC,OAAO,CAAC,IAAI,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE,CAAC;QAClE,MAAM,IAAI,KAAK,CAAC,2BAA2B,CAAC,CAAC;IAC/C,CAAC;IAED,SAAS;IACT,MAAM,cAAc,GAAG,EAAE,CAAC;IAC1B,KAAK,MAAM,IAAI,IAAI,OAAO,CAAC,KAAK,EAAE,CAAC;QACjC,MAAM,aAAa,GAAG,MAAM,mBAAmB,CAAC,IAAI,EAAE,KAAK,CAAC,CAAC;QAC7D,cAAc,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;IACrC,CAAC;IAED,QAAQ;IACR,IAAI,iBAAiB,GAAG,OAAO,CAAC,QAAQ,CAAC;IACzC,IAAI,cAAc,IAAI,OAAO,CAAC,QAAQ,EAAE,CAAC;QACvC,IAAI,CAAC;YACH,iBAAiB,GAAG,cAAc,CAAC,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;QAC7D,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,iBAAiB,GAAG,SAAS,CAAC;QAChC,CAAC;IACH,CAAC;IAED,OAAO;QACL,GAAG,OAAO;QACV,KAAK,EAAE,cAAc;QACrB,QAAQ,EAAE,iBAAiB;KAC5B,CAAC;AACJ,CAAC;AAED;;GAEG;AACH,KAAK,UAAU,mBAAmB,CAAC,IAAS,EAAE,KAA2B;IACvE,SAAS;IACT,IAAI,IAAI,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;QACzB,IAAI,OAAO,IAAI,CAAC,IAAI,KAAK,QAAQ,EAAE,CAAC;YAClC,MAAM,IAAI,KAAK,CAAC,4CAA4C,CAAC,CAAC;QAChE,CAAC;QACD,OAAO,IAAI,CAAC;IACd,CAAC;IAED,SAAS;IACT,IAAI,IAAI,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;QACzB,IAAI,CAAC,IAAI,CAAC,QAAQ,IAAI,CAAC,IAAI,CAAC,GAAG,EAAE,CAAC;YAChC,MAAM,IAAI,KAAK,CAAC,iDAAiD,CAAC,CAAC;QACrE,CAAC;QACD,OAAO,IAAI,CAAC;IACd,CAAC;IAED,WAAW;IACX,IAAI,IAAI,CAAC,IAAI,CAAC,UAAU,CAAC,OAAO,CAAC,IAAI,IAAI,CAAC,IAAI,KAAK,cAAc,EAAE,CAAC;QAClE,MAAM,QAAQ,GAAG,IAAI,CAAC,IAAI,KAAK,cAAc,CAAC,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC;QAE/F,qBAAqB;QACrB,IAAI,KAAK,IAAI,QAAQ,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC,EAAE,CAAC;YAC1C,OAAO,CAAC,IAAI,CAAC,QAAQ,QAAQ,8BAA8B,CAAC,CAAC;QAC/D,CAAC;QAED,OAAO,IAAI,CAAC;IACd,CAAC;IAED,SAAS;IACT,IAAI,IAAI,CAAC,IAAI,CAAC,UAAU,CAAC,OAAO,CAAC,EAAE,CAAC;QAClC,IAAI,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC;YACf,MAAM,IAAI,KAAK,CAAC,qCAAqC,CAAC,CAAC;QACzD,CAAC;QACD,OAAO,IAAI,CAAC;IACd,CAAC;IAED,OAAO,IAAI,CAAC;AACd,CAAC;AAED;;GAEG;AACH,SAAS,eAAe,CAAC,OAAkB;IACzC,MAAM,cAAc,GAAG,OAAO,CAAC,KAAK,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE;IACjD,YAAY;IACX,IAAI,IAAI,OAAO,IAAI,KAAK,QAAQ,IAAI,IAAI,CAAC,IAAI,CAC/C,CAAC;IAEF,OAAO;QACL,GAAG,OAAO;QACV,KAAK,EAAE,cAAc,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,cAAc,CAAC,CAAC,CAAC;YAClD,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,0CAA0C,EAAE;SACnE;KACF,CAAC;AACJ,CAAC;AAED,mCAAmC;AACnC,YAAY;AACZ,mCAAmC;AAEnC;;GAEG;AACH,MAAM,OAAO,mBAAoB,SAAQ,KAAK;IAC5C,YAAY,OAAe,EAAkB,OAAa;QACxD,KAAK,CAAC,OAAO,CAAC,CAAC;QAD4B,YAAO,GAAP,OAAO,CAAM;QAExD,IAAI,CAAC,IAAI,GAAG,qBAAqB,CAAC;IACpC,CAAC;CACF;AAiBD;;GAEG;AACH,MAAM,OAAO,oBAAoB;IAAjC;QACU,UAAK,GAAG,IAAI,GAAG,EAAuB,CAAC;QACvC,aAAQ,GAAG,IAAI,GAAG,EAA+C,CAAC;IAoD5E,CAAC;IAlDC,KAAK,CAAC,UAAU;QACd,MAAM,EAAE,GAAG,UAAU,EAAE,CAAC;QACxB,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC;QACvB,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,EAAE,EAAE,EAAE,SAAS,EAAE,IAAI,IAAI,EAAE,EAAE,CAAC,CAAC;QACjD,OAAO,EAAE,CAAC;IACZ,CAAC;IAED,KAAK,CAAC,QAAQ,CAAC,EAAU;QACvB,MAAM,QAAQ,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;QACpC,IAAI,CAAC,QAAQ,EAAE,CAAC;YACd,MAAM,IAAI,KAAK,CAAC,QAAQ,EAAE,YAAY,CAAC,CAAC;QAC1C,CAAC;QACD,OAAO,CAAC,GAAG,QAAQ,CAAC,CAAC;IACvB,CAAC;IAED,KAAK,CAAC,QAAQ,CAAC,EAAE,MAAM,EAAE,QAAQ,EAA6C;QAC5E,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,MAAM,EAAE,CAAC,GAAG,QAAQ,CAAC,CAAC,CAAC;QAEtC,QAAQ;QACR,MAAM,QAAQ,GAAG,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,MAAM,CAAC,IAAI,EAAE,CAAC;QACjD,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,MAAM,EAAE;YACxB,GAAG,QAAQ;YACX,SAAS,EAAE,IAAI,IAAI,EAAE;YACrB,KAAK,EAAG,QAAgB,CAAC,KAAK,IAAI,IAAI,CAAC,YAAY,CAAC,QAAQ,CAAC;SAC9D,CAAC,CAAC;IACL,CAAC;IAED,KAAK,CAAC,UAAU,CAAC,EAAU;QACzB,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;QACtB,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;IAC3B,CAAC;IAED,KAAK,CAAC,SAAS;QACb,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,OAAO,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,CAAC;YAC9D,EAAE;YACF,GAAG,IAAI;SACR,CAAC,CAAC,CAAC;IACN,CAAC;IAEO,YAAY,CAAC,QAAqB;QACxC,gBAAgB;QAChB,MAAM,gBAAgB,GAAG,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,MAAM,CAAC,CAAC;QAC/D,IAAI,gBAAgB,EAAE,CAAC;YACrB,MAAM,QAAQ,GAAG,gBAAgB,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,MAAM,CAAQ,CAAC;YAC5E,IAAI,QAAQ,EAAE,IAAI,EAAE,CAAC;gBACnB,OAAO,QAAQ,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,EAAE,EAAE,CAAC,GAAG,CAAC,QAAQ,CAAC,IAAI,CAAC,MAAM,GAAG,EAAE,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;YAC/E,CAAC;QACH,CAAC;QACD,OAAO,eAAe,CAAC;IACzB,CAAC;CACF;AAED;;GAEG;AACH,MAAM,OAAO,wBAAwB;IAArC;QACmB,cAAS,GAAG,0BAA0B,CAAC;IAgG1D,CAAC;IA9FC,KAAK,CAAC,UAAU;QACd,MAAM,EAAE,GAAG,UAAU,EAAE,CAAC;QACxB,MAAM,GAAG,GAAG,IAAI,CAAC,SAAS,GAAG,EAAE,CAAC;QAEhC,MAAM,QAAQ,GAAG;YACf,EAAE;YACF,QAAQ,EAAE,EAAE;YACZ,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;YACnC,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;SACpC,CAAC;QAEF,YAAY,CAAC,OAAO,CAAC,GAAG,EAAE,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC,CAAC;QACpD,OAAO,EAAE,CAAC;IACZ,CAAC;IAED,KAAK,CAAC,QAAQ,CAAC,EAAU;QACvB,MAAM,GAAG,GAAG,IAAI,CAAC,SAAS,GAAG,EAAE,CAAC;QAChC,MAAM,IAAI,GAAG,YAAY,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC;QAEvC,IAAI,CAAC,IAAI,EAAE,CAAC;YACV,MAAM,IAAI,KAAK,CAAC,QAAQ,EAAE,YAAY,CAAC,CAAC;QAC1C,CAAC;QAED,IAAI,CAAC;YACH,MAAM,QAAQ,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;YAClC,OAAO,QAAQ,CAAC,QAAQ,IAAI,EAAE,CAAC;QACjC,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,IAAI,KAAK,CAAC,8BAA8B,KAAK,EAAE,CAAC,CAAC;QACzD,CAAC;IACH,CAAC;IAED,KAAK,CAAC,QAAQ,CAAC,EAAE,MAAM,EAAE,QAAQ,EAA6C;QAC5E,MAAM,GAAG,GAAG,IAAI,CAAC,SAAS,GAAG,MAAM,CAAC;QACpC,MAAM,QAAQ,GAAG,YAAY,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC;QAE3C,MAAM,QAAQ,GAAQ;YACpB,EAAE,EAAE,MAAM;YACV,QAAQ;YACR,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;YACnC,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;SACpC,CAAC;QAEF,IAAI,QAAQ,EAAE,CAAC;YACb,IAAI,CAAC;gBACH,MAAM,cAAc,GAAG,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC;gBAC5C,QAAQ,CAAC,SAAS,GAAG,cAAc,CAAC,SAAS,IAAI,QAAQ,CAAC,SAAS,CAAC;YACtE,CAAC;YAAC,MAAM,CAAC;gBACP,2BAA2B;YAC7B,CAAC;QACH,CAAC;QAED,YAAY,CAAC,OAAO,CAAC,GAAG,EAAE,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC,CAAC;IACtD,CAAC;IAED,KAAK,CAAC,UAAU,CAAC,EAAU;QACzB,MAAM,GAAG,GAAG,IAAI,CAAC,SAAS,GAAG,EAAE,CAAC;QAChC,YAAY,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC;IAC/B,CAAC;IAED,KAAK,CAAC,SAAS;QACb,MAAM,KAAK,GAA2D,EAAE,CAAC;QAEzE,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,YAAY,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;YAC7C,MAAM,GAAG,GAAG,YAAY,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC;YAChC,IAAI,GAAG,EAAE,UAAU,CAAC,IAAI,CAAC,SAAS,CAAC,EAAE,CAAC;gBACpC,IAAI,CAAC;oBACH,MAAM,IAAI,GAAG,YAAY,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC;oBACvC,IAAI,IAAI,EAAE,CAAC;wBACT,MAAM,QAAQ,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;wBAClC,KAAK,CAAC,IAAI,CAAC;4BACT,EAAE,EAAE,QAAQ,CAAC,EAAE;4BACf,KAAK,EAAE,IAAI,CAAC,YAAY,CAAC,QAAQ,CAAC,QAAQ,IAAI,EAAE,CAAC;4BACjD,SAAS,EAAE,IAAI,IAAI,CAAC,QAAQ,CAAC,SAAS,IAAI,QAAQ,CAAC,SAAS,CAAC;yBAC9D,CAAC,CAAC;oBACL,CAAC;gBACH,CAAC;gBAAC,MAAM,CAAC;oBACP,gCAAgC;gBAClC,CAAC;YACH,CAAC;QACH,CAAC;QAED,OAAO,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,SAAS,CAAC,OAAO,EAAE,GAAG,CAAC,CAAC,SAAS,CAAC,OAAO,EAAE,CAAC,CAAC;IAC7E,CAAC;IAEO,YAAY,CAAC,QAAqB;QACxC,MAAM,gBAAgB,GAAG,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,MAAM,CAAC,CAAC;QAC/D,IAAI,gBAAgB,EAAE,CAAC;YACrB,MAAM,QAAQ,GAAG,gBAAgB,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,MAAM,CAAQ,CAAC;YAC5E,IAAI,QAAQ,EAAE,IAAI,EAAE,CAAC;gBACnB,OAAO,QAAQ,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,EAAE,EAAE,CAAC,GAAG,CAAC,QAAQ,CAAC,IAAI,CAAC,MAAM,GAAG,EAAE,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;YAC/E,CAAC;QACH,CAAC;QACD,OAAO,eAAe,CAAC;IACzB,CAAC;CACF"}

package/dist/utils/streaming.d.ts

@@ -0,0 +1,144 @@
+/**
+ * Entity Engine AI - Streaming Utils
+ *
+ * Stream processing utilities for message stream reading and processing
+ */
+import type { UIMessage } from 'ai';
+/**
+ * Read UI message stream options
+ */
+export interface ReadUIMessageStreamOptions {
+    stream: ReadableStream;
+    message?: UIMessage;
+    onMessage?: (message: UIMessage) => void;
+    onError?: (error: Error) => void;
+    onComplete?: (finalMessage: UIMessage) => void;
+}
+/**
+ * UI message stream reader
+ */
+export declare class UIMessageStreamReader {
+    private options;
+    private reader;
+    private currentMessage;
+    private isReading;
+    constructor(options: ReadUIMessageStreamOptions);
+    /**
+     * Start reading stream
+     */
+    read(): AsyncIterableIterator<UIMessage>;
+    /**
+     * Process stream chunk
+     */
+    private processChunk;
+    /**
+     * Handle stream data
+     */
+    private handleStreamData;
+    /**
+     * Handle message start
+     */
+    private handleMessageStart;
+    /**
+     * Handle message delta
+     */
+    private handleMessageDelta;
+    /**
+     * Handle message stop
+     */
+    private handleMessageStop;
+    /**
+     * Handle text delta
+     */
+    private handleTextDelta;
+    /**
+     * Handle tool call
+     */
+    private handleToolCall;
+    /**
+     * Handle tool result
+     */
+    private handleToolResult;
+    /**
+     * Generate ID
+     */
+    private generateId;
+    /**
+     * Stop reading
+     */
+    stop(): Promise<void>;
+    /**
+     * Cleanup resources
+     */
+    private cleanup;
+}
+/**
+ * Read UI message stream
+ */
+export declare function readUIMessageStream(options: ReadUIMessageStreamOptions): AsyncIterableIterator<UIMessage>;
+/**
+ * Stream resume options
+ */
+export interface StreamResumeOptions {
+    chatId: string;
+    messageId?: string;
+    endpoint?: string;
+    headers?: Record<string, string>;
+}
+/**
+ * Resume stream
+ */
+export declare function resumeStream({ chatId, messageId, endpoint, headers, }: StreamResumeOptions): Promise<ReadableStream>;
+/**
+ * Stream state
+ */
+export type StreamState = 'idle' | 'connecting' | 'streaming' | 'complete' | 'error' | 'aborted';
+/**
+ * Stream state manager
+ */
+export declare class StreamStateManager {
+    private state;
+    private listeners;
+    private abortController;
+    /**
+     * Get current state
+     */
+    getState(): StreamState;
+    /**
+     * Set state
+     */
+    setState(newState: StreamState): void;
+    /**
+     * Listen to state changes
+     */
+    onStateChange(listener: (state: StreamState) => void): () => void;
+    /**
+     * Start stream
+     */
+    startStream(): AbortController;
+    /**
+     * Abort stream
+     */
+    abortStream(): void;
+    /**
+     * Complete stream
+     */
+    completeStream(): void;
+    /**
+     * Stream error
+     */
+    errorStream(): void;
+    /**
+     * Reset state
+     */
+    reset(): void;
+    /**
+     * Notify listeners
+     */
+    private notifyListeners;
+    /**
+     * Dispose
+     */
+    dispose(): void;
+}
+//# sourceMappingURL=streaming.d.ts.map

package/dist/utils/streaming.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"streaming.d.ts","sourceRoot":"","sources":["../../src/utils/streaming.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,IAAI,CAAC;AAIpC;;GAEG;AACH,MAAM,WAAW,0BAA0B;IACzC,MAAM,EAAE,cAAc,CAAC;IACvB,OAAO,CAAC,EAAE,SAAS,CAAC;IACpB,SAAS,CAAC,EAAE,CAAC,OAAO,EAAE,SAAS,KAAK,IAAI,CAAC;IACzC,OAAO,CAAC,EAAE,CAAC,KAAK,EAAE,KAAK,KAAK,IAAI,CAAC;IACjC,UAAU,CAAC,EAAE,CAAC,YAAY,EAAE,SAAS,KAAK,IAAI,CAAC;CAChD;AAED;;GAEG;AACH,qBAAa,qBAAqB;IAKpB,OAAO,CAAC,OAAO;IAJ3B,OAAO,CAAC,MAAM,CAA4C;IAC1D,OAAO,CAAC,cAAc,CAA0B;IAChD,OAAO,CAAC,SAAS,CAAS;gBAEN,OAAO,EAAE,0BAA0B;IAIvD;;OAEG;IACI,IAAI,IAAI,qBAAqB,CAAC,SAAS,CAAC;IAoC/C;;OAEG;IACH,OAAO,CAAC,YAAY;IAwBpB;;OAEG;IACH,OAAO,CAAC,gBAAgB;IAyBxB;;OAEG;IACH,OAAO,CAAC,kBAAkB;IAY1B;;OAEG;IACH,OAAO,CAAC,kBAAkB;IAoB1B;;OAEG;IACH,OAAO,CAAC,iBAAiB;IAgBzB;;OAEG;IACH,OAAO,CAAC,eAAe;IAsBvB;;OAEG;IACH,OAAO,CAAC,cAAc;IAsBtB;;OAEG;IACH,OAAO,CAAC,gBAAgB;IAqBxB;;OAEG;IACH,OAAO,CAAC,UAAU;IAIlB;;OAEG;IACG,IAAI,IAAI,OAAO,CAAC,IAAI,CAAC;IAO3B;;OAEG;IACH,OAAO,CAAC,OAAO;CAOhB;AAED;;GAEG;AACH,wBAAuB,mBAAmB,CACxC,OAAO,EAAE,0BAA0B,GAClC,qBAAqB,CAAC,SAAS,CAAC,CAGlC;AAID;;GAEG;AACH,MAAM,WAAW,mBAAmB;IAClC,MAAM,EAAE,MAAM,CAAC;IACf,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,OAAO,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;CAClC;AAED;;GAEG;AACH,wBAAsB,YAAY,CAAC,EACjC,MAAM,EACN,SAAS,EACT,QAA6B,EAC7B,OAAY,GACb,EAAE,mBAAmB,GAAG,OAAO,CAAC,cAAc,CAAC,CAsB/C;AAID;;GAEG;AACH,MAAM,MAAM,WAAW,GAAG,MAAM,GAAG,YAAY,GAAG,WAAW,GAAG,UAAU,GAAG,OAAO,GAAG,SAAS,CAAC;AAEjG;;GAEG;AACH,qBAAa,kBAAkB;IAC7B,OAAO,CAAC,KAAK,CAAuB;IACpC,OAAO,CAAC,SAAS,CAA2C;IAC5D,OAAO,CAAC,eAAe,CAAgC;IAEvD;;OAEG;IACH,QAAQ,IAAI,WAAW;IAIvB;;OAEG;IACH,QAAQ,CAAC,QAAQ,EAAE,WAAW,GAAG,IAAI;IAOrC;;OAEG;IACH,aAAa,CAAC,QAAQ,EAAE,CAAC,KAAK,EAAE,WAAW,KAAK,IAAI,GAAG,MAAM,IAAI;IASjE;;OAEG;IACH,WAAW,IAAI,eAAe;IAM9B;;OAEG;IACH,WAAW,IAAI,IAAI;IAQnB;;OAEG;IACH,cAAc,IAAI,IAAI;IAKtB;;OAEG;IACH,WAAW,IAAI,IAAI;IAKnB;;OAEG;IACH,KAAK,IAAI,IAAI;IAKb;;OAEG;IACH,OAAO,CAAC,eAAe;IAUvB;;OAEG;IACH,OAAO,IAAI,IAAI;CAKhB"}