@assistant-ui/mcp-docs-server 0.3.2 → 0.3.4

package/LICENSE

@@ -1,6 +1,6 @@
 MIT License
 
-Copyright (c) 2025 AgentbaseAI Inc.
+Copyright (c) 2026 AgentbaseAI Inc.
 
 Permission is hereby granted, free of charge, to any person obtaining a copy
 of this software and associated documentation files (the "Software"), to deal

package/dist/index.d.ts

@@ -1,6 +1,4 @@
 import { runProxy } from "./proxy.js";
-//#region src/index.d.ts
-export declare function runServer(): Promise<void>;
-//#endregion
 export { runProxy };
+export declare function runServer(): Promise<void>;
 //# sourceMappingURL=index.d.ts.map

package/dist/index.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","names":[],"sources":["../src/index.ts"],"mappings":";;wBAIsB,aAAS"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,MAAM,YAAY,CAAC;AAEtC,OAAO,EAAE,QAAQ,EAAE,CAAC;AAEpB,wBAAsB,SAAS,kBAE9B"}

package/dist/proxy.d.ts

@@ -1,8 +1,6 @@
-//#region src/proxy.d.ts
-export declare function runProxy({ url, stdin, stdout }?: {
-  url?: URL;
-  stdin?: NodeJS.ReadableStream;
-  stdout?: NodeJS.WritableStream;
+export declare function runProxy({ url, stdin, stdout, }?: {
+    url?: URL;
+    stdin?: NodeJS.ReadableStream;
+    stdout?: NodeJS.WritableStream;
 }): Promise<void>;
-//#endregion
 //# sourceMappingURL=proxy.d.ts.map

package/dist/proxy.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"proxy.d.ts","names":[],"sources":["../src/proxy.ts"],"mappings":";wBAesB,WACpB,KACA,OACA;EAEA,MAAM;EACN,QAAQ,OAAO;EACf,SAAS,OAAO;IACZ"}
+{"version":3,"file":"proxy.d.ts","sourceRoot":"","sources":["../src/proxy.ts"],"names":[],"mappings":"AAeA,wBAAsB,QAAQ,CAAC,EAC7B,GAA8D,EAC9D,KAAqB,EACrB,MAAuB,GACxB,GAAE;IACD,GAAG,CAAC,EAAE,GAAG,CAAC;IACV,KAAK,CAAC,EAAE,MAAM,CAAC,cAAc,CAAC;IAC9B,MAAM,CAAC,EAAE,MAAM,CAAC,cAAc,CAAC;CAC3B,iBAsGL"}

package/dist/stdio.d.ts

@@ -1 +1,3 @@
-export {}
+#!/usr/bin/env node
+export {};
+//# sourceMappingURL=stdio.d.ts.map

package/dist/stdio.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"stdio.d.ts","sourceRoot":"","sources":["../src/stdio.ts"],"names":[],"mappings":""}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@assistant-ui/mcp-docs-server",
-  "version": "0.3.2",
+  "version": "0.3.4",
   "description": "Stdio proxy for the hosted assistant-ui documentation MCP server",
   "keywords": [
     "mcp",
@@ -31,15 +31,15 @@
   ],
   "sideEffects": false,
   "dependencies": {
-    "@modelcontextprotocol/client": "^2.0.0",
-    "@modelcontextprotocol/server": "^2.0.0"
+    "@modelcontextprotocol/client": "^2.3.1",
+    "@modelcontextprotocol/server": "^2.3.1"
   },
   "devDependencies": {
-    "@assistant-ui/x-buildutils": "0.0.28",
-    "@modelcontextprotocol/core": "^2.0.0",
-    "@types/node": "^26.5.1",
-    "tsx": "^4.23.13",
-    "vitest": "^5.0.1"
+    "@assistant-ui/x-buildutils": "0.0.30",
+    "@modelcontextprotocol/core": "^2.3.1",
+    "@types/node": "^26.6.4",
+    "tsx": "^4.23.15",
+    "vitest": "^5.0.3"
   },
   "publishConfig": {
     "access": "public",
@@ -58,6 +58,7 @@
     "build": "aui-build",
     "dev": "tsx src/stdio.ts",
     "test": "vitest run",
+    "test:coverage": "vitest run --coverage --coverage.reporter=text --coverage.reporter=html --coverage.reporter=json-summary",
     "typecheck": "tsc --noEmit",
     "test:watch": "vitest"
   }

package/src/proxy.test.ts

@@ -291,34 +291,48 @@ describe("runProxy", () => {
       required: ["text"],
       additionalProperties: false,
     });
-    const mcpServer = new McpServer({ name: "proxy-test", version: "1.0.0" });
-    mcpServer.registerTool(
-      "echo",
-      { description: "Echo text", inputSchema },
-      ({ text }) => ({ content: [{ type: "text", text }] }),
-    );
-    const transport = new WebStandardStreamableHTTPServerTransport({
-      sessionIdGenerator: undefined,
-      enableJsonResponse: true,
-    });
     let resolveInitialized = () => {};
     const initialized = new Promise<void>((resolve) => {
       resolveInitialized = resolve;
     });
-    mcpServer.server.oninitialized = resolveInitialized;
-    await mcpServer.connect(transport);
 
     const httpServer = createServer((request, response) => {
       void (async () => {
         const webRequest = await toWebRequest(request);
-        const webResponse =
-          webRequest.method === "GET"
-            ? Response.json({
-                name: "assistant-ui",
-                transport: "streamable-http",
-              })
-            : await transport.handleRequest(webRequest);
-        await writeWebResponse(webResponse, response);
+        if (webRequest.method === "GET") {
+          await writeWebResponse(
+            Response.json({
+              name: "assistant-ui",
+              transport: "streamable-http",
+            }),
+            response,
+          );
+          return;
+        }
+
+        const mcpServer = new McpServer({
+          name: "proxy-test",
+          version: "1.0.0",
+        });
+        mcpServer.registerTool(
+          "echo",
+          { description: "Echo text", inputSchema },
+          ({ text }) => ({ content: [{ type: "text", text }] }),
+        );
+        mcpServer.server.oninitialized = resolveInitialized;
+        const transport = new WebStandardStreamableHTTPServerTransport({
+          sessionIdGenerator: undefined,
+          enableJsonResponse: true,
+        });
+
+        try {
+          await mcpServer.connect(transport);
+          const webResponse = await transport.handleRequest(webRequest);
+          await writeWebResponse(webResponse, response);
+        } finally {
+          await transport.close();
+          await mcpServer.close();
+        }
       })().catch((error: unknown) => {
         response.statusCode = 500;
         response.end(error instanceof Error ? error.message : String(error));
@@ -378,7 +392,6 @@ describe("runProxy", () => {
       proxyStdout.destroy(new Error("Proxy test cleanup"));
       await proxy;
       proxyStdin.destroy();
-      await mcpServer.close();
       stderr.mockRestore();
     }
   });