Expose exact source documents through MCP

scripts/smoke-exact-source-document-mcp.ts

@@ -0,0 +1,54 @@
+import { mkdtemp, rm } from "node:fs/promises";
+import { tmpdir } from "node:os";
+import { join } from "node:path";
+
+import { Client } from "@modelcontextprotocol/sdk/client/index.js";
+import { InMemoryTransport } from "@modelcontextprotocol/sdk/inMemory.js";
+import {
+  FileSourceDocumentArchiveRepository,
+  SourceDocumentIngester,
+} from "@voidbot/rag";
+
+import { createVoidbotMcpServer } from "../apps/worker/src/mcp-server-factory";
+import { type VoidbotMcpContext } from "../apps/worker/src/mcp-server-context";
+
+async function main(): Promise<void> {
+  const root = await mkdtemp(join(tmpdir(), "voidbot-exact-source-"));
+  try {
+  const sources = new FileSourceDocumentArchiveRepository(join(root, "sources.json"));
+  await sources.syncRepoDocuments("AetheriaLore", [{
+    id: "AetheriaLore:places/forge.md",
+    repoName: "AetheriaLore",
+    path: "places/forge.md",
+    content: "John keeps the village forge.\nThe eastern road takes six hours.",
+    language: "markdown",
+    title: "The Village Forge",
+  }]);
+  const context = {
+    sourceArchiveRepository: sources,
+    sourceDocumentIngester: new SourceDocumentIngester(),
+  } as VoidbotMcpContext;
+  const server = createVoidbotMcpServer(context);
+  const client = new Client({ name: "exact-source-smoke", version: "1.0.0" });
+  const [clientTransport, serverTransport] = InMemoryTransport.createLinkedPair();
+  await Promise.all([server.connect(serverTransport), client.connect(clientTransport)]);
+  const result = await client.callTool({
+    name: "get_exact_source_document",
+    arguments: { sourceId: "AetheriaLore:places/forge.md" },
+  });
+  const typed = result.structuredContent as Record<string, unknown>;
+  if (typed.found !== true || typed.content !== "John keeps the village forge.\nThe eastern road takes six hours.") {
+    throw new Error(`exact source MCP contract failed: ${JSON.stringify(typed)}`);
+  }
+  await client.close();
+  await server.close();
+  console.log("Exact source MCP smoke passed.");
+  } finally {
+    await rm(root, { recursive: true, force: true });
+  }
+}
+
+void main().catch((error) => {
+  console.error(error);
+  process.exitCode = 1;
+});