@docrouter/mcp 0.2.0 → 0.2.2

package/dist/index.mjs

@@ -3,8 +3,8 @@ 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 { DocRouterAccount, DocRouterOrg } from '@docrouter/sdk';
+import { readFileSync, existsSync } from 'fs';
 import { dirname, join } from 'path';
 import { fileURLToPath } from 'url';
 
@@ -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,57 @@ 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
-  });
+async function initializeClient(config) {
+  try {
+    console.error("Resolving organization ID from token...");
+    const accountClient = new DocRouterAccount({
+      baseURL: config.baseURL,
+      accountToken: config.orgToken,
+      timeout: config.timeout,
+      retries: config.retries
+    });
+    const tokenResponse = await accountClient.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) {
@@ -887,9 +942,9 @@ 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" }
       },
-      required: ["promptId", "prompt"]
+      required: ["promptId", "content"]
     }
   },
   {
@@ -1475,9 +1530,29 @@ server.setRequestHandler(CallToolRequestSchema, async (request) => {
         };
       }
       case "update_prompt": {
+        const promptId = getArg(args, "promptId");
+        const content = getArg(args, "content");
+        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,
+            schema_id: currentPrompt.schema_id,
+            schema_version: currentPrompt.schema_version,
+            tag_ids: currentPrompt.tag_ids,
+            model: currentPrompt.model
+          }
         });
         return {
           content: [
@@ -1768,9 +1843,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 +1867,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 +1891,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 +1931,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");