@imenam/mcp-documentor 1.0.33 → 1.0.34

package/dist/src/index.js

@@ -33,6 +33,7 @@ import { HistoryStore } from "./core/history-store.js";
 import { listDocVersions } from "./tools/list-doc-versions.js";
 import { getDocVersion } from "./tools/get-doc-version.js";
 import { restoreDocVersion } from "./tools/restore-doc-version.js";
+import { reindexEmbeddings } from "./tools/reindex-embeddings.js";
 const __filename = fileURLToPath(import.meta.url);
 const __dirname = path.dirname(__filename);
 function findProjectRoot(startDir) {
@@ -709,6 +710,18 @@ server.setRequestHandler(ListToolsRequestSchema, async () => {
                     required: ["path", "version"],
                 },
             },
+            {
+                name: "reindex_embeddings",
+                description: "Generate missing embeddings for all documentation entries in a project (or all projects). Useful when importing existing documentation that was created outside the MCP. Use force: true to regenerate embeddings for all docs, including those already indexed.",
+                inputSchema: {
+                    type: "object",
+                    properties: {
+                        project: { type: "string", description: "Project to reindex (processes all projects if omitted)" },
+                        force: { type: "boolean", description: "If true, regenerates embeddings for all docs even if they are already indexed (default: false)" },
+                    },
+                    required: [],
+                },
+            },
         ],
     };
 });
@@ -754,6 +767,8 @@ server.setRequestHandler(CallToolRequestSchema, async (request) => {
                 return await getDocVersion(historyStore, storageEngine, args);
             case "restore_doc_version":
                 return await restoreDocVersion(historyStore, storageEngine, embeddingEngine, args);
+            case "reindex_embeddings":
+                return await reindexEmbeddings(storageEngine, embeddingEngine, embeddingStore, args);
             default:
                 throw new Error(`Tool not found: ${name}`);
         }

package/dist/src/tools/reindex-embeddings.d.ts

@@ -0,0 +1,16 @@
+import type { StorageEngine } from "../core/storage-engine.js";
+import type { EmbeddingEngine } from "../core/embedding-engine.js";
+import type { EmbeddingStore } from "../core/embedding-store.js";
+export declare function reindexEmbeddings(storageEngine: StorageEngine, embeddingEngine: EmbeddingEngine, embeddingStore: EmbeddingStore, args: unknown): Promise<{
+    isError: boolean;
+    content: {
+        type: string;
+        text: string;
+    }[];
+} | {
+    content: {
+        type: string;
+        text: string;
+    }[];
+    isError?: undefined;
+}>;

package/dist/src/tools/reindex-embeddings.js

@@ -0,0 +1,83 @@
+import { z } from "zod";
+const ReindexEmbeddingsArgsSchema = z.object({
+    project: z.string().optional(),
+    force: z.boolean().optional().default(false),
+});
+export async function reindexEmbeddings(storageEngine, embeddingEngine, embeddingStore, args) {
+    try {
+        if (!embeddingEngine.isConfigured()) {
+            return {
+                isError: true,
+                content: [
+                    {
+                        type: "text",
+                        text: "OpenAI API key is not configured. Set MCP_DOCUMENTOR_OPENAI_API_KEY in your .env file to enable embedding generation.",
+                    },
+                ],
+            };
+        }
+        const validated = ReindexEmbeddingsArgsSchema.parse(args);
+        const config = storageEngine.getConfig();
+        const projectsToProcess = validated.project
+            ? [validated.project]
+            : config.projects;
+        const summary = [];
+        for (const project of projectsToProcess) {
+            const docs = await storageEngine.listDocs(project);
+            if (docs.length === 0) {
+                summary.push({ project, indexed: 0, skipped: 0, failed: 0 });
+                continue;
+            }
+            const existingEmbeddings = embeddingStore.loadAll(project);
+            let indexed = 0;
+            let skipped = 0;
+            let failed = 0;
+            for (const docSummary of docs) {
+                const alreadyIndexed = existingEmbeddings.has(docSummary.id);
+                if (alreadyIndexed && !validated.force) {
+                    skipped++;
+                    continue;
+                }
+                const doc = await storageEngine.getDoc(docSummary.id, project);
+                if (!doc) {
+                    failed++;
+                    continue;
+                }
+                const embedding = await embeddingEngine.generateEmbedding(doc.content);
+                if (embedding) {
+                    await embeddingEngine.saveEmbedding(doc.id, embedding, project);
+                    indexed++;
+                }
+                else {
+                    failed++;
+                }
+            }
+            summary.push({ project, indexed, skipped, failed });
+        }
+        const lines = summary.map((s) => `• [${s.project}] ${s.indexed} indexed, ${s.skipped} skipped (already indexed), ${s.failed} failed`);
+        const totalIndexed = summary.reduce((acc, s) => acc + s.indexed, 0);
+        const totalSkipped = summary.reduce((acc, s) => acc + s.skipped, 0);
+        const totalFailed = summary.reduce((acc, s) => acc + s.failed, 0);
+        const resultText = [
+            `Reindex complete across ${summary.length} project(s):`,
+            ...lines,
+            "",
+            `Total: ${totalIndexed} indexed, ${totalSkipped} skipped, ${totalFailed} failed.`,
+            totalSkipped > 0 && !validated.force
+                ? "Tip: use force: true to regenerate all embeddings including existing ones."
+                : "",
+        ]
+            .filter((l) => l !== "")
+            .join("\n");
+        return {
+            content: [{ type: "text", text: resultText }],
+        };
+    }
+    catch (err) {
+        const message = err instanceof Error ? err.message : String(err);
+        return {
+            isError: true,
+            content: [{ type: "text", text: `Error during reindex: ${message}` }],
+        };
+    }
+}

package/dist/src/version.d.ts

@@ -1 +1 @@
-export declare const APP_VERSION = "1.0.33";
+export declare const APP_VERSION = "1.0.34";

package/dist/src/version.js

@@ -1,2 +1,2 @@
 // Auto-generated version file - do not edit manually
-export const APP_VERSION = '1.0.33';
+export const APP_VERSION = '1.0.34';

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@imenam/mcp-documentor",
-  "version": "1.0.33",
+  "version": "1.0.34",
   "main": "dist/src/index.js",
   "scripts": {
     "test": "echo \"Error: no test specified\" && exit 1",