@storybook/addon-mcp 0.1.7 → 0.1.8

package/dist/preset.js

@@ -11,7 +11,7 @@ import { buffer } from "node:stream/consumers";
 
 //#region package.json
 var name = "@storybook/addon-mcp";
-var version = "0.1.7";
+var version = "0.1.8";
 var description = "Help agents automatically write and test stories for your UI components";
 
 //#endregion
@@ -219,6 +219,60 @@ const getManifestStatus = async (options) => {
 	};
 };
 
+//#endregion
+//#region src/utils/estimate-tokens.ts
+/**
+* Checks if a character code is whitespace (space, tab, newline, carriage return)
+*
+* Checking char codes is slightly faster than using regex or string methods.
+*/
+function isWhitespace(code) {
+	return code === 32 || code === 9 || code === 10 || code === 13;
+}
+/**
+* Checks if a character code is alphanumeric or underscore
+* 0-9 (48-57), A-Z (65-90), a-z (97-122), underscore (95)
+*
+* Checking char codes is slightly faster than using regex or string methods.
+*/
+function isAlphanumeric(code) {
+	return code >= 48 && code <= 57 || code >= 65 && code <= 90 || code >= 97 && code <= 122 || code === 95;
+}
+/**
+* Estimates token count from text using a fast approximation.
+* Counts:
+* - Continuous whitespace as a single token
+* - Continuous alphanumeric sequences as single tokens
+* - Each special character as an individual token
+*
+* This is a cheap approximation suitable for telemetry purposes.
+*
+* @param text - The text to estimate token count for
+* @returns Estimated token count
+*/
+function estimateTokens(text) {
+	if (!text) return 0;
+	let tokenCount = 0;
+	let i = 0;
+	const len = text.length;
+	while (i < len) {
+		const code = text.charCodeAt(i);
+		if (isWhitespace(code)) {
+			tokenCount++;
+			i++;
+			while (i < len && isWhitespace(text.charCodeAt(i))) i++;
+		} else if (isAlphanumeric(code)) {
+			tokenCount++;
+			i++;
+			while (i < len && isAlphanumeric(text.charCodeAt(i))) i++;
+		} else {
+			tokenCount++;
+			i++;
+		}
+	}
+	return tokenCount;
+}
+
 //#endregion
 //#region src/mcp-handler.ts
 let transport;
@@ -266,22 +320,24 @@ const mcpServerHandler = async ({ req, res, options, addonOptions }) => {
 		disableTelemetry,
 		request: webRequest,
 		...!disableTelemetry && {
-			onListAllDocumentation: async ({ manifests }) => {
+			onListAllDocumentation: async ({ manifests, resultText }) => {
 				await collectTelemetry({
 					event: "tool:listAllDocumentation",
 					server,
 					toolset: "docs",
 					componentCount: Object.keys(manifests.componentManifest.components).length,
-					docsCount: Object.keys(manifests.docsManifest?.docs || {}).length
+					docsCount: Object.keys(manifests.docsManifest?.docs || {}).length,
+					resultTokenCount: estimateTokens(resultText)
 				});
 			},
-			onGetDocumentation: async ({ input, foundDocumentation }) => {
+			onGetDocumentation: async ({ input, foundDocumentation, resultText }) => {
 				await collectTelemetry({
 					event: "tool:getDocumentation",
 					server,
 					toolset: "docs",
 					componentId: input.id,
-					found: !!foundDocumentation
+					found: !!foundDocumentation,
+					resultTokenCount: estimateTokens(resultText ?? "")
 				});
 			}
 		}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@storybook/addon-mcp",
-  "version": "0.1.7",
+  "version": "0.1.8",
   "description": "Help agents automatically write and test stories for your UI components",
   "keywords": [
     "storybook-addon",
@@ -29,7 +29,7 @@
     "@tmcp/transport-http": "^0.8.0",
     "tmcp": "^1.16.0",
     "valibot": "1.2.0",
-    "@storybook/mcp": "0.2.0"
+    "@storybook/mcp": "0.2.1"
   },
   "devDependencies": {
     "storybook": "10.2.0-alpha.14"