@codeyam/codeyam-cli 0.1.0-staging.28f73cf → 0.1.0-staging.323686

package/packages/ai/src/lib/promptGenerators/gatherAttributesMap.js

@@ -47,15 +47,57 @@ function validAttribute(attribute) {
     }
     return true;
 }
-export function gatherAttributesMap(entity, reverse = false) {
+/**
+ * Gathers attributes from data structures for use in prompts.
+ *
+ * @param isolatedDataStructure - The entity's own data structure (determines which paths to include)
+ * @param mergedDataStructureOrEquivalentVars - Either the merged data structure (with richer types from
+ *   dependent analyses) OR equivalentSignatureVariables for backwards compatibility
+ * @param equivalentSignatureVariablesOrReverse - Either equivalentSignatureVariables OR reverse flag
+ * @param reverseArg - The reverse flag (only used when all 4 args provided)
+ */
+export function gatherAttributesMap(isolatedDataStructure, mergedDataStructureOrEquivalentVars, equivalentSignatureVariablesOrReverse, reverseArg) {
+    // Handle backwards compatibility: detect if called with old signature
+    // Old: (mergedDataStructure, equivalentSignatureVariables?, reverse?)
+    // New: (isolatedDataStructure, mergedDataStructure, equivalentSignatureVariables?, reverse?)
+    let mergedDataStructure;
+    let equivalentSignatureVariables;
+    let reverse;
+    if (mergedDataStructureOrEquivalentVars &&
+        typeof mergedDataStructureOrEquivalentVars === 'object' &&
+        'dependencySchemas' in mergedDataStructureOrEquivalentVars) {
+        // New signature: (isolated, merged, equivalentVars?, reverse?)
+        mergedDataStructure = mergedDataStructureOrEquivalentVars;
+        equivalentSignatureVariables =
+            equivalentSignatureVariablesOrReverse ??
+                {};
+        reverse = reverseArg ?? false;
+    }
+    else {
+        // Old signature: (merged, equivalentVars?, reverse?)
+        // Use isolatedDataStructure as both isolated and merged for backwards compatibility
+        mergedDataStructure = null; // Will use isolatedDataStructure for types
+        equivalentSignatureVariables =
+            mergedDataStructureOrEquivalentVars ??
+                {};
+        reverse =
+            equivalentSignatureVariablesOrReverse ?? false;
+    }
     const attributesMap = {};
     const associationMap = {};
     // Maps internalPath -> full data structure path (e.g., "useDiffModal().diffView")
     const dataStructurePathMap = {};
-    const { signatureSchema, equivalentSignatureVariables, dependencySchemas } = entity.metadata.isolatedDataStructure;
-    for (const key in signatureSchema) {
+    // Use isolatedDataStructure for paths, but look up types in mergedDataStructure when available
+    const isolatedSignatureSchema = isolatedDataStructure.signatureSchema;
+    const mergedSignatureSchema = mergedDataStructure?.signatureSchema ?? isolatedSignatureSchema;
+    const isolatedDependencySchemas = isolatedDataStructure.dependencySchemas;
+    const mergedDependencySchemas = mergedDataStructure?.dependencySchemas ?? isolatedDependencySchemas;
+    // Process signature schema - use isolatedDataStructure keys, but merged types
+    for (const key in isolatedSignatureSchema) {
         const keyParts = splitOutsideParenthesesAndArrays(key);
         if (keyParts[0].startsWith('signature')) {
+            // Use merged type if available, otherwise fall back to isolated type
+            const typeValue = mergedSignatureSchema[key] ?? isolatedSignatureSchema[key];
             const equivalentSignatureVariable = equivalentSignatureVariables[keyParts[0]];
             const equivalentSignatureVariableParts = splitOutsideParenthesesAndArrays(equivalentSignatureVariable);
             if (equivalentSignatureVariable &&
@@ -65,7 +107,7 @@ export function gatherAttributesMap(entity, reverse = false) {
                     equivalentKey,
                     ...keyParts.slice(1),
                 ]);
-                attributesMap[equivalentPath] = signatureSchema[key];
+                attributesMap[equivalentPath] = typeValue;
                 associationMap[reverse ? equivalentPath : key] = reverse
                     ? key
                     : equivalentPath;
@@ -74,7 +116,7 @@ export function gatherAttributesMap(entity, reverse = false) {
             }
             else {
                 const relativePath = joinParenthesesAndArrays(keyParts.slice(1));
-                attributesMap[relativePath] = signatureSchema[key];
+                attributesMap[relativePath] = typeValue;
                 associationMap[reverse ? relativePath : key] = reverse
                     ? key
                     : relativePath;
@@ -83,22 +125,35 @@ export function gatherAttributesMap(entity, reverse = false) {
             }
         }
     }
-    for (const filePath in dependencySchemas) {
-        for (const entityName in dependencySchemas[filePath]) {
-            const returnValueDataStructure = dependencySchemas[filePath][entityName].returnValueSchema ?? {};
-            // First pass: collect all function paths
+    // Process dependency schemas - iterate through isolatedDependencySchemas (for which paths to include)
+    // but use mergedDependencySchemas for types (richer type information)
+    for (const filePath in isolatedDependencySchemas) {
+        for (const entityName in isolatedDependencySchemas[filePath]) {
+            // Get the isolated return value schema (determines which paths exist)
+            const isolatedReturnValueSchema = isolatedDependencySchemas[filePath][entityName].returnValueSchema ?? {};
+            // Get the merged return value schema (for richer types)
+            const mergedReturnValueSchema = mergedDependencySchemas?.[filePath]?.[entityName]?.returnValueSchema ??
+                isolatedReturnValueSchema;
+            // First pass: collect all function paths from merged schema (for broader coverage)
             const functionPaths = [];
-            for (const path in returnValueDataStructure) {
-                if (returnValueDataStructure[path] === 'function') {
+            for (const path in mergedReturnValueSchema) {
+                if (mergedReturnValueSchema[path] === 'function') {
                     functionPaths.push(path);
                 }
             }
-            // Second pass: process all paths
-            for (const path in returnValueDataStructure) {
+            // Second pass: process paths from merged schema, but only include paths that
+            // are rooted in an isolated path (i.e., isolatedDataStructure knows about this dependency)
+            for (const path in mergedReturnValueSchema) {
                 const pathParts = splitOutsideParenthesesAndArrays(path);
                 if (pathParts.length < 2)
                     continue;
-                if (returnValueDataStructure[path] === 'function') {
+                // Filter out paths that contain signature[N] anywhere
+                // These represent function input parameters, not data from data sources
+                // Key attributes must reference data sources, not function inputs
+                if (pathParts.some((part) => part.startsWith('signature['))) {
+                    continue;
+                }
+                if (mergedReturnValueSchema[path] === 'function') {
                     continue;
                 }
                 // Check if this path is inside a function (but not a function return value we care about)
@@ -137,7 +192,7 @@ export function gatherAttributesMap(entity, reverse = false) {
                             const relativePath = path;
                             if (!validAttribute(relativePath))
                                 continue;
-                            attributesMap[relativePath] = returnValueDataStructure[path];
+                            attributesMap[relativePath] = mergedReturnValueSchema[path];
                             associationMap[reverse ? relativePath : path] = reverse
                                 ? path
                                 : relativePath;
@@ -153,7 +208,8 @@ export function gatherAttributesMap(entity, reverse = false) {
                     continue;
                 // Normalize the path for internal use (e.g., ['*'] -> '*')
                 const relativePath = normalizePathForInternalUse(rawRelativePath);
-                attributesMap[relativePath] = returnValueDataStructure[path];
+                // Use the type from mergedReturnValueSchema
+                attributesMap[relativePath] = mergedReturnValueSchema[path];
                 associationMap[reverse ? relativePath : path] = reverse
                     ? path
                     : relativePath;

package/packages/ai/src/lib/promptGenerators/gatherAttributesMap.js.map

@@ -1 +1 @@
-{"version":3,"file":"gatherAttributesMap.js","sourceRoot":"","sources":["../../../../../../../packages/ai/src/lib/promptGenerators/gatherAttributesMap.ts"],"names":[],"mappings":"AACA,OAAO,gBAAgB,MAAM,qBAAqB,CAAC;AACnD,OAAO,EACL,wBAAwB,EACxB,gCAAgC,GACjC,MAAM,4BAA4B,CAAC;AAEpC;;;;GAIG;AACH,SAAS,uBAAuB,CAAC,SAAiB;IAChD,0CAA0C;IAC1C,OAAO,sBAAsB,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;AAChD,CAAC;AAED;;;GAGG;AACH,SAAS,uBAAuB,CAAC,SAAiB;IAChD,MAAM,KAAK,GAAG,SAAS,CAAC,KAAK,CAAC,wBAAwB,CAAC,CAAC;IACxD,OAAO,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC;AACjC,CAAC;AAED;;;;GAIG;AACH,SAAS,2BAA2B,CAAC,IAAY;IAC/C,MAAM,SAAS,GAAG,uBAAuB,CAAC,IAAI,CAAC,CAAC;IAChD,OAAO,SAAS,KAAK,IAAI,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC;AAC/C,CAAC;AAED,SAAS,cAAc,CAAC,SAAiB;IACvC,IAAI,CAAC,SAAS,IAAI,SAAS,CAAC,MAAM,KAAK,CAAC;QAAE,OAAO,KAAK,CAAC;IAEvD,4DAA4D;IAC5D,oEAAoE;IACpE,IAAI,SAAS,CAAC,UAAU,CAAC,GAAG,CAAC,EAAE,CAAC;QAC9B,OAAO,uBAAuB,CAAC,SAAS,CAAC,CAAC;IAC5C,CAAC;IAED,MAAM,cAAc,GAAG,gCAAgC,CAAC,SAAS,CAAC,CAAC;IACnE,IACE,cAAc,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,EAAE,CAC3B,gBAAgB,CAAC,gBAAgB,CAAC,IAAI,CAAC,EAAE,GAAG,EAAE,GAAG,CAAC,CAAC,KAAK,CAAC,IAAI,CAAC,CAC/D,EACD,CAAC;QACD,OAAO,KAAK,CAAC;IACf,CAAC;IAED,2EAA2E;IAC3E,kEAAkE;IAClE,yCAAyC;IACzC,IACE,cAAc,CAAC,MAAM,GAAG,CAAC;QACzB,cAAc,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,UAAU,CAAC,YAAY,CAAC,CAAC,EACrE,CAAC;QACD,OAAO,KAAK,CAAC;IACf,CAAC;IAED,OAAO,IAAI,CAAC;AACd,CAAC;AAED,MAAM,UAAU,mBAAmB,CACjC,MAAgC,EAChC,OAAO,GAAG,KAAK;IAEf,MAAM,aAAa,GAA2B,EAAE,CAAC;IACjD,MAAM,cAAc,GAA2B,EAAE,CAAC;IAClD,kFAAkF;IAClF,MAAM,oBAAoB,GAA2B,EAAE,CAAC;IACxD,MAAM,EAAE,eAAe,EAAE,4BAA4B,EAAE,iBAAiB,EAAE,GACxE,MAAM,CAAC,QAAQ,CAAC,qBAAqB,CAAC;IACxC,KAAK,MAAM,GAAG,IAAI,eAAe,EAAE,CAAC;QAClC,MAAM,QAAQ,GAAG,gCAAgC,CAAC,GAAG,CAAC,CAAC;QACvD,IAAI,QAAQ,CAAC,CAAC,CAAC,CAAC,UAAU,CAAC,WAAW,CAAC,EAAE,CAAC;YACxC,MAAM,2BAA2B,GAC/B,4BAA4B,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC;YAC5C,MAAM,gCAAgC,GAAG,gCAAgC,CACvE,2BAA2B,CAC5B,CAAC;YACF,IACE,2BAA2B;gBAC3B,CAAC,gCAAgC,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,GAAG,CAAC,EAClD,CAAC;gBACD,MAAM,aAAa,GAAG,4BAA4B,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC;gBAChE,MAAM,cAAc,GAAG,wBAAwB,CAAC;oBAC9C,aAAa;oBACb,GAAG,QAAQ,CAAC,KAAK,CAAC,CAAC,CAAC;iBACrB,CAAC,CAAC;gBACH,aAAa,CAAC,cAAc,CAAC,GAAG,eAAe,CAAC,GAAG,CAAC,CAAC;gBACrD,cAAc,CAAC,OAAO,CAAC,CAAC,CAAC,cAAc,CAAC,CAAC,CAAC,GAAG,CAAC,GAAG,OAAO;oBACtD,CAAC,CAAC,GAAG;oBACL,CAAC,CAAC,cAAc,CAAC;gBACnB,+EAA+E;gBAC/E,oBAAoB,CAAC,cAAc,CAAC,GAAG,cAAc,CAAC;YACxD,CAAC;iBAAM,CAAC;gBACN,MAAM,YAAY,GAAG,wBAAwB,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC;gBACjE,aAAa,CAAC,YAAY,CAAC,GAAG,eAAe,CAAC,GAAG,CAAC,CAAC;gBACnD,cAAc,CAAC,OAAO,CAAC,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,GAAG,CAAC,GAAG,OAAO;oBACpD,CAAC,CAAC,GAAG;oBACL,CAAC,CAAC,YAAY,CAAC;gBACjB,+EAA+E;gBAC/E,oBAAoB,CAAC,YAAY,CAAC,GAAG,YAAY,CAAC;YACpD,CAAC;QACH,CAAC;IACH,CAAC;IAED,KAAK,MAAM,QAAQ,IAAI,iBAAiB,EAAE,CAAC;QACzC,KAAK,MAAM,UAAU,IAAI,iBAAiB,CAAC,QAAQ,CAAC,EAAE,CAAC;YACrD,MAAM,wBAAwB,GAC5B,iBAAiB,CAAC,QAAQ,CAAC,CAAC,UAAU,CAAC,CAAC,iBAAiB,IAAI,EAAE,CAAC;YAElE,yCAAyC;YACzC,MAAM,aAAa,GAAa,EAAE,CAAC;YACnC,KAAK,MAAM,IAAI,IAAI,wBAAwB,EAAE,CAAC;gBAC5C,IAAI,wBAAwB,CAAC,IAAI,CAAC,KAAK,UAAU,EAAE,CAAC;oBAClD,aAAa,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;gBAC3B,CAAC;YACH,CAAC;YAED,iCAAiC;YACjC,KAAK,MAAM,IAAI,IAAI,wBAAwB,EAAE,CAAC;gBAC5C,MAAM,SAAS,GAAG,gCAAgC,CAAC,IAAI,CAAC,CAAC;gBACzD,IAAI,SAAS,CAAC,MAAM,GAAG,CAAC;oBAAE,SAAS;gBAEnC,IAAI,wBAAwB,CAAC,IAAI,CAAC,KAAK,UAAU,EAAE,CAAC;oBAClD,SAAS;gBACX,CAAC;gBAED,0FAA0F;gBAC1F,0DAA0D;gBAC1D,MAAM,oBAAoB,GAAG,aAAa;qBACvC,MAAM,CAAC,CAAC,YAAY,EAAE,EAAE,CAAC,IAAI,CAAC,UAAU,CAAC,YAAY,CAAC,CAAC;qBACvD,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,MAAM,GAAG,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC;gBAC1C,IAAI,oBAAoB,EAAE,CAAC;oBACzB,6DAA6D;oBAC7D,MAAM,aAAa,GAAG,IAAI,CAAC,SAAS,CAAC,oBAAoB,CAAC,MAAM,CAAC,CAAC;oBAElE,IAAI,CAAC,aAAa,CAAC,UAAU,CAAC,0BAA0B,CAAC,EAAE,CAAC;wBAC1D,6EAA6E;wBAC7E,UAAU;wBACV,SAAS;oBACX,CAAC;oBAED,gEAAgE;oBAChE,MAAM,aAAa,GAAG,IAAI,CAAC,QAAQ,CAAC,0BAA0B,CAAC,CAAC;oBAChE,IAAI,aAAa,EAAE,CAAC;wBAClB,sFAAsF;wBACtF,uEAAuE;wBACvE,MAAM,iBAAiB,GACrB,gCAAgC,CAAC,oBAAoB,CAAC,CAAC;wBACzD,MAAM,gBAAgB,GACpB,iBAAiB,CAAC,MAAM,GAAG,CAAC;4BAC1B,CAAC,CAAC,iBAAiB,CAAC,iBAAiB,CAAC,MAAM,GAAG,CAAC,CAAC;4BACjD,CAAC,CAAC,IAAI,CAAC;wBACX,MAAM,qBAAqB,GACzB,iBAAiB,CAAC,MAAM,GAAG,CAAC;4BAC1B,CAAC,CAAC,iBAAiB,CAAC,iBAAiB,CAAC,MAAM,GAAG,CAAC,CAAC;4BACjD,CAAC,CAAC,IAAI,CAAC;wBAEX,gFAAgF;wBAChF,yCAAyC;wBACzC,MAAM,sBAAsB,GAC1B,gBAAgB;4BAChB,gBAAgB,CAAC,UAAU,CAAC,GAAG,CAAC;4BAChC,qBAAqB;4BACrB,qBAAqB,KAAK,yBAAyB,CAAC;wBAEtD,IAAI,sBAAsB,EAAE,CAAC;4BAC3B,wDAAwD;4BACxD,MAAM,YAAY,GAAG,IAAI,CAAC;4BAC1B,IAAI,CAAC,cAAc,CAAC,YAAY,CAAC;gCAAE,SAAS;4BAE5C,aAAa,CAAC,YAAY,CAAC,GAAG,wBAAwB,CAAC,IAAI,CAAC,CAAC;4BAC7D,cAAc,CAAC,OAAO,CAAC,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,OAAO;gCACrD,CAAC,CAAC,IAAI;gCACN,CAAC,CAAC,YAAY,CAAC;4BACjB,6EAA6E;4BAC7E,oBAAoB,CAAC,YAAY,CAAC,GAAG,YAAY,CAAC;4BAClD,SAAS;wBACX,CAAC;oBACH,CAAC;oBACD,sEAAsE;gBACxE,CAAC;gBAED,MAAM,eAAe,GAAG,wBAAwB,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC;gBACrE,IAAI,CAAC,cAAc,CAAC,eAAe,CAAC;oBAAE,SAAS;gBAE/C,2DAA2D;gBAC3D,MAAM,YAAY,GAAG,2BAA2B,CAAC,eAAe,CAAC,CAAC;gBAElE,aAAa,CAAC,YAAY,CAAC,GAAG,wBAAwB,CAAC,IAAI,CAAC,CAAC;gBAC7D,cAAc,CAAC,OAAO,CAAC,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,OAAO;oBACrD,CAAC,CAAC,IAAI;oBACN,CAAC,CAAC,YAAY,CAAC;gBAEjB,2DAA2D;gBAC3D,sEAAsE;gBACtE,MAAM,eAAe,GAAG,SAAS,CAAC,CAAC,CAAC,CAAC,CAAC,6DAA6D;gBACnG,MAAM,iBAAiB,GAAG,GAAG,eAAe,IAAI,YAAY,EAAE,CAAC;gBAC/D,oBAAoB,CAAC,YAAY,CAAC,GAAG,iBAAiB,CAAC;YACzD,CAAC;QACH,CAAC;IACH,CAAC;IAED,KAAK,MAAM,SAAS,IAAI,aAAa,EAAE,CAAC;QACtC,IAAI,CAAC,cAAc,CAAC,SAAS,CAAC,EAAE,CAAC;YAC/B,OAAO,aAAa,CAAC,SAAS,CAAC,CAAC;YAChC,OAAO,oBAAoB,CAAC,SAAS,CAAC,CAAC;QACzC,CAAC;IACH,CAAC;IAED,OAAO,EAAE,aAAa,EAAE,cAAc,EAAE,oBAAoB,EAAE,CAAC;AACjE,CAAC"}
+{"version":3,"file":"gatherAttributesMap.js","sourceRoot":"","sources":["../../../../../../../packages/ai/src/lib/promptGenerators/gatherAttributesMap.ts"],"names":[],"mappings":"AACA,OAAO,gBAAgB,MAAM,qBAAqB,CAAC;AACnD,OAAO,EACL,wBAAwB,EACxB,gCAAgC,GACjC,MAAM,4BAA4B,CAAC;AAEpC;;;;GAIG;AACH,SAAS,uBAAuB,CAAC,SAAiB;IAChD,0CAA0C;IAC1C,OAAO,sBAAsB,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;AAChD,CAAC;AAED;;;GAGG;AACH,SAAS,uBAAuB,CAAC,SAAiB;IAChD,MAAM,KAAK,GAAG,SAAS,CAAC,KAAK,CAAC,wBAAwB,CAAC,CAAC;IACxD,OAAO,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC;AACjC,CAAC;AAED;;;;GAIG;AACH,SAAS,2BAA2B,CAAC,IAAY;IAC/C,MAAM,SAAS,GAAG,uBAAuB,CAAC,IAAI,CAAC,CAAC;IAChD,OAAO,SAAS,KAAK,IAAI,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC;AAC/C,CAAC;AAED,SAAS,cAAc,CAAC,SAAiB;IACvC,IAAI,CAAC,SAAS,IAAI,SAAS,CAAC,MAAM,KAAK,CAAC;QAAE,OAAO,KAAK,CAAC;IAEvD,4DAA4D;IAC5D,oEAAoE;IACpE,IAAI,SAAS,CAAC,UAAU,CAAC,GAAG,CAAC,EAAE,CAAC;QAC9B,OAAO,uBAAuB,CAAC,SAAS,CAAC,CAAC;IAC5C,CAAC;IAED,MAAM,cAAc,GAAG,gCAAgC,CAAC,SAAS,CAAC,CAAC;IACnE,IACE,cAAc,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,EAAE,CAC3B,gBAAgB,CAAC,gBAAgB,CAAC,IAAI,CAAC,EAAE,GAAG,EAAE,GAAG,CAAC,CAAC,KAAK,CAAC,IAAI,CAAC,CAC/D,EACD,CAAC;QACD,OAAO,KAAK,CAAC;IACf,CAAC;IAED,2EAA2E;IAC3E,kEAAkE;IAClE,yCAAyC;IACzC,IACE,cAAc,CAAC,MAAM,GAAG,CAAC;QACzB,cAAc,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,UAAU,CAAC,YAAY,CAAC,CAAC,EACrE,CAAC;QACD,OAAO,KAAK,CAAC;IACf,CAAC;IAED,OAAO,IAAI,CAAC;AACd,CAAC;AAED;;;;;;;;GAQG;AACH,MAAM,UAAU,mBAAmB,CACjC,qBAA0E,EAC1E,mCAEiD,EACjD,qCAEW,EACX,UAAoB;IAEpB,sEAAsE;IACtE,sEAAsE;IACtE,6FAA6F;IAC7F,IAAI,mBAGI,CAAC;IACT,IAAI,4BAA2E,CAAC;IAChF,IAAI,OAAgB,CAAC;IAErB,IACE,mCAAmC;QACnC,OAAO,mCAAmC,KAAK,QAAQ;QACvD,mBAAmB,IAAI,mCAAmC,EAC1D,CAAC;QACD,+DAA+D;QAC/D,mBAAmB,GAAG,mCAGrB,CAAC;QACF,4BAA4B;YACzB,qCAAuF;gBACxF,EAAE,CAAC;QACL,OAAO,GAAG,UAAU,IAAI,KAAK,CAAC;IAChC,CAAC;SAAM,CAAC;QACN,qDAAqD;QACrD,oFAAoF;QACpF,mBAAmB,GAAG,IAAI,CAAC,CAAC,2CAA2C;QACvE,4BAA4B;YACzB,mCAAqF;gBACtF,EAAE,CAAC;QACL,OAAO;YACJ,qCAA6D,IAAI,KAAK,CAAC;IAC5E,CAAC;IAED,MAAM,aAAa,GAA2B,EAAE,CAAC;IACjD,MAAM,cAAc,GAA2B,EAAE,CAAC;IAClD,kFAAkF;IAClF,MAAM,oBAAoB,GAA2B,EAAE,CAAC;IAExD,+FAA+F;IAC/F,MAAM,uBAAuB,GAAG,qBAAqB,CAAC,eAAe,CAAC;IACtE,MAAM,qBAAqB,GACzB,mBAAmB,EAAE,eAAe,IAAI,uBAAuB,CAAC;IAClE,MAAM,yBAAyB,GAAG,qBAAqB,CAAC,iBAAiB,CAAC;IAC1E,MAAM,uBAAuB,GAC3B,mBAAmB,EAAE,iBAAiB,IAAI,yBAAyB,CAAC;IAEtE,8EAA8E;IAC9E,KAAK,MAAM,GAAG,IAAI,uBAAuB,EAAE,CAAC;QAC1C,MAAM,QAAQ,GAAG,gCAAgC,CAAC,GAAG,CAAC,CAAC;QACvD,IAAI,QAAQ,CAAC,CAAC,CAAC,CAAC,UAAU,CAAC,WAAW,CAAC,EAAE,CAAC;YACxC,qEAAqE;YACrE,MAAM,SAAS,GACb,qBAAqB,CAAC,GAAG,CAAC,IAAI,uBAAuB,CAAC,GAAG,CAAC,CAAC;YAC7D,MAAM,2BAA2B,GAC/B,4BAA4B,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC;YAC5C,MAAM,gCAAgC,GAAG,gCAAgC,CACvE,2BAA2B,CAC5B,CAAC;YACF,IACE,2BAA2B;gBAC3B,CAAC,gCAAgC,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,GAAG,CAAC,EAClD,CAAC;gBACD,MAAM,aAAa,GAAG,4BAA4B,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC;gBAChE,MAAM,cAAc,GAAG,wBAAwB,CAAC;oBAC9C,aAAa;oBACb,GAAG,QAAQ,CAAC,KAAK,CAAC,CAAC,CAAC;iBACrB,CAAC,CAAC;gBACH,aAAa,CAAC,cAAc,CAAC,GAAG,SAAS,CAAC;gBAC1C,cAAc,CAAC,OAAO,CAAC,CAAC,CAAC,cAAc,CAAC,CAAC,CAAC,GAAG,CAAC,GAAG,OAAO;oBACtD,CAAC,CAAC,GAAG;oBACL,CAAC,CAAC,cAAc,CAAC;gBACnB,+EAA+E;gBAC/E,oBAAoB,CAAC,cAAc,CAAC,GAAG,cAAc,CAAC;YACxD,CAAC;iBAAM,CAAC;gBACN,MAAM,YAAY,GAAG,wBAAwB,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC;gBACjE,aAAa,CAAC,YAAY,CAAC,GAAG,SAAS,CAAC;gBACxC,cAAc,CAAC,OAAO,CAAC,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,GAAG,CAAC,GAAG,OAAO;oBACpD,CAAC,CAAC,GAAG;oBACL,CAAC,CAAC,YAAY,CAAC;gBACjB,+EAA+E;gBAC/E,oBAAoB,CAAC,YAAY,CAAC,GAAG,YAAY,CAAC;YACpD,CAAC;QACH,CAAC;IACH,CAAC;IAED,sGAAsG;IACtG,sEAAsE;IACtE,KAAK,MAAM,QAAQ,IAAI,yBAAyB,EAAE,CAAC;QACjD,KAAK,MAAM,UAAU,IAAI,yBAAyB,CAAC,QAAQ,CAAC,EAAE,CAAC;YAC7D,sEAAsE;YACtE,MAAM,yBAAyB,GAC7B,yBAAyB,CAAC,QAAQ,CAAC,CAAC,UAAU,CAAC,CAAC,iBAAiB,IAAI,EAAE,CAAC;YAE1E,wDAAwD;YACxD,MAAM,uBAAuB,GAC3B,uBAAuB,EAAE,CAAC,QAAQ,CAAC,EAAE,CAAC,UAAU,CAAC,EAAE,iBAAiB;gBACpE,yBAAyB,CAAC;YAE5B,mFAAmF;YACnF,MAAM,aAAa,GAAa,EAAE,CAAC;YACnC,KAAK,MAAM,IAAI,IAAI,uBAAuB,EAAE,CAAC;gBAC3C,IAAI,uBAAuB,CAAC,IAAI,CAAC,KAAK,UAAU,EAAE,CAAC;oBACjD,aAAa,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;gBAC3B,CAAC;YACH,CAAC;YAED,6EAA6E;YAC7E,2FAA2F;YAC3F,KAAK,MAAM,IAAI,IAAI,uBAAuB,EAAE,CAAC;gBAC3C,MAAM,SAAS,GAAG,gCAAgC,CAAC,IAAI,CAAC,CAAC;gBACzD,IAAI,SAAS,CAAC,MAAM,GAAG,CAAC;oBAAE,SAAS;gBAEnC,sDAAsD;gBACtD,wEAAwE;gBACxE,kEAAkE;gBAClE,IAAI,SAAS,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,UAAU,CAAC,YAAY,CAAC,CAAC,EAAE,CAAC;oBAC5D,SAAS;gBACX,CAAC;gBAED,IAAI,uBAAuB,CAAC,IAAI,CAAC,KAAK,UAAU,EAAE,CAAC;oBACjD,SAAS;gBACX,CAAC;gBAED,0FAA0F;gBAC1F,0DAA0D;gBAC1D,MAAM,oBAAoB,GAAG,aAAa;qBACvC,MAAM,CAAC,CAAC,YAAY,EAAE,EAAE,CAAC,IAAI,CAAC,UAAU,CAAC,YAAY,CAAC,CAAC;qBACvD,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,MAAM,GAAG,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC;gBAC1C,IAAI,oBAAoB,EAAE,CAAC;oBACzB,6DAA6D;oBAC7D,MAAM,aAAa,GAAG,IAAI,CAAC,SAAS,CAAC,oBAAoB,CAAC,MAAM,CAAC,CAAC;oBAElE,IAAI,CAAC,aAAa,CAAC,UAAU,CAAC,0BAA0B,CAAC,EAAE,CAAC;wBAC1D,6EAA6E;wBAC7E,UAAU;wBACV,SAAS;oBACX,CAAC;oBAED,gEAAgE;oBAChE,MAAM,aAAa,GAAG,IAAI,CAAC,QAAQ,CAAC,0BAA0B,CAAC,CAAC;oBAChE,IAAI,aAAa,EAAE,CAAC;wBAClB,sFAAsF;wBACtF,uEAAuE;wBACvE,MAAM,iBAAiB,GACrB,gCAAgC,CAAC,oBAAoB,CAAC,CAAC;wBACzD,MAAM,gBAAgB,GACpB,iBAAiB,CAAC,MAAM,GAAG,CAAC;4BAC1B,CAAC,CAAC,iBAAiB,CAAC,iBAAiB,CAAC,MAAM,GAAG,CAAC,CAAC;4BACjD,CAAC,CAAC,IAAI,CAAC;wBACX,MAAM,qBAAqB,GACzB,iBAAiB,CAAC,MAAM,GAAG,CAAC;4BAC1B,CAAC,CAAC,iBAAiB,CAAC,iBAAiB,CAAC,MAAM,GAAG,CAAC,CAAC;4BACjD,CAAC,CAAC,IAAI,CAAC;wBAEX,gFAAgF;wBAChF,yCAAyC;wBACzC,MAAM,sBAAsB,GAC1B,gBAAgB;4BAChB,gBAAgB,CAAC,UAAU,CAAC,GAAG,CAAC;4BAChC,qBAAqB;4BACrB,qBAAqB,KAAK,yBAAyB,CAAC;wBAEtD,IAAI,sBAAsB,EAAE,CAAC;4BAC3B,wDAAwD;4BACxD,MAAM,YAAY,GAAG,IAAI,CAAC;4BAC1B,IAAI,CAAC,cAAc,CAAC,YAAY,CAAC;gCAAE,SAAS;4BAE5C,aAAa,CAAC,YAAY,CAAC,GAAG,uBAAuB,CAAC,IAAI,CAAC,CAAC;4BAC5D,cAAc,CAAC,OAAO,CAAC,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,OAAO;gCACrD,CAAC,CAAC,IAAI;gCACN,CAAC,CAAC,YAAY,CAAC;4BACjB,6EAA6E;4BAC7E,oBAAoB,CAAC,YAAY,CAAC,GAAG,YAAY,CAAC;4BAClD,SAAS;wBACX,CAAC;oBACH,CAAC;oBACD,sEAAsE;gBACxE,CAAC;gBAED,MAAM,eAAe,GAAG,wBAAwB,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC;gBACrE,IAAI,CAAC,cAAc,CAAC,eAAe,CAAC;oBAAE,SAAS;gBAE/C,2DAA2D;gBAC3D,MAAM,YAAY,GAAG,2BAA2B,CAAC,eAAe,CAAC,CAAC;gBAElE,4CAA4C;gBAC5C,aAAa,CAAC,YAAY,CAAC,GAAG,uBAAuB,CAAC,IAAI,CAAC,CAAC;gBAC5D,cAAc,CAAC,OAAO,CAAC,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,OAAO;oBACrD,CAAC,CAAC,IAAI;oBACN,CAAC,CAAC,YAAY,CAAC;gBAEjB,2DAA2D;gBAC3D,sEAAsE;gBACtE,MAAM,eAAe,GAAG,SAAS,CAAC,CAAC,CAAC,CAAC,CAAC,6DAA6D;gBACnG,MAAM,iBAAiB,GAAG,GAAG,eAAe,IAAI,YAAY,EAAE,CAAC;gBAC/D,oBAAoB,CAAC,YAAY,CAAC,GAAG,iBAAiB,CAAC;YACzD,CAAC;QACH,CAAC;IACH,CAAC;IAED,KAAK,MAAM,SAAS,IAAI,aAAa,EAAE,CAAC;QACtC,IAAI,CAAC,cAAc,CAAC,SAAS,CAAC,EAAE,CAAC;YAC/B,OAAO,aAAa,CAAC,SAAS,CAAC,CAAC;YAChC,OAAO,oBAAoB,CAAC,SAAS,CAAC,CAAC;QACzC,CAAC;IACH,CAAC;IAED,OAAO,EAAE,aAAa,EAAE,cAAc,EAAE,oBAAoB,EAAE,CAAC;AACjE,CAAC"}

package/packages/ai/src/lib/promptGenerators/generateChangesEntityScenarioDataGenerator.js

@@ -53,7 +53,7 @@ ${incompleteResponse}
     const scenarioJson = JSON.stringify({
         name: scenario.name,
         description: scenario.description,
-        keyAttributeInstructions: scenario.metadata.keyAttributeInstructions,
+        coveredFlows: scenario.metadata?.coveredFlows || [],
     }, null, 2);
     const existingJson = JSON.stringify({
         name: existingScenario.name,

package/packages/ai/src/lib/promptGenerators/generateChangesEntityScenarioDataGenerator.js.map

@@ -1 +1 @@
-{"version":3,"file":"generateChangesEntityScenarioDataGenerator.js","sourceRoot":"","sources":["../../../../../../../packages/ai/src/lib/promptGenerators/generateChangesEntityScenarioDataGenerator.ts"],"names":[],"mappings":"AAEA,SAAS,UAAU,CACjB,aAAqC,EACrC,QAA8B;IAE9B,IAAI,CAAC,aAAa,IAAI,CAAC,QAAQ;QAAE,OAAO;IAExC,MAAM,WAAW,GAAG,CAAC,SAAc,EAAE,IAAS,EAAO,EAAE;QACrD,IAAI,CAAC,SAAS,IAAI,CAAC,IAAI;YAAE,OAAO,IAAI,CAAC;QAErC,IAAI,OAAO,SAAS,KAAK,QAAQ,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,SAAS,CAAC,EAAE,CAAC;YAC/D,MAAM,YAAY,GAAQ,EAAE,CAAC;YAC7B,KAAK,MAAM,GAAG,IAAI,SAAS,EAAE,CAAC;gBAC5B,IAAI,GAAG,KAAK,kBAAkB,IAAI,GAAG,KAAK,iBAAiB,EAAE,CAAC;oBAC5D,YAAY,CAAC,GAAG,CAAC,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC;gBAChC,CAAC;qBAAM,IAAI,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,EAAE,CAAC;oBACpC,IAAI,OAAO,SAAS,CAAC,GAAG,CAAC,KAAK,QAAQ,EAAE,CAAC;wBACvC,YAAY,CAAC,GAAG,CAAC,GAAG,WAAW,CAAC,SAAS,CAAC,GAAG,CAAC,EAAE,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC;oBAC7D,CAAC;yBAAM,CAAC;wBACN,YAAY,CAAC,GAAG,CAAC,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC;oBAChC,CAAC;gBACH,CAAC;YACH,CAAC;YACD,OAAO,YAAY,CAAC;QACtB,CAAC;aAAM,IAAI,KAAK,CAAC,OAAO,CAAC,SAAS,CAAC,EAAE,CAAC;YACpC,MAAM,aAAa,GAAG,EAAE,CAAC;YACzB,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,IAAI,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;gBACrC,IAAI,OAAO,IAAI,CAAC,CAAC,CAAC,KAAK,QAAQ,EAAE,CAAC;oBAChC,aAAa,CAAC,IAAI,CAChB,WAAW,CAAC,SAAS,CAAC,CAAC,CAAC,IAAI,SAAS,CAAC,CAAC,CAAC,EAAE,IAAI,CAAC,CAAC,CAAC,CAAC,CACnD,CAAC;gBACJ,CAAC;qBAAM,CAAC;oBACN,aAAa,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC;gBAC9B,CAAC;YACH,CAAC;YACD,OAAO,aAAa,CAAC;QACvB,CAAC;QACD,OAAO,IAAI,CAAC;IACd,CAAC,CAAC;IAEF,OAAO;QACL,SAAS,EAAE,WAAW,CAAC,aAAa,CAAC,SAAS,EAAE,QAAQ,CAAC,aAAa,CAAC;QACvE,YAAY,EAAE,WAAW,CAAC,aAAa,CAAC,YAAY,EAAE,QAAQ,CAAC,QAAQ,CAAC;KACzE,CAAC;AACJ,CAAC;AAED,MAAM,CAAC,OAAO,UAAU,0CAA0C,CAChE,SAAiC,EACjC,QAAkB,EAClB,gBAA0B,EAC1B,mBAAkC,EAClC,kBAA2B;IAE3B,MAAM,iBAAiB,GAAG,kBAAkB;QAC1C,CAAC,CAAC;;;EAGJ,kBAAkB;;;CAGnB;QACG,CAAC,CAAC,EAAE,CAAC;IAEP,MAAM,YAAY,GAAG,IAAI,CAAC,SAAS,CACjC;QACE,IAAI,EAAE,QAAQ,CAAC,IAAI;QACnB,WAAW,EAAE,QAAQ,CAAC,WAAW;QACjC,wBAAwB,EAAE,QAAQ,CAAC,QAAQ,CAAC,wBAAwB;KACrE,EACD,IAAI,EACJ,CAAC,CACF,CAAC;IAEF,MAAM,YAAY,GAAG,IAAI,CAAC,SAAS,CACjC;QACE,IAAI,EAAE,gBAAgB,CAAC,IAAI;QAC3B,IAAI,EAAE,UAAU,CAAC,SAAS,EAAE,gBAAgB,CAAC,QAAQ,CAAC,IAAI,CAAC;KAC5D,EACD,IAAI,EACJ,CAAC,CACF,CAAC;IAEF,MAAM,cAAc,GAAG,mBAAmB;QACxC,CAAC,CAAC;;;EAGJ,IAAI,CAAC,SAAS,CAAC,mBAAmB,CAAC,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC;;CAElD;QACG,CAAC,CAAC,EAAE,CAAC;IAEP,MAAM,eAAe,GAAG,SAAS,CAAC,YAAY;QAC5C,CAAC,CAAC;;EAEJ,IAAI,CAAC,SAAS,CAAC,SAAS,CAAC,YAAY,EAAE,IAAI,EAAE,CAAC,CAAC;OAC1C;QACH,CAAC,CAAC,uDAAuD,CAAC;IAE5D,MAAM,gBAAgB,GAAG,SAAS,CAAC,SAAS;QAC1C,CAAC,CAAC;;EAEJ,IAAI,CAAC,SAAS,CAAC,SAAS,CAAC,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;OACvC;QACH,CAAC,CAAC,iEAAiE,CAAC;IAEtE,OAAO,GAAG,iBAAiB;;EAE3B,YAAY;;;;;EAKZ,YAAY;;EAEZ,cAAc;EACd,eAAe;;EAEf,gBAAgB,EAAE,CAAC;AACrB,CAAC"}
+{"version":3,"file":"generateChangesEntityScenarioDataGenerator.js","sourceRoot":"","sources":["../../../../../../../packages/ai/src/lib/promptGenerators/generateChangesEntityScenarioDataGenerator.ts"],"names":[],"mappings":"AAEA,SAAS,UAAU,CACjB,aAAqC,EACrC,QAA8B;IAE9B,IAAI,CAAC,aAAa,IAAI,CAAC,QAAQ;QAAE,OAAO;IAExC,MAAM,WAAW,GAAG,CAAC,SAAc,EAAE,IAAS,EAAO,EAAE;QACrD,IAAI,CAAC,SAAS,IAAI,CAAC,IAAI;YAAE,OAAO,IAAI,CAAC;QAErC,IAAI,OAAO,SAAS,KAAK,QAAQ,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,SAAS,CAAC,EAAE,CAAC;YAC/D,MAAM,YAAY,GAAQ,EAAE,CAAC;YAC7B,KAAK,MAAM,GAAG,IAAI,SAAS,EAAE,CAAC;gBAC5B,IAAI,GAAG,KAAK,kBAAkB,IAAI,GAAG,KAAK,iBAAiB,EAAE,CAAC;oBAC5D,YAAY,CAAC,GAAG,CAAC,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC;gBAChC,CAAC;qBAAM,IAAI,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,EAAE,CAAC;oBACpC,IAAI,OAAO,SAAS,CAAC,GAAG,CAAC,KAAK,QAAQ,EAAE,CAAC;wBACvC,YAAY,CAAC,GAAG,CAAC,GAAG,WAAW,CAAC,SAAS,CAAC,GAAG,CAAC,EAAE,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC;oBAC7D,CAAC;yBAAM,CAAC;wBACN,YAAY,CAAC,GAAG,CAAC,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC;oBAChC,CAAC;gBACH,CAAC;YACH,CAAC;YACD,OAAO,YAAY,CAAC;QACtB,CAAC;aAAM,IAAI,KAAK,CAAC,OAAO,CAAC,SAAS,CAAC,EAAE,CAAC;YACpC,MAAM,aAAa,GAAG,EAAE,CAAC;YACzB,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,IAAI,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;gBACrC,IAAI,OAAO,IAAI,CAAC,CAAC,CAAC,KAAK,QAAQ,EAAE,CAAC;oBAChC,aAAa,CAAC,IAAI,CAChB,WAAW,CAAC,SAAS,CAAC,CAAC,CAAC,IAAI,SAAS,CAAC,CAAC,CAAC,EAAE,IAAI,CAAC,CAAC,CAAC,CAAC,CACnD,CAAC;gBACJ,CAAC;qBAAM,CAAC;oBACN,aAAa,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC;gBAC9B,CAAC;YACH,CAAC;YACD,OAAO,aAAa,CAAC;QACvB,CAAC;QACD,OAAO,IAAI,CAAC;IACd,CAAC,CAAC;IAEF,OAAO;QACL,SAAS,EAAE,WAAW,CAAC,aAAa,CAAC,SAAS,EAAE,QAAQ,CAAC,aAAa,CAAC;QACvE,YAAY,EAAE,WAAW,CAAC,aAAa,CAAC,YAAY,EAAE,QAAQ,CAAC,QAAQ,CAAC;KACzE,CAAC;AACJ,CAAC;AAED,MAAM,CAAC,OAAO,UAAU,0CAA0C,CAChE,SAAiC,EACjC,QAAkB,EAClB,gBAA0B,EAC1B,mBAAkC,EAClC,kBAA2B;IAE3B,MAAM,iBAAiB,GAAG,kBAAkB;QAC1C,CAAC,CAAC;;;EAGJ,kBAAkB;;;CAGnB;QACG,CAAC,CAAC,EAAE,CAAC;IAEP,MAAM,YAAY,GAAG,IAAI,CAAC,SAAS,CACjC;QACE,IAAI,EAAE,QAAQ,CAAC,IAAI;QACnB,WAAW,EAAE,QAAQ,CAAC,WAAW;QACjC,YAAY,EAAE,QAAQ,CAAC,QAAQ,EAAE,YAAY,IAAI,EAAE;KACpD,EACD,IAAI,EACJ,CAAC,CACF,CAAC;IAEF,MAAM,YAAY,GAAG,IAAI,CAAC,SAAS,CACjC;QACE,IAAI,EAAE,gBAAgB,CAAC,IAAI;QAC3B,IAAI,EAAE,UAAU,CAAC,SAAS,EAAE,gBAAgB,CAAC,QAAQ,CAAC,IAAI,CAAC;KAC5D,EACD,IAAI,EACJ,CAAC,CACF,CAAC;IAEF,MAAM,cAAc,GAAG,mBAAmB;QACxC,CAAC,CAAC;;;EAGJ,IAAI,CAAC,SAAS,CAAC,mBAAmB,CAAC,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC;;CAElD;QACG,CAAC,CAAC,EAAE,CAAC;IAEP,MAAM,eAAe,GAAG,SAAS,CAAC,YAAY;QAC5C,CAAC,CAAC;;EAEJ,IAAI,CAAC,SAAS,CAAC,SAAS,CAAC,YAAY,EAAE,IAAI,EAAE,CAAC,CAAC;OAC1C;QACH,CAAC,CAAC,uDAAuD,CAAC;IAE5D,MAAM,gBAAgB,GAAG,SAAS,CAAC,SAAS;QAC1C,CAAC,CAAC;;EAEJ,IAAI,CAAC,SAAS,CAAC,SAAS,CAAC,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;OACvC;QACH,CAAC,CAAC,iEAAiE,CAAC;IAEtE,OAAO,GAAG,iBAAiB;;EAE3B,YAAY;;;;;EAKZ,YAAY;;EAEZ,cAAc;EACd,eAAe;;EAEf,gBAAgB,EAAE,CAAC;AACrB,CAAC"}

package/packages/ai/src/lib/promptGenerators/generateChangesEntityScenariosGenerator.js

@@ -1,32 +1,9 @@
-import gatherRelevantDependentKeyAttributes from "../gatherRelevantDependentKeyAttributes.js";
-export default function generateChangesEntityScenariosGenerator({ entity, keyAttributes, addedKeyAttributes, removedKeyAttributes, dependentAnalyses, existingScenarios, commitDiff, error, changedDataStructureFields = [], }) {
-    const relevantDependentKeyAttributesFlat = gatherRelevantDependentKeyAttributes(entity, dependentAnalyses);
-    // Restructure flat array into nested format: { [filePath]: { [entityName]: KeyAttribute[] } }
-    const relevantDependentKeyAttributes = {};
-    for (const attr of relevantDependentKeyAttributesFlat) {
-        const filePath = attr.filePath;
-        const entityName = attr.originalEntityName;
-        if (!filePath || !entityName)
-            continue;
-        if (!relevantDependentKeyAttributes[filePath]) {
-            relevantDependentKeyAttributes[filePath] = {};
-        }
-        if (!relevantDependentKeyAttributes[filePath][entityName]) {
-            relevantDependentKeyAttributes[filePath][entityName] = [];
-        }
-        relevantDependentKeyAttributes[filePath][entityName].push(attr);
-    }
-    const formatKeyAttrs = (attrs) => attrs.map((ka) => ({
-        dataStructurePath: ka.dataStructurePath,
-        localVariable: ka.internalPath,
-        description: ka.description,
-        validValueOptions: ka.validValueOptions,
-    }));
-    const addedSection = addedKeyAttributes?.length > 0
-        ? JSON.stringify(formatKeyAttrs(addedKeyAttributes), null, 2)
+export default function generateChangesEntityScenariosGenerator({ entity, executionFlows, addedFlows, removedFlows, existingScenarios, commitDiff, error, changedDataStructureFields = [], }) {
+    const addedSection = addedFlows && addedFlows.length > 0
+        ? JSON.stringify(addedFlows, null, 2)
         : 'None';
-    const removedSection = removedKeyAttributes?.length > 0
-        ? JSON.stringify(removedKeyAttributes, null, 2)
+    const removedSection = removedFlows && removedFlows.length > 0
+        ? JSON.stringify(removedFlows.map((f) => ({ id: f.id, name: f.name })), null, 2)
         : 'None';
     const changedFieldsSection = changedDataStructureFields.length > 0
         ? `
@@ -36,16 +13,16 @@ Fields that changed type, were added, or removed:
 ${JSON.stringify(changedDataStructureFields, null, 2)}
 \`\`\`
 
-**Update keyAttributeInstructions for affected scenarios** (e.g., if "stats.visualEntities" changed from array to number).
+**Update scenarios if their coveredFlows depend on these changed fields.**
 `
         : '';
     const existingScenariosJson = JSON.stringify(existingScenarios
-        .filter((s) => (error ? !!s.metadata.error : !s.metadata.error))
+        .filter((s) => (error ? !!s.metadata?.error : !s.metadata?.error))
         .map((s) => ({
         name: s.name,
         description: s.description,
-        keyAttributeInstructions: s.metadata.keyAttributeInstructions,
-        playwrightInstructions: s.metadata.playwrightInstructions,
+        coveredFlows: s.metadata?.coveredFlows || [],
+        playwrightInstructions: s.metadata?.playwrightInstructions,
     })), null, 2);
     const diffSection = commitDiff
         ? `
@@ -54,41 +31,27 @@ ${JSON.stringify(changedDataStructureFields, null, 2)}
 ${commitDiff}
 \`\`\``
         : '';
-    const dependencySection = !error &&
-        relevantDependentKeyAttributes &&
-        Object.keys(relevantDependentKeyAttributes).length > 0
-        ? `
-## Dependency Key Attributes
+    const executionFlowsSection = executionFlows && executionFlows.length > 0
+        ? `## Current Execution Flows
+All distinct outcomes/behaviors this component can currently produce:
 \`\`\`json
-${JSON.stringify(Object.keys(relevantDependentKeyAttributes).reduce((acc, path) => {
-            acc[path] = Object.keys(relevantDependentKeyAttributes[path]).reduce((entityAcc, entityName) => {
-                entityAcc[entityName] = relevantDependentKeyAttributes[path][entityName].map((ka) => ({
-                    dataStructurePath: ka.dataStructurePath,
-                    localVariable: ka.path || ka.internalPath,
-                    description: ka.description,
-                    validValueOptions: ka.validValueOptions,
-                }));
-                return entityAcc;
-            }, {});
-            return acc;
-        }, {}), null, 2)}
-\`\`\``
+${JSON.stringify(executionFlows, null, 2)}
+\`\`\`
+`
         : '';
-    return `## Added Key Attributes
+    return `## Added Execution Flows
+New flows that need scenario coverage:
 \`\`\`json
 ${addedSection}
 \`\`\`
 
-## Removed Key Attributes
+## Removed Execution Flows
+Flows that no longer exist (remove from scenarios):
 \`\`\`json
 ${removedSection}
 \`\`\`
 ${changedFieldsSection}
-## Current Key Attributes
-\`\`\`json
-${JSON.stringify(formatKeyAttrs(keyAttributes), null, 2)}
-\`\`\`
-
+${executionFlowsSection}
 ## Existing Scenarios
 \`\`\`json
 ${existingScenariosJson}
@@ -98,6 +61,6 @@ ${diffSection}
 ## Current Code
 \`\`\`typescript
 ${entity.code}
-\`\`\`${dependencySection}`;
+\`\`\``;
 }
 //# sourceMappingURL=generateChangesEntityScenariosGenerator.js.map

package/packages/ai/src/lib/promptGenerators/generateChangesEntityScenariosGenerator.js.map

@@ -1 +1 @@
-{"version":3,"file":"generateChangesEntityScenariosGenerator.js","sourceRoot":"","sources":["../../../../../../../packages/ai/src/lib/promptGenerators/generateChangesEntityScenariosGenerator.ts"],"names":[],"mappings":"AAMA,OAAO,oCAAoC,MAAM,yCAAyC,CAAC;AAc3F,MAAM,CAAC,OAAO,UAAU,uCAAuC,CAAC,EAC9D,MAAM,EACN,aAAa,EACb,kBAAkB,EAClB,oBAAoB,EACpB,iBAAiB,EACjB,iBAAiB,EACjB,UAAU,EACV,KAAK,EACL,0BAA0B,GAAG,EAAE,GACa;IAC5C,MAAM,kCAAkC,GACtC,oCAAoC,CAAC,MAAM,EAAE,iBAAiB,CAAC,CAAC;IAElE,8FAA8F;IAC9F,MAAM,8BAA8B,GAGhC,EAAE,CAAC;IACP,KAAK,MAAM,IAAI,IAAI,kCAAkC,EAAE,CAAC;QACtD,MAAM,QAAQ,GAAI,IAAY,CAAC,QAAQ,CAAC;QACxC,MAAM,UAAU,GAAI,IAAY,CAAC,kBAAkB,CAAC;QAEpD,IAAI,CAAC,QAAQ,IAAI,CAAC,UAAU;YAAE,SAAS;QAEvC,IAAI,CAAC,8BAA8B,CAAC,QAAQ,CAAC,EAAE,CAAC;YAC9C,8BAA8B,CAAC,QAAQ,CAAC,GAAG,EAAE,CAAC;QAChD,CAAC;QACD,IAAI,CAAC,8BAA8B,CAAC,QAAQ,CAAC,CAAC,UAAU,CAAC,EAAE,CAAC;YAC1D,8BAA8B,CAAC,QAAQ,CAAC,CAAC,UAAU,CAAC,GAAG,EAAE,CAAC;QAC5D,CAAC;QACD,8BAA8B,CAAC,QAAQ,CAAC,CAAC,UAAU,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IAClE,CAAC;IAED,MAAM,cAAc,GAAG,CAAC,KAA4C,EAAE,EAAE,CACtE,KAAK,CAAC,GAAG,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC;QACjB,iBAAiB,EAAE,EAAE,CAAC,iBAAiB;QACvC,aAAa,EAAE,EAAE,CAAC,YAAY;QAC9B,WAAW,EAAE,EAAE,CAAC,WAAW;QAC3B,iBAAiB,EAAE,EAAE,CAAC,iBAAiB;KACxC,CAAC,CAAC,CAAC;IAEN,MAAM,YAAY,GAChB,kBAAkB,EAAE,MAAM,GAAG,CAAC;QAC5B,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,cAAc,CAAC,kBAAkB,CAAC,EAAE,IAAI,EAAE,CAAC,CAAC;QAC7D,CAAC,CAAC,MAAM,CAAC;IAEb,MAAM,cAAc,GAClB,oBAAoB,EAAE,MAAM,GAAG,CAAC;QAC9B,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,oBAAoB,EAAE,IAAI,EAAE,CAAC,CAAC;QAC/C,CAAC,CAAC,MAAM,CAAC;IAEb,MAAM,oBAAoB,GACxB,0BAA0B,CAAC,MAAM,GAAG,CAAC;QACnC,CAAC,CAAC;;;;EAIN,IAAI,CAAC,SAAS,CAAC,0BAA0B,EAAE,IAAI,EAAE,CAAC,CAAC;;;;CAIpD;QACK,CAAC,CAAC,EAAE,CAAC;IAET,MAAM,qBAAqB,GAAG,IAAI,CAAC,SAAS,CAC1C,iBAAiB;SACd,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC;SAC/D,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;QACX,IAAI,EAAE,CAAC,CAAC,IAAI;QACZ,WAAW,EAAE,CAAC,CAAC,WAAW;QAC1B,wBAAwB,EAAE,CAAC,CAAC,QAAQ,CAAC,wBAAwB;QAC7D,sBAAsB,EAAE,CAAC,CAAC,QAAQ,CAAC,sBAAsB;KAC1D,CAAC,CAAC,EACL,IAAI,EACJ,CAAC,CACF,CAAC;IAEF,MAAM,WAAW,GAAG,UAAU;QAC5B,CAAC,CAAC;;;EAGJ,UAAU;OACL;QACH,CAAC,CAAC,EAAE,CAAC;IAEP,MAAM,iBAAiB,GACrB,CAAC,KAAK;QACN,8BAA8B;QAC9B,MAAM,CAAC,IAAI,CAAC,8BAA8B,CAAC,CAAC,MAAM,GAAG,CAAC;QACpD,CAAC,CAAC;;;EAGN,IAAI,CAAC,SAAS,CACd,MAAM,CAAC,IAAI,CAAC,8BAA8B,CAAC,CAAC,MAAM,CAAC,CAAC,GAAQ,EAAE,IAAI,EAAE,EAAE;YACpE,GAAG,CAAC,IAAI,CAAC,GAAG,MAAM,CAAC,IAAI,CAAC,8BAA8B,CAAC,IAAI,CAAC,CAAC,CAAC,MAAM,CAClE,CAAC,SAAc,EAAE,UAAU,EAAE,EAAE;gBAC7B,SAAS,CAAC,UAAU,CAAC,GAAG,8BAA8B,CAAC,IAAI,CAAC,CAC1D,UAAU,CACX,CAAC,GAAG,CAAC,CAAC,EAAO,EAAE,EAAE,CAAC,CAAC;oBAClB,iBAAiB,EAAE,EAAE,CAAC,iBAAiB;oBACvC,aAAa,EAAE,EAAE,CAAC,IAAI,IAAI,EAAE,CAAC,YAAY;oBACzC,WAAW,EAAE,EAAE,CAAC,WAAW;oBAC3B,iBAAiB,EAAE,EAAE,CAAC,iBAAiB;iBACxC,CAAC,CAAC,CAAC;gBACJ,OAAO,SAAS,CAAC;YACnB,CAAC,EACD,EAAE,CACH,CAAC;YACF,OAAO,GAAG,CAAC;QACb,CAAC,EAAE,EAAE,CAAC,EACN,IAAI,EACJ,CAAC,CACF;OACM;QACD,CAAC,CAAC,EAAE,CAAC;IAET,OAAO;;EAEP,YAAY;;;;;EAKZ,cAAc;;EAEd,oBAAoB;;;EAGpB,IAAI,CAAC,SAAS,CAAC,cAAc,CAAC,aAAa,CAAC,EAAE,IAAI,EAAE,CAAC,CAAC;;;;;EAKtD,qBAAqB;;EAErB,WAAW;;;;EAIX,MAAM,CAAC,IAAI;QACL,iBAAiB,EAAE,CAAC;AAC5B,CAAC"}
+{"version":3,"file":"generateChangesEntityScenariosGenerator.js","sourceRoot":"","sources":["../../../../../../../packages/ai/src/lib/promptGenerators/generateChangesEntityScenariosGenerator.ts"],"names":[],"mappings":"AAcA,MAAM,CAAC,OAAO,UAAU,uCAAuC,CAAC,EAC9D,MAAM,EACN,cAAc,EACd,UAAU,EACV,YAAY,EACZ,iBAAiB,EACjB,UAAU,EACV,KAAK,EACL,0BAA0B,GAAG,EAAE,GACa;IAC5C,MAAM,YAAY,GAChB,UAAU,IAAI,UAAU,CAAC,MAAM,GAAG,CAAC;QACjC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,UAAU,EAAE,IAAI,EAAE,CAAC,CAAC;QACrC,CAAC,CAAC,MAAM,CAAC;IAEb,MAAM,cAAc,GAClB,YAAY,IAAI,YAAY,CAAC,MAAM,GAAG,CAAC;QACrC,CAAC,CAAC,IAAI,CAAC,SAAS,CACZ,YAAY,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC,EAAE,EAAE,IAAI,EAAE,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC,EACrD,IAAI,EACJ,CAAC,CACF;QACH,CAAC,CAAC,MAAM,CAAC;IAEb,MAAM,oBAAoB,GACxB,0BAA0B,CAAC,MAAM,GAAG,CAAC;QACnC,CAAC,CAAC;;;;EAIN,IAAI,CAAC,SAAS,CAAC,0BAA0B,EAAE,IAAI,EAAE,CAAC,CAAC;;;;CAIpD;QACK,CAAC,CAAC,EAAE,CAAC;IAET,MAAM,qBAAqB,GAAG,IAAI,CAAC,SAAS,CAC1C,iBAAiB;SACd,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,QAAQ,EAAE,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,QAAQ,EAAE,KAAK,CAAC,CAAC;SACjE,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;QACX,IAAI,EAAE,CAAC,CAAC,IAAI;QACZ,WAAW,EAAE,CAAC,CAAC,WAAW;QAC1B,YAAY,EAAE,CAAC,CAAC,QAAQ,EAAE,YAAY,IAAI,EAAE;QAC5C,sBAAsB,EAAE,CAAC,CAAC,QAAQ,EAAE,sBAAsB;KAC3D,CAAC,CAAC,EACL,IAAI,EACJ,CAAC,CACF,CAAC;IAEF,MAAM,WAAW,GAAG,UAAU;QAC5B,CAAC,CAAC;;;EAGJ,UAAU;OACL;QACH,CAAC,CAAC,EAAE,CAAC;IAEP,MAAM,qBAAqB,GACzB,cAAc,IAAI,cAAc,CAAC,MAAM,GAAG,CAAC;QACzC,CAAC,CAAC;;;EAGN,IAAI,CAAC,SAAS,CAAC,cAAc,EAAE,IAAI,EAAE,CAAC,CAAC;;CAExC;QACK,CAAC,CAAC,EAAE,CAAC;IAET,OAAO;;;EAGP,YAAY;;;;;;EAMZ,cAAc;;EAEd,oBAAoB;EACpB,qBAAqB;;;EAGrB,qBAAqB;;EAErB,WAAW;;;;EAIX,MAAM,CAAC,IAAI;OACN,CAAC;AACR,CAAC"}

package/packages/ai/src/lib/promptGenerators/generateChunkPrompt.js

@@ -0,0 +1,54 @@
+import simplifyKeysForLLM from "./simplifyKeysForLLM.js";
+import noErrorAttributes from "./noErrorAttributes.js";
+/**
+ * Generate a prompt for a focused LLM call to generate mock data for a single chunk.
+ *
+ * Large data structures overwhelm LLMs, causing them to make mistakes on some keys.
+ * By processing smaller chunks, each key gets more focused attention.
+ */
+export default function generateChunkPrompt({ scenario, chunk, chunkIndex, totalChunks, relevantRequiredValues, }) {
+    // Clean up the schema for LLM consumption
+    const cleanedSchema = simplifyKeysForLLM(noErrorAttributes(chunk));
+    const keysInChunk = Object.keys(chunk || {});
+    // Build requirements section if there are relevant requiredValues
+    const requirementsSection = relevantRequiredValues.length > 0
+        ? `## Execution Flow Requirements for These Keys
+These specific values MUST be set to produce the correct scenario behavior:
+\`\`\`json
+${JSON.stringify(relevantRequiredValues, null, 2)}
+\`\`\`
+
+`
+        : '';
+    return `Generate mock data for chunk ${chunkIndex + 1} of ${totalChunks}.
+
+## Scenario
+\`\`\`json
+${JSON.stringify({ name: scenario.name, description: scenario.description }, null, 2)}
+\`\`\`
+
+${requirementsSection}## Keys in This Chunk (${keysInChunk.length} keys)
+Generate data for ONLY these keys:
+${keysInChunk.map((k) => `- \`${k}\``).join('\n')}
+
+## Schema for This Chunk
+\`\`\`json
+${JSON.stringify(cleanedSchema, null, 2)}
+\`\`\`
+
+## Instructions
+1. Match the EXACT structure shown in the schema
+2. Keys with special characters like \`*\` are LITERAL keys, not patterns
+3. If requiredValues are shown above, set values that satisfy them
+4. For paths like \`functionCall().functionCallReturnValue.property\`, the mock data key is just \`functionCall()\`
+
+Return ONLY a JSON object with this structure:
+\`\`\`json
+{
+  "mockData": {
+    // fill in ONLY the keys from this chunk
+  }
+}
+\`\`\``;
+}
+//# sourceMappingURL=generateChunkPrompt.js.map

package/packages/ai/src/lib/promptGenerators/generateChunkPrompt.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"generateChunkPrompt.js","sourceRoot":"","sources":["../../../../../../../packages/ai/src/lib/promptGenerators/generateChunkPrompt.ts"],"names":[],"mappings":"AAKA,OAAO,kBAAkB,MAAM,sBAAsB,CAAC;AACtD,OAAO,iBAAiB,MAAM,qBAAqB,CAAC;AAapD;;;;;GAKG;AACH,MAAM,CAAC,OAAO,UAAU,mBAAmB,CAAC,EAC1C,QAAQ,EACR,KAAK,EACL,UAAU,EACV,WAAW,EACX,sBAAsB,GACE;IACxB,0CAA0C;IAC1C,MAAM,aAAa,GAAG,kBAAkB,CACtC,iBAAiB,CAAC,KAAK,CAAuB,CAC/C,CAAC;IAEF,MAAM,WAAW,GAAG,MAAM,CAAC,IAAI,CAAC,KAAK,IAAI,EAAE,CAAC,CAAC;IAE7C,kEAAkE;IAClE,MAAM,mBAAmB,GACvB,sBAAsB,CAAC,MAAM,GAAG,CAAC;QAC/B,CAAC,CAAC;;;EAGN,IAAI,CAAC,SAAS,CAAC,sBAAsB,EAAE,IAAI,EAAE,CAAC,CAAC;;;CAGhD;QACK,CAAC,CAAC,EAAE,CAAC;IAET,OAAO,gCAAgC,UAAU,GAAG,CAAC,OAAO,WAAW;;;;EAIvE,IAAI,CAAC,SAAS,CAAC,EAAE,IAAI,EAAE,QAAQ,CAAC,IAAI,EAAE,WAAW,EAAE,QAAQ,CAAC,WAAW,EAAE,EAAE,IAAI,EAAE,CAAC,CAAC;;;EAGnF,mBAAmB,0BAA0B,WAAW,CAAC,MAAM;;EAE/D,WAAW,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC;;;;EAI/C,IAAI,CAAC,SAAS,CAAC,aAAa,EAAE,IAAI,EAAE,CAAC,CAAC;;;;;;;;;;;;;;;;OAgBjC,CAAC;AACR,CAAC"}

package/packages/ai/src/lib/promptGenerators/generateEntityScenarioDataGenerator.js

@@ -1,12 +1,57 @@
 import noErrorAttributes from "./noErrorAttributes.js";
-export default function generateEntityScenarioDataGenerator(structure, scenario, defaultScenarioData, incompleteResponse) {
+import simplifyKeysForLLM from "./simplifyKeysForLLM.js";
+/**
+ * Sort object keys by the size of their JSON-stringified values (smallest first).
+ * This ensures small keys appear before large keys in the LLM prompt, increasing
+ * the chance they get generated before the LLM loses focus on large schemas.
+ */
+function sortKeysBySize(obj) {
+    const entries = Object.entries(obj);
+    entries.sort(([, a], [, b]) => {
+        const sizeA = JSON.stringify(a).length;
+        const sizeB = JSON.stringify(b).length;
+        return sizeA - sizeB;
+    });
+    return Object.fromEntries(entries);
+}
+export default function generateEntityScenarioDataGenerator(structure, scenario, executionFlows, defaultScenarioData, incompleteResponse, options) {
     if (incompleteResponse) {
         return incompleteResponse;
     }
+    // Get the covered flows for this scenario and look up their requiredValues
+    const coveredFlowIds = scenario.metadata?.coveredFlows || [];
+    const coveredFlowRequirements = coveredFlowIds
+        .map((flowId) => {
+        const flow = executionFlows?.find((f) => f.id === flowId);
+        if (!flow)
+            return null;
+        return {
+            flowId: flow.id,
+            flowName: flow.name,
+            flowDescription: flow.description,
+            requiredValues: flow.requiredValues,
+        };
+    })
+        .filter(Boolean);
+    // Get blocking flows that are NOT covered by this scenario
+    // These should be explicitly avoided in the mock data
+    const blockingFlowsToAvoid = (executionFlows || [])
+        .filter((flow) => flow.blocksOtherFlows === true && !coveredFlowIds.includes(flow.id))
+        .map((flow) => ({
+        flowId: flow.id,
+        flowName: flow.name,
+        flowDescription: flow.description,
+        requiredValues: flow.requiredValues,
+    }));
     const scenarioJson = JSON.stringify({
         name: scenario.name,
         description: scenario.description,
-        keyAttributeInstructions: scenario.metadata.keyAttributeInstructions,
+        // Include the flow requirements to guide data generation
+        coveredFlows: coveredFlowRequirements,
+        // Include blocking flows to avoid
+        ...(blockingFlowsToAvoid.length > 0 && {
+            blockingFlowsToAvoid,
+        }),
     }, null, 2);
     const defaultDataSection = defaultScenarioData
         ? `
@@ -16,12 +61,29 @@ ${JSON.stringify(defaultScenarioData.data, null, 2)}
 \`\`\`
 `
         : '';
-    const mockDataSection = structure.dataForMocks
-        ? `## mockData Structure
+    // Simplify verbose function call keys for LLM consumption, then sort by size.
+    // The static analysis generates keys like "useQuery({ id: Number(id), onSuccess: async (quote) => { ...30 lines... } })"
+    // which confuse the LLM. simplifyKeysForLLM reduces them to "useQuery({ id, onSuccess: () => <truncated> })"
+    // while the reconcileMockDataKeys function will still match the simplified response back to original keys.
+    // Then sort keys by size (smallest first) so small keys appear before large keys,
+    // increasing the chance the LLM generates all keys for large schemas.
+    const cleanedDataForMocks = structure.dataForMocks
+        ? simplifyKeysForLLM(noErrorAttributes(structure.dataForMocks))
+        : null;
+    const sortedDataForMocks = cleanedDataForMocks &&
+        typeof cleanedDataForMocks === 'object' &&
+        !Array.isArray(cleanedDataForMocks)
+        ? sortKeysBySize(cleanedDataForMocks)
+        : cleanedDataForMocks;
+    // When mockData has already been generated via chunks, skip it in the main call
+    const mockDataSection = options?.mockDataAlreadyGenerated
+        ? '## mockData Structure\nmockData has been pre-generated—return `{}` for mockData.'
+        : sortedDataForMocks
+            ? `## mockData Structure
 \`\`\`json
-${JSON.stringify(noErrorAttributes(structure.dataForMocks), null, 2)}
+${JSON.stringify(sortedDataForMocks, null, 2)}
 \`\`\``
-        : '## mockData Structure\nNone—return `{}` for mockData.';
+            : '## mockData Structure\nNone—return `{}` for mockData.';
     const argumentsSection = structure.arguments
         ? `## argumentsData Structure
 \`\`\`json

package/packages/ai/src/lib/promptGenerators/generateEntityScenarioDataGenerator.js.map

@@ -1 +1 @@
-{"version":3,"file":"generateEntityScenarioDataGenerator.js","sourceRoot":"","sources":["../../../../../../../packages/ai/src/lib/promptGenerators/generateEntityScenarioDataGenerator.ts"],"names":[],"mappings":"AACA,OAAO,iBAAiB,MAAM,qBAAqB,CAAC;AAEpD,MAAM,CAAC,OAAO,UAAU,mCAAmC,CACzD,SAAiC,EACjC,QAAkB,EAClB,mBAAkC,EAClC,kBAA2B;IAE3B,IAAI,kBAAkB,EAAE,CAAC;QACvB,OAAO,kBAAkB,CAAC;IAC5B,CAAC;IAED,MAAM,YAAY,GAAG,IAAI,CAAC,SAAS,CACjC;QACE,IAAI,EAAE,QAAQ,CAAC,IAAI;QACnB,WAAW,EAAE,QAAQ,CAAC,WAAW;QACjC,wBAAwB,EAAE,QAAQ,CAAC,QAAQ,CAAC,wBAAwB;KACrE,EACD,IAAI,EACJ,CAAC,CACF,CAAC;IAEF,MAAM,kBAAkB,GAAG,mBAAmB;QAC5C,CAAC,CAAC;;;EAGJ,IAAI,CAAC,SAAS,CAAC,mBAAmB,CAAC,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC;;CAElD;QACG,CAAC,CAAC,EAAE,CAAC;IAEP,MAAM,eAAe,GAAG,SAAS,CAAC,YAAY;QAC5C,CAAC,CAAC;;EAEJ,IAAI,CAAC,SAAS,CAAC,iBAAiB,CAAC,SAAS,CAAC,YAAY,CAAC,EAAE,IAAI,EAAE,CAAC,CAAC;OAC7D;QACH,CAAC,CAAC,uDAAuD,CAAC;IAE5D,MAAM,gBAAgB,GAAG,SAAS,CAAC,SAAS;QAC1C,CAAC,CAAC;;EAEJ,IAAI,CAAC,SAAS,CAAC,iBAAiB,CAAC,SAAS,CAAC,SAAS,CAAC,EAAE,IAAI,EAAE,CAAC,CAAC;OAC1D;QACH,CAAC,CAAC,iEAAiE,CAAC;IAEtE,OAAO;;EAEP,YAAY;;EAEZ,kBAAkB;EAClB,eAAe;;EAEf,gBAAgB,EAAE,CAAC;AACrB,CAAC"}
+{"version":3,"file":"generateEntityScenarioDataGenerator.js","sourceRoot":"","sources":["../../../../../../../packages/ai/src/lib/promptGenerators/generateEntityScenarioDataGenerator.ts"],"names":[],"mappings":"AAOA,OAAO,iBAAiB,MAAM,qBAAqB,CAAC;AACpD,OAAO,kBAAkB,MAAM,sBAAsB,CAAC;AAEtD;;;;GAIG;AACH,SAAS,cAAc,CAAC,GAAwB;IAC9C,MAAM,OAAO,GAAG,MAAM,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC;IACpC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,EAAE,EAAE;QAC5B,MAAM,KAAK,GAAG,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC;QACvC,MAAM,KAAK,GAAG,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC;QACvC,OAAO,KAAK,GAAG,KAAK,CAAC;IACvB,CAAC,CAAC,CAAC;IACH,OAAO,MAAM,CAAC,WAAW,CAAC,OAAO,CAAC,CAAC;AACrC,CAAC;AAOD,MAAM,CAAC,OAAO,UAAU,mCAAmC,CACzD,SAAiC,EACjC,QAAkB,EAClB,cAA2C,EAC3C,mBAAkC,EAClC,kBAA2B,EAC3B,OAA+B;IAE/B,IAAI,kBAAkB,EAAE,CAAC;QACvB,OAAO,kBAAkB,CAAC;IAC5B,CAAC;IAED,2EAA2E;IAC3E,MAAM,cAAc,GAAG,QAAQ,CAAC,QAAQ,EAAE,YAAY,IAAI,EAAE,CAAC;IAC7D,MAAM,uBAAuB,GAAG,cAAc;SAC3C,GAAG,CAAC,CAAC,MAAM,EAAE,EAAE;QACd,MAAM,IAAI,GAAG,cAAc,EAAE,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,KAAK,MAAM,CAAC,CAAC;QAC1D,IAAI,CAAC,IAAI;YAAE,OAAO,IAAI,CAAC;QACvB,OAAO;YACL,MAAM,EAAE,IAAI,CAAC,EAAE;YACf,QAAQ,EAAE,IAAI,CAAC,IAAI;YACnB,eAAe,EAAE,IAAI,CAAC,WAAW;YACjC,cAAc,EAAE,IAAI,CAAC,cAAc;SACpC,CAAC;IACJ,CAAC,CAAC;SACD,MAAM,CAAC,OAAO,CAAC,CAAC;IAEnB,2DAA2D;IAC3D,sDAAsD;IACtD,MAAM,oBAAoB,GAAG,CAAC,cAAc,IAAI,EAAE,CAAC;SAChD,MAAM,CACL,CAAC,IAAI,EAAE,EAAE,CACP,IAAI,CAAC,gBAAgB,KAAK,IAAI,IAAI,CAAC,cAAc,CAAC,QAAQ,CAAC,IAAI,CAAC,EAAE,CAAC,CACtE;SACA,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;QACd,MAAM,EAAE,IAAI,CAAC,EAAE;QACf,QAAQ,EAAE,IAAI,CAAC,IAAI;QACnB,eAAe,EAAE,IAAI,CAAC,WAAW;QACjC,cAAc,EAAE,IAAI,CAAC,cAAc;KACpC,CAAC,CAAC,CAAC;IAEN,MAAM,YAAY,GAAG,IAAI,CAAC,SAAS,CACjC;QACE,IAAI,EAAE,QAAQ,CAAC,IAAI;QACnB,WAAW,EAAE,QAAQ,CAAC,WAAW;QACjC,yDAAyD;QACzD,YAAY,EAAE,uBAAuB;QACrC,kCAAkC;QAClC,GAAG,CAAC,oBAAoB,CAAC,MAAM,GAAG,CAAC,IAAI;YACrC,oBAAoB;SACrB,CAAC;KACH,EACD,IAAI,EACJ,CAAC,CACF,CAAC;IAEF,MAAM,kBAAkB,GAAG,mBAAmB;QAC5C,CAAC,CAAC;;;EAGJ,IAAI,CAAC,SAAS,CAAC,mBAAmB,CAAC,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC;;CAElD;QACG,CAAC,CAAC,EAAE,CAAC;IAEP,8EAA8E;IAC9E,yHAAyH;IACzH,6GAA6G;IAC7G,2GAA2G;IAC3G,kFAAkF;IAClF,sEAAsE;IACtE,MAAM,mBAAmB,GAAG,SAAS,CAAC,YAAY;QAChD,CAAC,CAAC,kBAAkB,CAChB,iBAAiB,CAAC,SAAS,CAAC,YAAY,CAAuB,CAChE;QACH,CAAC,CAAC,IAAI,CAAC;IACT,MAAM,kBAAkB,GACtB,mBAAmB;QACnB,OAAO,mBAAmB,KAAK,QAAQ;QACvC,CAAC,KAAK,CAAC,OAAO,CAAC,mBAAmB,CAAC;QACjC,CAAC,CAAC,cAAc,CAAC,mBAA0C,CAAC;QAC5D,CAAC,CAAC,mBAAmB,CAAC;IAE1B,gFAAgF;IAChF,MAAM,eAAe,GAAG,OAAO,EAAE,wBAAwB;QACvD,CAAC,CAAC,kFAAkF;QACpF,CAAC,CAAC,kBAAkB;YAClB,CAAC,CAAC;;EAEN,IAAI,CAAC,SAAS,CAAC,kBAAkB,EAAE,IAAI,EAAE,CAAC,CAAC;OACtC;YACD,CAAC,CAAC,uDAAuD,CAAC;IAE9D,MAAM,gBAAgB,GAAG,SAAS,CAAC,SAAS;QAC1C,CAAC,CAAC;;EAEJ,IAAI,CAAC,SAAS,CAAC,iBAAiB,CAAC,SAAS,CAAC,SAAS,CAAC,EAAE,IAAI,EAAE,CAAC,CAAC;OAC1D;QACH,CAAC,CAAC,iEAAiE,CAAC;IAEtE,OAAO;;EAEP,YAAY;;EAEZ,kBAAkB;EAClB,eAAe;;EAEf,gBAAgB,EAAE,CAAC;AACrB,CAAC"}

package/packages/ai/src/lib/promptGenerators/generateEntityScenariosGenerator.js

@@ -1,35 +1,17 @@
-export default function generateEntityScenariosGenerator({ entity, keyAttributes, error, }) {
-    const thisEntityKeyAttributes = keyAttributes?.filter((ka) => !ka.dependentEntityName);
-    const relevantDependentKeyAttributes = keyAttributes?.filter((ka) => ka.dependentEntityName);
-    const keyAttrsJson = JSON.stringify(thisEntityKeyAttributes.map((ka) => ({
-        dataStructurePath: ka.dataStructurePath,
-        localVariable: ka.internalPath,
-        description: ka.description,
-        validValueOptions: ka.validValueOptions,
-    })), null, 2);
-    const dependencySection = !error &&
-        relevantDependentKeyAttributes &&
-        relevantDependentKeyAttributes.length > 0
-        ? `
-## Dependency Key Attributes
+export default function generateEntityScenariosGenerator({ entity, executionFlows, error, }) {
+    // Build execution flows section
+    const executionFlowsSection = !error && executionFlows && executionFlows.length > 0
+        ? `## Execution Flows
+These are all the distinct outcomes/behaviors this component can produce. Create scenarios that maximize coverage of these flows.
 \`\`\`json
-${JSON.stringify(relevantDependentKeyAttributes.map((ka) => ({
-            dataStructurePath: ka.dataStructurePath,
-            localVariable: ka.internalPath,
-            forDependency: ka.dependentEntityName,
-            description: ka.description,
-            validValueOptions: ka.validValueOptions,
-        })), null, 2)}
-\`\`\``
-        : '';
-    return `## Key Attributes
-\`\`\`json
-${keyAttrsJson}
+${JSON.stringify(executionFlows, null, 2)}
 \`\`\`
 
-## Code
+`
+        : '';
+    return `${executionFlowsSection}## Code
 \`\`\`typescript
 ${entity.code}
-\`\`\`${dependencySection}`;
+\`\`\``;
 }
 //# sourceMappingURL=generateEntityScenariosGenerator.js.map

package/packages/ai/src/lib/promptGenerators/generateEntityScenariosGenerator.js.map

@@ -1 +1 @@
-{"version":3,"file":"generateEntityScenariosGenerator.js","sourceRoot":"","sources":["../../../../../../../packages/ai/src/lib/promptGenerators/generateEntityScenariosGenerator.ts"],"names":[],"mappings":"AAQA,MAAM,CAAC,OAAO,UAAU,gCAAgC,CAAC,EACvD,MAAM,EACN,aAAa,EACb,KAAK,GACgC;IACrC,MAAM,uBAAuB,GAAG,aAAa,EAAE,MAAM,CACnD,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC,EAAE,CAAC,mBAAmB,CAChC,CAAC;IAEF,MAAM,8BAA8B,GAAG,aAAa,EAAE,MAAM,CAC1D,CAAC,EAAE,EAAE,EAAE,CAAC,EAAE,CAAC,mBAAmB,CAC/B,CAAC;IAEF,MAAM,YAAY,GAAG,IAAI,CAAC,SAAS,CACjC,uBAAuB,CAAC,GAAG,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC;QACnC,iBAAiB,EAAE,EAAE,CAAC,iBAAiB;QACvC,aAAa,EAAE,EAAE,CAAC,YAAY;QAC9B,WAAW,EAAE,EAAE,CAAC,WAAW;QAC3B,iBAAiB,EAAE,EAAE,CAAC,iBAAiB;KACxC,CAAC,CAAC,EACH,IAAI,EACJ,CAAC,CACF,CAAC;IAEF,MAAM,iBAAiB,GACrB,CAAC,KAAK;QACN,8BAA8B;QAC9B,8BAA8B,CAAC,MAAM,GAAG,CAAC;QACvC,CAAC,CAAC;;;EAGN,IAAI,CAAC,SAAS,CACd,8BAA8B,CAAC,GAAG,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC;YAC1C,iBAAiB,EAAE,EAAE,CAAC,iBAAiB;YACvC,aAAa,EAAE,EAAE,CAAC,YAAY;YAC9B,aAAa,EAAE,EAAE,CAAC,mBAAmB;YACrC,WAAW,EAAE,EAAE,CAAC,WAAW;YAC3B,iBAAiB,EAAE,EAAE,CAAC,iBAAiB;SACxC,CAAC,CAAC,EACH,IAAI,EACJ,CAAC,CACF;OACM;QACD,CAAC,CAAC,EAAE,CAAC;IAET,OAAO;;EAEP,YAAY;;;;;EAKZ,MAAM,CAAC,IAAI;QACL,iBAAiB,EAAE,CAAC;AAC5B,CAAC"}
+{"version":3,"file":"generateEntityScenariosGenerator.js","sourceRoot":"","sources":["../../../../../../../packages/ai/src/lib/promptGenerators/generateEntityScenariosGenerator.ts"],"names":[],"mappings":"AAQA,MAAM,CAAC,OAAO,UAAU,gCAAgC,CAAC,EACvD,MAAM,EACN,cAAc,EACd,KAAK,GACgC;IACrC,gCAAgC;IAChC,MAAM,qBAAqB,GACzB,CAAC,KAAK,IAAI,cAAc,IAAI,cAAc,CAAC,MAAM,GAAG,CAAC;QACnD,CAAC,CAAC;;;EAGN,IAAI,CAAC,SAAS,CAAC,cAAc,EAAE,IAAI,EAAE,CAAC,CAAC;;;CAGxC;QACK,CAAC,CAAC,EAAE,CAAC;IAET,OAAO,GAAG,qBAAqB;;EAE/B,MAAM,CAAC,IAAI;OACN,CAAC;AACR,CAAC"}

package/packages/ai/src/lib/promptGenerators/generateMissingKeysPrompt.js

@@ -0,0 +1,45 @@
+/**
+ * Generate a prompt for the follow-up LLM call to fill in missing mockData keys.
+ * This is used when the initial Default Scenario response is missing some keys.
+ */
+export default function generateMissingKeysPrompt({ scenario, executionFlows, generatedMockData, missingSchema, }) {
+    const coveredFlowIds = scenario.metadata?.coveredFlows || [];
+    const coveredFlowRequirements = coveredFlowIds
+        .map((flowId) => {
+        const flow = executionFlows?.find((f) => f.id === flowId);
+        if (!flow)
+            return null;
+        return {
+            flowId: flow.id,
+            flowName: flow.name,
+            requiredValues: flow.requiredValues,
+        };
+    })
+        .filter(Boolean);
+    return `The initial response was missing some keys. Generate mock data for ONLY the missing keys listed below.
+
+## Scenario Context
+\`\`\`json
+${JSON.stringify({ name: scenario.name, description: scenario.description, coveredFlows: coveredFlowRequirements }, null, 2)}
+\`\`\`
+
+## Already Generated Data
+\`\`\`json
+${JSON.stringify(generatedMockData, null, 2)}
+\`\`\`
+
+## Missing Keys to Generate
+\`\`\`json
+${JSON.stringify(missingSchema, null, 2)}
+\`\`\`
+
+Return ONLY a JSON object with this structure:
+\`\`\`json
+{
+  "mockData": {
+    // fill in ONLY the missing keys above
+  }
+}
+\`\`\``;
+}
+//# sourceMappingURL=generateMissingKeysPrompt.js.map

package/packages/ai/src/lib/promptGenerators/generateMissingKeysPrompt.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"generateMissingKeysPrompt.js","sourceRoot":"","sources":["../../../../../../../packages/ai/src/lib/promptGenerators/generateMissingKeysPrompt.ts"],"names":[],"mappings":"AASA;;;GAGG;AACH,MAAM,CAAC,OAAO,UAAU,yBAAyB,CAAC,EAChD,QAAQ,EACR,cAAc,EACd,iBAAiB,EACjB,aAAa,GACiB;IAC9B,MAAM,cAAc,GAAG,QAAQ,CAAC,QAAQ,EAAE,YAAY,IAAI,EAAE,CAAC;IAC7D,MAAM,uBAAuB,GAAG,cAAc;SAC3C,GAAG,CAAC,CAAC,MAAM,EAAE,EAAE;QACd,MAAM,IAAI,GAAG,cAAc,EAAE,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,KAAK,MAAM,CAAC,CAAC;QAC1D,IAAI,CAAC,IAAI;YAAE,OAAO,IAAI,CAAC;QACvB,OAAO;YACL,MAAM,EAAE,IAAI,CAAC,EAAE;YACf,QAAQ,EAAE,IAAI,CAAC,IAAI;YACnB,cAAc,EAAE,IAAI,CAAC,cAAc;SACpC,CAAC;IACJ,CAAC,CAAC;SACD,MAAM,CAAC,OAAO,CAAC,CAAC;IAEnB,OAAO;;;;EAIP,IAAI,CAAC,SAAS,CAAC,EAAE,IAAI,EAAE,QAAQ,CAAC,IAAI,EAAE,WAAW,EAAE,QAAQ,CAAC,WAAW,EAAE,YAAY,EAAE,uBAAuB,EAAE,EAAE,IAAI,EAAE,CAAC,CAAC;;;;;EAK1H,IAAI,CAAC,SAAS,CAAC,iBAAiB,EAAE,IAAI,EAAE,CAAC,CAAC;;;;;EAK1C,IAAI,CAAC,SAAS,CAAC,aAAa,EAAE,IAAI,EAAE,CAAC,CAAC;;;;;;;;;;OAUjC,CAAC;AACR,CAAC"}