@docrouter/mcp 0.2.2 → 0.3.0

package/dist/index.mjs

@@ -4,8 +4,8 @@ import { StdioServerTransport } from '@modelcontextprotocol/sdk/server/stdio.js'
 import { ListToolsRequestSchema, CallToolRequestSchema } from '@modelcontextprotocol/sdk/types.js';
 import { z } from 'zod';
 import { DocRouterAccount, DocRouterOrg } from '@docrouter/sdk';
-import { readFileSync, existsSync } from 'fs';
-import { dirname, join } from 'path';
+import { readFileSync, existsSync, statSync, mkdirSync, writeFileSync } from 'fs';
+import { isAbsolute, resolve, extname, basename, join, dirname } from 'path';
 import { fileURLToPath } from 'url';
 
 var __require = /* @__PURE__ */ ((x) => typeof require !== "undefined" ? require : typeof Proxy !== "undefined" ? new Proxy(x, {
@@ -156,16 +156,19 @@ var server = new Server(
   }
 );
 var docrouterClient;
+var docrouterAccountClient;
+var orgToken;
 async function initializeClient(config) {
   try {
     console.error("Resolving organization ID from token...");
-    const accountClient = new DocRouterAccount({
+    docrouterAccountClient = new DocRouterAccount({
       baseURL: config.baseURL,
       accountToken: config.orgToken,
       timeout: config.timeout,
       retries: config.retries
     });
-    const tokenResponse = await accountClient.getOrganizationFromToken(config.orgToken);
+    orgToken = config.orgToken;
+    const tokenResponse = await docrouterAccountClient.getOrganizationFromToken(config.orgToken);
     const organizationId = tokenResponse.organization_id;
     if (!organizationId) {
       throw new Error("Token is an account-level token, not an organization-specific token. Please use an organization API token.");
@@ -562,7 +565,7 @@ var tools = [
   // ========== DOCUMENTS ==========
   {
     name: "upload_documents",
-    description: "Upload documents to DocRouter",
+    description: "Upload documents to DocRouter from file paths",
     inputSchema: {
       type: "object",
       properties: {
@@ -572,12 +575,12 @@ var tools = [
           items: {
             type: "object",
             properties: {
-              name: { type: "string", description: "Document name" },
-              content: { type: "string", description: "Base64 encoded document content (supports both plain base64 and data URLs)" },
+              file_path: { type: "string", description: "Path to the document file on disk" },
+              name: { type: "string", description: "Document name (optional, defaults to filename)" },
               tag_ids: { type: "array", items: { type: "string" }, description: "Optional list of tag IDs" },
               metadata: { type: "object", description: "Optional metadata" }
             },
-            required: ["name", "content"]
+            required: ["file_path"]
           }
         }
       },
@@ -600,12 +603,13 @@ var tools = [
   },
   {
     name: "get_document",
-    description: "Get document by ID from DocRouter",
+    description: "Get document metadata (state, tags, metadata) and optionally download the file to disk",
     inputSchema: {
       type: "object",
       properties: {
         documentId: { type: "string", description: "ID of the document to retrieve" },
-        fileType: { type: "string", description: "File type to retrieve (pdf, image, etc.)", default: "pdf" }
+        fileType: { type: "string", description: "File type to retrieve (original or pdf)", default: "original" },
+        save_path: { type: "string", description: "Optional file path or directory to save the document. If directory, uses original filename. If not provided, file is not downloaded." }
       },
       required: ["documentId"]
     }
@@ -942,9 +946,12 @@ var tools = [
       type: "object",
       properties: {
         promptId: { type: "string", description: "ID of the prompt" },
-        content: { type: "string", description: "Prompt content" }
+        content: { type: "string", description: "Prompt content" },
+        model: { type: "string", description: "LLM model to use" },
+        schema_id: { type: "string", description: "Schema ID to link" },
+        tag_ids: { type: "array", items: { type: "string" }, description: "Tag IDs that trigger this prompt" }
       },
-      required: ["promptId", "content"]
+      required: ["promptId"]
     }
   },
   {
@@ -1078,6 +1085,22 @@ var tools = [
       required: ["messages", "model"]
     }
   },
+  {
+    name: "get_organization",
+    description: "Get information about the current organization (name, type, ID)",
+    inputSchema: {
+      type: "object",
+      properties: {}
+    }
+  },
+  {
+    name: "list_llm_models",
+    description: "List enabled LLM models available for use in prompts for this organization",
+    inputSchema: {
+      type: "object",
+      properties: {}
+    }
+  },
   // ========== HELPER TOOLS ==========
   {
     name: "help",
@@ -1139,21 +1162,57 @@ server.setRequestHandler(CallToolRequestSchema, async (request) => {
         };
       }
       case "get_document": {
-        const result = await docrouterClient.getDocument({
-          documentId: getArg(args, "documentId"),
-          fileType: getArg(args, "fileType", "pdf")
+        const documentId = getArg(args, "documentId");
+        const fileType = getArg(args, "fileType", "original");
+        const savePath = getOptionalArg(args, "save_path");
+        const fileResult = await docrouterClient.getDocument({
+          documentId,
+          fileType
         });
-        const base64Content = Buffer.from(result.content).toString("base64");
+        const response = {
+          id: fileResult.id,
+          pdf_id: fileResult.pdf_id,
+          document_name: fileResult.document_name,
+          upload_date: fileResult.upload_date,
+          uploaded_by: fileResult.uploaded_by,
+          state: fileResult.state,
+          tag_ids: fileResult.tag_ids,
+          type: fileResult.type,
+          metadata: fileResult.metadata
+        };
+        if (savePath) {
+          let finalPath;
+          if (isAbsolute(savePath)) {
+            finalPath = savePath;
+          } else {
+            finalPath = resolve(process.cwd(), savePath);
+          }
+          let isDirectory = false;
+          try {
+            const stats = statSync(finalPath);
+            isDirectory = stats.isDirectory();
+          } catch {
+            isDirectory = savePath.endsWith("/") || savePath.endsWith("\\");
+          }
+          if (isDirectory) {
+            const extension = fileType === "pdf" ? ".pdf" : extname(fileResult.document_name) || ".pdf";
+            const fileName = basename(fileResult.document_name, extname(fileResult.document_name)) + extension;
+            finalPath = join(finalPath, fileName);
+          }
+          const targetDir = dirname(finalPath);
+          if (!existsSync(targetDir)) {
+            mkdirSync(targetDir, { recursive: true });
+          }
+          const fileBuffer = Buffer.from(fileResult.content);
+          writeFileSync(finalPath, fileBuffer);
+          response.saved_path = finalPath;
+          response.file_size = fileBuffer.length;
+        }
         return {
           content: [
             {
               type: "text",
-              text: JSON.stringify({
-                ...serializeDates(result),
-                content: base64Content,
-                content_type: "base64",
-                content_size: result.content.byteLength
-              }, null, 2)
+              text: JSON.stringify(response, null, 2)
             }
           ]
         };
@@ -1276,7 +1335,28 @@ server.setRequestHandler(CallToolRequestSchema, async (request) => {
       // ========== DOCUMENTS ==========
       case "upload_documents": {
         const documentsInput = getArg(args, "documents");
-        const result = await docrouterClient.uploadDocuments({ documents: documentsInput });
+        const documents = [];
+        for (const doc of documentsInput) {
+          let filePath;
+          if (isAbsolute(doc.file_path)) {
+            filePath = doc.file_path;
+          } else {
+            filePath = resolve(process.cwd(), doc.file_path);
+          }
+          if (!existsSync(filePath)) {
+            throw new Error(`File not found: ${filePath}`);
+          }
+          const fileBuffer = readFileSync(filePath);
+          const base64Content = fileBuffer.toString("base64");
+          const fileName = doc.name || filePath.split(/[/\\]/).pop() || "document";
+          documents.push({
+            name: fileName,
+            content: base64Content,
+            tag_ids: doc.tag_ids,
+            metadata: doc.metadata
+          });
+        }
+        const result = await docrouterClient.uploadDocuments({ documents });
         return {
           content: [
             {
@@ -1531,7 +1611,10 @@ server.setRequestHandler(CallToolRequestSchema, async (request) => {
       }
       case "update_prompt": {
         const promptId = getArg(args, "promptId");
-        const content = getArg(args, "content");
+        const content = getOptionalArg(args, "content");
+        const model = getOptionalArg(args, "model");
+        const schema_id = getOptionalArg(args, "schema_id");
+        const tag_ids = getOptionalArg(args, "tag_ids");
         let currentPrompt = null;
         const allPrompts = await docrouterClient.listPrompts({});
         for (const p of allPrompts.prompts) {
@@ -1547,11 +1630,11 @@ server.setRequestHandler(CallToolRequestSchema, async (request) => {
           promptId,
           prompt: {
             name: currentPrompt.name,
-            content,
-            schema_id: currentPrompt.schema_id,
+            content: content ?? currentPrompt.content,
+            schema_id: schema_id ?? currentPrompt.schema_id,
             schema_version: currentPrompt.schema_version,
-            tag_ids: currentPrompt.tag_ids,
-            model: currentPrompt.model
+            tag_ids: tag_ids ?? currentPrompt.tag_ids,
+            model: model ?? currentPrompt.model
           }
         });
         return {
@@ -1736,6 +1819,28 @@ server.setRequestHandler(CallToolRequestSchema, async (request) => {
           ]
         };
       }
+      case "get_organization": {
+        const result = await docrouterAccountClient.getOrganizationFromToken(orgToken);
+        return {
+          content: [
+            {
+              type: "text",
+              text: JSON.stringify(serializeDates(result), null, 2)
+            }
+          ]
+        };
+      }
+      case "list_llm_models": {
+        const result = await docrouterClient.listLLMModels();
+        return {
+          content: [
+            {
+              type: "text",
+              text: JSON.stringify(serializeDates(result), null, 2)
+            }
+          ]
+        };
+      }
       // ========== HELPER TOOLS ==========
       case "help": {
         const helpText = `
@@ -1746,7 +1851,7 @@ This server provides access to DocRouter resources and tools.
 ## Available Tools
 
 ### Documents
-- \`upload_documents(documents)\` - Upload documents
+- \`upload_documents(documents)\` - Upload documents from file paths
 - \`list_documents(skip, limit, tagIds, nameSearch, metadataSearch)\` - List documents
 - \`get_document(documentId, fileType)\` - Get document by ID
 - \`update_document(documentId, documentName, tagIds, metadata)\` - Update document
@@ -1800,6 +1905,10 @@ This server provides access to DocRouter resources and tools.
 ### LLM Chat
 - \`run_llm_chat(messages, model, temperature, max_tokens, stream)\` - Run chat
 
+### Organization
+- \`get_organization()\` - Get information about the current organization (name, type, ID)
+- \`list_llm_models()\` - List enabled LLM models available for use in prompts
+
 ### Help Tools
 - \`help()\` - Get general API help information
 - \`help_prompts()\` - Get detailed help on creating and configuring prompts