@codeyam/codeyam-cli 0.1.0-staging.bbe4da9 → 0.1.0-staging.d0ad4ae

package/packages/ai/src/lib/generateEntityKeyAttributes.js

@@ -5,14 +5,21 @@ import { awsLog } from "../../../../packages/utils/index.js";
 import { DEFAULT_LARGER_MODEL } from "./aiConfig.js";
 import { gatherAttributesMap } from "./promptGenerators/gatherAttributesMap.js";
 import { fillInDirectSchemaGapsAndUnknowns } from "./dataStructure/helpers/fillInSchemaGapsAndUnknowns.js";
+import { clearAttributesFromMapping } from "./dataStructure/helpers/cleanNonObjectFunctions.js";
 import isFrontend from "./isFrontend.js";
 import { formatConditionalUsagesForPrompt } from "./getConditionalUsagesFromCode.js";
-export default async function generateEntityKeyAttributes({ entity, model, }) {
+export default async function generateEntityKeyAttributes({ entity, mergedDataStructure, model, }) {
     awsLog(`CodeYam: Generating entity key attributes`, {
         filePath: entity.filePath,
         entityName: entity.name,
     });
-    const { attributesMap, associationMap, dataStructurePathMap } = gatherAttributesMap(entity, true);
+    const isolatedDataStructure = entity.metadata?.isolatedDataStructure;
+    const equivalentSignatureVariables = isolatedDataStructure?.equivalentSignatureVariables ?? {};
+    const { attributesMap, associationMap, dataStructurePathMap } = gatherAttributesMap(isolatedDataStructure ?? mergedDataStructure, mergedDataStructure, equivalentSignatureVariables, true);
+    // Clean the attributes map to remove primitive-returning array method paths
+    // like .includes().functionCallReturnValue, .indexOf().functionCallReturnValue, etc.
+    // These paths are not mockable and should not be presented as key attributes.
+    clearAttributesFromMapping(attributesMap);
     if (Object.keys(attributesMap).length === 0) {
         console.log(`CodeYam Error: No valid attributes found for ${entity.filePath} ${entity.name}`);
         return {
@@ -20,14 +27,22 @@ export default async function generateEntityKeyAttributes({ entity, model, }) {
             llmCall: null,
         };
     }
+    // Build merged schema for type inference (has more complete nested paths)
+    // Strip signature[N] prefix to match attributesMap format
+    const mergedSchemaForTypeLookup = Object.fromEntries(Object.entries(mergedDataStructure.signatureSchema ?? {})
+        .filter(([k]) => k.startsWith('signature['))
+        .map(([k, v]) => [k.replace(/^signature\[\d+\]\./, ''), v]));
     // Resolve unknown types using heuristics (e.g., "unknown | undefined" -> "string | undefined")
+    // Pass mergedSchema so checkIfObjectOrFunction can detect nested paths that prove a type is an object
     const resolvedAttributesMap = fillInDirectSchemaGapsAndUnknowns({
         scopeName: entity.name,
         schema: { ...attributesMap },
+        mergedSchema: mergedSchemaForTypeLookup,
     });
     // Use pre-computed conditional usages from metadata, or compute on-the-fly as fallback
     const conditionalUsages = entity.metadata?.isolatedDataStructure?.conditionalUsages ?? {};
-    const staticAnalysisContext = formatConditionalUsagesForPrompt(conditionalUsages);
+    const compoundConditionals = entity.metadata?.isolatedDataStructure?.compoundConditionals ?? [];
+    const staticAnalysisContext = formatConditionalUsagesForPrompt(conditionalUsages, compoundConditionals);
     const prompt = generateEntityKeyAttributesGenerator({
         code: entity.code,
         attributesMap: resolvedAttributesMap,
@@ -84,6 +99,89 @@ export default async function generateEntityKeyAttributes({ entity, model, }) {
         fullPath,
         internalPath,
     ]));
+    // Helper to extract source locations from conditional usages for a given path
+    const getSourceLocationsForPath = (path) => {
+        const usages = conditionalUsages[path] || [];
+        const locations = [];
+        for (const usage of usages.slice(0, 5)) {
+            // Limit to 5 locations
+            if (usage.sourceLocation) {
+                locations.push({
+                    entityName: entity.name,
+                    filePath: entity.filePath,
+                    lineNumber: usage.sourceLocation.lineNumber,
+                    column: usage.sourceLocation.column,
+                    codeSnippet: usage.sourceLocation.codeSnippet,
+                    usageType: 'conditional',
+                    description: `Used in ${usage.location} statement (${usage.conditionType} check)`,
+                });
+            }
+        }
+        return locations;
+    };
+    // Helper to find source locations by matching sourceDataPath
+    // This handles cases where local variables (e.g., displayState.hasError) trace back
+    // to data structure paths (e.g., getScenarioDisplayState().functionCallReturnValue.hasError)
+    const getSourceLocationsBySourceDataPath = (targetPath) => {
+        const locations = [];
+        // Search all conditional usages for ones whose sourceDataPath matches
+        for (const [, usages] of Object.entries(conditionalUsages)) {
+            for (const usage of usages) {
+                const sourceDataPath = usage
+                    .sourceDataPath;
+                if (!sourceDataPath || !usage.sourceLocation)
+                    continue;
+                // Check if sourceDataPath ends with the target path
+                // e.g., "LibraryFunctionPreview.signature[0].isOutdated" ends with "isOutdated"
+                // or "getScenarioDisplayState.getScenarioDisplayState(...).functionCallReturnValue.hasError" ends with "hasError"
+                const normalizedSourcePath = sourceDataPath
+                    .replace(/\.functionCallReturnValue/g, '')
+                    .replace(/\([^)]*\)/g, '()');
+                if (normalizedSourcePath.endsWith('.' + targetPath) ||
+                    normalizedSourcePath.endsWith('.' + targetPath.replace(/\./g, '.'))) {
+                    locations.push({
+                        entityName: entity.name,
+                        filePath: entity.filePath,
+                        lineNumber: usage.sourceLocation.lineNumber,
+                        column: usage.sourceLocation.column,
+                        codeSnippet: usage.sourceLocation.codeSnippet,
+                        usageType: 'conditional',
+                        description: `Used in ${usage.location} statement (${usage.conditionType} check)`,
+                    });
+                }
+            }
+        }
+        return locations.slice(0, 5); // Limit to 5
+    };
+    // Helper to find source locations from LLM-provided code snippets
+    const getSourceLocationsFromSnippets = (codeSnippets) => {
+        if (!codeSnippets || !Array.isArray(codeSnippets))
+            return [];
+        const locations = [];
+        const codeLines = entity.code.split('\n');
+        for (const snippet of codeSnippets.slice(0, 3)) {
+            // Limit to 3 snippets
+            if (!snippet || typeof snippet !== 'string')
+                continue;
+            // Normalize whitespace for matching
+            const normalizedSnippet = snippet.trim().replace(/\s+/g, ' ');
+            for (let i = 0; i < codeLines.length; i++) {
+                const normalizedLine = codeLines[i].replace(/\s+/g, ' ');
+                if (normalizedLine.includes(normalizedSnippet)) {
+                    locations.push({
+                        entityName: entity.name,
+                        filePath: entity.filePath,
+                        lineNumber: i + 1, // 1-based line numbers
+                        codeSnippet: codeLines[i].trim(),
+                        usageType: 'conditional',
+                        description: `Key usage identified by analysis`,
+                    });
+                    break; // Only find first occurrence of each snippet
+                }
+            }
+        }
+        return locations;
+    };
     const fullKeyAttributes = keyAttributes
         .filter(Boolean)
         .filter((keyAttribute, index, self) => keyAttribute?.fullPath &&
@@ -92,6 +190,37 @@ export default async function generateEntityKeyAttributes({ entity, model, }) {
         .filter((keyAttribute) => fullPathToInternalPath[keyAttribute.fullPath])
         .map((keyAttribute) => {
         const internalPath = fullPathToInternalPath[keyAttribute.fullPath];
+        // Get source locations from conditional usages
+        // Try multiple path formats since conditionalUsages might use different formats:
+        // - internalPath: the short path like "status"
+        // - fullPath: the LLM's path like "signature[0].status"
+        // - dataStructurePath: the external path like "useStatus().status"
+        // - Also try stripping signature[N] prefix since conditional usages use AST paths
+        let sourceLocations = getSourceLocationsForPath(internalPath);
+        if (sourceLocations.length === 0) {
+            sourceLocations = getSourceLocationsForPath(keyAttribute.fullPath);
+        }
+        if (sourceLocations.length === 0) {
+            const dataStructurePath = dataStructurePathMap[internalPath] || keyAttribute.fullPath;
+            sourceLocations = getSourceLocationsForPath(dataStructurePath);
+        }
+        // Also try the path without signature[N] prefix since AST uses local variable names
+        if (sourceLocations.length === 0) {
+            const pathWithoutSignature = internalPath.replace(/^signature\[\d+\]\./, '');
+            if (pathWithoutSignature !== internalPath) {
+                sourceLocations = getSourceLocationsForPath(pathWithoutSignature);
+            }
+        }
+        // Finally, search by sourceDataPath which traces local variables back to data sources
+        // This handles cases like displayState.hasError → getScenarioDisplayState().hasError
+        if (sourceLocations.length === 0) {
+            sourceLocations = getSourceLocationsBySourceDataPath(internalPath);
+        }
+        // Last resort: use LLM-provided code snippets to find source locations
+        // This ensures every key attribute the LLM identifies has a source location
+        if (sourceLocations.length === 0 && keyAttribute.codeSnippets) {
+            sourceLocations = getSourceLocationsFromSnippets(keyAttribute.codeSnippets);
+        }
         return {
             internalPath,
             externalPath: associationMap[internalPath],
@@ -99,6 +228,11 @@ export default async function generateEntityKeyAttributes({ entity, model, }) {
             description: keyAttribute.description,
             validValueOptions: keyAttribute.validValueOptions,
             errorValueOptions: keyAttribute.errorValueOptions,
+            // Preserve new fields from LLM response
+            valueType: keyAttribute.valueType,
+            dependencies: keyAttribute.dependencies,
+            // Add source locations from static analysis or LLM snippets
+            sourceLocations: sourceLocations.length > 0 ? sourceLocations : undefined,
         };
     });
     return {
@@ -126,15 +260,42 @@ NOT key attributes:
 
 We've provided a list of variables that are used in conditional statements in the code. These are very likely key attributes.
 
-Each conditional lists it's source and how it is used. This should help you identify which attributes to include and what values to suggest.
+Each conditional lists its source and how it is used. This should help you identify which attributes to include and what values to suggest.
+
+**IMPORTANT**: Only include attributes with explicit evidence of impact in THIS code:
+- Used directly in a conditional statement (if/ternary/switch)
+- Displayed or rendered directly (e.g., text content, list length)
+- Used in a computation that affects rendering
+
+Do NOT include attributes that are only passed as arguments to other functions without direct usage here. We analyze those functions separately and merge their key attributes automatically.
 
 ## Path Format
 
 Only return attributes from the "Attributes and Value Types" list — these are the only values we can control. Use the exact \`fullPath\` provided. Use \`functionCallReturnValue\` for function return values. Use \`[]\` for array items (e.g., \`items[]\`).
 
+## Dependencies and Compound Conditionals
+
+When attributes are used in && chains (compound conditionals), they form dependencies on each other:
+- Each attribute in the chain only has effect when ALL other conditions in the chain are satisfied
+- Include these as dependencies with the required value for the code path to execute
+
+Example: For \`a && b && !c\`:
+- \`a\` has dependencies: \`b\` must be truthy, \`c\` must be falsy
+- \`b\` has dependencies: \`a\` must be truthy, \`c\` must be falsy
+- \`c\` has dependencies: \`a\` must be truthy, \`b\` must be truthy (for the falsy check to matter)
+
+The static analysis section may show "Compound Conditionals" that list which conditions must be true together. Use this to populate the \`dependencies\` array with \`{ "path": "...", "requiredValue": "..." }\` entries.
+
+## Value Type
+
+Should reiterate the type of the attribute (e.g., "string", "object", "array", "boolean", etc.) based on the "Attributes and Value Types" list.
+
+All validValueOptions must be of this type.
+
 ## Valid Value Options
 
-- For \`validValueOptions\`, include boundary/extreme values that trigger different behavior 
+- For \`validValueOptions\`, include boundary/extreme values that trigger different behavior
+- Each of the \`validValueOptions\` must be the exact type that we provide in "Attributes and Value Types".
 - Be very specific with values. Do not make them ambiguous in any way
 — Do NOT include values that would cause unexpected/unhandled errors.
 - Ensure all values or value ranges that will trigger different rendering behavior are included
@@ -149,11 +310,18 @@ List all key attributes in order from most impactful to least impactful regardin
     {
       "fullPath": "signature[0].user",
       "description": "Controls whether the main form or redirect screen displays",
+      "codeSnippets": ["if (!user) return <Redirect />", "user.name"],
+      "dependencies": [
+        { "path": "signature[0].projectId", "requiredValue": "truthy", "description": "must be a valid project ID for 'user' to have effect" }
+      ],
+      "valueType": "object",
       "validValueOptions": ["an empty object {}", "null/undefined"]
     }
   ]
 }
 \`\`\`
+
+The \`codeSnippets\` field is required - provide 1-3 short code excerpts (under 80 chars each) showing exactly where and how this attribute is used. These snippets must appear verbatim in the code.
 `;
 const BACKEND_SYSTEM_MESSAGE = `You analyze backend functions to identify key data attributes that significantly impact behavior.
 
@@ -174,15 +342,42 @@ NOT key attributes:
 
 We've provided a list of variables that are used in conditional statements in the code. These are very likely key attributes.
 
-Each conditional lists it's source and how it is used. This should help you identify which attributes to include and what values to suggest.
+Each conditional lists its source and how it is used. This should help you identify which attributes to include and what values to suggest.
+
+**IMPORTANT**: Only include attributes with explicit evidence of impact in THIS code:
+- Used directly in a conditional statement (if/ternary/switch)
+- Displayed or rendered directly (e.g., text content, list length)
+- Used in a computation that affects rendering
+
+Do NOT include attributes that are only passed as arguments to other functions without direct usage here. We analyze those functions separately and merge their key attributes automatically.
 
 ## Path Format
 
 Only return attributes from the "Attributes and Value Types" list — these are the only values we can control. Use the exact \`fullPath\` provided. Use \`functionCallReturnValue\` for function return values. Use \`[]\` for array items (e.g., \`items[]\`).
 
+## Dependencies and Compound Conditionals
+
+When attributes are used in && chains (compound conditionals), they form dependencies on each other:
+- Each attribute in the chain only has effect when ALL other conditions in the chain are satisfied
+- Include these as dependencies with the required value for the code path to execute
+
+Example: For \`a && b && !c\`:
+- \`a\` has dependencies: \`b\` must be truthy, \`c\` must be falsy
+- \`b\` has dependencies: \`a\` must be truthy, \`c\` must be falsy
+- \`c\` has dependencies: \`a\` must be truthy, \`b\` must be truthy (for the falsy check to matter)
+
+The static analysis section may show "Compound Conditionals" that list which conditions must be true together. Use this to populate the \`dependencies\` array with \`{ "path": "...", "requiredValue": "..." }\` entries.
+
+## Value Type
+
+Should reiterate the type of the attribute (e.g., "string", "object", "array", "boolean", etc.) based on the "Attributes and Value Types" list.
+
+All validValueOptions must be of this type.
+
 ## Valid Value Options
 
-- For \`validValueOptions\`, include boundary/extreme values that trigger different behavior 
+- For \`validValueOptions\`, include boundary/extreme values that trigger different behavior
+- Each of the \`validValueOptions\` must be the exact type that we provide in "Attributes and Value Types".
 - Be very specific with values. Do not make them ambiguous in any way
 — Do NOT include values that would cause unexpected/unhandled errors.
 - Ensure all values or value ranges that will trigger different rendering behavior are included
@@ -197,10 +392,17 @@ List all key attributes in order from most impactful to least impactful using th
     {
       "fullPath": "signature[0].config.enabled",
       "description": "Controls whether the main processing logic runs",
+      "codeSnippets": ["if (!config.enabled) return", "enabled && processData()"],
+      "dependencies": [
+        { "path": "signature[0].analysisMode", "requiredValue": "full", "description": "must be 'full' for 'enabled' to have effect" }
+      ],
+      "valueType": "boolean",
       "validValueOptions": ["true", "false"]
     }
   ]
 }
 \`\`\`
+
+The \`codeSnippets\` field is required - provide 1-3 short code excerpts (under 80 chars each) showing exactly where and how this attribute is used. These snippets must appear verbatim in the code.
 `;
 //# sourceMappingURL=generateEntityKeyAttributes.js.map

package/packages/ai/src/lib/generateEntityKeyAttributes.js.map

@@ -1 +1 @@
-{"version":3,"file":"generateEntityKeyAttributes.js","sourceRoot":"","sources":["../../../../../../packages/ai/src/lib/generateEntityKeyAttributes.ts"],"names":[],"mappings":"AAAA,OAAO,cAAc,MAAM,kBAAkB,CAAC;AAC9C,OAAO,EAAE,WAAW,EAAE,MAAM,uBAAuB,CAAC;AAEpD,OAAO,oCAAoC,MAAM,yDAAyD,CAAC;AAC3G,OAAO,EAAE,MAAM,EAAE,MAAM,gBAAgB,CAAC;AACxC,OAAO,EAAE,oBAAoB,EAAE,MAAM,YAAY,CAAC;AAClD,OAAO,EAAE,mBAAmB,EAAE,MAAM,wCAAwC,CAAC;AAC7E,OAAO,EAAE,iCAAiC,EAAE,MAAM,qDAAqD,CAAC;AACxG,OAAO,UAAU,MAAM,cAAc,CAAC;AAEtC,OAAO,EAAE,gCAAgC,EAAE,MAAM,gCAAgC,CAAC;AAQlF,MAAM,CAAC,OAAO,CAAC,KAAK,UAAU,2BAA2B,CAAC,EACxD,MAAM,EACN,KAAK,GAC2B;IAChC,MAAM,CAAC,2CAA2C,EAAE;QAClD,QAAQ,EAAE,MAAM,CAAC,QAAQ;QACzB,UAAU,EAAE,MAAM,CAAC,IAAI;KACxB,CAAC,CAAC;IAEH,MAAM,EAAE,aAAa,EAAE,cAAc,EAAE,oBAAoB,EAAE,GAC3D,mBAAmB,CAAC,MAAM,EAAE,IAAI,CAAC,CAAC;IAEpC,IAAI,MAAM,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QAC5C,OAAO,CAAC,GAAG,CACT,gDAAgD,MAAM,CAAC,QAAQ,IAAI,MAAM,CAAC,IAAI,EAAE,CACjF,CAAC;QACF,OAAO;YACL,aAAa,EAAE,EAAE;YACjB,OAAO,EAAE,IAAI;SACd,CAAC;IACJ,CAAC;IAED,+FAA+F;IAC/F,MAAM,qBAAqB,GAAG,iCAAiC,CAAC;QAC9D,SAAS,EAAE,MAAM,CAAC,IAAI;QACtB,MAAM,EAAE,EAAE,GAAG,aAAa,EAAE;KAC7B,CAAC,CAAC;IAEH,uFAAuF;IACvF,MAAM,iBAAiB,GACrB,MAAM,CAAC,QAAQ,EAAE,qBAAqB,EAAE,iBAAiB,IAAI,EAAE,CAAC;IAClE,MAAM,qBAAqB,GACzB,gCAAgC,CAAC,iBAAiB,CAAC,CAAC;IAEtD,MAAM,MAAM,GAAG,oCAAoC,CAAC;QAClD,IAAI,EAAE,MAAM,CAAC,IAAI;QACjB,aAAa,EAAE,qBAAqB;QACpC,cAAc;QACd,qBAAqB;KACtB,CAAC,CAAC;IAEH,OAAO,CAAC,GAAG,CAAC,mDAAmD,CAAC,CAAC;IACjE,OAAO,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;IAEpB,MAAM,QAAQ,GAAG,MAAM,cAAc,CAAC;QACpC,IAAI,EAAE,6BAA6B;QACnC,aAAa,EAAE,UAAU,CAAC,MAAM,CAAC,QAAQ,CAAC;YACxC,CAAC,CAAC,uBAAuB;YACzB,CAAC,CAAC,sBAAsB;QAC1B,MAAM;QACN,KAAK,EAAE,KAAK,IAAI,oBAAoB;KACrC,CAAC,CAAC;IAEH,MAAM,OAAO,GAAG,MAAM,WAAW,CAAC;QAChC,WAAW,EAAE,QAAQ;QACrB,SAAS,EAAE,MAAM,CAAC,GAAG;QACrB,SAAS,EAAE;YACT,MAAM,EAAE;gBACN,IAAI,EAAE,MAAM,CAAC,IAAI;gBACjB,QAAQ,EAAE,MAAM,CAAC,QAAQ;gBACzB,IAAI,EAAE,MAAM,CAAC,IAAI;gBACjB,QAAQ,EAAE;oBACR,qBAAqB,EAAE,MAAM,CAAC,QAAQ,CAAC,qBAAqB;iBAC7D;aACF;YACD,KAAK;SACN;QACD,GAAG,QAAQ,CAAC,KAAK;KAClB,CAAC,CAAC;IAEH,MAAM,EAAE,UAAU,EAAE,mBAAmB,EAAE,GAAG,QAAQ,CAAC;IACrD,IAAI,CAAC,mBAAmB,EAAE,CAAC;QACzB,OAAO,CAAC,GAAG,CACT,mFAAmF,CACpF,CAAC;QACF,OAAO,IAAI,CAAC;IACd,CAAC;IAED,OAAO,CAAC,GAAG,CACT,WAAW,OAAO,CAAC,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC,CAAC,KAAK,KAAK,MAAM,CAAC,QAAQ,IAAI,MAAM,CAAC,IAAI,sBAAsB,CACjG,CAAC;IACF,OAAO,CAAC,GAAG,CAAC,mBAAmB,CAAC,CAAC;IAEjC,IAAI,aAAa,GAAG,SAAS,CAAC;IAC9B,IAAI,CAAC;QACH,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,mBAAmB,CAAC,CAAC;QAC/C,aAAa,GAAG,MAAM,EAAE,aAAa,CAAC;IACxC,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,OAAO,CAAC,GAAG,CACT,2EAA2E,MAAM,CAAC,QAAQ,IAAI,MAAM,CAAC,IAAI,GAAG,EAC5G,KAAK,CACN,CAAC;IACJ,CAAC;IAED,IAAI,CAAC,aAAa,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,aAAa,CAAC,EAAE,CAAC;QACpD,OAAO,EAAE,aAAa,EAAE,EAAE,EAAE,OAAO,EAAE,CAAC;IACxC,CAAC;IAED,iDAAiD;IACjD,kFAAkF;IAClF,MAAM,sBAAsB,GAAG,MAAM,CAAC,WAAW,CAC/C,MAAM,CAAC,OAAO,CAAC,cAAc,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,YAAY,EAAE,QAAQ,CAAC,EAAE,EAAE,CAAC;QAC/D,QAAQ;QACR,YAAY;KACb,CAAC,CACH,CAAC;IAEF,MAAM,iBAAiB,GAAG,aAAa;SACpC,MAAM,CAAC,OAAO,CAAC;SACf,MAAM,CACL,CAAC,YAAY,EAAE,KAAK,EAAE,IAAI,EAAE,EAAE,CAC5B,YAAY,EAAE,QAAQ;QACtB,IAAI,CAAC,SAAS,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,EAAE,EAAE,QAAQ,KAAK,YAAY,CAAC,QAAQ,CAAC;YAC5D,KAAK,CACV;SACA,MAAM,CAAC,CAAC,YAAY,EAAE,EAAE,CAAC,sBAAsB,CAAC,YAAY,CAAC,QAAQ,CAAC,CAAC;SACvE,GAAG,CAAC,CAAC,YAAY,EAAE,EAAE;QACpB,MAAM,YAAY,GAAG,sBAAsB,CAAC,YAAY,CAAC,QAAQ,CAAC,CAAC;QACnE,OAAO;YACL,YAAY;YACZ,YAAY,EAAE,cAAc,CAAC,YAAY,CAAC;YAC1C,iBAAiB,EACf,oBAAoB,CAAC,YAAY,CAAC,IAAI,YAAY,CAAC,QAAQ;YAC7D,WAAW,EAAE,YAAY,CAAC,WAAW;YACrC,iBAAiB,EAAE,YAAY,CAAC,iBAAiB;YACjD,iBAAiB,EAAE,YAAY,CAAC,iBAAiB;SAClD,CAAC;IACJ,CAAC,CAAC,CAAC;IAEL,OAAO;QACL,aAAa,EAAE,iBAAiB;QAChC,OAAO;KACR,CAAC;AACJ,CAAC;AAED,MAAM,uBAAuB,GAAG;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;CAgD/B,CAAC;AAEF,MAAM,sBAAsB,GAAG;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;CA+C9B,CAAC"}
+{"version":3,"file":"generateEntityKeyAttributes.js","sourceRoot":"","sources":["../../../../../../packages/ai/src/lib/generateEntityKeyAttributes.ts"],"names":[],"mappings":"AAAA,OAAO,cAAc,MAAM,kBAAkB,CAAC;AAC9C,OAAO,EAAE,WAAW,EAAE,MAAM,uBAAuB,CAAC;AAEpD,OAAO,oCAAoC,MAAM,yDAAyD,CAAC;AAC3G,OAAO,EAAE,MAAM,EAAE,MAAM,gBAAgB,CAAC;AACxC,OAAO,EAAE,oBAAoB,EAAE,MAAM,YAAY,CAAC;AAClD,OAAO,EAAE,mBAAmB,EAAE,MAAM,wCAAwC,CAAC;AAC7E,OAAO,EAAE,iCAAiC,EAAE,MAAM,qDAAqD,CAAC;AACxG,OAAO,EAAE,0BAA0B,EAAE,MAAM,iDAAiD,CAAC;AAC7F,OAAO,UAAU,MAAM,cAAc,CAAC;AAEtC,OAAO,EAAE,gCAAgC,EAAE,MAAM,gCAAgC,CAAC;AASlF,MAAM,CAAC,OAAO,CAAC,KAAK,UAAU,2BAA2B,CAAC,EACxD,MAAM,EACN,mBAAmB,EACnB,KAAK,GAC2B;IAChC,MAAM,CAAC,2CAA2C,EAAE;QAClD,QAAQ,EAAE,MAAM,CAAC,QAAQ;QACzB,UAAU,EAAE,MAAM,CAAC,IAAI;KACxB,CAAC,CAAC;IAEH,MAAM,qBAAqB,GAAG,MAAM,CAAC,QAAQ,EAAE,qBAAqB,CAAC;IACrE,MAAM,4BAA4B,GAChC,qBAAqB,EAAE,4BAA4B,IAAI,EAAE,CAAC;IAC5D,MAAM,EAAE,aAAa,EAAE,cAAc,EAAE,oBAAoB,EAAE,GAC3D,mBAAmB,CACjB,qBAAqB,IAAI,mBAAmB,EAC5C,mBAAmB,EACnB,4BAA4B,EAC5B,IAAI,CACL,CAAC;IAEJ,4EAA4E;IAC5E,qFAAqF;IACrF,8EAA8E;IAC9E,0BAA0B,CAAC,aAAa,CAAC,CAAC;IAE1C,IAAI,MAAM,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QAC5C,OAAO,CAAC,GAAG,CACT,gDAAgD,MAAM,CAAC,QAAQ,IAAI,MAAM,CAAC,IAAI,EAAE,CACjF,CAAC;QACF,OAAO;YACL,aAAa,EAAE,EAAE;YACjB,OAAO,EAAE,IAAI;SACd,CAAC;IACJ,CAAC;IAED,0EAA0E;IAC1E,0DAA0D;IAC1D,MAAM,yBAAyB,GAAG,MAAM,CAAC,WAAW,CAClD,MAAM,CAAC,OAAO,CAAC,mBAAmB,CAAC,eAAe,IAAI,EAAE,CAAC;SACtD,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,UAAU,CAAC,YAAY,CAAC,CAAC;SAC3C,GAAG,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,qBAAqB,EAAE,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC,CAC9D,CAAC;IAEF,+FAA+F;IAC/F,sGAAsG;IACtG,MAAM,qBAAqB,GAAG,iCAAiC,CAAC;QAC9D,SAAS,EAAE,MAAM,CAAC,IAAI;QACtB,MAAM,EAAE,EAAE,GAAG,aAAa,EAAE;QAC5B,YAAY,EAAE,yBAAyB;KACxC,CAAC,CAAC;IAEH,uFAAuF;IACvF,MAAM,iBAAiB,GACrB,MAAM,CAAC,QAAQ,EAAE,qBAAqB,EAAE,iBAAiB,IAAI,EAAE,CAAC;IAClE,MAAM,oBAAoB,GACxB,MAAM,CAAC,QAAQ,EAAE,qBAAqB,EAAE,oBAAoB,IAAI,EAAE,CAAC;IACrE,MAAM,qBAAqB,GAAG,gCAAgC,CAC5D,iBAAiB,EACjB,oBAAoB,CACrB,CAAC;IAEF,MAAM,MAAM,GAAG,oCAAoC,CAAC;QAClD,IAAI,EAAE,MAAM,CAAC,IAAI;QACjB,aAAa,EAAE,qBAAqB;QACpC,cAAc;QACd,qBAAqB;KACtB,CAAC,CAAC;IAEH,OAAO,CAAC,GAAG,CAAC,mDAAmD,CAAC,CAAC;IACjE,OAAO,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;IAEpB,MAAM,QAAQ,GAAG,MAAM,cAAc,CAAC;QACpC,IAAI,EAAE,6BAA6B;QACnC,aAAa,EAAE,UAAU,CAAC,MAAM,CAAC,QAAQ,CAAC;YACxC,CAAC,CAAC,uBAAuB;YACzB,CAAC,CAAC,sBAAsB;QAC1B,MAAM;QACN,KAAK,EAAE,KAAK,IAAI,oBAAoB;KACrC,CAAC,CAAC;IAEH,MAAM,OAAO,GAAG,MAAM,WAAW,CAAC;QAChC,WAAW,EAAE,QAAQ;QACrB,SAAS,EAAE,MAAM,CAAC,GAAG;QACrB,SAAS,EAAE;YACT,MAAM,EAAE;gBACN,IAAI,EAAE,MAAM,CAAC,IAAI;gBACjB,QAAQ,EAAE,MAAM,CAAC,QAAQ;gBACzB,IAAI,EAAE,MAAM,CAAC,IAAI;gBACjB,QAAQ,EAAE;oBACR,qBAAqB,EAAE,MAAM,CAAC,QAAQ,CAAC,qBAAqB;iBAC7D;aACF;YACD,KAAK;SACN;QACD,GAAG,QAAQ,CAAC,KAAK;KAClB,CAAC,CAAC;IAEH,MAAM,EAAE,UAAU,EAAE,mBAAmB,EAAE,GAAG,QAAQ,CAAC;IACrD,IAAI,CAAC,mBAAmB,EAAE,CAAC;QACzB,OAAO,CAAC,GAAG,CACT,mFAAmF,CACpF,CAAC;QACF,OAAO,IAAI,CAAC;IACd,CAAC;IAED,OAAO,CAAC,GAAG,CACT,WAAW,OAAO,CAAC,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC,CAAC,KAAK,KAAK,MAAM,CAAC,QAAQ,IAAI,MAAM,CAAC,IAAI,sBAAsB,CACjG,CAAC;IACF,OAAO,CAAC,GAAG,CAAC,mBAAmB,CAAC,CAAC;IAEjC,IAAI,aAAa,GAAG,SAAS,CAAC;IAC9B,IAAI,CAAC;QACH,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,mBAAmB,CAAC,CAAC;QAC/C,aAAa,GAAG,MAAM,EAAE,aAAa,CAAC;IACxC,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,OAAO,CAAC,GAAG,CACT,2EAA2E,MAAM,CAAC,QAAQ,IAAI,MAAM,CAAC,IAAI,GAAG,EAC5G,KAAK,CACN,CAAC;IACJ,CAAC;IAED,IAAI,CAAC,aAAa,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,aAAa,CAAC,EAAE,CAAC;QACpD,OAAO,EAAE,aAAa,EAAE,EAAE,EAAE,OAAO,EAAE,CAAC;IACxC,CAAC;IAED,iDAAiD;IACjD,kFAAkF;IAClF,MAAM,sBAAsB,GAAG,MAAM,CAAC,WAAW,CAC/C,MAAM,CAAC,OAAO,CAAC,cAAc,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,YAAY,EAAE,QAAQ,CAAC,EAAE,EAAE,CAAC;QAC/D,QAAQ;QACR,YAAY;KACb,CAAC,CACH,CAAC;IAEF,8EAA8E;IAC9E,MAAM,yBAAyB,GAAG,CAAC,IAAY,EAAoB,EAAE;QACnE,MAAM,MAAM,GAAG,iBAAiB,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC;QAC7C,MAAM,SAAS,GAAqB,EAAE,CAAC;QAEvC,KAAK,MAAM,KAAK,IAAI,MAAM,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,CAAC;YACvC,uBAAuB;YACvB,IAAI,KAAK,CAAC,cAAc,EAAE,CAAC;gBACzB,SAAS,CAAC,IAAI,CAAC;oBACb,UAAU,EAAE,MAAM,CAAC,IAAI;oBACvB,QAAQ,EAAE,MAAM,CAAC,QAAQ;oBACzB,UAAU,EAAE,KAAK,CAAC,cAAc,CAAC,UAAU;oBAC3C,MAAM,EAAE,KAAK,CAAC,cAAc,CAAC,MAAM;oBACnC,WAAW,EAAE,KAAK,CAAC,cAAc,CAAC,WAAW;oBAC7C,SAAS,EAAE,aAAa;oBACxB,WAAW,EAAE,WAAW,KAAK,CAAC,QAAQ,eAAe,KAAK,CAAC,aAAa,SAAS;iBAClF,CAAC,CAAC;YACL,CAAC;QACH,CAAC;QAED,OAAO,SAAS,CAAC;IACnB,CAAC,CAAC;IAEF,6DAA6D;IAC7D,oFAAoF;IACpF,6FAA6F;IAC7F,MAAM,kCAAkC,GAAG,CACzC,UAAkB,EACA,EAAE;QACpB,MAAM,SAAS,GAAqB,EAAE,CAAC;QAEvC,sEAAsE;QACtE,KAAK,MAAM,CAAC,EAAE,MAAM,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,iBAAiB,CAAC,EAAE,CAAC;YAC3D,KAAK,MAAM,KAAK,IAAI,MAAM,EAAE,CAAC;gBAC3B,MAAM,cAAc,GAAI,KAAqC;qBAC1D,cAAc,CAAC;gBAClB,IAAI,CAAC,cAAc,IAAI,CAAC,KAAK,CAAC,cAAc;oBAAE,SAAS;gBAEvD,oDAAoD;gBACpD,gFAAgF;gBAChF,kHAAkH;gBAClH,MAAM,oBAAoB,GAAG,cAAc;qBACxC,OAAO,CAAC,4BAA4B,EAAE,EAAE,CAAC;qBACzC,OAAO,CAAC,YAAY,EAAE,IAAI,CAAC,CAAC;gBAE/B,IACE,oBAAoB,CAAC,QAAQ,CAAC,GAAG,GAAG,UAAU,CAAC;oBAC/C,oBAAoB,CAAC,QAAQ,CAAC,GAAG,GAAG,UAAU,CAAC,OAAO,CAAC,KAAK,EAAE,GAAG,CAAC,CAAC,EACnE,CAAC;oBACD,SAAS,CAAC,IAAI,CAAC;wBACb,UAAU,EAAE,MAAM,CAAC,IAAI;wBACvB,QAAQ,EAAE,MAAM,CAAC,QAAQ;wBACzB,UAAU,EAAE,KAAK,CAAC,cAAc,CAAC,UAAU;wBAC3C,MAAM,EAAE,KAAK,CAAC,cAAc,CAAC,MAAM;wBACnC,WAAW,EAAE,KAAK,CAAC,cAAc,CAAC,WAAW;wBAC7C,SAAS,EAAE,aAAa;wBACxB,WAAW,EAAE,WAAW,KAAK,CAAC,QAAQ,eAAe,KAAK,CAAC,aAAa,SAAS;qBAClF,CAAC,CAAC;gBACL,CAAC;YACH,CAAC;QACH,CAAC;QAED,OAAO,SAAS,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,aAAa;IAC7C,CAAC,CAAC;IAEF,kEAAkE;IAClE,MAAM,8BAA8B,GAAG,CACrC,YAAkC,EAChB,EAAE;QACpB,IAAI,CAAC,YAAY,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,YAAY,CAAC;YAAE,OAAO,EAAE,CAAC;QAE7D,MAAM,SAAS,GAAqB,EAAE,CAAC;QACvC,MAAM,SAAS,GAAG,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;QAE1C,KAAK,MAAM,OAAO,IAAI,YAAY,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,CAAC;YAC/C,sBAAsB;YACtB,IAAI,CAAC,OAAO,IAAI,OAAO,OAAO,KAAK,QAAQ;gBAAE,SAAS;YAEtD,oCAAoC;YACpC,MAAM,iBAAiB,GAAG,OAAO,CAAC,IAAI,EAAE,CAAC,OAAO,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;YAE9D,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,SAAS,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;gBAC1C,MAAM,cAAc,GAAG,SAAS,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;gBACzD,IAAI,cAAc,CAAC,QAAQ,CAAC,iBAAiB,CAAC,EAAE,CAAC;oBAC/C,SAAS,CAAC,IAAI,CAAC;wBACb,UAAU,EAAE,MAAM,CAAC,IAAI;wBACvB,QAAQ,EAAE,MAAM,CAAC,QAAQ;wBACzB,UAAU,EAAE,CAAC,GAAG,CAAC,EAAE,uBAAuB;wBAC1C,WAAW,EAAE,SAAS,CAAC,CAAC,CAAC,CAAC,IAAI,EAAE;wBAChC,SAAS,EAAE,aAAa;wBACxB,WAAW,EAAE,kCAAkC;qBAChD,CAAC,CAAC;oBACH,MAAM,CAAC,6CAA6C;gBACtD,CAAC;YACH,CAAC;QACH,CAAC;QAED,OAAO,SAAS,CAAC;IACnB,CAAC,CAAC;IAEF,MAAM,iBAAiB,GAAG,aAAa;SACpC,MAAM,CAAC,OAAO,CAAC;SACf,MAAM,CACL,CAAC,YAAY,EAAE,KAAK,EAAE,IAAI,EAAE,EAAE,CAC5B,YAAY,EAAE,QAAQ;QACtB,IAAI,CAAC,SAAS,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,EAAE,EAAE,QAAQ,KAAK,YAAY,CAAC,QAAQ,CAAC;YAC5D,KAAK,CACV;SACA,MAAM,CAAC,CAAC,YAAY,EAAE,EAAE,CAAC,sBAAsB,CAAC,YAAY,CAAC,QAAQ,CAAC,CAAC;SACvE,GAAG,CAAC,CAAC,YAAY,EAAE,EAAE;QACpB,MAAM,YAAY,GAAG,sBAAsB,CAAC,YAAY,CAAC,QAAQ,CAAC,CAAC;QAEnE,+CAA+C;QAC/C,iFAAiF;QACjF,+CAA+C;QAC/C,wDAAwD;QACxD,mEAAmE;QACnE,kFAAkF;QAClF,IAAI,eAAe,GAAG,yBAAyB,CAAC,YAAY,CAAC,CAAC;QAC9D,IAAI,eAAe,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACjC,eAAe,GAAG,yBAAyB,CAAC,YAAY,CAAC,QAAQ,CAAC,CAAC;QACrE,CAAC;QACD,IAAI,eAAe,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACjC,MAAM,iBAAiB,GACrB,oBAAoB,CAAC,YAAY,CAAC,IAAI,YAAY,CAAC,QAAQ,CAAC;YAC9D,eAAe,GAAG,yBAAyB,CAAC,iBAAiB,CAAC,CAAC;QACjE,CAAC;QACD,oFAAoF;QACpF,IAAI,eAAe,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACjC,MAAM,oBAAoB,GAAG,YAAY,CAAC,OAAO,CAC/C,qBAAqB,EACrB,EAAE,CACH,CAAC;YACF,IAAI,oBAAoB,KAAK,YAAY,EAAE,CAAC;gBAC1C,eAAe,GAAG,yBAAyB,CAAC,oBAAoB,CAAC,CAAC;YACpE,CAAC;QACH,CAAC;QACD,sFAAsF;QACtF,qFAAqF;QACrF,IAAI,eAAe,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACjC,eAAe,GAAG,kCAAkC,CAAC,YAAY,CAAC,CAAC;QACrE,CAAC;QAED,uEAAuE;QACvE,4EAA4E;QAC5E,IAAI,eAAe,CAAC,MAAM,KAAK,CAAC,IAAI,YAAY,CAAC,YAAY,EAAE,CAAC;YAC9D,eAAe,GAAG,8BAA8B,CAC9C,YAAY,CAAC,YAAY,CAC1B,CAAC;QACJ,CAAC;QAED,OAAO;YACL,YAAY;YACZ,YAAY,EAAE,cAAc,CAAC,YAAY,CAAC;YAC1C,iBAAiB,EACf,oBAAoB,CAAC,YAAY,CAAC,IAAI,YAAY,CAAC,QAAQ;YAC7D,WAAW,EAAE,YAAY,CAAC,WAAW;YACrC,iBAAiB,EAAE,YAAY,CAAC,iBAAiB;YACjD,iBAAiB,EAAE,YAAY,CAAC,iBAAiB;YACjD,wCAAwC;YACxC,SAAS,EAAE,YAAY,CAAC,SAAS;YACjC,YAAY,EAAE,YAAY,CAAC,YAAY;YACvC,4DAA4D;YAC5D,eAAe,EACb,eAAe,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,SAAS;SAC3D,CAAC;IACJ,CAAC,CAAC,CAAC;IAEL,OAAO;QACL,aAAa,EAAE,iBAAiB;QAChC,OAAO;KACR,CAAC;AACJ,CAAC;AAED,MAAM,uBAAuB,GAAG;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;CAkF/B,CAAC;AAEF,MAAM,sBAAsB,GAAG;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;CAiF9B,CAAC"}

package/packages/ai/src/lib/generateEntityScenarioData.js

@@ -4,6 +4,7 @@ import { saveLlmCall } from "../../../../packages/aws/dynamodb/index.js";
 import validateJson from "./validateJson.js";
 import { awsLog, awsLogDebugLevel } from "../../../../packages/utils/index.js";
 import { parseJsonSafe } from "../../../../packages/ai/index.js";
+import convertNullToUndefinedBySchema from "./dataStructure/helpers/convertNullToUndefinedBySchema.js";
 const DEFAULT_SCENARIO_NAME = 'Default Scenario';
 export async function generateDataForScenario({ entity, structure, scenario, defaultScenarioData, incompleteResponse, analysis, model, }) {
     awsLogDebugLevel(1, `Generating data for ${entity.name}: ${scenario.name}`);
@@ -83,8 +84,9 @@ export async function generateDataForScenario({ entity, structure, scenario, def
     };
     if (structure.dataForMocks && !fullScenarioData.data.argumentsData) {
         fullScenarioData.data.argumentsData = [];
-        if (structure.arguments && !fullScenarioData.data.argumentsData) {
-            fullScenarioData.data.argumentsData = [];
+        // Populate argumentsData from structure.arguments using top-level data values
+        // Bug fix: removed redundant !argumentsData check that was always false after setting to []
+        if (structure.arguments) {
             for (let i = 0; i < structure.arguments.length; ++i) {
                 if (!fullScenarioData.data.argumentsData[i]) {
                     fullScenarioData.data.argumentsData[i] = {};
@@ -103,6 +105,20 @@ export async function generateDataForScenario({ entity, structure, scenario, def
             fullScenarioData.data.mockData[propKey] = dataAsAny[propKey];
         }
     }
+    // Convert null values to undefined based on schema type constraints.
+    // LLM uses null for "no value" (JSON doesn't support undefined), but TypeScript
+    // types like "string | undefined" don't accept null. This converts null→undefined
+    // for fields typed as "T | undefined" (but preserves null for "T | null").
+    if (structure.dataForMocks && fullScenarioData.data.mockData) {
+        convertNullToUndefinedBySchema(fullScenarioData.data.mockData, structure.dataForMocks);
+    }
+    if (structure.arguments && fullScenarioData.data.argumentsData) {
+        for (let i = 0; i < fullScenarioData.data.argumentsData.length; i++) {
+            if (structure.arguments[i]) {
+                convertNullToUndefinedBySchema(fullScenarioData.data.argumentsData[i], structure.arguments[i]);
+            }
+        }
+    }
     return {
         scenarioData: fullScenarioData,
         llmCall: { name: scenario.name, id: llmCall.id },
@@ -181,9 +197,9 @@ export const generateSystemMessage = (scenarioName, defaultScenario) => {
     const scenarioType = defaultScenario
         ? `## Default Scenario
 Generate COMPLETE, robust data for the entire data structure.
-- Fill ALL fields with realistic values (except error attributes)
-- Arrays should have 2-3 items
-- Don't skip nested attributes
+- Fill ALL fields with realistic values (except error attributes and key attributes set to null or undefined)
+- Arrays should have 3-5 items to provide realistic test data variety
+- Don't skip nested attributes unless the key attributes specify a parent attribute should be null or undefined
 - This provides the baseline data for all other scenarios`
         : `## Non-Default Scenario
 Generate ONLY the differences from the default scenario.
@@ -213,16 +229,19 @@ Use for relative dates. Code runs in Node (no browser APIs, no external librarie
 \`\`\`
 Use simple elements only (\`<div>\`, \`<span>\`). No custom components.
 
-## Function Keys
-Preserve function call syntax exactly as written in the structure:
-\`\`\`json
-{
-  "mockData": {
-    "useUser()": { "user": { "name": "John" } },
-    "from().select()": [{ "id": "1" }]
-  }
-}
-\`\`\`
+### Arrays
+- Arrays should have many items (at least 4) unless specified otherwise
+- Each item must follow the exact structure provided
+- In general we want robust data, not minimal data unless specified otherwise
+
+## CRITICAL: Preserve Exact Structure
+Your response MUST mirror the EXACT nested structure provided in mockData Structure.
+- Do NOT reorganize, split, or create duplicate keys
+- The hierarchy of nested objects must match exactly what was provided unless overridden by scenario rules
+- Only change the leaf VALUES (replacing type descriptions like "string" with actual data like "hello")
+- Copy the key strings EXACTLY from the structure
+- Do NOT modify type parameters, arguments, or any part of the key
+- The keys preserve the exact function call as written in the original code
 
 ## Response Format
 \`\`\`json
@@ -240,7 +259,7 @@ Preserve function call syntax exactly as written in the structure:
 ## Rules
 - Valid JSON only—no raw code outside markers
 - No \`undefined\`—use \`null\` or omit
-- No data references (can't use \`posts[0]\` elsewhere—duplicate the value)
+- No data references (can't use \`posts[0]\` elsewhere — duplicate the value)
 - Scenario name must match exactly: "${scenarioName}"
 - Empty mockData: \`{}\`, empty argumentsData: \`[]\`
 `;

package/packages/ai/src/lib/generateEntityScenarioData.js.map

@@ -1 +1 @@
-{"version":3,"file":"generateEntityScenarioData.js","sourceRoot":"","sources":["../../../../../../packages/ai/src/lib/generateEntityScenarioData.ts"],"names":[],"mappings":"AAAA,OAAO,cAAc,MAAM,kBAAkB,CAAC;AAC9C,OAAO,mCAAmC,MAAM,wDAAwD,CAAC;AACzG,OAAO,EAAE,WAAW,EAAE,MAAM,uBAAuB,CAAC;AAQpD,OAAO,YAAY,MAAM,gBAAgB,CAAC;AAC1C,OAAO,EAAE,MAAM,EAAE,gBAAgB,EAAE,MAAM,gBAAgB,CAAC;AAC1D,OAAO,EAAM,aAAa,EAAE,MAAM,aAAa,CAAC;AAEhD,MAAM,qBAAqB,GAAG,kBAAkB,CAAC;AAqBjD,MAAM,CAAC,KAAK,UAAU,uBAAuB,CAAC,EAC5C,MAAM,EACN,SAAS,EACT,QAAQ,EACR,mBAAmB,EACnB,kBAAkB,EAClB,QAAQ,EACR,KAAK,GACuB;IAC5B,gBAAgB,CAAC,CAAC,EAAE,uBAAuB,MAAM,CAAC,IAAI,KAAK,QAAQ,CAAC,IAAI,EAAE,CAAC,CAAC;IAE5E,MAAM,MAAM,GAAG,mCAAmC,CAChD,SAAS,EACT,QAAQ,EACR,mBAAmB,EACnB,kBAAkB,CACnB,CAAC;IAEF,MAAM,SAAS,GAAG,QAAQ,CAAC,IAAI,KAAK,qBAAqB,CAAC;IAC1D,MAAM,QAAQ,GAAG,MAAM,cAAc,CAAC;QACpC,IAAI,EAAE,4BAA4B;QAClC,aAAa,EAAE,kBAAkB;YAC/B,CAAC,CAAC,+BAA+B,CAAC,QAAQ,CAAC,IAAI,EAAE,SAAS,CAAC;YAC3D,CAAC,CAAC,qBAAqB,CAAC,QAAQ,CAAC,IAAI,EAAE,SAAS,CAAC;QACnD,MAAM;QACN,KAAK;KACN,CAAC,CAAC;IAEH,MAAM,OAAO,GAAG,MAAM,WAAW,CAAC;QAChC,WAAW,EAAE,UAAU;QACvB,SAAS,EAAE,QAAQ,CAAC,EAAE;QACtB,SAAS,EAAE;YACT,MAAM,EAAE;gBACN,IAAI,EAAE,MAAM,CAAC,IAAI;gBACjB,QAAQ,EAAE,MAAM,CAAC,QAAQ;gBACzB,QAAQ,EAAE;oBACR,WAAW,EAAE,MAAM,CAAC,QAAQ,EAAE,WAAW;iBAC1C;aACF;YACD,SAAS;YACT,QAAQ;YACR,mBAAmB;YACnB,kBAAkB;YAClB,QAAQ,EAAE;gBACR,EAAE,EAAE,QAAQ,CAAC,EAAE;gBACf,QAAQ,EAAE;oBACR,qBAAqB,EAAE;wBACrB,4BAA4B,EAC1B,MAAM,CAAC,QAAQ,CAAC,qBAAqB;6BAClC,4BAA4B;qBAClC;iBACF;aACF;YACD,KAAK;SACN;QACD,GAAG,QAAQ,CAAC,KAAK;KAClB,CAAC,CAAC;IAEH,MAAM,EAAE,UAAU,EAAE,YAAY,EAAE,GAAG,QAAQ,CAAC;IAE9C,IAAI,CAAC,UAAU,EAAE,CAAC;QAChB,OAAO,CAAC,GAAG,CACT,oEAAoE,CACrE,CAAC;QACF,OAAO,IAAI,CAAC;IACd,CAAC;IAED,MAAM,CACJ,WAAW,OAAO,CAAC,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC,CAAC,KAAK,KAAK,MAAM,CAAC,QAAQ,IAAI,MAAM,CAAC,QAAQ,EAAE,WAAW,IAAI,MAAM,CAAC,IAAI,8CAA8C,EACxJ;QACE,YAAY;QACZ,UAAU;KACX,CACF,CAAC;IAEF,MAAM,SAAS,GAAG,YAAY,CAAC,UAAU,CAAC,CAAC;IAE3C,MAAM,MAAM,GAAG,aAAa,CAAC,SAAS,CAAC,CAAC;IACxC,IAAI,CAAC,MAAM,IAAI,OAAO,MAAM,KAAK,QAAQ,IAAI,CAAC,CAAC,cAAc,IAAI,MAAM,CAAC,EAAE,CAAC;QACzE,OAAO,CAAC,GAAG,CAAC,wDAAwD,EAAE;YACpE,UAAU,EAAE,MAAM,CAAC,IAAI;YACvB,YAAY,EAAE,QAAQ,CAAC,IAAI;YAC3B,SAAS,EAAE,CAAC,CAAC,MAAM;YACnB,UAAU,EAAE,OAAO,MAAM;YACzB,eAAe,EAAE,MAAM,IAAI,cAAc,IAAI,MAAM;YACnD,iBAAiB,EAAE,UAAU,CAAC,SAAS,CAAC,CAAC,EAAE,GAAG,CAAC;SAChD,CAAC,CAAC;QACH,OAAO,IAAI,CAAC;IACd,CAAC;IAED,MAAM,EAAE,YAAY,EAAE,8BAA8B,EAAE,GAAG,MAExD,CAAC;IAEF,OAAO,CAAC,GAAG,CAAC,4DAA4D,EAAE;QACxE,UAAU,EAAE,MAAM,CAAC,IAAI;QACvB,YAAY,EAAE,QAAQ,CAAC,IAAI;QAC3B,kBAAkB,EAAE,8BAA8B,CAAC,YAAY;QAC/D,OAAO,EAAE,CAAC,CAAC,8BAA8B,CAAC,IAAI;QAC9C,QAAQ,EAAE,8BAA8B,CAAC,IAAI;YAC3C,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,8BAA8B,CAAC,IAAI,CAAC;YAClD,CAAC,CAAC,EAAE;KACP,CAAC,CAAC;IAEH,MAAM,gBAAgB,GAAiB;QACrC,GAAG,8BAA8B;QACjC,mBAAmB,EAAE,QAAQ,EAAE,WAAW,IAAI,EAAE;KACjC,CAAC;IAElB,IAAI,SAAS,CAAC,YAAY,IAAI,CAAC,gBAAgB,CAAC,IAAI,CAAC,aAAa,EAAE,CAAC;QACnE,gBAAgB,CAAC,IAAI,CAAC,aAAa,GAAG,EAAE,CAAC;QACzC,IAAI,SAAS,CAAC,SAAS,IAAI,CAAC,gBAAgB,CAAC,IAAI,CAAC,aAAa,EAAE,CAAC;YAChE,gBAAgB,CAAC,IAAI,CAAC,aAAa,GAAG,EAAE,CAAC;YACzC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,SAAS,CAAC,SAAS,CAAC,MAAM,EAAE,EAAE,CAAC,EAAE,CAAC;gBACpD,IAAI,CAAC,gBAAgB,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC,CAAC,EAAE,CAAC;oBAC5C,gBAAgB,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC,CAAC,GAAG,EAAE,CAAC;gBAC9C,CAAC;gBACD,KAAK,MAAM,MAAM,IAAI,MAAM,CAAC,IAAI,CAAC,SAAS,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;oBACzD,MAAM,SAAS,GAAG,gBAAgB,CAAC,IAAW,CAAC;oBAC/C,gBAAgB,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,GAAG,SAAS,CAAC,MAAM,CAAC,CAAC;gBACrE,CAAC;YACH,CAAC;QACH,CAAC;IACH,CAAC;IAED,IAAI,SAAS,CAAC,YAAY,IAAI,CAAC,gBAAgB,CAAC,IAAI,CAAC,QAAQ,EAAE,CAAC;QAC9D,gBAAgB,CAAC,IAAI,CAAC,QAAQ,GAAG,EAAE,CAAC;QACpC,KAAK,MAAM,OAAO,IAAI,MAAM,CAAC,IAAI,CAAC,SAAS,CAAC,SAAS,CAAC,EAAE,CAAC;YACvD,MAAM,SAAS,GAAG,gBAAgB,CAAC,IAAW,CAAC;YAC/C,gBAAgB,CAAC,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC,GAAG,SAAS,CAAC,OAAO,CAAC,CAAC;QAC/D,CAAC;IACH,CAAC;IAED,OAAO;QACL,YAAY,EAAE,gBAAgB;QAC9B,OAAO,EAAE,EAAE,IAAI,EAAE,QAAQ,CAAC,IAAI,EAAE,EAAE,EAAE,OAAO,CAAC,EAAE,EAAE;KACjD,CAAC;AACJ,CAAC;AAED,MAAM,CAAC,OAAO,CAAC,KAAK,UAAU,0BAA0B,CAAC,EACvD,MAAM,EACN,SAAS,EACT,SAAS,EACT,kBAAkB,EAClB,QAAQ,EACR,KAAK,GAC0B;IAC/B,IAAI,SAAS,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QAC3B,OAAO,EAAE,aAAa,EAAE,EAAE,EAAE,QAAQ,EAAE,EAAE,EAAE,CAAC;IAC7C,CAAC;IAED,IAAI,CAAC;QACH,MAAM,aAAa,GAAG,EAAE,CAAC;QACzB,MAAM,QAAQ,GAAG,EAAE,CAAC;QAEpB,MAAM,eAAe,GAAG,SAAS,CAAC,IAAI,CACpC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,qBAAqB,CACxC,CAAC;QAEF,MAAM,qBAAqB,GAAG,MAAM,uBAAuB,CAAC;YAC1D,MAAM;YACN,SAAS;YACT,QAAQ,EAAE,eAAe;YACzB,kBAAkB;YAClB,QAAQ;YACR,KAAK;SACN,CAAC,CAAC;QAEH,IAAI,CAAC,qBAAqB,EAAE,CAAC;YAC3B,OAAO,EAAE,aAAa,EAAE,EAAE,EAAE,QAAQ,EAAE,EAAE,EAAE,CAAC;QAC7C,CAAC;QAED,MAAM,mBAAmB,GAAG,qBAAqB,CAAC,YAAY,CAAC;QAE/D,aAAa,CAAC,IAAI,CAAC,mBAAmB,CAAC,CAAC;QACxC,QAAQ,CAAC,IAAI,CAAC,qBAAqB,CAAC,OAAO,CAAC,CAAC;QAE7C,MAAM,mBAAmB,GAAG,SAAS,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,KAAK,eAAe,CAAC,CAAC;QAE3E,MAAM,OAAO,GAAG,MAAM,OAAO,CAAC,GAAG,CAC/B,mBAAmB,CAAC,GAAG,CAAC,KAAK,EAAE,QAAQ,EAAE,EAAE;YACzC,OAAO,uBAAuB,CAAC;gBAC7B,MAAM;gBACN,SAAS;gBACT,QAAQ;gBACR,mBAAmB;gBACnB,kBAAkB;gBAClB,QAAQ;gBACR,KAAK;aACN,CAAC,CAAC;QACL,CAAC,CAAC,CACH,CAAC;QAEF,qDAAqD;QACrD,MAAM,YAAY,GAAG,OAAO,CAAC,MAAM,CACjC,CAAC,MAAM,EAAwC,EAAE,CAAC,MAAM,KAAK,IAAI,CAClE,CAAC;QAEF,MAAM,SAAS,GAAG,OAAO,CAAC,MAAM,GAAG,YAAY,CAAC,MAAM,CAAC;QACvD,IAAI,SAAS,GAAG,CAAC,EAAE,CAAC;YAClB,MAAM,CACJ,eAAe,SAAS,OAAO,OAAO,CAAC,MAAM,sDAAsD,MAAM,CAAC,IAAI,EAAE,CACjH,CAAC;QACJ,CAAC;QAED,aAAa,CAAC,IAAI,CAAC,GAAG,YAAY,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC,CAAC;QACzE,QAAQ,CAAC,IAAI,CAAC,GAAG,YAAY,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC;QAE/D,OAAO,CAAC,GAAG,CAAC,mDAAmD,EAAE;YAC/D,QAAQ,EAAE,MAAM,CAAC,QAAQ;YACzB,UAAU,EAAE,MAAM,CAAC,IAAI;YACvB,cAAc,EAAE,SAAS,CAAC,MAAM;YAChC,wBAAwB,EAAE,CAAC,CAAC,qBAAqB;YACjD,uBAAuB,EAAE,SAAS,CAAC,MAAM,GAAG,CAAC;YAC7C,qBAAqB,EAAE,OAAO,CAAC,MAAM;YACrC,WAAW,EAAE,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,KAAK,IAAI,CAAC,CAAC,MAAM;YACrD,YAAY,EAAE,YAAY,CAAC,MAAM;YACjC,uBAAuB,EAAE,aAAa,CAAC,MAAM;SAC9C,CAAC,CAAC;QAEH,MAAM,CACJ,6BAA6B,EAC7B,IAAI,CAAC,SAAS,CACZ;YACE,QAAQ,EAAE,MAAM,CAAC,QAAQ;YACzB,UAAU,EAAE,MAAM,CAAC,IAAI;YACvB,aAAa,EAAE,aAAa,CAAC,GAAG,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC;gBACxC,YAAY,EAAE,EAAE,CAAC,YAAY;gBAC7B,OAAO,EAAE,CAAC,CAAC,EAAE,CAAC,IAAI;gBAClB,QAAQ,EAAE,EAAE,CAAC,IAAI,CAAC,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,EAAE;aAC9C,CAAC,CAAC;SACJ,EACD,IAAI,EACJ,CAAC,CACF,CACF,CAAC;QAEF,OAAO,EAAE,aAAa,EAAE,QAAQ,EAAE,CAAC;IACrC,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,OAAO,CAAC,GAAG,CAAC,iDAAiD,EAAE,KAAK,CAAC,CAAC;QACtE,MAAM,KAAK,CAAC;IACd,CAAC;AACH,CAAC;AAED,MAAM,CAAC,MAAM,qBAAqB,GAAG,CACnC,YAAoB,EACpB,eAAwB,EACxB,EAAE;IACF,MAAM,YAAY,GAAG,eAAe;QAClC,CAAC,CAAC;;;;;0DAKoD;QACtD,CAAC,CAAC;;;;gDAI0C,CAAC;IAE/C,OAAO;;EAEP,YAAY;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;uBAoCS,YAAY;;;;;;;;;;;;;uCAaI,YAAY;;CAElD,CAAC;AACF,CAAC,CAAC;AAEF,MAAM,CAAC,MAAM,+BAA+B,GAAG,CAC7C,YAAoB,EACpB,SAAkB,EAClB,EAAE,CAAC;;;;;;EAMH,qBAAqB,CAAC,YAAY,EAAE,SAAS,CAAC;;CAE/C,CAAC"}
+{"version":3,"file":"generateEntityScenarioData.js","sourceRoot":"","sources":["../../../../../../packages/ai/src/lib/generateEntityScenarioData.ts"],"names":[],"mappings":"AAAA,OAAO,cAAc,MAAM,kBAAkB,CAAC;AAC9C,OAAO,mCAAmC,MAAM,wDAAwD,CAAC;AACzG,OAAO,EAAE,WAAW,EAAE,MAAM,uBAAuB,CAAC;AAQpD,OAAO,YAAY,MAAM,gBAAgB,CAAC;AAC1C,OAAO,EAAE,MAAM,EAAE,gBAAgB,EAAE,MAAM,gBAAgB,CAAC;AAC1D,OAAO,EAAM,aAAa,EAAE,MAAM,aAAa,CAAC;AAChD,OAAO,8BAA8B,MAAM,wDAAwD,CAAC;AAEpG,MAAM,qBAAqB,GAAG,kBAAkB,CAAC;AAqBjD,MAAM,CAAC,KAAK,UAAU,uBAAuB,CAAC,EAC5C,MAAM,EACN,SAAS,EACT,QAAQ,EACR,mBAAmB,EACnB,kBAAkB,EAClB,QAAQ,EACR,KAAK,GACuB;IAC5B,gBAAgB,CAAC,CAAC,EAAE,uBAAuB,MAAM,CAAC,IAAI,KAAK,QAAQ,CAAC,IAAI,EAAE,CAAC,CAAC;IAE5E,MAAM,MAAM,GAAG,mCAAmC,CAChD,SAAS,EACT,QAAQ,EACR,mBAAmB,EACnB,kBAAkB,CACnB,CAAC;IAEF,MAAM,SAAS,GAAG,QAAQ,CAAC,IAAI,KAAK,qBAAqB,CAAC;IAC1D,MAAM,QAAQ,GAAG,MAAM,cAAc,CAAC;QACpC,IAAI,EAAE,4BAA4B;QAClC,aAAa,EAAE,kBAAkB;YAC/B,CAAC,CAAC,+BAA+B,CAAC,QAAQ,CAAC,IAAI,EAAE,SAAS,CAAC;YAC3D,CAAC,CAAC,qBAAqB,CAAC,QAAQ,CAAC,IAAI,EAAE,SAAS,CAAC;QACnD,MAAM;QACN,KAAK;KACN,CAAC,CAAC;IAEH,MAAM,OAAO,GAAG,MAAM,WAAW,CAAC;QAChC,WAAW,EAAE,UAAU;QACvB,SAAS,EAAE,QAAQ,CAAC,EAAE;QACtB,SAAS,EAAE;YACT,MAAM,EAAE;gBACN,IAAI,EAAE,MAAM,CAAC,IAAI;gBACjB,QAAQ,EAAE,MAAM,CAAC,QAAQ;gBACzB,QAAQ,EAAE;oBACR,WAAW,EAAE,MAAM,CAAC,QAAQ,EAAE,WAAW;iBAC1C;aACF;YACD,SAAS;YACT,QAAQ;YACR,mBAAmB;YACnB,kBAAkB;YAClB,QAAQ,EAAE;gBACR,EAAE,EAAE,QAAQ,CAAC,EAAE;gBACf,QAAQ,EAAE;oBACR,qBAAqB,EAAE;wBACrB,4BAA4B,EAC1B,MAAM,CAAC,QAAQ,CAAC,qBAAqB;6BAClC,4BAA4B;qBAClC;iBACF;aACF;YACD,KAAK;SACN;QACD,GAAG,QAAQ,CAAC,KAAK;KAClB,CAAC,CAAC;IAEH,MAAM,EAAE,UAAU,EAAE,YAAY,EAAE,GAAG,QAAQ,CAAC;IAE9C,IAAI,CAAC,UAAU,EAAE,CAAC;QAChB,OAAO,CAAC,GAAG,CACT,oEAAoE,CACrE,CAAC;QACF,OAAO,IAAI,CAAC;IACd,CAAC;IAED,MAAM,CACJ,WAAW,OAAO,CAAC,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC,CAAC,KAAK,KAAK,MAAM,CAAC,QAAQ,IAAI,MAAM,CAAC,QAAQ,EAAE,WAAW,IAAI,MAAM,CAAC,IAAI,8CAA8C,EACxJ;QACE,YAAY;QACZ,UAAU;KACX,CACF,CAAC;IAEF,MAAM,SAAS,GAAG,YAAY,CAAC,UAAU,CAAC,CAAC;IAE3C,MAAM,MAAM,GAAG,aAAa,CAAC,SAAS,CAAC,CAAC;IACxC,IAAI,CAAC,MAAM,IAAI,OAAO,MAAM,KAAK,QAAQ,IAAI,CAAC,CAAC,cAAc,IAAI,MAAM,CAAC,EAAE,CAAC;QACzE,OAAO,CAAC,GAAG,CAAC,wDAAwD,EAAE;YACpE,UAAU,EAAE,MAAM,CAAC,IAAI;YACvB,YAAY,EAAE,QAAQ,CAAC,IAAI;YAC3B,SAAS,EAAE,CAAC,CAAC,MAAM;YACnB,UAAU,EAAE,OAAO,MAAM;YACzB,eAAe,EAAE,MAAM,IAAI,cAAc,IAAI,MAAM;YACnD,iBAAiB,EAAE,UAAU,CAAC,SAAS,CAAC,CAAC,EAAE,GAAG,CAAC;SAChD,CAAC,CAAC;QACH,OAAO,IAAI,CAAC;IACd,CAAC;IAED,MAAM,EAAE,YAAY,EAAE,8BAA8B,EAAE,GAAG,MAExD,CAAC;IAEF,OAAO,CAAC,GAAG,CAAC,4DAA4D,EAAE;QACxE,UAAU,EAAE,MAAM,CAAC,IAAI;QACvB,YAAY,EAAE,QAAQ,CAAC,IAAI;QAC3B,kBAAkB,EAAE,8BAA8B,CAAC,YAAY;QAC/D,OAAO,EAAE,CAAC,CAAC,8BAA8B,CAAC,IAAI;QAC9C,QAAQ,EAAE,8BAA8B,CAAC,IAAI;YAC3C,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,8BAA8B,CAAC,IAAI,CAAC;YAClD,CAAC,CAAC,EAAE;KACP,CAAC,CAAC;IAEH,MAAM,gBAAgB,GAAiB;QACrC,GAAG,8BAA8B;QACjC,mBAAmB,EAAE,QAAQ,EAAE,WAAW,IAAI,EAAE;KACjC,CAAC;IAElB,IAAI,SAAS,CAAC,YAAY,IAAI,CAAC,gBAAgB,CAAC,IAAI,CAAC,aAAa,EAAE,CAAC;QACnE,gBAAgB,CAAC,IAAI,CAAC,aAAa,GAAG,EAAE,CAAC;QACzC,8EAA8E;QAC9E,4FAA4F;QAC5F,IAAI,SAAS,CAAC,SAAS,EAAE,CAAC;YACxB,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,SAAS,CAAC,SAAS,CAAC,MAAM,EAAE,EAAE,CAAC,EAAE,CAAC;gBACpD,IAAI,CAAC,gBAAgB,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC,CAAC,EAAE,CAAC;oBAC5C,gBAAgB,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC,CAAC,GAAG,EAAE,CAAC;gBAC9C,CAAC;gBACD,KAAK,MAAM,MAAM,IAAI,MAAM,CAAC,IAAI,CAAC,SAAS,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;oBACzD,MAAM,SAAS,GAAG,gBAAgB,CAAC,IAAW,CAAC;oBAC/C,gBAAgB,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,GAAG,SAAS,CAAC,MAAM,CAAC,CAAC;gBACrE,CAAC;YACH,CAAC;QACH,CAAC;IACH,CAAC;IAED,IAAI,SAAS,CAAC,YAAY,IAAI,CAAC,gBAAgB,CAAC,IAAI,CAAC,QAAQ,EAAE,CAAC;QAC9D,gBAAgB,CAAC,IAAI,CAAC,QAAQ,GAAG,EAAE,CAAC;QACpC,KAAK,MAAM,OAAO,IAAI,MAAM,CAAC,IAAI,CAAC,SAAS,CAAC,SAAS,CAAC,EAAE,CAAC;YACvD,MAAM,SAAS,GAAG,gBAAgB,CAAC,IAAW,CAAC;YAC/C,gBAAgB,CAAC,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC,GAAG,SAAS,CAAC,OAAO,CAAC,CAAC;QAC/D,CAAC;IACH,CAAC;IAED,qEAAqE;IACrE,gFAAgF;IAChF,kFAAkF;IAClF,2EAA2E;IAC3E,IAAI,SAAS,CAAC,YAAY,IAAI,gBAAgB,CAAC,IAAI,CAAC,QAAQ,EAAE,CAAC;QAC7D,8BAA8B,CAC5B,gBAAgB,CAAC,IAAI,CAAC,QAAQ,EAC9B,SAAS,CAAC,YAAY,CACvB,CAAC;IACJ,CAAC;IACD,IAAI,SAAS,CAAC,SAAS,IAAI,gBAAgB,CAAC,IAAI,CAAC,aAAa,EAAE,CAAC;QAC/D,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,gBAAgB,CAAC,IAAI,CAAC,aAAa,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;YACpE,IAAI,SAAS,CAAC,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC;gBAC3B,8BAA8B,CAC5B,gBAAgB,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC,CAAC,EACtC,SAAS,CAAC,SAAS,CAAC,CAAC,CAAC,CACvB,CAAC;YACJ,CAAC;QACH,CAAC;IACH,CAAC;IAED,OAAO;QACL,YAAY,EAAE,gBAAgB;QAC9B,OAAO,EAAE,EAAE,IAAI,EAAE,QAAQ,CAAC,IAAI,EAAE,EAAE,EAAE,OAAO,CAAC,EAAE,EAAE;KACjD,CAAC;AACJ,CAAC;AAED,MAAM,CAAC,OAAO,CAAC,KAAK,UAAU,0BAA0B,CAAC,EACvD,MAAM,EACN,SAAS,EACT,SAAS,EACT,kBAAkB,EAClB,QAAQ,EACR,KAAK,GAC0B;IAC/B,IAAI,SAAS,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QAC3B,OAAO,EAAE,aAAa,EAAE,EAAE,EAAE,QAAQ,EAAE,EAAE,EAAE,CAAC;IAC7C,CAAC;IAED,IAAI,CAAC;QACH,MAAM,aAAa,GAAG,EAAE,CAAC;QACzB,MAAM,QAAQ,GAAG,EAAE,CAAC;QAEpB,MAAM,eAAe,GAAG,SAAS,CAAC,IAAI,CACpC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,qBAAqB,CACxC,CAAC;QAEF,MAAM,qBAAqB,GAAG,MAAM,uBAAuB,CAAC;YAC1D,MAAM;YACN,SAAS;YACT,QAAQ,EAAE,eAAe;YACzB,kBAAkB;YAClB,QAAQ;YACR,KAAK;SACN,CAAC,CAAC;QAEH,IAAI,CAAC,qBAAqB,EAAE,CAAC;YAC3B,OAAO,EAAE,aAAa,EAAE,EAAE,EAAE,QAAQ,EAAE,EAAE,EAAE,CAAC;QAC7C,CAAC;QAED,MAAM,mBAAmB,GAAG,qBAAqB,CAAC,YAAY,CAAC;QAE/D,aAAa,CAAC,IAAI,CAAC,mBAAmB,CAAC,CAAC;QACxC,QAAQ,CAAC,IAAI,CAAC,qBAAqB,CAAC,OAAO,CAAC,CAAC;QAE7C,MAAM,mBAAmB,GAAG,SAAS,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,KAAK,eAAe,CAAC,CAAC;QAE3E,MAAM,OAAO,GAAG,MAAM,OAAO,CAAC,GAAG,CAC/B,mBAAmB,CAAC,GAAG,CAAC,KAAK,EAAE,QAAQ,EAAE,EAAE;YACzC,OAAO,uBAAuB,CAAC;gBAC7B,MAAM;gBACN,SAAS;gBACT,QAAQ;gBACR,mBAAmB;gBACnB,kBAAkB;gBAClB,QAAQ;gBACR,KAAK;aACN,CAAC,CAAC;QACL,CAAC,CAAC,CACH,CAAC;QAEF,qDAAqD;QACrD,MAAM,YAAY,GAAG,OAAO,CAAC,MAAM,CACjC,CAAC,MAAM,EAAwC,EAAE,CAAC,MAAM,KAAK,IAAI,CAClE,CAAC;QAEF,MAAM,SAAS,GAAG,OAAO,CAAC,MAAM,GAAG,YAAY,CAAC,MAAM,CAAC;QACvD,IAAI,SAAS,GAAG,CAAC,EAAE,CAAC;YAClB,MAAM,CACJ,eAAe,SAAS,OAAO,OAAO,CAAC,MAAM,sDAAsD,MAAM,CAAC,IAAI,EAAE,CACjH,CAAC;QACJ,CAAC;QAED,aAAa,CAAC,IAAI,CAAC,GAAG,YAAY,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC,CAAC;QACzE,QAAQ,CAAC,IAAI,CAAC,GAAG,YAAY,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC;QAE/D,OAAO,CAAC,GAAG,CAAC,mDAAmD,EAAE;YAC/D,QAAQ,EAAE,MAAM,CAAC,QAAQ;YACzB,UAAU,EAAE,MAAM,CAAC,IAAI;YACvB,cAAc,EAAE,SAAS,CAAC,MAAM;YAChC,wBAAwB,EAAE,CAAC,CAAC,qBAAqB;YACjD,uBAAuB,EAAE,SAAS,CAAC,MAAM,GAAG,CAAC;YAC7C,qBAAqB,EAAE,OAAO,CAAC,MAAM;YACrC,WAAW,EAAE,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,KAAK,IAAI,CAAC,CAAC,MAAM;YACrD,YAAY,EAAE,YAAY,CAAC,MAAM;YACjC,uBAAuB,EAAE,aAAa,CAAC,MAAM;SAC9C,CAAC,CAAC;QAEH,MAAM,CACJ,6BAA6B,EAC7B,IAAI,CAAC,SAAS,CACZ;YACE,QAAQ,EAAE,MAAM,CAAC,QAAQ;YACzB,UAAU,EAAE,MAAM,CAAC,IAAI;YACvB,aAAa,EAAE,aAAa,CAAC,GAAG,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC;gBACxC,YAAY,EAAE,EAAE,CAAC,YAAY;gBAC7B,OAAO,EAAE,CAAC,CAAC,EAAE,CAAC,IAAI;gBAClB,QAAQ,EAAE,EAAE,CAAC,IAAI,CAAC,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,EAAE;aAC9C,CAAC,CAAC;SACJ,EACD,IAAI,EACJ,CAAC,CACF,CACF,CAAC;QAEF,OAAO,EAAE,aAAa,EAAE,QAAQ,EAAE,CAAC;IACrC,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,OAAO,CAAC,GAAG,CAAC,iDAAiD,EAAE,KAAK,CAAC,CAAC;QACtE,MAAM,KAAK,CAAC;IACd,CAAC;AACH,CAAC;AAED,MAAM,CAAC,MAAM,qBAAqB,GAAG,CACnC,YAAoB,EACpB,eAAwB,EACxB,EAAE;IACF,MAAM,YAAY,GAAG,eAAe;QAClC,CAAC,CAAC;;;;;0DAKoD;QACtD,CAAC,CAAC;;;;gDAI0C,CAAC;IAE/C,OAAO;;EAEP,YAAY;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;uBAuCS,YAAY;;;;;;;;;;;;;uCAaI,YAAY;;CAElD,CAAC;AACF,CAAC,CAAC;AAEF,MAAM,CAAC,MAAM,+BAA+B,GAAG,CAC7C,YAAoB,EACpB,SAAkB,EAClB,EAAE,CAAC;;;;;;EAMH,qBAAqB,CAAC,YAAY,EAAE,SAAS,CAAC;;CAE/C,CAAC"}

package/packages/ai/src/lib/generateEntityScenarios.js

@@ -125,6 +125,17 @@ Focus scenarios on demonstrating this specific behavior change.
 
 ${contextSection}${scenarioCountText}
 
+## Coverage Goals (IMPORTANT)
+Your primary goal is to MAXIMIZE coverage of key attributes across all scenarios:
+
+1. **Use ALL validValueOptions**: Each key attribute has indexed \`validValueOptions\`. Across your scenarios, try to use EVERY option at least once. If an attribute has 3 options (indices 0, 1, 2), create scenarios that collectively cover all 3.
+
+2. **Prioritize attributes with dependencies**: Key attributes with a \`dependencies\` array are HIGH PRIORITY - they control conditionally-rendered content. Create scenarios that satisfy these dependencies to reveal gated UI/behavior.
+
+3. **Cover conditional branches**: Use \`codeUsages\` to identify all conditional paths. If code shows \`status === 'active'\` and \`status === 'pending'\`, ensure scenarios cover BOTH values.
+
+4. **Vary values strategically**: Don't repeat the same validValueOption across scenarios unless necessary. Use different indices to maximize coverage.
+
 ## Default Scenario Rules
 The FIRST scenario MUST be named exactly "${DEFAULT_SCENARIO_NAME}" (case-sensitive).
 - Fill ALL key attributes using values from their \`validValueOptions\` list
@@ -132,23 +143,69 @@ The FIRST scenario MUST be named exactly "${DEFAULT_SCENARIO_NAME}" (case-sensit
 - Ignore error-related attributes (set to null)
 - NEVER include \`playwrightInstructions\` - we need a screenshot BEFORE any user interaction
 - Do NOT add playwrightInstructions even if the component has forms/buttons - default shows initial state only
+- **For attributes WITH dependencies**: The Default scenario should satisfy ALL dependencies so the attribute's effect is visible
 
 ## Other Scenarios
 - Each produces DISTINCT output (visual: different screenshot; backend: different return/side-effect)
+- **Use DIFFERENT validValueOptions** than the Default scenario - this is how we achieve coverage
 - Only specify attributes that DIFFER from default
+- Target uncovered validValueOptions: check which indices you haven't used yet
 - May include \`playwrightInstructions\` for user interactions (clicks, form fills)
 
+## Using codeUsages for Branch Coverage
+Key attributes may include a \`codeUsages\` array showing exactly where they're used in conditionals:
+\`\`\`json
+{
+  "codeUsages": [
+    { "line": 45, "code": "if (status === 'active')", "context": "Used in if statement (comparison check)" }
+  ]
+}
+\`\`\`
+Use this to ensure scenarios cover all conditional branches. If you see \`status === 'active'\`, create scenarios for both active AND non-active states.
+
+## Handling Dependencies (Compound Conditionals) - HIGH PRIORITY
+Key attributes with \`dependencies\` are the MOST IMPORTANT to cover because they reveal conditionally-gated content that only appears when multiple conditions are satisfied together.
+
+**Why this matters**: If you see \`items.length > 0 && !isLoading && !hasError\`, the content only renders when ALL THREE conditions are true. Missing even one dependency means the UI won't show.
+
+**Strategy for dependent attributes**:
+1. **Default scenario**: MUST satisfy ALL dependencies of the highest-impact key attributes
+2. **Other scenarios**: Can vary the dependent values to show alternative states (e.g., what happens when isLoading=true)
+
+When a key attribute has dependencies like:
+\`\`\`json
+{
+  "dependencies": [
+    { "path": "signature[0].isLoading", "requiredValue": false },
+    { "path": "signature[0].hasError", "requiredValue": false }
+  ]
+}
+\`\`\`
+The Default scenario MUST set BOTH dependent attributes to their required values so the key attribute's effect is visible.
+
+**Creating scenarios for dependent pathways**:
+- Create at least one scenario where ALL dependencies are satisfied (usually Default)
+- Consider creating additional scenarios that intentionally break dependencies to show the "hidden" state
+
 ## keyAttributeInstructions Format
-Use \`dataStructurePath\` as key, include both variable name and instruction:
+Use \`dataStructurePath\` as key. Include variable name, instruction, AND a valueOptionRef that references which validValueOption you're using:
 \`\`\`json
 {
   "useDiffModal().diffView": {
     "localVariable": "diffView",
-    "instruction": "Set to 'split' mode"
+    "instruction": "Set to 'split' mode",
+    "valueOptionRef": { "index": 0, "refType": "exact" }
   }
 }
 \`\`\`
 
+### valueOptionRef Types
+- \`"exact"\`: Uses the validValueOption at this index exactly
+- \`"derived"\`: Based on this validValueOption but modified (include \`derivation\` field explaining how)
+- \`"custom"\`: Value not from validValueOptions list (use sparingly)
+
+The \`index\` refers to the validValueOptions array position (0 = first option).
+
 ## playwrightInstructions (optional, non-default only)
 \`\`\`json
 {
@@ -176,24 +233,63 @@ Use when: data only appears after user action (button click, form submit), or ke
   "dataScenarios": [
     {
       "name": "${DEFAULT_SCENARIO_NAME}",
-      "testName": "it(\"displays notifications with unread indicator\")",
-      "description": "Default state with 3 unread and 2 read notifications",
+      "testName": "it(\"displays items list with data\")",
+      "description": "Shows item list when data is loaded and no errors",
       "keyAttributeInstructions": {
-        "useNotifications().notifications[]": { "localVariable": "notifications[]", "instruction": "Create 5 notifications" },
-        "useNotifications().notifications[].read": { "localVariable": "notifications[].read", "instruction": "false for 3, true for 2" }
+        "useData().items[]": {
+          "localVariable": "items[]",
+          "instruction": "Array with 3 items",
+          "valueOptionRef": { "index": 0, "refType": "exact" }
+        },
+        "useData().isLoading": {
+          "localVariable": "isLoading",
+          "instruction": "false - satisfies dependency for items to show",
+          "valueOptionRef": { "index": 1, "refType": "exact" }
+        },
+        "useData().status": {
+          "localVariable": "status",
+          "instruction": "Set to 'success'",
+          "valueOptionRef": { "index": 0, "refType": "exact" }
+        }
+      }
+    },
+    {
+      "name": "Loading state",
+      "testName": "it(\"shows spinner when loading\")",
+      "description": "Data is loading - items hidden, spinner shown",
+      "keyAttributeInstructions": {
+        "useData().isLoading": {
+          "localVariable": "isLoading",
+          "instruction": "true - breaks dependency, items won't show",
+          "valueOptionRef": { "index": 0, "refType": "exact" }
+        },
+        "useData().status": {
+          "localVariable": "status",
+          "instruction": "Set to 'loading' for coverage of loading state",
+          "valueOptionRef": { "index": 1, "refType": "exact" }
+        }
       }
     },
     {
       "name": "Empty state",
-      "testName": "it(\"shows placeholder when no notifications\")",
-      "description": "User has no notifications",
+      "testName": "it(\"shows empty message when no items\")",
+      "description": "No items to display",
       "keyAttributeInstructions": {
-        "useNotifications().notifications[]": { "localVariable": "notifications[]", "instruction": "Empty array" }
+        "useData().items[]": {
+          "localVariable": "items[]",
+          "instruction": "Empty array [] to show empty state",
+          "valueOptionRef": { "index": 1, "refType": "exact" }
+        }
       }
     }
   ]
 }
 \`\`\`
+Note how the example:
+- Default satisfies ALL dependencies (isLoading=false) so items content shows
+- "Loading state" uses a different valueOptionRef for status to cover that code path
+- "Empty state" covers the empty array case to maximize coverage
+- Instructions describe the VALUE, not the index (e.g., "Empty array []" not "uses index 1")
 `;
 }
 const SYSTEM_MESSAGE = generateSystemMessage();