Skip to content

Commit f40da8f

Browse files
CopilotJReinhold
andauthored
Add toolset property to telemetry payloads in addon-mcp (#78)
* Initial plan * Add toolset property to all telemetry payloads in addon-mcp Co-authored-by: JReinhold <5678122+JReinhold@users.noreply.github.com> * add changeset --------- Co-authored-by: copilot-swe-agent[bot] <198982749+Copilot@users.noreply.github.com> Co-authored-by: JReinhold <5678122+JReinhold@users.noreply.github.com> Co-authored-by: Jeppe Reinhold <jeppe@reinhold.is> Co-authored-by: Jeppe Reinhold <jeppe@chromatic.com>
1 parent 2c6f0ee commit f40da8f

File tree

6 files changed

+11
-0
lines changed

6 files changed

+11
-0
lines changed

.changeset/clean-bats-say.md

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,5 @@
1+
---
2+
'@storybook/addon-mcp': patch
3+
---
4+
5+
Add toolset to telemetry payload

packages/addon-mcp/src/mcp-handler.ts

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -104,13 +104,15 @@ export const mcpServerHandler = async ({
104104
await collectTelemetry({
105105
event: 'tool:listAllComponents',
106106
server,
107+
toolset: 'docs',
107108
componentCount: Object.keys(manifest.components).length,
108109
});
109110
},
110111
onGetComponentDocumentation: async ({ input, foundComponent }) => {
111112
await collectTelemetry({
112113
event: 'tool:getComponentDocumentation',
113114
server,
115+
toolset: 'docs',
114116
componentId: input.componentId,
115117
found: !!foundComponent,
116118
});

packages/addon-mcp/src/tools/get-story-urls.test.ts

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -279,6 +279,7 @@ describe('getStoryUrlsTool', () => {
279279
expect(collectTelemetrySpy).toHaveBeenCalledWith({
280280
event: 'tool:getStoryUrls',
281281
server,
282+
toolset: 'dev',
282283
inputStoryCount: 1,
283284
outputStoryCount: 1,
284285
});

packages/addon-mcp/src/tools/get-story-urls.ts

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -83,6 +83,7 @@ export async function addGetStoryUrlsTool(
8383
await collectTelemetry({
8484
event: 'tool:getStoryUrls',
8585
server,
86+
toolset: 'dev',
8687
inputStoryCount: input.stories.length,
8788
outputStoryCount: foundStoryCount,
8889
});

packages/addon-mcp/src/tools/get-ui-building-instructions.test.ts

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -196,6 +196,7 @@ describe('getUIBuildingInstructionsTool', () => {
196196
expect(collectTelemetrySpy).toHaveBeenCalledWith({
197197
event: 'tool:getUIBuildingInstructions',
198198
server,
199+
toolset: 'dev',
199200
});
200201
});
201202

packages/addon-mcp/src/tools/get-ui-building-instructions.ts

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -32,6 +32,7 @@ export async function addGetUIBuildingInstructionsTool(
3232
await collectTelemetry({
3333
event: 'tool:getUIBuildingInstructions',
3434
server,
35+
toolset: 'dev',
3536
});
3637
}
3738

0 commit comments

Comments
 (0)