@magic-ingredients/tiny-brain-local 0.3.10

package/dist/core/mcp-server.js

@@ -0,0 +1,295 @@
+/**
+ * Local MCP Server
+ *
+ * Simplified MCP server for local execution without deployment complexity
+ */
+import { Server as McpServer } from '@modelcontextprotocol/sdk/server/index.js';
+import { ListToolsRequestSchema, CallToolRequestSchema, ListPromptsRequestSchema, GetPromptRequestSchema, SetLevelRequestSchema, } from '@modelcontextprotocol/sdk/types.js';
+import { LocalFilesystemStorageAdapter } from '../storage/local-filesystem-adapter.js';
+import { FileLogger } from './file-logger.js';
+import { ToolRegistry } from '../tools/tool-registry.js';
+import { PromptRegistry } from '../prompts/prompt-registry.js';
+/**
+ * MCP Server for local execution
+ */
+export class MCPServer {
+    server;
+    storage;
+    logger;
+    config;
+    sessionId;
+    activePersona = 'default';
+    userId = 'local-user';
+    personaChangeListeners = [];
+    instanceId = `mcp-${Date.now()}-${Math.random().toString(36).substring(2, 9)}`;
+    constructor(config = {}) {
+        this.config = {
+            name: 'tiny-brain-local',
+            version: '0.3.7',
+            dataDir: process.env.TINY_BRAIN_DATA_DIR || '~/.tiny-brain',
+            logLevel: process.env.LOG_LEVEL || 'info',
+            watchAutoOpen: true,
+            ...config,
+        };
+        // Create logger - file logger for local (stdio transport requires no console output)
+        this.logger = new FileLogger({
+            logLevel: this.config.logLevel,
+            enabled: true,
+        });
+        // Create storage - always local filesystem for local package
+        this.storage = new LocalFilesystemStorageAdapter({
+            baseDir: this.config.dataDir,
+        });
+        // Generate session ID
+        this.sessionId = `local-${Date.now()}-${Math.random().toString(36).substring(2, 9)}`;
+        // Create MCP server instance
+        this.server = new McpServer({
+            name: this.config.name,
+            version: this.config.version,
+        }, {
+            capabilities: {
+                tools: {},
+                prompts: {},
+                logging: {},
+            },
+        });
+        // Set up handlers
+        this.setupHandlers();
+    }
+    setupHandlers() {
+        // Tool handlers
+        this.server.setRequestHandler(ListToolsRequestSchema, this.handleListTools.bind(this));
+        this.server.setRequestHandler(CallToolRequestSchema, this.handleCallTool.bind(this));
+        // Prompt handlers  
+        this.server.setRequestHandler(ListPromptsRequestSchema, this.handleListPrompts.bind(this));
+        this.server.setRequestHandler(GetPromptRequestSchema, this.handleGetPrompt.bind(this));
+        // Logging handler
+        this.server.setRequestHandler(SetLevelRequestSchema, this.handleSetLogLevel.bind(this));
+    }
+    /**
+     * Get server configuration
+     */
+    getConfig() {
+        return this.config;
+    }
+    /**
+     * Initialize the server
+     */
+    async initialize() {
+        this.logger.info('Initializing Local MCP Server', {
+            dataDir: this.config.dataDir,
+            logLevel: this.config.logLevel,
+        });
+        try {
+            // Initialize user storage first
+            await this.storage.initializeUserStorage(this.userId);
+            // Load default persona if available
+            try {
+                const personas = await this.storage.listPersonas();
+                if (personas.length > 0) {
+                    this.activePersona = personas[0];
+                    this.logger.info(`Loaded default persona: ${this.activePersona}`);
+                }
+                else {
+                    // Create default persona
+                    await this.storage.storePersonaFile('default', 'profile.json', JSON.stringify({
+                        id: 'default',
+                        name: 'Default',
+                        description: 'Default persona',
+                        created: new Date().toISOString(),
+                    }, null, 2), this.userId);
+                    this.activePersona = 'default';
+                    this.logger.info('Created default persona');
+                }
+            }
+            catch (error) {
+                this.logger.warn('Could not load personas, using default', error);
+                this.activePersona = 'default';
+            }
+        }
+        catch (error) {
+            this.logger.error('Failed to initialize storage', error);
+            throw error;
+        }
+    }
+    /**
+     * Connect to transport
+     */
+    async connect(transport) {
+        await this.server.connect(transport);
+        this.logger.info('Connected to transport');
+    }
+    /**
+     * Shutdown the server
+     */
+    async shutdown() {
+        this.logger.info('Shutting down Local MCP Server');
+        await this.server.close();
+    }
+    // Tool handlers
+    async handleListTools(_request) {
+        const tools = ToolRegistry.getTools();
+        this.logger.debug(`Listing ${tools.length} tools`);
+        return { tools };
+    }
+    async handleCallTool(request) {
+        const { name, arguments: args } = request.params;
+        this.logger.info(`Calling tool: ${name}`, { args });
+        // Create request context - simplified for local
+        const activePersona = this.activePersona ? {
+            id: this.activePersona,
+            userContent: {
+                profile: undefined,
+                background: undefined,
+                expertise: undefined,
+                preferences: undefined
+            }
+        } : undefined;
+        const context = {
+            sessionId: this.sessionId,
+            userId: (typeof args?.userId === 'string' ? args.userId : undefined) || this.userId,
+            storage: this.storage,
+            logger: this.logger.child(name),
+            input: args || {},
+            conversationId: `conv-${this.sessionId}`,
+            enhancementActive: false,
+            requestId: `req-${Date.now()}`,
+            timestamp: new Date(),
+            deploymentMode: 'local',
+            activePersona,
+            updateActivePersona: (persona) => {
+                const newPersonaId = persona?.id || 'default';
+                this.activePersona = newPersonaId;
+                this.logger.info(`[${this.instanceId}] Active persona updated to: ${newPersonaId}, notifying ${this.personaChangeListeners.length} listeners`);
+                // Notify all registered listeners
+                for (const listener of this.personaChangeListeners) {
+                    try {
+                        const result = listener(newPersonaId);
+                        if (result instanceof Promise) {
+                            result.catch(err => this.logger.error('Persona change listener error:', err));
+                        }
+                    }
+                    catch (err) {
+                        this.logger.error('Persona change listener error:', err);
+                    }
+                }
+            },
+            getCurrentActivePersona: () => this.activePersona,
+            personaChangeListeners: this.personaChangeListeners,
+        };
+        // Get the tool
+        const ToolClass = ToolRegistry.getToolByName(name);
+        if (!ToolClass) {
+            const error = `Unknown tool: ${name}`;
+            this.logger.error(error);
+            return {
+                content: [{ type: 'text', text: error }],
+                isError: true,
+            };
+        }
+        try {
+            // Execute the tool
+            const result = await ToolClass.execute(args || {}, context);
+            // Result already has content and isError properties
+            if (!result.isError) {
+                this.logger.debug(`Tool ${name} executed successfully`, {
+                    contentLength: result.content?.length,
+                    hasContent: !!result.content
+                });
+            }
+            else {
+                this.logger.warn(`Tool ${name} returned error`, { result });
+            }
+            return result;
+        }
+        catch (error) {
+            this.logger.error(`Tool execution failed: ${name}`, error);
+            return {
+                content: [
+                    {
+                        type: 'text',
+                        text: `Tool execution failed: ${error instanceof Error ? error.message : 'Unknown error'}`,
+                    },
+                ],
+                isError: true,
+            };
+        }
+    }
+    // Prompt handlers
+    async handleListPrompts(_request) {
+        const prompts = PromptRegistry.getPrompts();
+        this.logger.debug(`Listing ${prompts.length} prompts`);
+        return { prompts };
+    }
+    async handleGetPrompt(request) {
+        const { name, arguments: args } = request.params;
+        this.logger.info(`Getting prompt: ${name}`);
+        // Create request context
+        const activePersona = this.activePersona ? {
+            id: this.activePersona,
+            userContent: {
+                profile: undefined,
+                background: undefined,
+                expertise: undefined,
+                preferences: undefined
+            }
+        } : undefined;
+        const context = {
+            sessionId: this.sessionId,
+            userId: this.userId,
+            storage: this.storage,
+            logger: this.logger.child(`prompt:${name}`),
+            input: args || {},
+            conversationId: `conv-${this.sessionId}`,
+            enhancementActive: false,
+            requestId: `req-${Date.now()}`,
+            timestamp: new Date(),
+            deploymentMode: 'local',
+            activePersona,
+            updateActivePersona: (persona) => {
+                const newPersonaId = persona?.id || 'default';
+                this.activePersona = newPersonaId;
+                this.logger.info(`[${this.instanceId}] Active persona updated to: ${newPersonaId}, notifying ${this.personaChangeListeners.length} listeners`);
+                // Notify all registered listeners
+                for (const listener of this.personaChangeListeners) {
+                    try {
+                        const result = listener(newPersonaId);
+                        if (result instanceof Promise) {
+                            result.catch(err => this.logger.error('Persona change listener error:', err));
+                        }
+                    }
+                    catch (err) {
+                        this.logger.error('Persona change listener error:', err);
+                    }
+                }
+            },
+            getCurrentActivePersona: () => this.activePersona,
+            personaChangeListeners: this.personaChangeListeners,
+        };
+        // Get the prompt class
+        const PromptClass = PromptRegistry.getPromptByName(name);
+        if (!PromptClass) {
+            const error = `Unknown prompt: ${name}`;
+            this.logger.error(error);
+            throw new Error(error);
+        }
+        try {
+            // Execute the prompt handler
+            const result = await PromptClass.handle(name, args || {}, context);
+            this.logger.debug(`Prompt ${name} generated`);
+            return result;
+        }
+        catch (error) {
+            this.logger.error(`Error executing prompt ${name}:`, error);
+            throw error;
+        }
+    }
+    // Logging handler
+    async handleSetLogLevel(request) {
+        const { level } = request.params;
+        this.logger.setLogLevel(level);
+        this.logger.info(`Log level set to: ${level}`);
+        return {};
+    }
+}
+//# sourceMappingURL=mcp-server.js.map

package/dist/core/mcp-server.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"mcp-server.js","sourceRoot":"","sources":["../../src/core/mcp-server.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,MAAM,IAAI,SAAS,EAAE,MAAM,2CAA2C,CAAC;AAChF,OAAO,EACL,sBAAsB,EACtB,qBAAqB,EACrB,wBAAwB,EACxB,sBAAsB,EACtB,qBAAqB,GACtB,MAAM,oCAAoC,CAAC;AAc5C,OAAO,EAAE,6BAA6B,EAAE,MAAM,wCAAwC,CAAC;AACvF,OAAO,EAAE,UAAU,EAAE,MAAM,kBAAkB,CAAC;AAC9C,OAAO,EAAE,YAAY,EAAE,MAAM,2BAA2B,CAAC;AACzD,OAAO,EAAE,cAAc,EAAE,MAAM,+BAA+B,CAAC;AAa/D;;GAEG;AACH,MAAM,OAAO,SAAS;IACZ,MAAM,CAAY;IAClB,OAAO,CAA2B;IAClC,MAAM,CAAa;IACnB,MAAM,CAAY;IAClB,SAAS,CAAS;IAClB,aAAa,GAAW,SAAS,CAAC;IAClC,MAAM,GAAW,YAAY,CAAC;IAC9B,sBAAsB,GAAuD,EAAE,CAAC;IAChF,UAAU,GAAW,OAAO,IAAI,CAAC,GAAG,EAAE,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,SAAS,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,CAAC;IAE/F,YAAY,SAAoB,EAAE;QAChC,IAAI,CAAC,MAAM,GAAG;YACZ,IAAI,EAAE,kBAAkB;YACxB,OAAO,EAAE,OAAO;YAChB,OAAO,EAAE,OAAO,CAAC,GAAG,CAAC,mBAAmB,IAAI,eAAe;YAC3D,QAAQ,EAAG,OAAO,CAAC,GAAG,CAAC,SAAsB,IAAI,MAAM;YACvD,aAAa,EAAE,IAAI;YACnB,GAAG,MAAM;SACV,CAAC;QAEF,qFAAqF;QACrF,IAAI,CAAC,MAAM,GAAG,IAAI,UAAU,CAAC;YAC3B,QAAQ,EAAE,IAAI,CAAC,MAAM,CAAC,QAAQ;YAC9B,OAAO,EAAE,IAAI;SACd,CAAC,CAAC;QAEH,6DAA6D;QAC7D,IAAI,CAAC,OAAO,GAAG,IAAI,6BAA6B,CAAC;YAC/C,OAAO,EAAE,IAAI,CAAC,MAAM,CAAC,OAAiB;SACvC,CAAC,CAAC;QAEH,sBAAsB;QACtB,IAAI,CAAC,SAAS,GAAG,SAAS,IAAI,CAAC,GAAG,EAAE,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,SAAS,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,CAAC;QAErF,6BAA6B;QAC7B,IAAI,CAAC,MAAM,GAAG,IAAI,SAAS,CACzB;YACE,IAAI,EAAE,IAAI,CAAC,MAAM,CAAC,IAAc;YAChC,OAAO,EAAE,IAAI,CAAC,MAAM,CAAC,OAAiB;SACvC,EACD;YACE,YAAY,EAAE;gBACZ,KAAK,EAAE,EAAE;gBACT,OAAO,EAAE,EAAE;gBACX,OAAO,EAAE,EAAE;aACZ;SACF,CACF,CAAC;QAEF,kBAAkB;QAClB,IAAI,CAAC,aAAa,EAAE,CAAC;IACvB,CAAC;IAEO,aAAa;QACnB,gBAAgB;QAChB,IAAI,CAAC,MAAM,CAAC,iBAAiB,CAAC,sBAAsB,EAAE,IAAI,CAAC,eAAe,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC;QACvF,IAAI,CAAC,MAAM,CAAC,iBAAiB,CAAC,qBAAqB,EAAE,IAAI,CAAC,cAAc,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC;QAErF,oBAAoB;QACpB,IAAI,CAAC,MAAM,CAAC,iBAAiB,CAAC,wBAAwB,EAAE,IAAI,CAAC,iBAAiB,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC;QAC3F,IAAI,CAAC,MAAM,CAAC,iBAAiB,CAAC,sBAAsB,EAAE,IAAI,CAAC,eAAe,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC;QAEvF,kBAAkB;QAClB,IAAI,CAAC,MAAM,CAAC,iBAAiB,CAAC,qBAAqB,EAAE,IAAI,CAAC,iBAAiB,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC;IAC1F,CAAC;IAED;;OAEG;IACH,SAAS;QACP,OAAO,IAAI,CAAC,MAAM,CAAC;IACrB,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,UAAU;QACd,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,+BAA+B,EAAE;YAChD,OAAO,EAAE,IAAI,CAAC,MAAM,CAAC,OAAO;YAC5B,QAAQ,EAAE,IAAI,CAAC,MAAM,CAAC,QAAQ;SAC/B,CAAC,CAAC;QAEH,IAAI,CAAC;YACH,gCAAgC;YAChC,MAAM,IAAI,CAAC,OAAO,CAAC,qBAAqB,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;YAEtD,oCAAoC;YACpC,IAAI,CAAC;gBACH,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,YAAY,EAAE,CAAC;gBACnD,IAAI,QAAQ,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;oBACxB,IAAI,CAAC,aAAa,GAAG,QAAQ,CAAC,CAAC,CAAC,CAAC;oBACjC,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,2BAA2B,IAAI,CAAC,aAAa,EAAE,CAAC,CAAC;gBACpE,CAAC;qBAAM,CAAC;oBACN,yBAAyB;oBACzB,MAAM,IAAI,CAAC,OAAO,CAAC,gBAAgB,CACjC,SAAS,EACT,cAAc,EACd,IAAI,CAAC,SAAS,CAAC;wBACb,EAAE,EAAE,SAAS;wBACb,IAAI,EAAE,SAAS;wBACf,WAAW,EAAE,iBAAiB;wBAC9B,OAAO,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;qBAClC,EAAE,IAAI,EAAE,CAAC,CAAC,EACX,IAAI,CAAC,MAAM,CACZ,CAAC;oBACF,IAAI,CAAC,aAAa,GAAG,SAAS,CAAC;oBAC/B,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,yBAAyB,CAAC,CAAC;gBAC9C,CAAC;YACH,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,wCAAwC,EAAE,KAAK,CAAC,CAAC;gBAClE,IAAI,CAAC,aAAa,GAAG,SAAS,CAAC;YACjC,CAAC;QACH,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,8BAA8B,EAAE,KAAK,CAAC,CAAC;YACzD,MAAM,KAAK,CAAC;QACd,CAAC;IACH,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,OAAO,CAAC,SAAc;QAC1B,MAAM,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;QACrC,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,wBAAwB,CAAC,CAAC;IAC7C,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,QAAQ;QACZ,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,gCAAgC,CAAC,CAAC;QACnD,MAAM,IAAI,CAAC,MAAM,CAAC,KAAK,EAAE,CAAC;IAC5B,CAAC;IAED,gBAAgB;IACR,KAAK,CAAC,eAAe,CAAC,QAA0B;QACtD,MAAM,KAAK,GAAG,YAAY,CAAC,QAAQ,EAAE,CAAC;QACtC,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,WAAW,KAAK,CAAC,MAAM,QAAQ,CAAC,CAAC;QACnD,OAAO,EAAE,KAAK,EAAE,CAAC;IACnB,CAAC;IAEO,KAAK,CAAC,cAAc,CAAC,OAAwB;QACnD,MAAM,EAAE,IAAI,EAAE,SAAS,EAAE,IAAI,EAAE,GAAG,OAAO,CAAC,MAAM,CAAC;QAEjD,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,iBAAiB,IAAI,EAAE,EAAE,EAAE,IAAI,EAAE,CAAC,CAAC;QAEpD,gDAAgD;QAChD,MAAM,aAAa,GAA+B,IAAI,CAAC,aAAa,CAAC,CAAC,CAAC;YACrE,EAAE,EAAE,IAAI,CAAC,aAAa;YACtB,WAAW,EAAE;gBACX,OAAO,EAAE,SAAS;gBAClB,UAAU,EAAE,SAAS;gBACrB,SAAS,EAAE,SAAS;gBACpB,WAAW,EAAE,SAAS;aACvB;SACF,CAAC,CAAC,CAAC,SAAS,CAAC;QAEd,MAAM,OAAO,GAAmB;YAC9B,SAAS,EAAE,IAAI,CAAC,SAAS;YACzB,MAAM,EAAE,CAAC,OAAO,IAAI,EAAE,MAAM,KAAK,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,SAAS,CAAC,IAAI,IAAI,CAAC,MAAM;YACnF,OAAO,EAAE,IAAI,CAAC,OAAO;YACrB,MAAM,EAAE,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC;YAC/B,KAAK,EAAE,IAAI,IAAI,EAAE;YACjB,cAAc,EAAE,QAAQ,IAAI,CAAC,SAAS,EAAE;YACxC,iBAAiB,EAAE,KAAK;YACxB,SAAS,EAAE,OAAO,IAAI,CAAC,GAAG,EAAE,EAAE;YAC9B,SAAS,EAAE,IAAI,IAAI,EAAE;YACrB,cAAc,EAAE,OAAO;YACvB,aAAa;YACb,mBAAmB,EAAE,CAAC,OAAmC,EAAE,EAAE;gBAC3D,MAAM,YAAY,GAAG,OAAO,EAAE,EAAE,IAAI,SAAS,CAAC;gBAC9C,IAAI,CAAC,aAAa,GAAG,YAAY,CAAC;gBAClC,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,IAAI,IAAI,CAAC,UAAU,gCAAgC,YAAY,eAAe,IAAI,CAAC,sBAAsB,CAAC,MAAM,YAAY,CAAC,CAAC;gBAC/I,kCAAkC;gBAClC,KAAK,MAAM,QAAQ,IAAI,IAAI,CAAC,sBAAsB,EAAE,CAAC;oBACnD,IAAI,CAAC;wBACH,MAAM,MAAM,GAAG,QAAQ,CAAC,YAAY,CAAC,CAAC;wBACtC,IAAI,MAAM,YAAY,OAAO,EAAE,CAAC;4BAC9B,MAAM,CAAC,KAAK,CAAC,GAAG,CAAC,EAAE,CAAC,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,gCAAgC,EAAE,GAAG,CAAC,CAAC,CAAC;wBAChF,CAAC;oBACH,CAAC;oBAAC,OAAO,GAAG,EAAE,CAAC;wBACb,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,gCAAgC,EAAE,GAAG,CAAC,CAAC;oBAC3D,CAAC;gBACH,CAAC;YACH,CAAC;YACD,uBAAuB,EAAE,GAAG,EAAE,CAAC,IAAI,CAAC,aAAa;YACjD,sBAAsB,EAAE,IAAI,CAAC,sBAAsB;SACpD,CAAC;QAEF,eAAe;QACf,MAAM,SAAS,GAAG,YAAY,CAAC,aAAa,CAAC,IAAI,CAAC,CAAC;QACnD,IAAI,CAAC,SAAS,EAAE,CAAC;YACf,MAAM,KAAK,GAAG,iBAAiB,IAAI,EAAE,CAAC;YACtC,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;YACzB,OAAO;gBACL,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,KAAK,EAAE,CAAC;gBACxC,OAAO,EAAE,IAAI;aACd,CAAC;QACJ,CAAC;QAED,IAAI,CAAC;YACH,mBAAmB;YACnB,MAAM,MAAM,GAAG,MAAM,SAAS,CAAC,OAAO,CAAC,IAAI,IAAI,EAAE,EAAE,OAAO,CAAC,CAAC;YAE5D,oDAAoD;YACpD,IAAI,CAAC,MAAM,CAAC,OAAO,EAAE,CAAC;gBACpB,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,QAAQ,IAAI,wBAAwB,EAAE;oBACtD,aAAa,EAAE,MAAM,CAAC,OAAO,EAAE,MAAM;oBACrC,UAAU,EAAE,CAAC,CAAC,MAAM,CAAC,OAAO;iBAC7B,CAAC,CAAC;YACL,CAAC;iBAAM,CAAC;gBACN,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,QAAQ,IAAI,iBAAiB,EAAE,EAAE,MAAM,EAAE,CAAC,CAAC;YAC9D,CAAC;YACD,OAAO,MAAM,CAAC;QAChB,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,0BAA0B,IAAI,EAAE,EAAE,KAAK,CAAC,CAAC;YAC3D,OAAO;gBACL,OAAO,EAAE;oBACP;wBACE,IAAI,EAAE,MAAM;wBACZ,IAAI,EAAE,0BAA0B,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,eAAe,EAAE;qBAC3F;iBACF;gBACD,OAAO,EAAE,IAAI;aACd,CAAC;QACJ,CAAC;IACH,CAAC;IAED,kBAAkB;IACV,KAAK,CAAC,iBAAiB,CAAC,QAA4B;QAC1D,MAAM,OAAO,GAAG,cAAc,CAAC,UAAU,EAAE,CAAC;QAC5C,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,WAAW,OAAO,CAAC,MAAM,UAAU,CAAC,CAAC;QACvD,OAAO,EAAE,OAAO,EAAE,CAAC;IACrB,CAAC;IAEO,KAAK,CAAC,eAAe,CAAC,OAAyB;QACrD,MAAM,EAAE,IAAI,EAAE,SAAS,EAAE,IAAI,EAAE,GAAG,OAAO,CAAC,MAAM,CAAC;QAEjD,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,mBAAmB,IAAI,EAAE,CAAC,CAAC;QAE5C,yBAAyB;QACzB,MAAM,aAAa,GAA+B,IAAI,CAAC,aAAa,CAAC,CAAC,CAAC;YACrE,EAAE,EAAE,IAAI,CAAC,aAAa;YACtB,WAAW,EAAE;gBACX,OAAO,EAAE,SAAS;gBAClB,UAAU,EAAE,SAAS;gBACrB,SAAS,EAAE,SAAS;gBACpB,WAAW,EAAE,SAAS;aACvB;SACF,CAAC,CAAC,CAAC,SAAS,CAAC;QAEd,MAAM,OAAO,GAAmB;YAC9B,SAAS,EAAE,IAAI,CAAC,SAAS;YACzB,MAAM,EAAE,IAAI,CAAC,MAAM;YACnB,OAAO,EAAE,IAAI,CAAC,OAAO;YACrB,MAAM,EAAE,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,UAAU,IAAI,EAAE,CAAC;YAC3C,KAAK,EAAE,IAAI,IAAI,EAAE;YACjB,cAAc,EAAE,QAAQ,IAAI,CAAC,SAAS,EAAE;YACxC,iBAAiB,EAAE,KAAK;YACxB,SAAS,EAAE,OAAO,IAAI,CAAC,GAAG,EAAE,EAAE;YAC9B,SAAS,EAAE,IAAI,IAAI,EAAE;YACrB,cAAc,EAAE,OAAO;YACvB,aAAa;YACb,mBAAmB,EAAE,CAAC,OAAmC,EAAE,EAAE;gBAC3D,MAAM,YAAY,GAAG,OAAO,EAAE,EAAE,IAAI,SAAS,CAAC;gBAC9C,IAAI,CAAC,aAAa,GAAG,YAAY,CAAC;gBAClC,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,IAAI,IAAI,CAAC,UAAU,gCAAgC,YAAY,eAAe,IAAI,CAAC,sBAAsB,CAAC,MAAM,YAAY,CAAC,CAAC;gBAC/I,kCAAkC;gBAClC,KAAK,MAAM,QAAQ,IAAI,IAAI,CAAC,sBAAsB,EAAE,CAAC;oBACnD,IAAI,CAAC;wBACH,MAAM,MAAM,GAAG,QAAQ,CAAC,YAAY,CAAC,CAAC;wBACtC,IAAI,MAAM,YAAY,OAAO,EAAE,CAAC;4BAC9B,MAAM,CAAC,KAAK,CAAC,GAAG,CAAC,EAAE,CAAC,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,gCAAgC,EAAE,GAAG,CAAC,CAAC,CAAC;wBAChF,CAAC;oBACH,CAAC;oBAAC,OAAO,GAAG,EAAE,CAAC;wBACb,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,gCAAgC,EAAE,GAAG,CAAC,CAAC;oBAC3D,CAAC;gBACH,CAAC;YACH,CAAC;YACD,uBAAuB,EAAE,GAAG,EAAE,CAAC,IAAI,CAAC,aAAa;YACjD,sBAAsB,EAAE,IAAI,CAAC,sBAAsB;SACpD,CAAC;QAEF,uBAAuB;QACvB,MAAM,WAAW,GAAG,cAAc,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC;QACzD,IAAI,CAAC,WAAW,EAAE,CAAC;YACjB,MAAM,KAAK,GAAG,mBAAmB,IAAI,EAAE,CAAC;YACxC,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;YACzB,MAAM,IAAI,KAAK,CAAC,KAAK,CAAC,CAAC;QACzB,CAAC;QAED,IAAI,CAAC;YACH,6BAA6B;YAC7B,MAAM,MAAM,GAAG,MAAM,WAAW,CAAC,MAAM,CAAC,IAAI,EAAE,IAAI,IAAI,EAAE,EAAE,OAAO,CAAC,CAAC;YACnE,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,UAAU,IAAI,YAAY,CAAC,CAAC;YAC9C,OAAO,MAAyB,CAAC;QACnC,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,0BAA0B,IAAI,GAAG,EAAE,KAAK,CAAC,CAAC;YAC5D,MAAM,KAAK,CAAC;QACd,CAAC;IACH,CAAC;IAED,kBAAkB;IACV,KAAK,CAAC,iBAAiB,CAAC,OAAY;QAC1C,MAAM,EAAE,KAAK,EAAE,GAAG,OAAO,CAAC,MAAM,CAAC;QAEjC,IAAI,CAAC,MAAM,CAAC,WAAW,CAAC,KAAiB,CAAC,CAAC;QAC3C,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,qBAAqB,KAAK,EAAE,CAAC,CAAC;QAE/C,OAAO,EAAE,CAAC;IACZ,CAAC;CACF"}

package/dist/index.d.ts

@@ -0,0 +1,8 @@
+/**
+ * Tiny Brain Local MCP Server
+ *
+ * Local implementation of the Tiny Brain AI assistant
+ * using the Model Context Protocol (MCP)
+ */
+export {};
+//# sourceMappingURL=index.d.ts.map

package/dist/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA;;;;;GAKG"}

package/dist/index.js

@@ -0,0 +1,37 @@
+/**
+ * Tiny Brain Local MCP Server
+ *
+ * Local implementation of the Tiny Brain AI assistant
+ * using the Model Context Protocol (MCP)
+ */
+import { StdioServerTransport } from '@modelcontextprotocol/sdk/server/stdio.js';
+import { MCPServer } from './core/mcp-server.js';
+async function main() {
+    // Create MCP server with config from environment
+    const server = new MCPServer({
+        name: 'tiny-brain-local',
+        version: '0.3.7',
+        dataDir: process.env.TINY_BRAIN_DATA_DIR,
+        logLevel: process.env.LOG_LEVEL,
+    });
+    // Initialize server
+    await server.initialize();
+    // Create and connect transport
+    const transport = new StdioServerTransport();
+    await server.connect(transport);
+    // Handle shutdown
+    process.on('SIGINT', async () => {
+        await server.shutdown();
+        process.exit(0);
+    });
+    process.on('SIGTERM', async () => {
+        await server.shutdown();
+        process.exit(0);
+    });
+}
+// Run the server
+main().catch((error) => {
+    console.error('Failed to start server:', error);
+    process.exit(1);
+});
+//# sourceMappingURL=index.js.map

package/dist/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAEH,OAAO,EAAE,oBAAoB,EAAE,MAAM,2CAA2C,CAAC;AACjF,OAAO,EAAE,SAAS,EAAE,MAAM,sBAAsB,CAAC;AAEjD,KAAK,UAAU,IAAI;IACjB,iDAAiD;IACjD,MAAM,MAAM,GAAG,IAAI,SAAS,CAAC;QAC3B,IAAI,EAAE,kBAAkB;QACxB,OAAO,EAAE,OAAO;QAChB,OAAO,EAAE,OAAO,CAAC,GAAG,CAAC,mBAAmB;QACxC,QAAQ,EAAE,OAAO,CAAC,GAAG,CAAC,SAAgB;KACvC,CAAC,CAAC;IAEH,oBAAoB;IACpB,MAAM,MAAM,CAAC,UAAU,EAAE,CAAC;IAE1B,+BAA+B;IAC/B,MAAM,SAAS,GAAG,IAAI,oBAAoB,EAAE,CAAC;IAC7C,MAAM,MAAM,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;IAEhC,kBAAkB;IAClB,OAAO,CAAC,EAAE,CAAC,QAAQ,EAAE,KAAK,IAAI,EAAE;QAC9B,MAAM,MAAM,CAAC,QAAQ,EAAE,CAAC;QACxB,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IAClB,CAAC,CAAC,CAAC;IAEH,OAAO,CAAC,EAAE,CAAC,SAAS,EAAE,KAAK,IAAI,EAAE;QAC/B,MAAM,MAAM,CAAC,QAAQ,EAAE,CAAC;QACxB,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IAClB,CAAC,CAAC,CAAC;AACL,CAAC;AAED,iBAAiB;AACjB,IAAI,EAAE,CAAC,KAAK,CAAC,CAAC,KAAK,EAAE,EAAE;IACrB,OAAO,CAAC,KAAK,CAAC,yBAAyB,EAAE,KAAK,CAAC,CAAC;IAChD,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;AAClB,CAAC,CAAC,CAAC"}

package/dist/prompts/index.d.ts

@@ -0,0 +1,39 @@
+/**
+ * Prompt System Types and Interfaces
+ */
+import type { RequestContext } from '../types/request-context.js';
+import type { Prompt } from '@modelcontextprotocol/sdk/types.js';
+/**
+ * Response format for all prompts
+ */
+export interface PromptResponse {
+    description?: string;
+    messages: Array<{
+        role: 'user' | 'assistant';
+        content: {
+            type: 'text';
+            text: string;
+        };
+    }>;
+}
+/**
+ * Interface that all prompt classes must implement
+ */
+export interface TinyBrainPrompt {
+    /**
+     * Get the prompt definitions for this prompt class
+     */
+    getPromptDefinitions(): Prompt[];
+    /**
+     * Handle a prompt execution
+     */
+    handle(name: string, args: Record<string, unknown>, context: RequestContext): Promise<PromptResponse>;
+}
+/**
+ * Type for prompt class constructors
+ */
+export type PromptClass = {
+    getPromptDefinitions(): Prompt[];
+    handle(name: string, args: Record<string, unknown>, context: RequestContext): Promise<PromptResponse>;
+};
+//# sourceMappingURL=index.d.ts.map

package/dist/prompts/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/prompts/index.ts"],"names":[],"mappings":"AAAA;;GAEG;AAEH,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,6BAA6B,CAAC;AAClE,OAAO,KAAK,EAAE,MAAM,EAAE,MAAM,oCAAoC,CAAC;AAEjE;;GAEG;AACH,MAAM,WAAW,cAAc;IAC7B,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,QAAQ,EAAE,KAAK,CAAC;QACd,IAAI,EAAE,MAAM,GAAG,WAAW,CAAC;QAC3B,OAAO,EAAE;YACP,IAAI,EAAE,MAAM,CAAC;YACb,IAAI,EAAE,MAAM,CAAC;SACd,CAAC;KACH,CAAC,CAAC;CACJ;AAED;;GAEG;AACH,MAAM,WAAW,eAAe;IAC9B;;OAEG;IACH,oBAAoB,IAAI,MAAM,EAAE,CAAC;IAEjC;;OAEG;IACH,MAAM,CACJ,IAAI,EAAE,MAAM,EACZ,IAAI,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,EAC7B,OAAO,EAAE,cAAc,GACtB,OAAO,CAAC,cAAc,CAAC,CAAC;CAC5B;AAED;;GAEG;AACH,MAAM,MAAM,WAAW,GAAG;IACxB,oBAAoB,IAAI,MAAM,EAAE,CAAC;IACjC,MAAM,CACJ,IAAI,EAAE,MAAM,EACZ,IAAI,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,EAC7B,OAAO,EAAE,cAAc,GACtB,OAAO,CAAC,cAAc,CAAC,CAAC;CAC5B,CAAC"}

package/dist/prompts/index.js

@@ -0,0 +1,5 @@
+/**
+ * Prompt System Types and Interfaces
+ */
+export {};
+//# sourceMappingURL=index.js.map

package/dist/prompts/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/prompts/index.ts"],"names":[],"mappings":"AAAA;;GAEG"}

package/dist/prompts/memory/memory.prompt.d.ts

@@ -0,0 +1,32 @@
+/**
+ * Memory Prompt Handler
+ *
+ * Thin dispatcher for memory-related prompt executions
+ * Validates args with Zod, dispatches to MemoryService, formats responses
+ */
+import type { RequestContext } from '../../types/request-context.js';
+import type { Prompt } from '@modelcontextprotocol/sdk/types.js';
+import type { PromptResponse } from '../index.js';
+export declare class MemoryPrompt {
+    /**
+     * Get all memory-related prompt definitions
+     */
+    static getPromptDefinitions(): Prompt[];
+    /**
+     * Main handler that routes to specific prompt handlers
+     */
+    static handle(name: string, args: Record<string, unknown>, context: RequestContext): Promise<PromptResponse>;
+    /**
+     * Handle remember prompt - stores information in appropriate memory tier
+     */
+    static handleRemember(args: Record<string, unknown>, context: RequestContext): Promise<PromptResponse>;
+    /**
+     * Handle recall prompt - searches memory across tiers
+     */
+    static handleRecall(args: Record<string, unknown>, context: RequestContext): Promise<PromptResponse>;
+    /**
+     * Handle forget prompt - removes information from memory
+     */
+    static handleForget(args: Record<string, unknown>, context: RequestContext): Promise<PromptResponse>;
+}
+//# sourceMappingURL=memory.prompt.d.ts.map

package/dist/prompts/memory/memory.prompt.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"memory.prompt.d.ts","sourceRoot":"","sources":["../../../src/prompts/memory/memory.prompt.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAGH,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,gCAAgC,CAAC;AACrE,OAAO,KAAK,EAAE,MAAM,EAAE,MAAM,oCAAoC,CAAC;AACjE,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,aAAa,CAAC;AAiBlD,qBAAa,YAAY;IACvB;;OAEG;IACH,MAAM,CAAC,oBAAoB,IAAI,MAAM,EAAE;IA4CvC;;OAEG;WACU,MAAM,CACjB,IAAI,EAAE,MAAM,EACZ,IAAI,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,EAC7B,OAAO,EAAE,cAAc,GACtB,OAAO,CAAC,cAAc,CAAC;IAa1B;;OAEG;WACU,cAAc,CACzB,IAAI,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,EAC7B,OAAO,EAAE,cAAc,GACtB,OAAO,CAAC,cAAc,CAAC;IA4C1B;;OAEG;WACU,YAAY,CACvB,IAAI,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,EAC7B,OAAO,EAAE,cAAc,GACtB,OAAO,CAAC,cAAc,CAAC;IA4C1B;;OAEG;WACU,YAAY,CACvB,IAAI,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,EAC7B,OAAO,EAAE,cAAc,GACtB,OAAO,CAAC,cAAc,CAAC;CAwC3B"}

package/dist/prompts/memory/memory.prompt.js

@@ -0,0 +1,204 @@
+/**
+ * Memory Prompt Handler
+ *
+ * Thin dispatcher for memory-related prompt executions
+ * Validates args with Zod, dispatches to MemoryService, formats responses
+ */
+import { z } from 'zod';
+import { MemoryService } from '@magic-ingredients/tiny-brain-core';
+// Zod schemas for each prompt handler
+const RememberArgsSchema = z.object({
+    fact: z.string(),
+    category: z.string().optional(),
+});
+const RecallArgsSchema = z.object({
+    query: z.string(),
+});
+const ForgetArgsSchema = z.object({
+    query: z.string(),
+});
+export class MemoryPrompt {
+    /**
+     * Get all memory-related prompt definitions
+     */
+    static getPromptDefinitions() {
+        return [
+            {
+                name: 'remember',
+                description: '💾 Store information in personal or conversation memory with smart classification',
+                arguments: [
+                    {
+                        name: 'fact',
+                        description: 'The information you want to remember',
+                        required: true,
+                    },
+                    {
+                        name: 'category',
+                        description: 'Category to store this in (e.g., preferences, project_details)',
+                        required: false,
+                    },
+                ],
+            },
+            {
+                name: 'recall',
+                description: '🔍 Search through personal and conversation memory',
+                arguments: [
+                    {
+                        name: 'query',
+                        description: 'What you want to search for',
+                        required: true,
+                    },
+                ],
+            },
+            {
+                name: 'forget',
+                description: '🗑️ Remove information from memory',
+                arguments: [
+                    {
+                        name: 'query',
+                        description: 'What information to forget',
+                        required: true,
+                    },
+                ],
+            },
+        ];
+    }
+    /**
+     * Main handler that routes to specific prompt handlers
+     */
+    static async handle(name, args, context) {
+        switch (name) {
+            case 'remember':
+                return await MemoryPrompt.handleRemember(args, context);
+            case 'recall':
+                return await MemoryPrompt.handleRecall(args, context);
+            case 'forget':
+                return await MemoryPrompt.handleForget(args, context);
+            default:
+                throw new Error(`Unknown memory prompt: ${name}`);
+        }
+    }
+    /**
+     * Handle remember prompt - stores information in appropriate memory tier
+     */
+    static async handleRemember(args, context) {
+        try {
+            // Validate args with Zod
+            const validatedArgs = RememberArgsSchema.parse(args);
+            // Create memory service with context
+            const memoryService = new MemoryService(context);
+            // Call service method with validated args
+            const result = await memoryService.rememberFact(validatedArgs.fact, validatedArgs.category);
+            // Format response
+            const resultText = `${result.message}\n\nExtracted:\n- Entities: ${result.entities.map((e) => `${e.name} (${e.entityType})`).join(', ')}\n- Relations: ${result.relations.map((r) => `${r.from} ${r.relationType} ${r.to}`).join(', ')}\n- Context: ${result.context || 'general'}`;
+            return {
+                description: 'Remember information with smart classification',
+                messages: [
+                    {
+                        role: 'user',
+                        content: {
+                            type: 'text',
+                            text: `Remember: ${validatedArgs.fact}${validatedArgs.category ? ` (category: ${validatedArgs.category})` : ''}`,
+                        },
+                    },
+                    {
+                        role: 'assistant',
+                        content: {
+                            type: 'text',
+                            text: resultText,
+                        },
+                    },
+                ],
+            };
+        }
+        catch (error) {
+            if (error instanceof z.ZodError) {
+                throw new Error(`Invalid arguments: ${error.errors.map(e => e.message).join(', ')}`);
+            }
+            throw error;
+        }
+    }
+    /**
+     * Handle recall prompt - searches memory across tiers
+     */
+    static async handleRecall(args, context) {
+        try {
+            // Validate args with Zod
+            const validatedArgs = RecallArgsSchema.parse(args);
+            // Create memory service with context
+            const memoryService = new MemoryService(context);
+            // Call service method with validated args
+            const result = await memoryService.recallMemories(validatedArgs.query);
+            // Format response
+            const resultText = result.results.length > 0
+                ? `Found ${result.totalMatches} relevant memories:\n\n${JSON.stringify(result.results, null, 2)}`
+                : `No memories found for query: ${validatedArgs.query}`;
+            return {
+                description: 'Search both personal and conversation memory',
+                messages: [
+                    {
+                        role: 'user',
+                        content: {
+                            type: 'text',
+                            text: `What do you know about: ${validatedArgs.query}`,
+                        },
+                    },
+                    {
+                        role: 'assistant',
+                        content: {
+                            type: 'text',
+                            text: resultText,
+                        },
+                    },
+                ],
+            };
+        }
+        catch (error) {
+            if (error instanceof z.ZodError) {
+                throw new Error(`Invalid arguments: ${error.errors.map(e => e.message).join(', ')}`);
+            }
+            throw error;
+        }
+    }
+    /**
+     * Handle forget prompt - removes information from memory
+     */
+    static async handleForget(args, context) {
+        try {
+            // Validate args with Zod
+            const validatedArgs = ForgetArgsSchema.parse(args);
+            // Create memory service with context
+            const memoryService = new MemoryService(context);
+            // Call service method with validated args
+            const result = await memoryService.forgetMemories(validatedArgs.query);
+            // Format response
+            const resultText = result.message;
+            return {
+                description: 'Remove information from memory',
+                messages: [
+                    {
+                        role: 'user',
+                        content: {
+                            type: 'text',
+                            text: `Forget: ${validatedArgs.query}`,
+                        },
+                    },
+                    {
+                        role: 'assistant',
+                        content: {
+                            type: 'text',
+                            text: resultText,
+                        },
+                    },
+                ],
+            };
+        }
+        catch (error) {
+            if (error instanceof z.ZodError) {
+                throw new Error(`Invalid arguments: ${error.errors.map(e => e.message).join(', ')}`);
+            }
+            throw error;
+        }
+    }
+}
+//# sourceMappingURL=memory.prompt.js.map