@code-rag/mcp-server 0.1.6 → 0.1.8

package/dist/server.d.ts

@@ -9,6 +9,7 @@ export interface CodeRAGServerOptions {
 export declare class CodeRAGServer {
     private readonly server;
     private readonly rootDir;
+    private runtime;
     private config;
     private store;
     private hybridSearch;

package/dist/server.js

@@ -2,9 +2,8 @@ import { Server } from '@modelcontextprotocol/sdk/server/index.js';
 import { StdioServerTransport } from '@modelcontextprotocol/sdk/server/stdio.js';
 import { SSEServerTransport } from '@modelcontextprotocol/sdk/server/sse.js';
 import { CallToolRequestSchema, ListToolsRequestSchema, } from '@modelcontextprotocol/sdk/types.js';
-import { loadConfig, checkIndexExists, OllamaEmbeddingProvider, LanceDBStore, BM25Index, HybridSearch, DependencyGraph, ContextExpander, CrossEncoderReRanker, } from '@code-rag/core';
-import { readFile } from 'node:fs/promises';
-import { join, resolve } from 'node:path';
+import { loadConfig, checkIndexExists, createRuntime, } from '@code-rag/core';
+import { resolve } from 'node:path';
 import { createServer } from 'node:http';
 import { URL } from 'node:url';
 import { handleSearch } from './tools/search.js';
@@ -27,6 +26,7 @@ The MCP server will start automatically once indexing is complete.`;
 export class CodeRAGServer {
     server;
     rootDir;
+    runtime = null;
     config = null;
     store = null;
     hybridSearch = null;
@@ -68,66 +68,26 @@ export class CodeRAGServer {
      */
     async initialize() {
         try {
+            // Check index existence before full init
             const configResult = await loadConfig(this.rootDir);
-            if (configResult.isErr()) {
-                // eslint-disable-next-line no-console
-                console.error(`[coderag] Config load failed: ${configResult.error.message}`);
-                return;
+            if (configResult.isOk()) {
+                const storagePath = resolve(this.rootDir, configResult.value.storage.path);
+                if (storagePath.startsWith(resolve(this.rootDir))) {
+                    this.indexCheck = await checkIndexExists(storagePath);
+                }
             }
-            this.config = configResult.value;
-            // Create embedding provider
-            const embeddingProvider = new OllamaEmbeddingProvider({
-                model: this.config.embedding.model,
-                dimensions: this.config.embedding.dimensions,
-            });
-            // Create LanceDB store (validate path stays within rootDir)
-            const storagePath = resolve(this.rootDir, this.config.storage.path);
-            if (!storagePath.startsWith(resolve(this.rootDir))) {
+            const runtimeResult = await createRuntime({ rootDir: this.rootDir });
+            if (runtimeResult.isErr()) {
                 // eslint-disable-next-line no-console
-                console.error('[coderag] Storage path escapes project root');
+                console.error(`[coderag] ${runtimeResult.error.message}`);
                 return;
             }
-            // Check index existence before connecting
-            this.indexCheck = await checkIndexExists(storagePath);
-            this.store = new LanceDBStore(storagePath, this.config.embedding.dimensions);
-            await this.store.connect();
-            // Create BM25 index -- try to load from stored data
-            let bm25Index = new BM25Index();
-            const bm25Path = join(storagePath, 'bm25-index.json');
-            try {
-                const bm25Data = await readFile(bm25Path, 'utf-8');
-                bm25Index = BM25Index.deserialize(bm25Data);
-            }
-            catch {
-                // No saved BM25 index, start empty
-            }
-            // Create HybridSearch
-            this.hybridSearch = new HybridSearch(this.store, bm25Index, embeddingProvider, this.config.search);
-            // Create re-ranker if enabled
-            if (this.config.reranker?.enabled) {
-                this.reranker = new CrossEncoderReRanker({
-                    model: this.config.reranker.model,
-                    topN: this.config.reranker.topN,
-                });
-            }
-            // Load dependency graph if available
-            let graph = new DependencyGraph();
-            const graphPath = join(storagePath, 'graph.json');
-            try {
-                const graphData = await readFile(graphPath, 'utf-8');
-                const parsed = JSON.parse(graphData);
-                graph = DependencyGraph.fromJSON(parsed);
-            }
-            catch {
-                // No saved graph, start empty
-            }
-            // Create retrieval services
-            const chunkLookup = (_chunkId) => {
-                // In a full implementation this would look up chunks by ID.
-                // For now, the context expander will only work with chunks found via search.
-                return undefined;
-            };
-            this.contextExpander = new ContextExpander(graph, chunkLookup);
+            this.runtime = runtimeResult.value;
+            this.config = this.runtime.config;
+            this.store = this.runtime.store;
+            this.hybridSearch = this.runtime.hybridSearch;
+            this.contextExpander = this.runtime.contextExpander;
+            this.reranker = this.runtime.reranker;
         }
         catch (error) {
             const message = error instanceof Error ? error.message : 'Unknown error';
@@ -412,6 +372,11 @@ export class CodeRAGServer {
             });
             this.httpServer = null;
         }
+        // Close the runtime (LanceDB, etc.)
+        if (this.runtime) {
+            this.runtime.close();
+            this.runtime = null;
+        }
     }
     /** Expose for testing. */
     getServer() {

package/dist/tools/context.js

@@ -77,7 +77,7 @@ export async function handleContext(args, hybridSearch, contextExpander) {
             };
         }
         // Step 2: Expand context via dependency graph
-        const expanded = contextExpander.expand(results);
+        const expanded = await contextExpander.expand(results);
         // Step 3: Assemble within token budget (use caller's max_tokens)
         const optimizer = new TokenBudgetOptimizer({ maxTokens: max_tokens });
         const assembled = optimizer.assemble(expanded);

package/dist/tools/explain.js

@@ -112,7 +112,7 @@ export async function handleExplain(args, hybridSearch, contextExpander) {
         };
         // Add related symbols from context expander when in detailed mode
         if (detail_level === 'detailed' && contextExpander) {
-            const expanded = contextExpander.expand(results);
+            const expanded = await contextExpander.expand(results);
             const relatedSymbols = expanded.relatedChunks.map((related) => {
                 return related.chunk.metadata.name ?? related.chunk.chunk?.filePath ?? 'unknown';
             });

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@code-rag/mcp-server",
-  "version": "0.1.6",
+  "version": "0.1.8",
   "description": "MCP server for CodeRAG — exposes codebase search, context, and status tools via Model Context Protocol",
   "type": "module",
   "license": "MIT",
@@ -41,7 +41,7 @@
   "dependencies": {
     "@modelcontextprotocol/sdk": "^1.5.0",
     "zod": "^4.3.6",
-    "@code-rag/core": "0.1.6"
+    "@code-rag/core": "0.1.8"
   },
   "devDependencies": {
     "@types/node": "^22.13.4",