@knowall-ai/mcp-neo4j-agent-memory/build/server.js

Graph-based memory system for AI agents. Store people, places, organizations as nodes. Build semantic relationships (KNOWS, WORKS_AT, CREATED). Word-tokenized search finds partial matches. Filter by date, traverse relationships, maintain temporal context.

import { Server } from '@modelcontextprotocol/sdk/server/index.js';
import { StdioServerTransport } from '@modelcontextprotocol/sdk/server/stdio.js';
import { CallToolRequestSchema, ListToolsRequestSchema } from '@modelcontextprotocol/sdk/types.js';
import { Neo4jClient } from './neo4j-client.js';
import { tools } from './tools/definitions.js';
import { handleToolCall } from './handlers/index.js';
export class Neo4jServer {
    server;
    neo4j;
    constructor(config) {
        this.server = new Server({
            name: 'mcp-neo4j-agent-memory',
            version: '1.0.0',
        }, {
            capabilities: {
                tools: {},
            },
        });
        this.neo4j = config ? new Neo4jClient(config.uri, config.username, config.password, config.database) : null;
        this.setupToolHandlers();
        // Error handling
        this.server.onerror = (error) => console.error('[MCP Error]', error);
        process.on('SIGINT', async () => {
            await this.close();
            process.exit(0);
        });
    }
    setupToolHandlers() {
        // Tool list handler
        this.server.setRequestHandler(ListToolsRequestSchema, async () => ({
            tools,
        }));
        // Tool execution handler
        this.server.setRequestHandler(CallToolRequestSchema, async (request) => {
            if (!this.neo4j) {
                return {
                    content: [
                        {
                            type: 'text',
                            text: 'Neo4j connection not configured. Please set NEO4J_URI, NEO4J_USERNAME, and NEO4J_PASSWORD environment variables.',
                        },
                    ],
                    isError: true,
                };
            }
            const { name, arguments: args } = request.params;
            return handleToolCall(name, args, this.neo4j);
        });
    }
    async run() {
        const transport = new StdioServerTransport();
        await this.server.connect(transport);
        console.error('Neo4j MCP server running on stdio');
    }
    async close() {
        if (this.neo4j) {
            await this.neo4j.close();
        }
        await this.server.close();
    }
}