@docrouter/mcp 0.2.1 → 0.3.0

package/dist/index.mjs

@@ -3,9 +3,9 @@ import { Server } from '@modelcontextprotocol/sdk/server/index.js';
 import { StdioServerTransport } from '@modelcontextprotocol/sdk/server/stdio.js';
 import { ListToolsRequestSchema, CallToolRequestSchema } from '@modelcontextprotocol/sdk/types.js';
 import { z } from 'zod';
-import { DocRouterOrg } from '@docrouter/sdk';
-import { readFileSync } from 'fs';
-import { dirname, join } from 'path';
+import { DocRouterAccount, DocRouterOrg } from '@docrouter/sdk';
+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, {
@@ -16,11 +16,36 @@ var __require = /* @__PURE__ */ ((x) => typeof require !== "undefined" ? require
 });
 var ConfigSchema = z.object({
   baseURL: z.string().default("https://app.docrouter.ai/fastapi"),
-  organizationId: z.string(),
   orgToken: z.string(),
   timeout: z.number().default(3e4),
   retries: z.number().default(3)
 });
+function resolveKnowledgeBasePath(filename) {
+  const currentFile = fileURLToPath(import.meta.url);
+  const currentDir = dirname(currentFile);
+  if (currentDir.endsWith("dist")) {
+    return join(currentDir, "docs/knowledge_base", filename);
+  }
+  if (currentDir.endsWith("src")) {
+    const distPath = join(currentDir, "..", "dist", "docs/knowledge_base", filename);
+    if (existsSync(distPath)) {
+      return distPath;
+    }
+    throw new Error(
+      `Knowledge base file not found: ${filename}
+Expected at: ${distPath}
+Please run 'npm run build' first to generate the knowledge base files.`
+    );
+  }
+  const fallbackPath = join(currentDir, "docs/knowledge_base", filename);
+  if (existsSync(fallbackPath)) {
+    return fallbackPath;
+  }
+  throw new Error(
+    `Knowledge base file not found: ${filename}
+Searched in: ${join(currentDir, "docs/knowledge_base", filename)}`
+  );
+}
 function showTools() {
   const toolNames = tools.map((tool) => tool.name).sort();
   console.log(toolNames.join("\n"));
@@ -51,8 +76,7 @@ DESCRIPTION:
 OPTIONS:
     --url <URL>              DocRouter API base URL
                             (default: https://app.docrouter.ai/fastapi)
-    --org-id <ID>            DocRouter organization ID
-    --org-token <TOKEN>      DocRouter organization API token
+    --org-token <TOKEN>      DocRouter organization API token (required)
     --timeout <MS>           Request timeout in milliseconds (default: 30000)
     --retries <COUNT>        Number of retry attempts (default: 3)
     --tools                  List all supported MCP tools
@@ -61,24 +85,22 @@ OPTIONS:
 
 ENVIRONMENT VARIABLES:
     DOCROUTER_API_URL        DocRouter API base URL
-    DOCROUTER_ORG_ID         DocRouter organization ID
-    DOCROUTER_ORG_API_TOKEN  DocRouter organization API token
+    DOCROUTER_ORG_API_TOKEN  DocRouter organization API token (required)
 
 EXAMPLES:
-    # Using command line arguments
-    docrouter-mcp --org-id "org123" --org-token "token456"
+    # Using command line arguments (organization ID will be resolved from token)
+    docrouter-mcp --org-token "token456"
 
-    # Using environment variables
-    export DOCROUTER_ORG_ID="org123"
+    # Using environment variables (organization ID will be resolved from token)
     export DOCROUTER_ORG_API_TOKEN="token456"
     docrouter-mcp
 
     # With custom API URL
-    docrouter-mcp --url "https://custom.docrouter.ai/fastapi" --org-id "org123" --org-token "token456"
+    docrouter-mcp --url "https://custom.docrouter.ai/fastapi" --org-token "token456"
 
 REQUIRED:
-    Either provide --org-id and --org-token as command line arguments,
-    or set DOCROUTER_ORG_ID and DOCROUTER_ORG_API_TOKEN environment variables.
+    DOCROUTER_ORG_API_TOKEN environment variable or --org-token argument.
+    Organization ID will be automatically resolved from the token.
 
 For more information about DocRouter, visit: https://docrouter.ai
 `);
@@ -106,9 +128,6 @@ function parseConfig() {
         case "url":
           config.baseURL = value;
           break;
-        case "org-id":
-          config.organizationId = value;
-          break;
         case "org-token":
           config.orgToken = value;
           break;
@@ -122,7 +141,6 @@ function parseConfig() {
     }
   }
   config.baseURL = config.baseURL || process.env.DOCROUTER_API_URL || "https://app.docrouter.ai/fastapi";
-  config.organizationId = config.organizationId || process.env.DOCROUTER_ORG_ID || "";
   config.orgToken = config.orgToken || process.env.DOCROUTER_ORG_API_TOKEN || "";
   return ConfigSchema.parse(config);
 }
@@ -138,20 +156,60 @@ var server = new Server(
   }
 );
 var docrouterClient;
-function initializeClient(config) {
-  docrouterClient = new DocRouterOrg({
-    baseURL: config.baseURL,
-    orgToken: config.orgToken,
-    organizationId: config.organizationId,
-    timeout: config.timeout,
-    retries: config.retries
-  });
+var docrouterAccountClient;
+var orgToken;
+async function initializeClient(config) {
+  try {
+    console.error("Resolving organization ID from token...");
+    docrouterAccountClient = new DocRouterAccount({
+      baseURL: config.baseURL,
+      accountToken: config.orgToken,
+      timeout: config.timeout,
+      retries: config.retries
+    });
+    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.");
+    }
+    console.error(`Resolved organization ID: ${organizationId}`);
+    docrouterClient = new DocRouterOrg({
+      baseURL: config.baseURL,
+      orgToken: config.orgToken,
+      organizationId,
+      timeout: config.timeout,
+      retries: config.retries
+    });
+  } catch (error) {
+    throw new Error(`Failed to resolve organization ID from token: ${error instanceof Error ? error.message : "Unknown error"}`);
+  }
 }
 function handleError(error) {
   if (error instanceof Error) {
-    return JSON.stringify({ error: error.message }, null, 2);
+    const apiError = error;
+    const errorObj = {
+      error: error.message
+    };
+    if (apiError.status !== void 0) {
+      errorObj.status = apiError.status;
+    }
+    if (apiError.code !== void 0) {
+      errorObj.code = apiError.code;
+    }
+    if (apiError.details !== void 0) {
+      errorObj.details = apiError.details;
+    }
+    return JSON.stringify(errorObj, null, 2);
   }
-  return JSON.stringify({ error: "Unknown error occurred" }, null, 2);
+  if (typeof error === "object" && error !== null) {
+    try {
+      return JSON.stringify({ error: "Request failed", details: error }, null, 2);
+    } catch {
+      return JSON.stringify({ error: "Unknown error occurred (could not serialize)" }, null, 2);
+    }
+  }
+  return JSON.stringify({ error: String(error) || "Unknown error occurred" }, null, 2);
 }
 function serializeDates(obj) {
   if (obj === null || obj === void 0) {
@@ -507,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: {
@@ -517,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"]
           }
         }
       },
@@ -545,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"]
     }
@@ -887,9 +946,12 @@ var tools = [
       type: "object",
       properties: {
         promptId: { type: "string", description: "ID of the prompt" },
-        prompt: { type: "object", description: "Prompt data" }
+        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", "prompt"]
+      required: ["promptId"]
     }
   },
   {
@@ -1023,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",
@@ -1084,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)
             }
           ]
         };
@@ -1221,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: [
             {
@@ -1475,9 +1610,32 @@ server.setRequestHandler(CallToolRequestSchema, async (request) => {
         };
       }
       case "update_prompt": {
+        const promptId = getArg(args, "promptId");
+        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) {
+          if (p.prompt_id === promptId) {
+            currentPrompt = p;
+            break;
+          }
+        }
+        if (!currentPrompt) {
+          throw new Error(`Prompt with ID ${promptId} not found`);
+        }
         const result = await docrouterClient.updatePrompt({
-          promptId: getArg(args, "promptId"),
-          prompt: getArg(args, "prompt")
+          promptId,
+          prompt: {
+            name: currentPrompt.name,
+            content: content ?? currentPrompt.content,
+            schema_id: schema_id ?? currentPrompt.schema_id,
+            schema_version: currentPrompt.schema_version,
+            tag_ids: tag_ids ?? currentPrompt.tag_ids,
+            model: model ?? currentPrompt.model
+          }
         });
         return {
           content: [
@@ -1661,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 = `
@@ -1671,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
@@ -1725,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
@@ -1768,9 +1952,7 @@ This server provides access to DocRouter resources and tools.
       }
       case "help_prompts": {
         try {
-          const currentFile = fileURLToPath(import.meta.url);
-          const currentDir = dirname(currentFile);
-          const promptsPath = join(currentDir, "docs/knowledge_base/prompts.md");
+          const promptsPath = resolveKnowledgeBasePath("prompts.md");
           const promptsContent = readFileSync(promptsPath, "utf-8");
           return {
             content: [
@@ -1794,9 +1976,7 @@ This server provides access to DocRouter resources and tools.
       }
       case "help_schemas": {
         try {
-          const currentFile = fileURLToPath(import.meta.url);
-          const currentDir = dirname(currentFile);
-          const schemasPath = join(currentDir, "docs/knowledge_base/schemas.md");
+          const schemasPath = resolveKnowledgeBasePath("schemas.md");
           const schemasContent = readFileSync(schemasPath, "utf-8");
           return {
             content: [
@@ -1820,9 +2000,7 @@ This server provides access to DocRouter resources and tools.
       }
       case "help_forms": {
         try {
-          const currentFile = fileURLToPath(import.meta.url);
-          const currentDir = dirname(currentFile);
-          const formsPath = join(currentDir, "docs/knowledge_base/forms.md");
+          const formsPath = resolveKnowledgeBasePath("forms.md");
           const formsContent = readFileSync(formsPath, "utf-8");
           return {
             content: [
@@ -1862,12 +2040,13 @@ This server provides access to DocRouter resources and tools.
 async function main() {
   try {
     const config = parseConfig();
-    if (!config.organizationId || !config.orgToken) {
-      console.error("Error: DOCROUTER_ORG_ID and DOCROUTER_ORG_API_TOKEN environment variables are required");
-      console.error("Or provide them as command line arguments: --org-id <id> --org-token <token>");
+    if (!config.orgToken) {
+      console.error("Error: DOCROUTER_ORG_API_TOKEN environment variable is required");
+      console.error("Or provide it as command line argument: --org-token <token>");
+      console.error("Organization ID will be automatically resolved from the token.");
       process.exit(1);
     }
-    initializeClient(config);
+    await initializeClient(config);
     const transport = new StdioServerTransport();
     await server.connect(transport);
     console.error("DocRouter MCP server started successfully");