@upstash/context7-mcp 2.2.4 → 2.2.5

package/dist/index.js

@@ -216,6 +216,49 @@ Do not call this tool more than 3 times per question.`,
     });
     return server;
 }
+const GLOBAL_ALIASES = {
+    query: ["userQuery", "question"],
+};
+// Tool-scoped aliases, for keys that are canonical on one tool but a
+// hallucination on another (e.g. `libraryName` is canonical for
+// `resolve-library-id`, so we only rewrite it on `query-docs` calls).
+const TOOL_ALIASES = {
+    "query-docs": {
+        libraryId: ["context7CompatibleLibraryID", "libraryID", "libraryName"],
+    },
+};
+function applyAliases(args, aliases) {
+    for (const [canonical, alternatives] of Object.entries(aliases)) {
+        if (canonical in args)
+            continue;
+        for (const alt of alternatives) {
+            if (alt in args) {
+                args[canonical] = args[alt];
+                delete args[alt];
+                break;
+            }
+        }
+    }
+}
+// Install BEFORE `server.connect(transport)`: the SDK's `Protocol.connect()`
+// captures the existing `onmessage` and chains its dispatch handler over it,
+// so our hook runs first on every incoming JSON-RPC message.
+function installTransportArgAliasing(transport) {
+    transport.onmessage = (message) => {
+        const msg = message;
+        if (msg.method !== "tools/call")
+            return;
+        const args = msg.params?.arguments;
+        if (!args || typeof args !== "object")
+            return;
+        const argsRecord = args;
+        applyAliases(argsRecord, GLOBAL_ALIASES);
+        const toolName = msg.params?.name;
+        if (toolName && toolName in TOOL_ALIASES) {
+            applyAliases(argsRecord, TOOL_ALIASES[toolName]);
+        }
+    };
+}
 async function main() {
     const transportType = TRANSPORT_TYPE;
     if (transportType === "http") {
@@ -317,6 +360,7 @@ async function main() {
                     server.close();
                 });
                 await requestContext.run(context, async () => {
+                    installTransportArgAliasing(transport);
                     await server.connect(transport);
                     await transport.handleRequest(req, res, req.body);
                 });
@@ -412,6 +456,9 @@ async function main() {
     }
     else {
         stdioApiKey = cliOptions.apiKey || process.env.CONTEXT7_API_KEY;
+        process.stdin.on("end", () => process.exit(0));
+        process.stdin.on("close", () => process.exit(0));
+        process.on("SIGHUP", () => process.exit(0));
         const transport = new StdioServerTransport();
         const server = createMcpServer();
         // Capture client info from MCP initialize handshake (stdio only — HTTP
@@ -425,6 +472,7 @@ async function main() {
                 };
             }
         };
+        installTransportArgAliasing(transport);
         await server.connect(transport);
         console.error(`Context7 Documentation MCP Server v${SERVER_VERSION} running on stdio`);
     }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@upstash/context7-mcp",
-  "version": "2.2.4",
+  "version": "2.2.5",
   "mcpName": "io.github.upstash/context7",
   "description": "MCP server for Context7",
   "repository": {