@code-rag/api-server 0.1.6 → 0.1.8

package/dist/routes/context.js

@@ -60,7 +60,7 @@ export function createContextRouter(deps) {
                 return;
             }
             // Expand context via dependency graph
-            const expanded = deps.contextExpander.expand(results);
+            const expanded = await deps.contextExpander.expand(results);
             // Assemble within token budget
             const optimizer = new TokenBudgetOptimizer({ maxTokens: max_tokens });
             const assembled = optimizer.assemble(expanded);

package/dist/routes/viewer.d.ts

@@ -1,74 +1,10 @@
 import { Router } from 'express';
-import type { LanceDBStore, CodeRAGConfig, HybridSearch, DependencyGraph, GraphNode, GraphEdge } from '@code-rag/core';
-export interface ViewerStatsResponse {
-    data: {
-        chunkCount: number;
-        fileCount: number;
-        languages: Record<string, number>;
-        storageBytes: number | null;
-        lastIndexed: string | null;
-    };
-}
-export interface ChunkSummary {
-    id: string;
-    filePath: string;
-    chunkType: string;
-    name: string;
-    language: string;
-    startLine: number;
-    endLine: number;
-    contentPreview: string;
-}
-export interface PaginationMeta {
-    page: number;
-    pageSize: number;
-    total: number;
-    totalPages: number;
-}
-export interface ChunkDetail {
-    id: string;
-    filePath: string;
-    chunkType: string;
-    name: string;
-    language: string;
-    startLine: number;
-    endLine: number;
-    content: string;
-    nlSummary: string;
-    metadata: Record<string, unknown>;
-    vector?: number[];
-}
-export interface GraphResponse {
-    data: {
-        nodes: GraphNode[];
-        edges: GraphEdge[];
-    };
-}
-export interface ViewerSearchResult {
-    chunkId: string;
-    filePath: string;
-    chunkType: string;
-    name: string;
-    content: string;
-    nlSummary: string;
-    score: number;
-    method: string;
-}
-export interface ViewerSearchResponse {
-    data: {
-        results: ViewerSearchResult[];
-        timing: {
-            totalMs: number;
-        };
-    };
-}
-export interface EmbeddingPoint {
-    id: string;
-    filePath: string;
-    chunkType: string;
-    language: string;
-    vector: number[];
-}
+import type { LanceDBStore, CodeRAGConfig, HybridSearch, DependencyGraph } from '@code-rag/core';
+import type { ViewerStatsResponse, ViewerChunksResponse, ViewerChunkDetailResponse, ViewerSearchResponse, ViewerGraphResponse, ViewerEmbeddingsResponse, ViewerChunkSummary as ChunkSummary, ViewerPaginationMeta as PaginationMeta, ViewerChunkDetail as ChunkDetail, ViewerSearchResultType as ViewerSearchResult, ViewerEmbeddingPoint as EmbeddingPoint } from '@code-rag/core';
+export type { ViewerStatsResponse, ViewerChunksResponse, ViewerChunkDetailResponse, ViewerSearchResponse, ViewerGraphResponse, ViewerEmbeddingsResponse, };
+export type { ChunkSummary, ChunkDetail, PaginationMeta, ViewerSearchResult, EmbeddingPoint };
+/** @deprecated Use ViewerGraphResponse from @code-rag/core instead */
+export type GraphResponse = ViewerGraphResponse;
 export interface ViewerDeps {
     readonly getStore: () => LanceDBStore | null;
     readonly getConfig: () => CodeRAGConfig | null;

package/dist/server.d.ts

@@ -19,6 +19,7 @@ export declare class ApiServer {
     private readonly rootDir;
     private readonly port;
     private httpServer;
+    private runtime;
     private config;
     private store;
     private hybridSearch;

package/dist/server.js

@@ -1,8 +1,6 @@
 import express from 'express';
 import { createServer } from 'node:http';
-import { loadConfig, OllamaEmbeddingProvider, LanceDBStore, BM25Index, HybridSearch, DependencyGraph, ContextExpander, CrossEncoderReRanker, } from '@code-rag/core';
-import { readFile } from 'node:fs/promises';
-import { join, resolve } from 'node:path';
+import { createRuntime, DependencyGraph, } from '@code-rag/core';
 import { parseApiKeys, createAuthMiddleware } from './middleware/auth.js';
 import { createRateLimitMiddleware, parseRateLimitConfig } from './middleware/rate-limit.js';
 import { createSearchRouter } from './routes/search.js';
@@ -22,6 +20,7 @@ export class ApiServer {
     port;
     httpServer = null;
     // Core services (populated after initialize())
+    runtime = null;
     config = null;
     store = null;
     hybridSearch = null;
@@ -129,62 +128,23 @@ export class ApiServer {
      */
     async initialize() {
         try {
-            const configResult = await loadConfig(this.rootDir);
-            if (configResult.isErr()) {
+            const runtimeResult = await createRuntime({ rootDir: this.rootDir });
+            if (runtimeResult.isErr()) {
                 // eslint-disable-next-line no-console
-                console.error(`[api-server] Config load failed: ${configResult.error.message}`);
+                console.error(`[api-server] ${runtimeResult.error.message}`);
                 return;
             }
-            this.config = configResult.value;
-            // Create embedding provider
-            const embeddingProvider = new OllamaEmbeddingProvider({
-                model: this.config.embedding.model,
-                dimensions: this.config.embedding.dimensions,
-            });
-            // Create LanceDB store
-            const storagePath = resolve(this.rootDir, this.config.storage.path);
-            if (!storagePath.startsWith(resolve(this.rootDir))) {
-                // eslint-disable-next-line no-console
-                console.error('[api-server] Storage path escapes project root');
-                return;
-            }
-            this.store = new LanceDBStore(storagePath, this.config.embedding.dimensions);
-            await this.store.connect();
-            // Load BM25 index
-            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
+            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;
+            // Expose graph as DependencyGraph for viewer API (getAllNodes, getAllEdges)
+            const graphInstance = this.runtime.graph;
+            if (graphInstance instanceof DependencyGraph) {
+                this.graph = graphInstance;
             }
-            // 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
-            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
-            }
-            // Store graph for viewer API
-            this.graph = graph;
-            // Create context expander
-            const chunkLookup = (_chunkId) => undefined;
-            this.contextExpander = new ContextExpander(graph, chunkLookup);
         }
         catch (error) {
             const message = error instanceof Error ? error.message : 'Unknown error';
@@ -219,6 +179,11 @@ export class ApiServer {
             });
             this.httpServer = null;
         }
+        // Close the runtime (LanceDB, etc.)
+        if (this.runtime) {
+            this.runtime.close();
+            this.runtime = null;
+        }
     }
     /** Expose Express app for testing with supertest. */
     getApp() {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@code-rag/api-server",
-  "version": "0.1.6",
+  "version": "0.1.8",
   "description": "REST API server for CodeRAG — Express-based server with authentication, rate limiting, and OpenAPI docs",
   "type": "module",
   "license": "MIT",
@@ -41,7 +41,7 @@
   "dependencies": {
     "express": "^4.21.2",
     "zod": "^4.3.6",
-    "@code-rag/core": "0.1.6"
+    "@code-rag/core": "0.1.8"
   },
   "devDependencies": {
     "@types/express": "^5.0.2",