@cmdoss/memwal-sdk 0.6.0

package/dist/services/ChatService.js

@@ -0,0 +1,319 @@
+/**
+ * ChatService handles all chat-related operations including session management,
+ * message sending, and streaming responses with memory context integration.
+ */
+export class ChatService {
+    constructor(apiClient) {
+        this.apiClient = apiClient;
+    }
+    /**
+     * Get all chat sessions for a user
+     */
+    async getSessions(userAddress) {
+        try {
+            const response = await this.apiClient.getChatSessions(userAddress);
+            return {
+                success: response.success,
+                sessions: response.data?.sessions || [],
+                message: response.message
+            };
+        }
+        catch (error) {
+            console.error('Failed to get chat sessions:', error);
+            throw new Error(`Failed to get chat sessions: ${error instanceof Error ? error.message : String(error)}`);
+        }
+    }
+    /**
+     * Get a specific chat session with messages
+     */
+    async getSession(sessionId, userAddress) {
+        try {
+            const response = await this.apiClient.getChatSession(sessionId, userAddress);
+            return {
+                success: response.success,
+                session: response.data,
+                message: response.message
+            };
+        }
+        catch (error) {
+            console.error(`Failed to get session ${sessionId}:`, error);
+            throw new Error(`Failed to get session: ${error instanceof Error ? error.message : String(error)}`);
+        }
+    }
+    /**
+     * Create a new chat session
+     */
+    async createSession(request) {
+        try {
+            const response = await this.apiClient.createChatSession(request);
+            return {
+                success: response.success,
+                session: response.data,
+                message: response.message
+            };
+        }
+        catch (error) {
+            console.error('Failed to create chat session:', error);
+            throw new Error(`Failed to create chat session: ${error instanceof Error ? error.message : String(error)}`);
+        }
+    }
+    /**
+     * Delete a chat session
+     */
+    async deleteSession(sessionId, userAddress) {
+        try {
+            const response = await this.apiClient.deleteChatSession(sessionId, userAddress);
+            return {
+                success: response.success,
+                message: response.message
+            };
+        }
+        catch (error) {
+            console.error(`Failed to delete session ${sessionId}:`, error);
+            throw new Error(`Failed to delete session: ${error instanceof Error ? error.message : String(error)}`);
+        }
+    }
+    /**
+     * Update the title of a chat session
+     */
+    async updateSessionTitle(sessionId, userAddress, title) {
+        try {
+            const response = await this.apiClient.updateChatSessionTitle(sessionId, userAddress, title);
+            return {
+                success: response.success,
+                message: response.message
+            };
+        }
+        catch (error) {
+            console.error(`Failed to update session title for ${sessionId}:`, error);
+            throw new Error(`Failed to update session title: ${error instanceof Error ? error.message : String(error)}`);
+        }
+    }
+    /**
+     * Add a message to a chat session
+     */
+    async addMessage(sessionId, request) {
+        try {
+            const response = await this.apiClient.addMessageToSession(sessionId, request.content, request.type, request.userAddress);
+            return {
+                success: response.success,
+                message: response.message
+            };
+        }
+        catch (error) {
+            console.error(`Failed to add message to session ${sessionId}:`, error);
+            throw new Error(`Failed to add message: ${error instanceof Error ? error.message : String(error)}`);
+        }
+    }
+    /**
+     * Save a summary for a chat session
+     */
+    async saveSummary(request) {
+        try {
+            const response = await this.apiClient.saveChatSummary(request.sessionId, request.summary, request.userAddress);
+            return {
+                success: response.success,
+                message: response.message
+            };
+        }
+        catch (error) {
+            console.error('Failed to save chat summary:', error);
+            throw new Error(`Failed to save summary: ${error instanceof Error ? error.message : String(error)}`);
+        }
+    }
+    /**
+     * Send a non-streaming chat message
+     */
+    async sendMessage(request) {
+        try {
+            const response = await this.apiClient.sendChatMessage({
+                text: request.text,
+                userId: request.userId,
+                sessionId: request.sessionId,
+                model: request.model,
+                userAddress: request.userAddress,
+                memoryContext: request.memoryContext
+            });
+            return response.data;
+        }
+        catch (error) {
+            console.error('Failed to send chat message:', error);
+            throw new Error(`Failed to send message: ${error instanceof Error ? error.message : String(error)}`);
+        }
+    }
+    /**
+     * Stream chat responses using Server-Sent Events
+     * Returns a promise that resolves when the stream completes
+     */
+    async streamChat(request, options = {}) {
+        const { onMessage, onThinking, onMemory, onError, onDone, abortController } = options;
+        return new Promise((resolve, reject) => {
+            try {
+                // Since EventSource only supports GET, we'll use fetch for streaming POST
+                this.streamChatWithFetch(request, options)
+                    .then(resolve)
+                    .catch(reject);
+            }
+            catch (error) {
+                console.error('Failed to initialize chat stream:', error);
+                reject(new Error(`Failed to stream chat: ${error instanceof Error ? error.message : String(error)}`));
+            }
+        });
+    }
+    /**
+     * Stream chat using fetch API with streaming response
+     */
+    async streamChatWithFetch(request, options = {}) {
+        const { onMessage, onThinking, onMemory, onError, onDone, abortController } = options;
+        try {
+            const response = await fetch(`${this.apiClient.baseURL}/chat/stream`, {
+                method: 'POST',
+                headers: {
+                    'Content-Type': 'application/json',
+                    'Accept': 'text/event-stream',
+                    ...this.apiClient.defaultHeaders,
+                },
+                body: JSON.stringify(request),
+                signal: abortController?.signal,
+            });
+            if (!response.ok) {
+                throw new Error(`HTTP error! status: ${response.status}`);
+            }
+            const reader = response.body?.getReader();
+            if (!reader) {
+                throw new Error('Response body is not readable');
+            }
+            const decoder = new TextDecoder();
+            let buffer = '';
+            try {
+                while (true) {
+                    const { done, value } = await reader.read();
+                    if (done) {
+                        onDone?.();
+                        break;
+                    }
+                    buffer += decoder.decode(value, { stream: true });
+                    const lines = buffer.split('\n');
+                    buffer = lines.pop() || ''; // Keep incomplete line in buffer
+                    for (const line of lines) {
+                        if (line.startsWith('data: ')) {
+                            const eventData = line.slice(6).trim();
+                            if (eventData === '')
+                                continue; // Empty data line
+                            try {
+                                const event = JSON.parse(eventData);
+                                switch (event.type) {
+                                    case 'message':
+                                        onMessage?.(event);
+                                        break;
+                                    case 'thinking':
+                                        onThinking?.(event);
+                                        break;
+                                    case 'memory':
+                                        onMemory?.(event);
+                                        break;
+                                    case 'error':
+                                        onError?.(event);
+                                        break;
+                                    case 'done':
+                                        onDone?.();
+                                        return;
+                                    default:
+                                        console.warn('Unknown event type:', event.type);
+                                }
+                            }
+                            catch (parseError) {
+                                console.error('Failed to parse SSE event:', parseError, 'Data:', eventData);
+                            }
+                        }
+                    }
+                }
+            }
+            finally {
+                reader.releaseLock();
+            }
+        }
+        catch (error) {
+            console.error('Streaming chat error:', error);
+            onError?.({
+                type: 'error',
+                data: JSON.stringify({ message: error instanceof Error ? error.message : String(error) }),
+                timestamp: new Date().toISOString(),
+            });
+            throw error;
+        }
+    }
+    /**
+     * Convenient method to stream chat with simple callback
+     */
+    async streamChatSimple(request, onMessage, onError, onDone) {
+        return this.streamChat(request, {
+            onMessage: (event) => {
+                try {
+                    const data = JSON.parse(event.data);
+                    if (data.content) {
+                        onMessage(data.content);
+                    }
+                }
+                catch (error) {
+                    console.error('Failed to parse message event:', error);
+                }
+            },
+            onError: (event) => {
+                try {
+                    const data = JSON.parse(event.data);
+                    onError?.(data.message || 'Unknown error');
+                }
+                catch (error) {
+                    onError?.(event.data);
+                }
+            },
+            onDone,
+        });
+    }
+    /**
+     * Create a convenient chat interface with session management
+     */
+    async createChatInterface(userAddress, modelName = process.env.AI_CHAT_MODEL || 'google/gemini-2.5-flash') {
+        const sessionResponse = await this.createSession({
+            userAddress,
+            modelName,
+            title: 'New Chat',
+        });
+        const sessionId = sessionResponse.session.id;
+        return {
+            sessionId,
+            session: sessionResponse.session,
+            sendMessage: async (text) => {
+                return this.sendMessage({
+                    text,
+                    userId: userAddress,
+                    sessionId,
+                    userAddress,
+                });
+            },
+            streamMessage: async (text, options = {}) => {
+                return this.streamChat({
+                    text,
+                    userId: userAddress,
+                    sessionId,
+                    userAddress,
+                }, options);
+            },
+            updateTitle: async (title) => {
+                return this.updateSessionTitle(sessionId, userAddress, title);
+            },
+            addMessage: async (content, type = 'user') => {
+                return this.addMessage(sessionId, {
+                    content,
+                    type,
+                    userAddress,
+                });
+            },
+            delete: async () => {
+                return this.deleteSession(sessionId, userAddress);
+            },
+        };
+    }
+}
+//# sourceMappingURL=ChatService.js.map

package/dist/services/ChatService.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"ChatService.js","sourceRoot":"","sources":["../../src/services/ChatService.ts"],"names":[],"mappings":"AAgBA;;;GAGG;AACH,MAAM,OAAO,WAAW;IACtB,YAAoB,SAAuB;QAAvB,cAAS,GAAT,SAAS,CAAc;IAAG,CAAC;IAE/C;;OAEG;IACH,KAAK,CAAC,WAAW,CAAC,WAAmB;QACnC,IAAI,CAAC;YACH,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,SAAS,CAAC,eAAe,CAAC,WAAW,CAAC,CAAC;YACnE,OAAO;gBACL,OAAO,EAAE,QAAQ,CAAC,OAAO;gBACzB,QAAQ,EAAE,QAAQ,CAAC,IAAI,EAAE,QAAQ,IAAI,EAAE;gBACvC,OAAO,EAAE,QAAQ,CAAC,OAAO;aAC1B,CAAC;QACJ,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,CAAC,KAAK,CAAC,8BAA8B,EAAE,KAAK,CAAC,CAAC;YACrD,MAAM,IAAI,KAAK,CAAC,gCAAgC,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;QAC5G,CAAC;IACH,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,UAAU,CAAC,SAAiB,EAAE,WAAmB;QACrD,IAAI,CAAC;YACH,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,SAAS,CAAC,cAAc,CAAC,SAAS,EAAE,WAAW,CAAC,CAAC;YAC7E,OAAO;gBACL,OAAO,EAAE,QAAQ,CAAC,OAAO;gBACzB,OAAO,EAAE,QAAQ,CAAC,IAAK;gBACvB,OAAO,EAAE,QAAQ,CAAC,OAAO;aAC1B,CAAC;QACJ,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,CAAC,KAAK,CAAC,yBAAyB,SAAS,GAAG,EAAE,KAAK,CAAC,CAAC;YAC5D,MAAM,IAAI,KAAK,CAAC,0BAA0B,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;QACtG,CAAC;IACH,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,aAAa,CAAC,OAAiC;QACnD,IAAI,CAAC;YACH,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,SAAS,CAAC,iBAAiB,CAAC,OAAO,CAAC,CAAC;YACjE,OAAO;gBACL,OAAO,EAAE,QAAQ,CAAC,OAAO;gBACzB,OAAO,EAAE,QAAQ,CAAC,IAAK;gBACvB,OAAO,EAAE,QAAQ,CAAC,OAAO;aAC1B,CAAC;QACJ,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,CAAC,KAAK,CAAC,gCAAgC,EAAE,KAAK,CAAC,CAAC;YACvD,MAAM,IAAI,KAAK,CAAC,kCAAkC,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;QAC9G,CAAC;IACH,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,aAAa,CAAC,SAAiB,EAAE,WAAmB;QACxD,IAAI,CAAC;YACH,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,SAAS,CAAC,iBAAiB,CAAC,SAAS,EAAE,WAAW,CAAC,CAAC;YAChF,OAAO;gBACL,OAAO,EAAE,QAAQ,CAAC,OAAO;gBACzB,OAAO,EAAE,QAAQ,CAAC,OAAO;aAC1B,CAAC;QACJ,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,CAAC,KAAK,CAAC,4BAA4B,SAAS,GAAG,EAAE,KAAK,CAAC,CAAC;YAC/D,MAAM,IAAI,KAAK,CAAC,6BAA6B,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;QACzG,CAAC;IACH,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,kBAAkB,CACtB,SAAiB,EACjB,WAAmB,EACnB,KAAa;QAEb,IAAI,CAAC;YACH,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,SAAS,CAAC,sBAAsB,CAAC,SAAS,EAAE,WAAW,EAAE,KAAK,CAAC,CAAC;YAC5F,OAAO;gBACL,OAAO,EAAE,QAAQ,CAAC,OAAO;gBACzB,OAAO,EAAE,QAAQ,CAAC,OAAO;aAC1B,CAAC;QACJ,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,CAAC,KAAK,CAAC,sCAAsC,SAAS,GAAG,EAAE,KAAK,CAAC,CAAC;YACzE,MAAM,IAAI,KAAK,CAAC,mCAAmC,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;QAC/G,CAAC;IACH,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,UAAU,CAAC,SAAiB,EAAE,OAA0B;QAC5D,IAAI,CAAC;YACH,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,SAAS,CAAC,mBAAmB,CAAC,SAAS,EAAE,OAAO,CAAC,OAAO,EAAE,OAAO,CAAC,IAAI,EAAE,OAAO,CAAC,WAAW,CAAC,CAAC;YACzH,OAAO;gBACL,OAAO,EAAE,QAAQ,CAAC,OAAO;gBACzB,OAAO,EAAE,QAAQ,CAAC,OAAO;aAC1B,CAAC;QACJ,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,CAAC,KAAK,CAAC,oCAAoC,SAAS,GAAG,EAAE,KAAK,CAAC,CAAC;YACvE,MAAM,IAAI,KAAK,CAAC,0BAA0B,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;QACtG,CAAC;IACH,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,WAAW,CAAC,OAA2B;QAC3C,IAAI,CAAC;YACH,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,SAAS,CAAC,eAAe,CAAC,OAAO,CAAC,SAAS,EAAE,OAAO,CAAC,OAAO,EAAE,OAAO,CAAC,WAAW,CAAC,CAAC;YAC/G,OAAO;gBACL,OAAO,EAAE,QAAQ,CAAC,OAAO;gBACzB,OAAO,EAAE,QAAQ,CAAC,OAAO;aAC1B,CAAC;QACJ,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,CAAC,KAAK,CAAC,8BAA8B,EAAE,KAAK,CAAC,CAAC;YACrD,MAAM,IAAI,KAAK,CAAC,2BAA2B,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;QACvG,CAAC;IACH,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,WAAW,CAAC,OAA2B;QAC3C,IAAI,CAAC;YACH,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,SAAS,CAAC,eAAe,CAAC;gBACpD,IAAI,EAAE,OAAO,CAAC,IAAI;gBAClB,MAAM,EAAE,OAAO,CAAC,MAAM;gBACtB,SAAS,EAAE,OAAO,CAAC,SAAS;gBAC5B,KAAK,EAAE,OAAO,CAAC,KAAK;gBACpB,WAAW,EAAE,OAAO,CAAC,WAAW;gBAChC,aAAa,EAAE,OAAO,CAAC,aAAa;aACrC,CAAC,CAAC;YACH,OAAO,QAAQ,CAAC,IAAK,CAAC;QACxB,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,CAAC,KAAK,CAAC,8BAA8B,EAAE,KAAK,CAAC,CAAC;YACrD,MAAM,IAAI,KAAK,CAAC,2BAA2B,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;QACvG,CAAC;IACH,CAAC;IAED;;;OAGG;IACH,KAAK,CAAC,UAAU,CAAC,OAA2B,EAAE,UAA6B,EAAE;QAC3E,MAAM,EACJ,SAAS,EACT,UAAU,EACV,QAAQ,EACR,OAAO,EACP,MAAM,EACN,eAAe,EAChB,GAAG,OAAO,CAAC;QAEZ,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;YACrC,IAAI,CAAC;gBACH,0EAA0E;gBAC1E,IAAI,CAAC,mBAAmB,CAAC,OAAO,EAAE,OAAO,CAAC;qBACvC,IAAI,CAAC,OAAO,CAAC;qBACb,KAAK,CAAC,MAAM,CAAC,CAAC;YACnB,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,OAAO,CAAC,KAAK,CAAC,mCAAmC,EAAE,KAAK,CAAC,CAAC;gBAC1D,MAAM,CAAC,IAAI,KAAK,CAAC,0BAA0B,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC,CAAC;YACxG,CAAC;QACH,CAAC,CAAC,CAAC;IACL,CAAC;IAED;;OAEG;IACK,KAAK,CAAC,mBAAmB,CAAC,OAA2B,EAAE,UAA6B,EAAE;QAC5F,MAAM,EACJ,SAAS,EACT,UAAU,EACV,QAAQ,EACR,OAAO,EACP,MAAM,EACN,eAAe,EAChB,GAAG,OAAO,CAAC;QAEZ,IAAI,CAAC;YACH,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,GAAG,IAAI,CAAC,SAAS,CAAC,OAAO,cAAc,EAAE;gBACpE,MAAM,EAAE,MAAM;gBACd,OAAO,EAAE;oBACP,cAAc,EAAE,kBAAkB;oBAClC,QAAQ,EAAE,mBAAmB;oBAC7B,GAAG,IAAI,CAAC,SAAS,CAAC,cAAc;iBACjC;gBACD,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC;gBAC7B,MAAM,EAAE,eAAe,EAAE,MAAM;aAChC,CAAC,CAAC;YAEH,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;gBACjB,MAAM,IAAI,KAAK,CAAC,uBAAuB,QAAQ,CAAC,MAAM,EAAE,CAAC,CAAC;YAC5D,CAAC;YAED,MAAM,MAAM,GAAG,QAAQ,CAAC,IAAI,EAAE,SAAS,EAAE,CAAC;YAC1C,IAAI,CAAC,MAAM,EAAE,CAAC;gBACZ,MAAM,IAAI,KAAK,CAAC,+BAA+B,CAAC,CAAC;YACnD,CAAC;YAED,MAAM,OAAO,GAAG,IAAI,WAAW,EAAE,CAAC;YAClC,IAAI,MAAM,GAAG,EAAE,CAAC;YAEhB,IAAI,CAAC;gBACH,OAAO,IAAI,EAAE,CAAC;oBACZ,MAAM,EAAE,IAAI,EAAE,KAAK,EAAE,GAAG,MAAM,MAAM,CAAC,IAAI,EAAE,CAAC;oBAE5C,IAAI,IAAI,EAAE,CAAC;wBACT,MAAM,EAAE,EAAE,CAAC;wBACX,MAAM;oBACR,CAAC;oBAED,MAAM,IAAI,OAAO,CAAC,MAAM,CAAC,KAAK,EAAE,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC;oBAClD,MAAM,KAAK,GAAG,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;oBACjC,MAAM,GAAG,KAAK,CAAC,GAAG,EAAE,IAAI,EAAE,CAAC,CAAC,iCAAiC;oBAE7D,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;wBACzB,IAAI,IAAI,CAAC,UAAU,CAAC,QAAQ,CAAC,EAAE,CAAC;4BAC9B,MAAM,SAAS,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC;4BAEvC,IAAI,SAAS,KAAK,EAAE;gCAAE,SAAS,CAAC,kBAAkB;4BAElD,IAAI,CAAC;gCACH,MAAM,KAAK,GAAoB,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC;gCAErD,QAAQ,KAAK,CAAC,IAAI,EAAE,CAAC;oCACnB,KAAK,SAAS;wCACZ,SAAS,EAAE,CAAC,KAAK,CAAC,CAAC;wCACnB,MAAM;oCACR,KAAK,UAAU;wCACb,UAAU,EAAE,CAAC,KAAK,CAAC,CAAC;wCACpB,MAAM;oCACR,KAAK,QAAQ;wCACX,QAAQ,EAAE,CAAC,KAAK,CAAC,CAAC;wCAClB,MAAM;oCACR,KAAK,OAAO;wCACV,OAAO,EAAE,CAAC,KAAK,CAAC,CAAC;wCACjB,MAAM;oCACR,KAAK,MAAM;wCACT,MAAM,EAAE,EAAE,CAAC;wCACX,OAAO;oCACT;wCACE,OAAO,CAAC,IAAI,CAAC,qBAAqB,EAAE,KAAK,CAAC,IAAI,CAAC,CAAC;gCACpD,CAAC;4BACH,CAAC;4BAAC,OAAO,UAAU,EAAE,CAAC;gCACpB,OAAO,CAAC,KAAK,CAAC,4BAA4B,EAAE,UAAU,EAAE,OAAO,EAAE,SAAS,CAAC,CAAC;4BAC9E,CAAC;wBACH,CAAC;oBACH,CAAC;gBACH,CAAC;YACH,CAAC;oBAAS,CAAC;gBACT,MAAM,CAAC,WAAW,EAAE,CAAC;YACvB,CAAC;QACH,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,CAAC,KAAK,CAAC,uBAAuB,EAAE,KAAK,CAAC,CAAC;YAC9C,OAAO,EAAE,CAAC;gBACR,IAAI,EAAE,OAAO;gBACb,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,EAAE,OAAO,EAAE,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE,CAAC;gBACzF,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;aACpC,CAAC,CAAC;YACH,MAAM,KAAK,CAAC;QACd,CAAC;IACH,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,gBAAgB,CACpB,OAA2B,EAC3B,SAAoC,EACpC,OAAiC,EACjC,MAAmB;QAEnB,OAAO,IAAI,CAAC,UAAU,CAAC,OAAO,EAAE;YAC9B,SAAS,EAAE,CAAC,KAAK,EAAE,EAAE;gBACnB,IAAI,CAAC;oBACH,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;oBACpC,IAAI,IAAI,CAAC,OAAO,EAAE,CAAC;wBACjB,SAAS,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;oBAC1B,CAAC;gBACH,CAAC;gBAAC,OAAO,KAAK,EAAE,CAAC;oBACf,OAAO,CAAC,KAAK,CAAC,gCAAgC,EAAE,KAAK,CAAC,CAAC;gBACzD,CAAC;YACH,CAAC;YACD,OAAO,EAAE,CAAC,KAAK,EAAE,EAAE;gBACjB,IAAI,CAAC;oBACH,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;oBACpC,OAAO,EAAE,CAAC,IAAI,CAAC,OAAO,IAAI,eAAe,CAAC,CAAC;gBAC7C,CAAC;gBAAC,OAAO,KAAK,EAAE,CAAC;oBACf,OAAO,EAAE,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;gBACxB,CAAC;YACH,CAAC;YACD,MAAM;SACP,CAAC,CAAC;IACL,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,mBAAmB,CAAC,WAAmB,EAAE,YAAoB,OAAO,CAAC,GAAG,CAAC,aAAa,IAAI,yBAAyB;QACvH,MAAM,eAAe,GAAG,MAAM,IAAI,CAAC,aAAa,CAAC;YAC/C,WAAW;YACX,SAAS;YACT,KAAK,EAAE,UAAU;SAClB,CAAC,CAAC;QAEH,MAAM,SAAS,GAAG,eAAe,CAAC,OAAO,CAAC,EAAE,CAAC;QAE7C,OAAO;YACL,SAAS;YACT,OAAO,EAAE,eAAe,CAAC,OAAO;YAEhC,WAAW,EAAE,KAAK,EAAE,IAAY,EAAE,EAAE;gBAClC,OAAO,IAAI,CAAC,WAAW,CAAC;oBACtB,IAAI;oBACJ,MAAM,EAAE,WAAW;oBACnB,SAAS;oBACT,WAAW;iBACZ,CAAC,CAAC;YACL,CAAC;YAED,aAAa,EAAE,KAAK,EAAE,IAAY,EAAE,UAA6B,EAAE,EAAE,EAAE;gBACrE,OAAO,IAAI,CAAC,UAAU,CAAC;oBACrB,IAAI;oBACJ,MAAM,EAAE,WAAW;oBACnB,SAAS;oBACT,WAAW;iBACZ,EAAE,OAAO,CAAC,CAAC;YACd,CAAC;YAED,WAAW,EAAE,KAAK,EAAE,KAAa,EAAE,EAAE;gBACnC,OAAO,IAAI,CAAC,kBAAkB,CAAC,SAAS,EAAE,WAAW,EAAE,KAAK,CAAC,CAAC;YAChE,CAAC;YAED,UAAU,EAAE,KAAK,EAAE,OAAe,EAAE,OAAwC,MAAM,EAAE,EAAE;gBACpF,OAAO,IAAI,CAAC,UAAU,CAAC,SAAS,EAAE;oBAChC,OAAO;oBACP,IAAI;oBACJ,WAAW;iBACZ,CAAC,CAAC;YACL,CAAC;YAED,MAAM,EAAE,KAAK,IAAI,EAAE;gBACjB,OAAO,IAAI,CAAC,aAAa,CAAC,SAAS,EAAE,WAAW,CAAC,CAAC;YACpD,CAAC;SACF,CAAC;IACJ,CAAC;CACF"}

package/dist/services/ClassifierService.d.ts

@@ -0,0 +1,104 @@
+/**
+ * ClassifierService - Content Classification and Filtering
+ *
+ * Determines if content should be saved as memory using pattern matching
+ * and AI classification. Provides category classification for content organization.
+ */
+import { EmbeddingService } from './EmbeddingService';
+import type { ClientWithCoreApi, PDWConfig } from '../types';
+export interface ClassificationResult {
+    shouldSave: boolean;
+    confidence: number;
+    category: string;
+    reasoning: string;
+}
+export interface ClassificationOptions {
+    useAI?: boolean;
+    confidenceThreshold?: number;
+    categories?: string[];
+    patterns?: RegExp[];
+    /**
+     * If true, only explicit memory commands trigger save (e.g., "remember that", "store in memory")
+     * Default: true - to give users control over what gets saved
+     */
+    explicitOnly?: boolean;
+}
+export interface PatternAnalysisResult {
+    patterns: string[];
+    categories: string[];
+    matches: Array<{
+        pattern: string;
+        match: string;
+        category: string;
+        confidence: number;
+    }>;
+}
+/**
+ * Content classification service for determining memory worthiness
+ */
+export declare class ClassifierService {
+    private client?;
+    private config?;
+    private embeddingService?;
+    private aiApiKey?;
+    private readonly explicitPatterns;
+    private readonly factPatterns;
+    private readonly categoryMap;
+    constructor(client?: ClientWithCoreApi | undefined, config?: PDWConfig | undefined, embeddingService?: EmbeddingService, aiApiKey?: string);
+    /**
+     * Determine if a message contains information worth saving
+     * @param message User message to classify
+     * @param options Classification options
+     * @returns Classification result
+     */
+    shouldSaveMemory(message: string, options?: ClassificationOptions): Promise<ClassificationResult>;
+    /**
+     * Classify content into categories
+     * @param content Content to classify
+     * @param options Classification options
+     * @returns Category string
+     */
+    classifyContent(content: string, options?: ClassificationOptions): Promise<string>;
+    /**
+     * Analyze patterns in text
+     * @param text Text to analyze
+     * @param patterns Optional custom patterns
+     * @returns Pattern analysis result
+     */
+    analyzePatterns(text: string, patterns?: RegExp[]): Promise<PatternAnalysisResult>;
+    /**
+     * Match patterns against text
+     * @param text Text to match
+     * @param patterns Patterns to use
+     * @returns Match result
+     */
+    private matchPatterns;
+    /**
+     * Get category for a regex pattern
+     * @param patternString String representation of the regex
+     * @returns Category string
+     */
+    private getCategoryForPattern;
+    /**
+     * Calculate confidence score for pattern match
+     * @param pattern The regex pattern
+     * @param match The matched text
+     * @returns Confidence score (0-1)
+     */
+    private calculatePatternConfidence;
+    /**
+     * Use AI for classification (fallback/enhancement)
+     * @param message Message to classify
+     * @param options Classification options
+     * @returns Classification result
+     */
+    private classifyWithAI;
+    /**
+     * Classify using embeddings (semantic similarity approach)
+     * @param message Message to classify
+     * @param options Classification options
+     * @returns Classification result
+     */
+    private classifyWithEmbeddings;
+}
+//# sourceMappingURL=ClassifierService.d.ts.map

package/dist/services/ClassifierService.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"ClassifierService.d.ts","sourceRoot":"","sources":["../../src/services/ClassifierService.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAEH,OAAO,EAAE,gBAAgB,EAAE,MAAM,oBAAoB,CAAC;AACtD,OAAO,KAAK,EAAE,iBAAiB,EAAE,SAAS,EAAE,MAAM,UAAU,CAAC;AAE7D,MAAM,WAAW,oBAAoB;IACnC,UAAU,EAAE,OAAO,CAAC;IACpB,UAAU,EAAE,MAAM,CAAC;IACnB,QAAQ,EAAE,MAAM,CAAC;IACjB,SAAS,EAAE,MAAM,CAAC;CACnB;AAED,MAAM,WAAW,qBAAqB;IACpC,KAAK,CAAC,EAAE,OAAO,CAAC;IAChB,mBAAmB,CAAC,EAAE,MAAM,CAAC;IAC7B,UAAU,CAAC,EAAE,MAAM,EAAE,CAAC;IACtB,QAAQ,CAAC,EAAE,MAAM,EAAE,CAAC;IACpB;;;OAGG;IACH,YAAY,CAAC,EAAE,OAAO,CAAC;CACxB;AAED,MAAM,WAAW,qBAAqB;IACpC,QAAQ,EAAE,MAAM,EAAE,CAAC;IACnB,UAAU,EAAE,MAAM,EAAE,CAAC;IACrB,OAAO,EAAE,KAAK,CAAC;QACb,OAAO,EAAE,MAAM,CAAC;QAChB,KAAK,EAAE,MAAM,CAAC;QACd,QAAQ,EAAE,MAAM,CAAC;QACjB,UAAU,EAAE,MAAM,CAAC;KACpB,CAAC,CAAC;CACJ;AAED;;GAEG;AACH,qBAAa,iBAAiB;IAkG1B,OAAO,CAAC,MAAM,CAAC;IACf,OAAO,CAAC,MAAM,CAAC;IAlGjB,OAAO,CAAC,gBAAgB,CAAC,CAAmB;IAC5C,OAAO,CAAC,QAAQ,CAAC,CAAS;IAG1B,OAAO,CAAC,QAAQ,CAAC,gBAAgB,CAY/B;IAGF,OAAO,CAAC,QAAQ,CAAC,YAAY,CAgC3B;IAGF,OAAO,CAAC,QAAQ,CAAC,WAAW,CAwC1B;gBAGQ,MAAM,CAAC,EAAE,iBAAiB,YAAA,EAC1B,MAAM,CAAC,EAAE,SAAS,YAAA,EAC1B,gBAAgB,CAAC,EAAE,gBAAgB,EACnC,QAAQ,CAAC,EAAE,MAAM;IAMnB;;;;;OAKG;IACG,gBAAgB,CAAC,OAAO,EAAE,MAAM,EAAE,OAAO,GAAE,qBAA0B,GAAG,OAAO,CAAC,oBAAoB,CAAC;IA6E3G;;;;;OAKG;IACG,eAAe,CAAC,OAAO,EAAE,MAAM,EAAE,OAAO,GAAE,qBAA0B,GAAG,OAAO,CAAC,MAAM,CAAC;IAK5F;;;;;OAKG;IACG,eAAe,CAAC,IAAI,EAAE,MAAM,EAAE,QAAQ,GAAE,MAAM,EAAsB,GAAG,OAAO,CAAC,qBAAqB,CAAC;IAyC3G;;;;;OAKG;IACH,OAAO,CAAC,aAAa;IA6CrB;;;;OAIG;IACH,OAAO,CAAC,qBAAqB;IAmB7B;;;;;OAKG;IACH,OAAO,CAAC,0BAA0B;IA8BlC;;;;;OAKG;YACW,cAAc;IAoB5B;;;;;OAKG;YACW,sBAAsB;CA+BrC"}