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

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@assistant-ui/mcp-docs-server",
-  "version": "0.3.3",
+  "version": "0.3.4",
   "description": "Stdio proxy for the hosted assistant-ui documentation MCP server",
   "keywords": [
     "mcp",
@@ -31,13 +31,13 @@
   ],
   "sideEffects": false,
   "dependencies": {
-    "@modelcontextprotocol/client": "^2.2.0",
-    "@modelcontextprotocol/server": "^2.2.0"
+    "@modelcontextprotocol/client": "^2.3.1",
+    "@modelcontextprotocol/server": "^2.3.1"
   },
   "devDependencies": {
     "@assistant-ui/x-buildutils": "0.0.30",
-    "@modelcontextprotocol/core": "^2.2.0",
-    "@types/node": "^26.6.3",
+    "@modelcontextprotocol/core": "^2.3.1",
+    "@types/node": "^26.6.4",
     "tsx": "^4.23.15",
     "vitest": "^5.0.3"
   },
@@ -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();
     }
   });