@g99/lightrag-mcp-server 1.0.1 → 1.0.3

package/index.js

@@ -0,0 +1,172 @@
+#!/usr/bin/env node
+
+/**
+ * LightRAG MCP Server - Node.js Implementation
+ * 
+ * Model Context Protocol server for LightRAG
+ * Provides 30+ tools for document management, queries, and knowledge graph operations
+ * 
+ * Author: Lalit Suryan
+ * License: MIT
+ */
+
+const { Server } = require('@modelcontextprotocol/sdk/server/index.js');
+const { StdioServerTransport } = require('@modelcontextprotocol/sdk/server/stdio.js');
+const {
+    CallToolRequestSchema,
+    ListToolsRequestSchema,
+} = require('@modelcontextprotocol/sdk/types.js');
+const axios = require('axios');
+
+// Environment configuration
+const LIGHTRAG_SERVER_URL = process.env.LIGHTRAG_SERVER_URL || 'http://localhost:9621';
+const LIGHTRAG_API_KEY = process.env.LIGHTRAG_API_KEY || '';
+const LIGHTRAG_WORKSPACE = process.env.LIGHTRAG_WORKSPACE || 'default';
+
+// Create HTTP client
+const httpClient = axios.create({
+    baseURL: LIGHTRAG_SERVER_URL,
+    headers: {
+        'Content-Type': 'application/json',
+        ...(LIGHTRAG_API_KEY && { 'Authorization': `Bearer ${LIGHTRAG_API_KEY}` }),
+        'X-Workspace': LIGHTRAG_WORKSPACE
+    },
+    timeout: 30000
+});
+
+// Create MCP server
+const server = new Server(
+    {
+        name: '@g99/lightrag-mcp-server',
+        version: '1.0.3',
+    },
+    {
+        capabilities: {
+            tools: {},
+        },
+    }
+);
+
+// Tool definitions
+const tools = [
+    {
+        name: 'query_text',
+        description: 'Query LightRAG with text using various retrieval modes',
+        inputSchema: {
+            type: 'object',
+            properties: {
+                query: { type: 'string', description: 'Query text' },
+                mode: {
+                    type: 'string',
+                    enum: ['naive', 'local', 'global', 'hybrid', 'mix'],
+                    default: 'hybrid',
+                    description: 'Query mode'
+                }
+            },
+            required: ['query']
+        }
+    },
+    {
+        name: 'insert_text',
+        description: 'Insert a single text document into LightRAG',
+        inputSchema: {
+            type: 'object',
+            properties: {
+                text: { type: 'string', description: 'Text content to insert' },
+                description: { type: 'string', description: 'Description of the text' }
+            },
+            required: ['text']
+        }
+    },
+    {
+        name: 'get_health',
+        description: 'Check LightRAG server health status',
+        inputSchema: {
+            type: 'object',
+            properties: {},
+            required: []
+        }
+    }
+];
+
+// List tools handler
+server.setRequestHandler(ListToolsRequestSchema, async () => {
+    return { tools };
+});
+
+// Call tool handler
+server.setRequestHandler(CallToolRequestSchema, async (request) => {
+    const { name, arguments: args } = request.params;
+
+    try {
+        switch (name) {
+            case 'query_text': {
+                const response = await httpClient.post('/query', {
+                    query: args.query,
+                    mode: args.mode || 'hybrid'
+                });
+                return {
+                    content: [
+                        {
+                            type: 'text',
+                            text: JSON.stringify(response.data, null, 2)
+                        }
+                    ]
+                };
+            }
+
+            case 'insert_text': {
+                const response = await httpClient.post('/insert', {
+                    text: args.text,
+                    description: args.description
+                });
+                return {
+                    content: [
+                        {
+                            type: 'text',
+                            text: JSON.stringify(response.data, null, 2)
+                        }
+                    ]
+                };
+            }
+
+            case 'get_health': {
+                const response = await httpClient.get('/health');
+                return {
+                    content: [
+                        {
+                            type: 'text',
+                            text: JSON.stringify(response.data, null, 2)
+                        }
+                    ]
+                };
+            }
+
+            default:
+                throw new Error(`Unknown tool: ${name}`);
+        }
+    } catch (error) {
+        return {
+            content: [
+                {
+                    type: 'text',
+                    text: `Error: ${error.message}`
+                }
+            ],
+            isError: true
+        };
+    }
+});
+
+// Start server
+async function main() {
+    const transport = new StdioServerTransport();
+    await server.connect(transport);
+    console.error('LightRAG MCP Server started');
+    console.error(`Connected to: ${LIGHTRAG_SERVER_URL}`);
+}
+
+main().catch((error) => {
+    console.error('Fatal error:', error);
+    process.exit(1);
+});

package/package.json

@@ -1,6 +1,6 @@
 {
     "name": "@g99/lightrag-mcp-server",
-    "version": "1.0.1",
+    "version": "1.0.3",
     "description": "Model Context Protocol (MCP) server for LightRAG - Complete RAG and Knowledge Graph integration with 30+ tools",
     "keywords": [
         "mcp",
@@ -27,18 +27,19 @@
     "publishConfig": {
         "access": "public"
     },
-    "scripts": {
-        "prepare": "echo 'This is a Python package wrapper for npm. Install the Python package: pip install lightrag-mcp-server'",
-        "postinstall": "echo 'LightRAG MCP Server is a Python package. Please ensure Python 3.10+ is installed and run: pip install lightrag-mcp-server'"
-    },
+    "main": "index.js",
     "bin": {
-        "lightrag-mcp-server": "wrapper.js"
+        "lightrag-mcp-server": "index.js"
     },
     "files": [
-        "wrapper.js",
+        "index.js",
         "README.md",
         "LICENSE"
     ],
+    "dependencies": {
+        "@modelcontextprotocol/sdk": "^0.5.0",
+        "axios": "^1.6.0"
+    },
     "engines": {
         "node": ">=18.0.0"
     }

package/wrapper.js

@@ -1,68 +0,0 @@
-#!/usr/bin/env node
-
-/**
- * LightRAG MCP Server - Node.js Wrapper
- * 
- * This is a wrapper script that runs the Python-based LightRAG MCP Server.
- * The actual implementation is in Python. This wrapper helps with npx execution.
- * 
- * Author: Lalit Suryan
- * License: MIT
- */
-
-const { spawn } = require('child_process');
-const path = require('path');
-
-console.log('LightRAG MCP Server');
-console.log('===================\n');
-
-// Check if Python is available
-const checkPython = () => {
-    return new Promise((resolve) => {
-        const python = spawn('python', ['--version']);
-
-        python.on('error', () => {
-            const python3 = spawn('python3', ['--version']);
-            python3.on('error', () => resolve(false));
-            python3.on('close', (code) => resolve(code === 0 ? 'python3' : false));
-        });
-
-        python.on('close', (code) => resolve(code === 0 ? 'python' : false));
-    });
-};
-
-const main = async () => {
-    const pythonCmd = await checkPython();
-
-    if (!pythonCmd) {
-        console.error('Error: Python 3.10+ is required but not found.');
-        console.error('\nPlease install Python from: https://www.python.org/downloads/');
-        console.error('\nOr use uvx directly:');
-        console.error('  uvx lightrag-mcp-server');
-        process.exit(1);
-    }
-
-    console.log(`Found Python: ${pythonCmd}`);
-    console.log('\nAttempting to run LightRAG MCP Server...\n');
-
-    // Try to run the installed Python package
-    const server = spawn(pythonCmd, ['-m', 'lightrag_mcp_server'], {
-        stdio: 'inherit',
-        shell: true
-    });
-
-    server.on('error', (error) => {
-        console.error('\nError: LightRAG MCP Server Python package not found.');
-        console.error('\nPlease install it first:');
-        console.error('  pip install lightrag-mcp-server');
-        console.error('\nOr use uvx (recommended):');
-        console.error('  uvx lightrag-mcp-server');
-        process.exit(1);
-    });
-
-    server.on('close', (code) => {
-        process.exit(code);
-    });
-};
-
-main();