@yasserkhanorg/e2e-agents 1.8.4 → 1.9.5

package/dist/esm/prompts/generation.js

@@ -1,6 +1,8 @@
 // Copyright (c) 2015-present Mattermost, Inc. All Rights Reserved.
 // See LICENSE.txt for license information.
 import { formatApiSurfaceForPrompt } from '../knowledge/api_surface.js';
+import { sanitizeForPrompt } from '../crew/sanitize.js';
+import { isMattermostProfile } from './generation_profile.js';
 function resolveRelevantPageObjects(apiSurface, decision) {
     const relevant = [];
     const familyHints = [
@@ -21,6 +23,8 @@ function resolveRelevantPageObjects(apiSurface, decision) {
     return [...new Set(relevant)].slice(0, 10);
 }
 export function buildGenerationPrompt(ctx) {
+    const profile = ctx.profile;
+    const isMM = profile ? isMattermostProfile(profile) : true;
     const relevantClasses = resolveRelevantPageObjects(ctx.apiSurface, ctx.decision);
     const apiBlock = relevantClasses.length > 0
         ? formatApiSurfaceForPrompt(ctx.apiSurface, relevantClasses)
@@ -35,73 +39,187 @@ export function buildGenerationPrompt(ctx) {
         ? `Create a NEW spec file at: ${ctx.specPath}`
         : `ADD scenarios to the EXISTING spec at: ${ctx.specPath}`;
     const routeFamilyTag = ctx.decision.routeFamily;
+    // Build prompt based on profile
+    const projectName = profile?.projectName || 'Mattermost';
+    const testFramework = profile?.testFramework || 'Playwright';
+    const importStatement = profile?.importStatement || '@mattermost/playwright-lib';
+    // API test mode prompt
+    if (profile?.testMode === 'api') {
+        return buildApiTestPrompt(ctx, profile, scenariosBlock, routeFamilyTag);
+    }
+    // Build rules from profile conventions or use Mattermost defaults
+    const rules = isMM
+        ? [
+            `1. Import ONLY from "${importStatement}" — no other test framework imports.`,
+            '2. Every test must call `await pw.initSetup()` first.',
+            '3. Use `await pw.testBrowser.login(user)` to log in — never hardcode credentials.',
+            '4. Use ONLY page object methods listed above. Do NOT invent methods that are not listed.',
+            '5. If a method is not available, use `page.getByRole()` or `page.getByTestId()`.',
+            `6. Tag every test: {tag: '@${routeFamilyTag}'}`,
+            '7. Write one test per scenario with a descriptive name of what the user does and what is verified.',
+            `8. Use \`expect\` from "${importStatement}" — do NOT import from "@playwright/test".`,
+            '9. Include the copyright header for new files.',
+            '10. NEVER fabricate test IDs (MM-TXXXX). Use descriptive names only.',
+        ]
+        : [
+            ...(profile?.conventions || []).map((c, i) => `${i + 1}. ${c}`),
+            `${(profile?.conventions?.length || 0) + 1}. Use ONLY page object methods listed above. Do NOT invent methods that are not listed.`,
+            `${(profile?.conventions?.length || 0) + 2}. If a method is not available, use \`page.getByRole()\` or \`page.getByTestId()\`.`,
+            `${(profile?.conventions?.length || 0) + 3}. Tag every test: {tag: '@${routeFamilyTag}'}`,
+        ];
+    // Build example block
+    const exampleBlock = isMM
+        ? [
+            'EXAMPLE SPEC STRUCTURE:',
+            '```typescript',
+            '// Copyright (c) 2015-present Mattermost, Inc. All Rights Reserved.',
+            '// See LICENSE.txt for license information.',
+            '',
+            `import {expect, test} from '${importStatement}';`,
+            '',
+            'test(',
+            "    'descriptive name of what is tested',",
+            `    {tag: '@${routeFamilyTag}'},`,
+            '    async ({pw}) => {',
+            '        const {user} = await pw.initSetup();',
+            '        const {channelsPage} = await pw.testBrowser.login(user);',
+            '        await channelsPage.goto();',
+            '        await channelsPage.toBeVisible();',
+            '        // test steps...',
+            '    },',
+            ');',
+            '```',
+        ]
+        : [
+            'EXAMPLE SPEC STRUCTURE:',
+            '```typescript',
+            ...(profile?.copyrightHeader ? [profile.copyrightHeader, ''] : []),
+            `import {test, expect} from '${importStatement}';`,
+            '',
+            'test(',
+            "    'descriptive name of what is tested',",
+            `    {tag: '@${routeFamilyTag}'},`,
+            '    async ({page}) => {',
+            '        // test steps...',
+            '    },',
+            ');',
+            '```',
+        ];
     return [
-        'You are generating Mattermost Playwright E2E test code.',
+        `You are generating ${projectName} ${testFramework} E2E test code.`,
         '',
         `TASK: ${modeInstruction}`,
         '',
-        `FLOW: ${ctx.decision.flowName}`,
+        `FLOW: ${sanitizeForPrompt(ctx.decision.flowName)}`,
         `Route Family: ${ctx.decision.routeFamily}${ctx.decision.featureId ? ` / ${ctx.decision.featureId}` : ''}`,
         `Route: ${ctx.decision.specificRoute || '(not specified)'}`,
         `Priority: ${ctx.decision.priority}`,
-        `Evidence: ${ctx.decision.evidence}`,
+        `Evidence: ${sanitizeForPrompt(ctx.decision.evidence)}`,
         '',
         'SCENARIOS TO IMPLEMENT:',
         scenariosBlock || '  (implement core user actions for this flow)',
         '',
         'USER ACTIONS:',
-        ctx.decision.userActions.map((a) => `  - ${a}`).join('\n') || '  (none specified)',
+        ctx.decision.userActions.map((a) => `  - ${sanitizeForPrompt(a)}`).join('\n') || '  (none specified)',
         '',
         'AVAILABLE PAGE OBJECTS AND METHODS:',
         apiBlock,
         existingBlock,
         '',
         'MANDATORY RULES:',
-        '1. Import ONLY from "@mattermost/playwright-lib" — no other test framework imports.',
-        '2. Every test must call `await pw.initSetup()` first.',
-        '3. Use `await pw.testBrowser.login(user)` to log in — never hardcode credentials.',
-        '4. Use ONLY page object methods listed above. Do NOT invent methods that are not listed.',
-        '5. If a method is not available, use `page.getByRole()` or `page.getByTestId()`.',
-        `6. Tag every test: {tag: '@${routeFamilyTag}'}`,
-        '7. Write one test per scenario with a descriptive name of what the user does and what is verified.',
-        '8. Use `expect` from "@mattermost/playwright-lib" — do NOT import from "@playwright/test".',
-        '9. Include the copyright header for new files.',
-        '10. NEVER fabricate test IDs (MM-TXXXX). Use descriptive names only.',
+        ...rules,
         '',
-        'EXAMPLE SPEC STRUCTURE:',
-        '```typescript',
-        '// Copyright (c) 2015-present Mattermost, Inc. All Rights Reserved.',
-        '// See LICENSE.txt for license information.',
+        ...exampleBlock,
         '',
-        "import {expect, test} from '@mattermost/playwright-lib';",
+        'Return ONLY the TypeScript code. No explanations, no markdown fences.',
+    ].join('\n');
+}
+function buildApiTestPrompt(ctx, profile, scenariosBlock, routeFamilyTag) {
+    const modeInstruction = ctx.mode === 'create_spec'
+        ? `Create a NEW test file at: ${ctx.specPath}`
+        : `ADD test cases to the EXISTING file at: ${ctx.specPath}`;
+    const existingBlock = ctx.existingSpecContent
+        ? `\nEXISTING FILE (extend this):\n\`\`\`typescript\n${ctx.existingSpecContent}\n\`\`\``
+        : '';
+    return [
+        `You are generating ${profile.projectName} API test code using ${profile.testFramework}.`,
         '',
-        'test(',
-        "    'descriptive name of what is tested',",
-        `    {tag: '@${routeFamilyTag}'},`,
-        '    async ({pw}) => {',
-        '        const {user} = await pw.initSetup();',
-        '        const {channelsPage} = await pw.testBrowser.login(user);',
-        '        await channelsPage.goto();',
-        '        await channelsPage.toBeVisible();',
-        '        // test steps...',
-        '    },',
-        ');',
-        '```',
+        `TASK: ${modeInstruction}`,
         '',
-        'Return ONLY the TypeScript code. No explanations, no markdown fences.',
+        `FLOW: ${sanitizeForPrompt(ctx.decision.flowName)}`,
+        `Route Family: ${ctx.decision.routeFamily}${ctx.decision.featureId ? ` / ${ctx.decision.featureId}` : ''}`,
+        `Endpoint: ${ctx.decision.specificRoute || '(not specified)'}`,
+        `Priority: ${ctx.decision.priority}`,
+        `Evidence: ${sanitizeForPrompt(ctx.decision.evidence)}`,
+        '',
+        'SCENARIOS TO IMPLEMENT:',
+        scenariosBlock || '  (implement core API endpoint tests)',
+        '',
+        'USER ACTIONS:',
+        ctx.decision.userActions.map((a) => `  - ${sanitizeForPrompt(a)}`).join('\n') || '  (none specified)',
+        existingBlock,
+        '',
+        'MANDATORY RULES:',
+        ...profile.conventions.map((c, i) => `${i + 1}. ${c}`),
+        `${profile.conventions.length + 1}. Tag every test: {tag: '@${routeFamilyTag}'}`,
+        '',
+        'EXAMPLE TEST STRUCTURE:',
+        ...(profile.testFramework.toLowerCase().includes('pytest')
+            ? [
+                '```python',
+                ...(profile.copyrightHeader ? [profile.copyrightHeader, ''] : []),
+                'import pytest',
+                'import requests',
+                '',
+                `BASE_URL = 'http://localhost:3000'`,
+                '',
+                '',
+                `class Test${ctx.decision.routeFamily.replace(/[^a-zA-Z0-9]/g, '')}:`,
+                "    def test_should_return_200_for_valid_request(self):",
+                `        res = requests.get(f'{BASE_URL}${ctx.decision.specificRoute || '/api/endpoint'}')`,
+                '        assert res.status_code == 200',
+                '```',
+            ]
+            : [
+                '```typescript',
+                ...(profile.copyrightHeader ? [profile.copyrightHeader, ''] : []),
+                `import {describe, it, expect} from '${profile.importStatement}';`,
+                "import supertest from 'supertest';",
+                '',
+                "const request = supertest('http://localhost:3000');",
+                '',
+                `describe('${ctx.decision.routeFamily}', () => {`,
+                "    it('should return 200 for valid request', async () => {",
+                `        const res = await request.get('${ctx.decision.specificRoute || '/api/endpoint'}');`,
+                '        expect(res.status).toBe(200);',
+                '    });',
+                '});',
+                '```',
+            ]),
+        '',
+        ...(profile.testFramework.toLowerCase().includes('pytest')
+            ? ['Return ONLY the Python code. No explanations, no markdown fences.']
+            : ['Return ONLY the TypeScript code. No explanations, no markdown fences.']),
     ].join('\n');
 }
-export function parseGenerationResponse(text, expectedPath, mode, flowId) {
+export function parseGenerationResponse(text, expectedPath, mode, flowId, profile) {
     let code = text.trim();
     const fenced = code.match(/^```(?:typescript|ts)?\s*([\s\S]*?)```\s*$/i);
     if (fenced) {
         code = fenced[1].trim();
     }
-    if (!code.includes('test(')) {
+    if (!code.includes('test(') && !code.includes('it(') && !code.includes('describe(')) {
         return null;
     }
-    if (!code.includes('@mattermost/playwright-lib')) {
-        code = `import {expect, test} from '@mattermost/playwright-lib';\n\n${code}`;
+    const importStatement = profile?.importStatement || '@mattermost/playwright-lib';
+    // Auto-add import if missing
+    if (!code.includes(importStatement)) {
+        if (profile?.testMode === 'api') {
+            code = `import {describe, it, expect} from '${importStatement}';\n\n${code}`;
+        }
+        else {
+            code = `import {expect, test} from '${importStatement}';\n\n${code}`;
+        }
     }
     return { specPath: expectedPath, code, mode, flowId };
 }
@@ -121,6 +239,10 @@ const BUILT_IN_METHODS = new Set([
     'initSetup', 'login', 'waitUntil', 'skipIfNoLicense', 'ensureLicense',
     'random', 'duration', 'isOutsideRemoteUserHour', 'setTimeout',
     'skip', 'fixme', 'slow', 'fail',
+    // API testing built-ins
+    'get', 'post', 'put', 'patch', 'delete', 'send', 'set', 'query',
+    'toBe', 'toEqual', 'toBeDefined', 'toContain', 'toHaveProperty',
+    'toMatchObject', 'toHaveLength', 'toBeTruthy', 'toBeFalsy',
 ]);
 /**
  * Returns method names that appear in generated code but do not exist in the API surface.

package/dist/esm/prompts/generation_profile.js

@@ -0,0 +1,147 @@
+// Copyright (c) 2015-present Mattermost, Inc. All Rights Reserved.
+// See LICENSE.txt for license information.
+import { UI_FRAMEWORKS, API_FRAMEWORKS } from '../adapters/framework_adapter.js';
+const MATTERMOST_PROFILE = {
+    projectName: 'Mattermost',
+    testFramework: 'Playwright',
+    importStatement: '@mattermost/playwright-lib',
+    conventions: [
+        'Import ONLY from "@mattermost/playwright-lib" — no other test framework imports.',
+        'Every test must call `await pw.initSetup()` first.',
+        'Use `await pw.testBrowser.login(user)` to log in — never hardcode credentials.',
+        'Use `expect` from "@mattermost/playwright-lib" — do NOT import from "@playwright/test".',
+        'Include the copyright header for new files.',
+    ],
+    copyrightHeader: [
+        '// Copyright (c) 2015-present Mattermost, Inc. All Rights Reserved.',
+        '// See LICENSE.txt for license information.',
+    ].join('\n'),
+    testMode: 'ui',
+};
+const DEFAULT_PLAYWRIGHT_PROFILE = {
+    projectName: 'Project',
+    testFramework: 'Playwright',
+    importStatement: '@playwright/test',
+    conventions: [
+        'Import from "@playwright/test" for test and expect.',
+        'Use page fixtures provided by Playwright.',
+        'Prefer ARIA roles and data-testid attributes for selectors.',
+        'Write one test per scenario with a descriptive name.',
+    ],
+    testMode: 'ui',
+};
+const DEFAULT_API_PROFILE = {
+    projectName: 'Project',
+    testFramework: 'vitest + supertest',
+    importStatement: 'vitest',
+    conventions: [
+        'Import from "vitest" for test and expect.',
+        'Use supertest for HTTP request assertions.',
+        'Validate response status codes, headers, and body structure.',
+        'Test both success and error paths for each endpoint.',
+    ],
+    testMode: 'api',
+};
+/**
+ * Resolves the generation profile from config and optional KG metadata.
+ * - If profile='mattermost' or Mattermost is detected, returns Mattermost profile.
+ * - If KG is present, derives project-specific profile from it.
+ * - Otherwise, returns generic Playwright profile.
+ */
+export function resolveGenerationProfile(config, kg) {
+    // Explicit Mattermost profile
+    if (config?.profile === 'mattermost') {
+        return { ...MATTERMOST_PROFILE };
+    }
+    // KG-based profile derivation
+    if (kg) {
+        const frameworks = kg.project.frameworks.map((f) => f.toLowerCase());
+        const isMattermost = kg.project.name.toLowerCase().includes('mattermost') ||
+            frameworks.includes('@mattermost/playwright-lib');
+        if (isMattermost) {
+            return { ...MATTERMOST_PROFILE };
+        }
+        const testMode = config?.testMode || deriveTestMode(frameworks);
+        const testFramework = deriveTestFramework(frameworks, testMode);
+        const importStatement = deriveImportStatement(frameworks, testMode);
+        return {
+            projectName: kg.project.name || 'Project',
+            testFramework,
+            importStatement,
+            conventions: buildConventions(testFramework, importStatement, testMode),
+            testMode,
+        };
+    }
+    // Default profiles based on test mode
+    if (config?.testMode === 'api') {
+        return { ...DEFAULT_API_PROFILE };
+    }
+    return { ...DEFAULT_PLAYWRIGHT_PROFILE };
+}
+/**
+ * Checks if a profile is the Mattermost profile (for backward compatibility checks).
+ */
+export function isMattermostProfile(profile) {
+    return profile.importStatement === '@mattermost/playwright-lib';
+}
+// ---------------------------------------------------------------------------
+// Internal helpers
+// ---------------------------------------------------------------------------
+function deriveTestMode(frameworks) {
+    const uiSet = new Set(UI_FRAMEWORKS);
+    const apiSet = new Set(API_FRAMEWORKS);
+    const hasUiFramework = frameworks.some((f) => uiSet.has(f));
+    const hasApiFramework = frameworks.some((f) => apiSet.has(f));
+    if (hasUiFramework && hasApiFramework)
+        return 'both';
+    if (hasApiFramework && !hasUiFramework)
+        return 'api';
+    return 'ui';
+}
+function deriveTestFramework(frameworks, testMode) {
+    if (testMode === 'api') {
+        if (frameworks.includes('pytest'))
+            return 'pytest';
+        if (frameworks.includes('jest'))
+            return 'jest + supertest';
+        return 'vitest + supertest';
+    }
+    if (frameworks.includes('cypress'))
+        return 'Cypress';
+    if (frameworks.includes('selenium'))
+        return 'Selenium';
+    return 'Playwright';
+}
+function deriveImportStatement(frameworks, testMode) {
+    if (testMode === 'api') {
+        if (frameworks.includes('pytest'))
+            return 'pytest';
+        if (frameworks.includes('jest'))
+            return 'jest';
+        return 'vitest';
+    }
+    if (frameworks.includes('cypress'))
+        return 'cypress';
+    return '@playwright/test';
+}
+function buildConventions(testFramework, importStatement, testMode) {
+    const conventions = [];
+    if (testMode === 'api' || testMode === 'both') {
+        conventions.push(`Import from "${importStatement}" for test and expect.`);
+        conventions.push('Validate response status codes, headers, and body structure.');
+        conventions.push('Test both success and error paths for each endpoint.');
+    }
+    if (testMode === 'ui' || testMode === 'both') {
+        if (testFramework.includes('Playwright')) {
+            conventions.push('Import from "@playwright/test" for test and expect.');
+            conventions.push('Use page fixtures provided by Playwright.');
+        }
+        else if (testFramework.includes('Cypress')) {
+            conventions.push('Use cy.* commands for browser interaction.');
+        }
+        conventions.push('Prefer ARIA roles and data-testid attributes for selectors.');
+    }
+    conventions.push('Write one test per scenario with a descriptive name of what the user does and what is verified.');
+    conventions.push('NEVER fabricate test IDs. Use descriptive names only.');
+    return conventions;
+}

package/dist/esm/prompts/heal.js

@@ -1,5 +1,7 @@
 // Copyright (c) 2015-present Mattermost, Inc. All Rights Reserved.
 // See LICENSE.txt for license information.
+import { sanitizeForPrompt } from '../crew/sanitize.js';
+import { isMattermostProfile } from './generation_profile.js';
 /**
  * Builds a route-family-aware heal prompt for the playwright-test-healer agent.
  * Enriches the base healer constraints with flow context so the agent understands
@@ -10,24 +12,43 @@ export function buildHealPrompt(ctx) {
         ? [
             '',
             'FLOW CONTEXT (use to understand test intent — do not change test objectives):',
-            `  Flow: ${ctx.decision.flowName}`,
+            `  Flow: ${sanitizeForPrompt(ctx.decision.flowName)}`,
             `  Route Family: ${ctx.decision.routeFamily}${ctx.decision.featureId ? ` / ${ctx.decision.featureId}` : ''}`,
             `  Route: ${ctx.decision.specificRoute || '(family-level)'}`,
-            `  User Actions: ${ctx.decision.userActions.join('; ') || 'not specified'}`,
-            `  Evidence: ${ctx.decision.evidence}`,
+            `  User Actions: ${sanitizeForPrompt(ctx.decision.userActions.join('; ')) || 'not specified'}`,
+            `  Evidence: ${sanitizeForPrompt(ctx.decision.evidence)}`,
         ].join('\n')
         : '';
     const statusNote = ctx.status === 'flaky'
         ? 'This test is FLAKY (passes sometimes, fails other times). Look for race conditions, missing waits, or order-dependent state.'
         : 'This test is FAILING consistently. The selector, URL, or API call is likely broken.';
     const failureBlock = ctx.failureDetail
-        ? `\nFailure detail:\n${ctx.failureDetail}`
+        ? `\nFailure detail:\n${sanitizeForPrompt(ctx.failureDetail)}`
         : '';
     const consoleBlock = ctx.consoleErrors && ctx.consoleErrors.length > 0
-        ? `\nRecent console errors from test run:\n${ctx.consoleErrors.slice(-3).map((e) => `  - ${e}`).join('\n')}`
+        ? `\nRecent console errors from test run:\n${ctx.consoleErrors.slice(-3).map((e) => `  - ${sanitizeForPrompt(e)}`).join('\n')}`
         : '';
+    const importLib = ctx.profile?.importStatement || '@mattermost/playwright-lib';
+    const isMM = ctx.profile ? isMattermostProfile(ctx.profile) : true;
+    const projectLabel = ctx.profile?.projectName || 'Mattermost';
+    const frameworkLabel = ctx.profile?.testFramework || 'Playwright';
+    const constraints = isMM
+        ? [
+            `- Import ONLY from "${importLib}". Do not use "@playwright/test" directly.`,
+            '- Do not use test.describe or test.only.',
+            '- Keep a single tag string matching the route family (e.g. "@channels", "@scheduled_posts").',
+            `- Use only existing ${projectLabel} ${frameworkLabel} fixture and page-object APIs.`,
+            '- Do NOT invent new pw.* clients or page object methods that do not exist.',
+            '- Avoid brittle class selectors (.backstage-navbar, .admin-console__wrapper, .left-panel, .panel-card).',
+        ]
+        : [
+            `- Import from "${importLib}".`,
+            '- Keep a single tag string matching the route family.',
+            `- Use only existing ${projectLabel} ${frameworkLabel} page-object APIs.`,
+            '- Do NOT invent page object methods that do not exist.',
+        ];
     return [
-        'Heal this specific Playwright test file and keep edits minimal.',
+        `Heal this specific ${frameworkLabel} test file and keep edits minimal.`,
         '',
         `Target test file: ${ctx.specPath}`,
         `Status: ${ctx.status.toUpperCase()} — ${statusNote}`,
@@ -36,12 +57,7 @@ export function buildHealPrompt(ctx) {
         flowBlock,
         '',
         'Healing constraints (must follow):',
-        '- Import ONLY from "@mattermost/playwright-lib". Do not use "@playwright/test" directly.',
-        '- Do not use test.describe or test.only.',
-        '- Keep a single tag string matching the route family (e.g. "@channels", "@scheduled_posts").',
-        '- Use only existing Mattermost Playwright fixture and page-object APIs.',
-        '- Do NOT invent new pw.* clients or page object methods that do not exist.',
-        '- Avoid brittle class selectors (.backstage-navbar, .admin-console__wrapper, .left-panel, .panel-card).',
+        ...constraints,
         '- Prefer stable assertions using URL patterns, data-testid attributes, ARIA roles, and page-object methods.',
         '- For flaky tests: add explicit waits (waitFor, expect().toBeVisible()) before interactions.',
         '- Keep the test intent and scenario unchanged — only fix what is broken.',
@@ -54,9 +70,11 @@ export function buildHealPrompt(ctx) {
  * Builds a minimal quality-fix prompt for spec files that fail content validation
  * (e.g. contain test.describe, test.only, wrong imports).
  */
-export function buildQualityFixPrompt(specPath, qualityIssues) {
+export function buildQualityFixPrompt(specPath, qualityIssues, profile) {
+    const importLib = profile?.importStatement || '@mattermost/playwright-lib';
+    const frameworkLabel = profile?.testFramework || 'Playwright';
     return [
-        'Fix quality issues in this Playwright spec file. Make minimal edits only.',
+        `Fix quality issues in this ${frameworkLabel} spec file. Make minimal edits only.`,
         '',
         `Target file: ${specPath}`,
         '',
@@ -64,7 +82,7 @@ export function buildQualityFixPrompt(specPath, qualityIssues) {
         ...qualityIssues.map((issue) => `  - ${issue}`),
         '',
         'Rules:',
-        '- Import only from "@mattermost/playwright-lib".',
+        `- Import only from "${importLib}".`,
         '- Remove test.describe wrappers (flatten to top-level test() calls).',
         '- Remove test.only calls.',
         '- Ensure each test has exactly one tag string.',

package/dist/esm/prompts/impact.js

@@ -1,5 +1,6 @@
 // Copyright (c) 2015-present Mattermost, Inc. All Rights Reserved.
 // See LICENSE.txt for license information.
+import { extractJsonFromResponse } from './json_extract.js';
 import { formatSpecsForPrompt } from '../knowledge/spec_index.js';
 import { formatApiSurfaceForPrompt } from '../knowledge/api_surface.js';
 export function buildImpactPrompt(ctx) {
@@ -23,7 +24,7 @@ export function buildImpactPrompt(ctx) {
     })
         .join('\n\n');
     return [
-        'You are analyzing code changes in Mattermost to identify impacted user-facing flows.',
+        `You are analyzing code changes in ${ctx.projectName || 'Mattermost'} to identify impacted user-facing flows.`,
         '',
         `ROUTE FAMILY: ${ctx.family.id}`,
         `ROUTES: ${familyRoutes}`,
@@ -58,25 +59,5 @@ export function buildImpactPrompt(ctx) {
     ].filter(Boolean).join('\n');
 }
 export function parseImpactResponse(text) {
-    // Try to extract JSON from the response
-    const fenced = text.match(/```(?:json)?\s*([\s\S]*?)```/i);
-    const candidates = fenced ? [fenced[1], text] : [text];
-    for (const candidate of candidates) {
-        const start = candidate.indexOf('{');
-        const end = candidate.lastIndexOf('}');
-        if (start < 0 || end <= start) {
-            continue;
-        }
-        const raw = candidate.slice(start, end + 1);
-        try {
-            const parsed = JSON.parse(raw);
-            if (parsed && Array.isArray(parsed.flows)) {
-                return parsed;
-            }
-        }
-        catch {
-            continue;
-        }
-    }
-    return null;
+    return extractJsonFromResponse(text, (obj) => obj != null && typeof obj === 'object' && Array.isArray(obj.flows));
 }

package/dist/esm/prompts/json_extract.js

@@ -0,0 +1,36 @@
+// Copyright (c) 2015-present Mattermost, Inc. All Rights Reserved.
+// See LICENSE.txt for license information.
+/**
+ * Shared JSON extraction from LLM text responses.
+ * Handles fenced code blocks, bare JSON, and partial text.
+ */
+/**
+ * Extract and parse JSON from LLM response text.
+ * Tries fenced code blocks first, then raw text.
+ * Returns null if no valid JSON found.
+ *
+ * @param text - Raw LLM response text
+ * @param validate - Predicate to check if parsed object has the expected shape
+ */
+export function extractJsonFromResponse(text, validate) {
+    const fenced = text.match(/```(?:json)?\s*([\s\S]*?)```/i);
+    const candidates = fenced ? [fenced[1], text] : [text];
+    for (const candidate of candidates) {
+        const start = candidate.indexOf('{');
+        const end = candidate.lastIndexOf('}');
+        if (start < 0 || end <= start) {
+            continue;
+        }
+        const raw = candidate.slice(start, end + 1);
+        try {
+            const parsed = JSON.parse(raw);
+            if (validate(parsed)) {
+                return parsed;
+            }
+        }
+        catch {
+            continue;
+        }
+    }
+    return null;
+}

package/dist/esm/prompts/strategist.js

@@ -1,6 +1,7 @@
 // Copyright (c) 2015-present Mattermost, Inc. All Rights Reserved.
 // See LICENSE.txt for license information.
 import { sanitizeForPrompt } from '../crew/sanitize.js';
+import { extractJsonFromResponse } from './json_extract.js';
 export function buildStrategistPrompt(ctx) {
     const flowsBlock = ctx.impactedFlows
         .map((f) => {
@@ -56,24 +57,5 @@ export function buildStrategistPrompt(ctx) {
     ].join('\n');
 }
 export function parseStrategistResponse(text) {
-    const fenced = text.match(/```(?:json)?\s*([\s\S]*?)```/i);
-    const candidates = fenced ? [fenced[1], text] : [text];
-    for (const candidate of candidates) {
-        const start = candidate.indexOf('{');
-        const end = candidate.lastIndexOf('}');
-        if (start < 0 || end <= start) {
-            continue;
-        }
-        const raw = candidate.slice(start, end + 1);
-        try {
-            const parsed = JSON.parse(raw);
-            if (parsed && Array.isArray(parsed.strategy)) {
-                return parsed;
-            }
-        }
-        catch {
-            continue;
-        }
-    }
-    return null;
+    return extractJsonFromResponse(text, (obj) => obj != null && typeof obj === 'object' && Array.isArray(obj.strategy));
 }

package/dist/esm/prompts/test-designer.js

@@ -1,5 +1,6 @@
 // Copyright (c) 2015-present Mattermost, Inc. All Rights Reserved.
 // See LICENSE.txt for license information.
+import { extractJsonFromResponse } from './json_extract.js';
 import { formatApiSurfaceForPrompt } from '../knowledge/api_surface.js';
 import { sanitizeForPrompt } from '../crew/sanitize.js';
 export function buildTestDesignerPrompt(ctx) {
@@ -24,7 +25,7 @@ export function buildTestDesignerPrompt(ctx) {
         : 'None detected.';
     const categories = ctx.strategy.testCategories.join(', ');
     return [
-        'You are a senior QA engineer designing comprehensive test cases for a Mattermost user flow.',
+        `You are a senior QA engineer designing comprehensive test cases for a ${ctx.profile?.projectName || 'Mattermost'} user flow.`,
         '',
         `FLOW: ${ctx.flow.flowName}`,
         `Flow ID: ${ctx.flow.flowId}`,
@@ -84,24 +85,8 @@ export function buildTestDesignerPrompt(ctx) {
     ].join('\n');
 }
 export function parseTestDesignerResponse(text) {
-    const fenced = text.match(/```(?:json)?\s*([\s\S]*?)```/i);
-    const candidates = fenced ? [fenced[1], text] : [text];
-    for (const candidate of candidates) {
-        const start = candidate.indexOf('{');
-        const end = candidate.lastIndexOf('}');
-        if (start < 0 || end <= start) {
-            continue;
-        }
-        const raw = candidate.slice(start, end + 1);
-        try {
-            const parsed = JSON.parse(raw);
-            if (parsed?.testDesign?.testCases && Array.isArray(parsed.testDesign.testCases)) {
-                return parsed;
-            }
-        }
-        catch {
-            continue;
-        }
-    }
-    return null;
+    return extractJsonFromResponse(text, (obj) => {
+        const r = obj;
+        return r?.testDesign?.testCases != null && Array.isArray(r.testDesign.testCases);
+    });
 }