diff --git a/.changeset/clean-bats-say.md b/.changeset/clean-bats-say.md new file mode 100644 index 00000000..d7d6ae3b --- /dev/null +++ b/.changeset/clean-bats-say.md @@ -0,0 +1,5 @@ +--- +'@storybook/addon-mcp': patch +--- + +Add toolset to telemetry payload diff --git a/packages/addon-mcp/src/mcp-handler.ts b/packages/addon-mcp/src/mcp-handler.ts index 8fe7b5f7..12c12719 100644 --- a/packages/addon-mcp/src/mcp-handler.ts +++ b/packages/addon-mcp/src/mcp-handler.ts @@ -104,6 +104,7 @@ export const mcpServerHandler = async ({ await collectTelemetry({ event: 'tool:listAllComponents', server, + toolset: 'docs', componentCount: Object.keys(manifest.components).length, }); }, @@ -111,6 +112,7 @@ export const mcpServerHandler = async ({ await collectTelemetry({ event: 'tool:getComponentDocumentation', server, + toolset: 'docs', componentId: input.componentId, found: !!foundComponent, }); diff --git a/packages/addon-mcp/src/tools/get-story-urls.test.ts b/packages/addon-mcp/src/tools/get-story-urls.test.ts index 71c58b05..f2d30021 100644 --- a/packages/addon-mcp/src/tools/get-story-urls.test.ts +++ b/packages/addon-mcp/src/tools/get-story-urls.test.ts @@ -279,6 +279,7 @@ describe('getStoryUrlsTool', () => { expect(collectTelemetrySpy).toHaveBeenCalledWith({ event: 'tool:getStoryUrls', server, + toolset: 'dev', inputStoryCount: 1, outputStoryCount: 1, }); diff --git a/packages/addon-mcp/src/tools/get-story-urls.ts b/packages/addon-mcp/src/tools/get-story-urls.ts index ede5f542..999f6cbd 100644 --- a/packages/addon-mcp/src/tools/get-story-urls.ts +++ b/packages/addon-mcp/src/tools/get-story-urls.ts @@ -83,6 +83,7 @@ export async function addGetStoryUrlsTool( await collectTelemetry({ event: 'tool:getStoryUrls', server, + toolset: 'dev', inputStoryCount: input.stories.length, outputStoryCount: foundStoryCount, }); diff --git a/packages/addon-mcp/src/tools/get-ui-building-instructions.test.ts b/packages/addon-mcp/src/tools/get-ui-building-instructions.test.ts index d7ab175a..2ca74dd5 100644 --- a/packages/addon-mcp/src/tools/get-ui-building-instructions.test.ts +++ b/packages/addon-mcp/src/tools/get-ui-building-instructions.test.ts @@ -196,6 +196,7 @@ describe('getUIBuildingInstructionsTool', () => { expect(collectTelemetrySpy).toHaveBeenCalledWith({ event: 'tool:getUIBuildingInstructions', server, + toolset: 'dev', }); }); diff --git a/packages/addon-mcp/src/tools/get-ui-building-instructions.ts b/packages/addon-mcp/src/tools/get-ui-building-instructions.ts index 4f1dee10..03ca386f 100644 --- a/packages/addon-mcp/src/tools/get-ui-building-instructions.ts +++ b/packages/addon-mcp/src/tools/get-ui-building-instructions.ts @@ -32,6 +32,7 @@ export async function addGetUIBuildingInstructionsTool( await collectTelemetry({ event: 'tool:getUIBuildingInstructions', server, + toolset: 'dev', }); }