@codeyam/codeyam-cli 0.1.0-staging.1669d45 → 0.1.0-staging.27d5a59

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

@@ -0,0 +1,335 @@
+/**
+ * Simplify verbose function call keys for LLM consumption.
+ *
+ * The static analysis generates keys that include entire function call signatures
+ * with all arguments, including multi-line callback bodies. For example:
+ *
+ * ```
+ * "useQuery({ id: Number(id), pagination: debouncedPagination, onSuccess: async (quote) => { ...30 lines... } })"
+ * ```
+ *
+ * These verbose keys confuse the LLM and cause it to miss generating data for some keys.
+ * This function simplifies keys to just the essential parts:
+ *
+ * ```
+ * "useQuery({ id, pagination, onSuccess: () => <truncated> })"
+ * ```
+ *
+ * The reconcileMockDataKeys function already does fuzzy matching, so the LLM's
+ * simplified response keys will still match back to the original verbose keys.
+ */
+export default function simplifyKeysForLLM(data) {
+    if (typeof data !== 'object' || data === null) {
+        return data;
+    }
+    const result = {};
+    for (const key of Object.keys(data)) {
+        const simplifiedKey = simplifyFunctionCallKey(key);
+        const value = data[key];
+        // Recursively simplify nested objects
+        if (typeof value === 'object' && value !== null && !Array.isArray(value)) {
+            result[simplifiedKey] = simplifyKeysForLLM(value);
+        }
+        else if (Array.isArray(value)) {
+            // For arrays, recursively simplify each object element
+            result[simplifiedKey] = value.map((item) => typeof item === 'object' && item !== null && !Array.isArray(item)
+                ? simplifyKeysForLLM(item)
+                : item);
+        }
+        else {
+            result[simplifiedKey] = value;
+        }
+    }
+    return result;
+}
+/**
+ * Simplify a single function call key.
+ *
+ * Rules:
+ * 1. If key doesn't look like a function call, return as-is
+ * 2. For function calls with object arguments:
+ *    - Keep property names
+ *    - Replace complex values with simplified versions
+ *    - Replace callbacks with () => ...
+ * 3. Cap total length at MAX_KEY_LENGTH
+ */
+function simplifyFunctionCallKey(key) {
+    // If it doesn't look like a function call, return as-is
+    if (!key.includes('(')) {
+        return key;
+    }
+    // Extract function name and arguments
+    const funcNameMatch = key.match(/^([a-zA-Z_][a-zA-Z0-9_]*)\s*\(/);
+    if (!funcNameMatch) {
+        return key;
+    }
+    const funcName = funcNameMatch[1];
+    const argsStart = key.indexOf('(');
+    const argsContent = key.slice(argsStart + 1, -1); // Remove outer parens
+    // If args are empty or very short, return as-is
+    if (argsContent.trim().length < 50) {
+        return key;
+    }
+    // Parse and simplify the arguments
+    const simplifiedArgs = simplifyArguments(argsContent);
+    const simplified = `${funcName}(${simplifiedArgs})`;
+    // Cap length - use <truncated> instead of ... to prevent LLMs from
+    // interpreting it as a fill-in placeholder (Issue #14)
+    const MAX_KEY_LENGTH = 150;
+    if (simplified.length > MAX_KEY_LENGTH) {
+        return simplified.slice(0, MAX_KEY_LENGTH - 11) + '<truncated>';
+    }
+    return simplified;
+}
+/**
+ * Simplify function call arguments.
+ *
+ * Handles multiple arguments separated by commas, where each argument
+ * could be an object literal, a simple value, or a complex expression.
+ */
+function simplifyArguments(argsContent) {
+    const trimmed = argsContent.trim();
+    // Split arguments at top-level commas (not inside nested structures)
+    const args = splitTopLevelArguments(trimmed);
+    const simplifiedArgs = args.map((arg) => {
+        const argTrimmed = arg.trim();
+        // Check if it's an object literal (starts with {)
+        if (argTrimmed.startsWith('{')) {
+            return simplifyObjectLiteral(argTrimmed);
+        }
+        // For non-object arguments, just truncate if needed
+        // Use <truncated> instead of ... to prevent LLMs from interpreting as fill-in
+        if (argTrimmed.length > 30) {
+            return argTrimmed.slice(0, 19) + '<truncated>';
+        }
+        return argTrimmed;
+    });
+    return simplifiedArgs.join(', ');
+}
+/**
+ * Split arguments at top-level commas, respecting nested braces/parens.
+ */
+function splitTopLevelArguments(argsContent) {
+    const args = [];
+    let current = '';
+    let depth = 0;
+    let inString = false;
+    let stringChar = '';
+    for (let i = 0; i < argsContent.length; i++) {
+        const char = argsContent[i];
+        const prevChar = i > 0 ? argsContent[i - 1] : '';
+        // Handle string boundaries
+        if ((char === '"' || char === "'" || char === '`') && prevChar !== '\\') {
+            if (!inString) {
+                inString = true;
+                stringChar = char;
+            }
+            else if (char === stringChar) {
+                inString = false;
+            }
+        }
+        if (!inString) {
+            if (char === '{' || char === '(' || char === '[') {
+                depth++;
+            }
+            else if (char === '}' || char === ')' || char === ']') {
+                depth--;
+            }
+            else if (char === ',' && depth === 0) {
+                // End of argument
+                if (current.trim()) {
+                    args.push(current.trim());
+                }
+                current = '';
+                continue;
+            }
+        }
+        current += char;
+    }
+    // Don't forget the last argument
+    if (current.trim()) {
+        args.push(current.trim());
+    }
+    return args;
+}
+/**
+ * Simplify an object literal, keeping property names but simplifying values.
+ */
+function simplifyObjectLiteral(objStr) {
+    // Extract property names and their values
+    const properties = extractObjectProperties(objStr);
+    if (properties.length === 0) {
+        return '{ }';
+    }
+    const simplifiedProps = properties.map(({ name, value }) => {
+        const simplifiedValue = simplifyPropertyValue(value);
+        // If value is same as name (shorthand), just use name
+        if (simplifiedValue === name) {
+            return name;
+        }
+        return `${name}: ${simplifiedValue}`;
+    });
+    return `{ ${simplifiedProps.join(', ')} }`;
+}
+/**
+ * Extract property names and values from an object literal string.
+ */
+function extractObjectProperties(objStr) {
+    const properties = [];
+    // Remove outer braces and trim
+    let content = objStr.trim();
+    if (content.startsWith('{'))
+        content = content.slice(1);
+    if (content.endsWith('}'))
+        content = content.slice(0, -1);
+    content = content.trim();
+    // Parse properties, handling nested structures
+    let current = '';
+    let depth = 0;
+    let inString = false;
+    let stringChar = '';
+    for (let i = 0; i < content.length; i++) {
+        const char = content[i];
+        const prevChar = i > 0 ? content[i - 1] : '';
+        // Handle string boundaries
+        if ((char === '"' || char === "'" || char === '`') && prevChar !== '\\') {
+            if (!inString) {
+                inString = true;
+                stringChar = char;
+            }
+            else if (char === stringChar) {
+                inString = false;
+            }
+        }
+        if (!inString) {
+            if (char === '{' || char === '(' || char === '[') {
+                depth++;
+            }
+            else if (char === '}' || char === ')' || char === ']') {
+                depth--;
+            }
+            else if (char === ',' && depth === 0) {
+                // End of property
+                const prop = parseProperty(current.trim());
+                if (prop)
+                    properties.push(prop);
+                current = '';
+                continue;
+            }
+        }
+        current += char;
+    }
+    // Don't forget the last property
+    if (current.trim()) {
+        const prop = parseProperty(current.trim());
+        if (prop)
+            properties.push(prop);
+    }
+    return properties;
+}
+/**
+ * Parse a single property string into name and value.
+ */
+function parseProperty(propStr) {
+    // Handle shorthand properties (just a name)
+    const shorthandMatch = propStr.match(/^([a-zA-Z_][a-zA-Z0-9_]*)$/);
+    if (shorthandMatch) {
+        return { name: shorthandMatch[1], value: shorthandMatch[1] };
+    }
+    // Handle name: value
+    const colonIndex = findPropertyColon(propStr);
+    if (colonIndex === -1) {
+        return null;
+    }
+    const name = propStr.slice(0, colonIndex).trim();
+    const value = propStr.slice(colonIndex + 1).trim();
+    // Clean up property name (remove quotes if present)
+    const cleanName = name.replace(/^['"]|['"]$/g, '');
+    return { name: cleanName, value };
+}
+/**
+ * Find the colon that separates property name from value.
+ * Must handle cases like { 'key': value } and { key: value }
+ */
+function findPropertyColon(propStr) {
+    let depth = 0;
+    let inString = false;
+    let stringChar = '';
+    for (let i = 0; i < propStr.length; i++) {
+        const char = propStr[i];
+        const prevChar = i > 0 ? propStr[i - 1] : '';
+        if ((char === '"' || char === "'" || char === '`') && prevChar !== '\\') {
+            if (!inString) {
+                inString = true;
+                stringChar = char;
+            }
+            else if (char === stringChar) {
+                inString = false;
+            }
+        }
+        if (!inString) {
+            if (char === '{' || char === '(' || char === '[') {
+                depth++;
+            }
+            else if (char === '}' || char === ')' || char === ']') {
+                depth--;
+            }
+            else if (char === ':' && depth === 0) {
+                return i;
+            }
+        }
+    }
+    return -1;
+}
+/**
+ * Simplify a property value.
+ */
+function simplifyPropertyValue(value) {
+    const trimmed = value.trim();
+    // If it's a simple identifier, return as-is
+    if (/^[a-zA-Z_][a-zA-Z0-9_]*$/.test(trimmed)) {
+        return trimmed;
+    }
+    // If it's a simple literal (number, boolean, null, undefined), return as-is
+    if (/^(true|false|null|undefined|\d+(\.\d+)?)$/.test(trimmed)) {
+        return trimmed;
+    }
+    // If it's a short string literal, return as-is
+    if ((trimmed.startsWith('"') || trimmed.startsWith("'")) &&
+        trimmed.length < 30) {
+        return trimmed;
+    }
+    // If it looks like a function/arrow function, simplify
+    // Use <truncated> instead of ... to prevent LLMs from interpreting as fill-in
+    if (trimmed.startsWith('function') ||
+        trimmed.startsWith('async function') ||
+        trimmed.includes('=>') ||
+        (trimmed.startsWith('(') && trimmed.includes('=>'))) {
+        return '() => <truncated>';
+    }
+    // If it's an object literal, recursively simplify
+    if (trimmed.startsWith('{')) {
+        const inner = simplifyObjectLiteral(trimmed);
+        if (inner.length > 50) {
+            return '{ <truncated> }';
+        }
+        return inner;
+    }
+    // If it's an array, simplify
+    // Use <truncated> instead of ... to prevent LLMs from interpreting as fill-in
+    if (trimmed.startsWith('[')) {
+        return '[<truncated>]';
+    }
+    // For complex expressions, extract the first identifier or truncate
+    // Use <truncated> instead of ... to prevent LLMs from interpreting as fill-in
+    const identMatch = trimmed.match(/^([a-zA-Z_][a-zA-Z0-9_]*)/);
+    if (identMatch && trimmed.length > 30) {
+        return identMatch[1] + '<truncated>';
+    }
+    // Truncate long values
+    if (trimmed.length > 30) {
+        return trimmed.slice(0, 19) + '<truncated>';
+    }
+    return trimmed;
+}
+//# sourceMappingURL=simplifyKeysForLLM.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"simplifyKeysForLLM.js","sourceRoot":"","sources":["../../../../../../../packages/ai/src/lib/promptGenerators/simplifyKeysForLLM.ts"],"names":[],"mappings":"AAEA;;;;;;;;;;;;;;;;;;;GAmBG;AACH,MAAM,CAAC,OAAO,UAAU,kBAAkB,CACxC,IAAwB;IAExB,IAAI,OAAO,IAAI,KAAK,QAAQ,IAAI,IAAI,KAAK,IAAI,EAAE,CAAC;QAC9C,OAAO,IAAI,CAAC;IACd,CAAC;IAED,MAAM,MAAM,GAAuB,EAAE,CAAC;IAEtC,KAAK,MAAM,GAAG,IAAI,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC;QACpC,MAAM,aAAa,GAAG,uBAAuB,CAAC,GAAG,CAAC,CAAC;QACnD,MAAM,KAAK,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC;QAExB,sCAAsC;QACtC,IAAI,OAAO,KAAK,KAAK,QAAQ,IAAI,KAAK,KAAK,IAAI,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE,CAAC;YACzE,MAAM,CAAC,aAAa,CAAC,GAAG,kBAAkB,CAAC,KAA2B,CAAC,CAAC;QAC1E,CAAC;aAAM,IAAI,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE,CAAC;YAChC,uDAAuD;YACvD,MAAM,CAAC,aAAa,CAAC,GAAG,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CACzC,OAAO,IAAI,KAAK,QAAQ,IAAI,IAAI,KAAK,IAAI,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC;gBAC/D,CAAC,CAAC,kBAAkB,CAAC,IAA0B,CAAC;gBAChD,CAAC,CAAC,IAAI,CACe,CAAC;QAC5B,CAAC;aAAM,CAAC;YACN,MAAM,CAAC,aAAa,CAAC,GAAG,KAAK,CAAC;QAChC,CAAC;IACH,CAAC;IAED,OAAO,MAAM,CAAC;AAChB,CAAC;AAED;;;;;;;;;;GAUG;AACH,SAAS,uBAAuB,CAAC,GAAW;IAC1C,wDAAwD;IACxD,IAAI,CAAC,GAAG,CAAC,QAAQ,CAAC,GAAG,CAAC,EAAE,CAAC;QACvB,OAAO,GAAG,CAAC;IACb,CAAC;IAED,sCAAsC;IACtC,MAAM,aAAa,GAAG,GAAG,CAAC,KAAK,CAAC,gCAAgC,CAAC,CAAC;IAClE,IAAI,CAAC,aAAa,EAAE,CAAC;QACnB,OAAO,GAAG,CAAC;IACb,CAAC;IAED,MAAM,QAAQ,GAAG,aAAa,CAAC,CAAC,CAAC,CAAC;IAClC,MAAM,SAAS,GAAG,GAAG,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC;IACnC,MAAM,WAAW,GAAG,GAAG,CAAC,KAAK,CAAC,SAAS,GAAG,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,sBAAsB;IAExE,gDAAgD;IAChD,IAAI,WAAW,CAAC,IAAI,EAAE,CAAC,MAAM,GAAG,EAAE,EAAE,CAAC;QACnC,OAAO,GAAG,CAAC;IACb,CAAC;IAED,mCAAmC;IACnC,MAAM,cAAc,GAAG,iBAAiB,CAAC,WAAW,CAAC,CAAC;IACtD,MAAM,UAAU,GAAG,GAAG,QAAQ,IAAI,cAAc,GAAG,CAAC;IAEpD,mEAAmE;IACnE,uDAAuD;IACvD,MAAM,cAAc,GAAG,GAAG,CAAC;IAC3B,IAAI,UAAU,CAAC,MAAM,GAAG,cAAc,EAAE,CAAC;QACvC,OAAO,UAAU,CAAC,KAAK,CAAC,CAAC,EAAE,cAAc,GAAG,EAAE,CAAC,GAAG,aAAa,CAAC;IAClE,CAAC;IAED,OAAO,UAAU,CAAC;AACpB,CAAC;AAED;;;;;GAKG;AACH,SAAS,iBAAiB,CAAC,WAAmB;IAC5C,MAAM,OAAO,GAAG,WAAW,CAAC,IAAI,EAAE,CAAC;IAEnC,qEAAqE;IACrE,MAAM,IAAI,GAAG,sBAAsB,CAAC,OAAO,CAAC,CAAC;IAE7C,MAAM,cAAc,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE;QACtC,MAAM,UAAU,GAAG,GAAG,CAAC,IAAI,EAAE,CAAC;QAE9B,kDAAkD;QAClD,IAAI,UAAU,CAAC,UAAU,CAAC,GAAG,CAAC,EAAE,CAAC;YAC/B,OAAO,qBAAqB,CAAC,UAAU,CAAC,CAAC;QAC3C,CAAC;QAED,oDAAoD;QACpD,8EAA8E;QAC9E,IAAI,UAAU,CAAC,MAAM,GAAG,EAAE,EAAE,CAAC;YAC3B,OAAO,UAAU,CAAC,KAAK,CAAC,CAAC,EAAE,EAAE,CAAC,GAAG,aAAa,CAAC;QACjD,CAAC;QAED,OAAO,UAAU,CAAC;IACpB,CAAC,CAAC,CAAC;IAEH,OAAO,cAAc,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;AACnC,CAAC;AAED;;GAEG;AACH,SAAS,sBAAsB,CAAC,WAAmB;IACjD,MAAM,IAAI,GAAa,EAAE,CAAC;IAC1B,IAAI,OAAO,GAAG,EAAE,CAAC;IACjB,IAAI,KAAK,GAAG,CAAC,CAAC;IACd,IAAI,QAAQ,GAAG,KAAK,CAAC;IACrB,IAAI,UAAU,GAAG,EAAE,CAAC;IAEpB,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,WAAW,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;QAC5C,MAAM,IAAI,GAAG,WAAW,CAAC,CAAC,CAAC,CAAC;QAC5B,MAAM,QAAQ,GAAG,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;QAEjD,2BAA2B;QAC3B,IAAI,CAAC,IAAI,KAAK,GAAG,IAAI,IAAI,KAAK,GAAG,IAAI,IAAI,KAAK,GAAG,CAAC,IAAI,QAAQ,KAAK,IAAI,EAAE,CAAC;YACxE,IAAI,CAAC,QAAQ,EAAE,CAAC;gBACd,QAAQ,GAAG,IAAI,CAAC;gBAChB,UAAU,GAAG,IAAI,CAAC;YACpB,CAAC;iBAAM,IAAI,IAAI,KAAK,UAAU,EAAE,CAAC;gBAC/B,QAAQ,GAAG,KAAK,CAAC;YACnB,CAAC;QACH,CAAC;QAED,IAAI,CAAC,QAAQ,EAAE,CAAC;YACd,IAAI,IAAI,KAAK,GAAG,IAAI,IAAI,KAAK,GAAG,IAAI,IAAI,KAAK,GAAG,EAAE,CAAC;gBACjD,KAAK,EAAE,CAAC;YACV,CAAC;iBAAM,IAAI,IAAI,KAAK,GAAG,IAAI,IAAI,KAAK,GAAG,IAAI,IAAI,KAAK,GAAG,EAAE,CAAC;gBACxD,KAAK,EAAE,CAAC;YACV,CAAC;iBAAM,IAAI,IAAI,KAAK,GAAG,IAAI,KAAK,KAAK,CAAC,EAAE,CAAC;gBACvC,kBAAkB;gBAClB,IAAI,OAAO,CAAC,IAAI,EAAE,EAAE,CAAC;oBACnB,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,IAAI,EAAE,CAAC,CAAC;gBAC5B,CAAC;gBACD,OAAO,GAAG,EAAE,CAAC;gBACb,SAAS;YACX,CAAC;QACH,CAAC;QAED,OAAO,IAAI,IAAI,CAAC;IAClB,CAAC;IAED,iCAAiC;IACjC,IAAI,OAAO,CAAC,IAAI,EAAE,EAAE,CAAC;QACnB,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,IAAI,EAAE,CAAC,CAAC;IAC5B,CAAC;IAED,OAAO,IAAI,CAAC;AACd,CAAC;AAED;;GAEG;AACH,SAAS,qBAAqB,CAAC,MAAc;IAC3C,0CAA0C;IAC1C,MAAM,UAAU,GAAG,uBAAuB,CAAC,MAAM,CAAC,CAAC;IAEnD,IAAI,UAAU,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QAC5B,OAAO,KAAK,CAAC;IACf,CAAC;IAED,MAAM,eAAe,GAAG,UAAU,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,EAAE,KAAK,EAAE,EAAE,EAAE;QACzD,MAAM,eAAe,GAAG,qBAAqB,CAAC,KAAK,CAAC,CAAC;QACrD,sDAAsD;QACtD,IAAI,eAAe,KAAK,IAAI,EAAE,CAAC;YAC7B,OAAO,IAAI,CAAC;QACd,CAAC;QACD,OAAO,GAAG,IAAI,KAAK,eAAe,EAAE,CAAC;IACvC,CAAC,CAAC,CAAC;IAEH,OAAO,KAAK,eAAe,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC;AAC7C,CAAC;AAED;;GAEG;AACH,SAAS,uBAAuB,CAC9B,MAAc;IAEd,MAAM,UAAU,GAA2C,EAAE,CAAC;IAE9D,+BAA+B;IAC/B,IAAI,OAAO,GAAG,MAAM,CAAC,IAAI,EAAE,CAAC;IAC5B,IAAI,OAAO,CAAC,UAAU,CAAC,GAAG,CAAC;QAAE,OAAO,GAAG,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;IACxD,IAAI,OAAO,CAAC,QAAQ,CAAC,GAAG,CAAC;QAAE,OAAO,GAAG,OAAO,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC;IAC1D,OAAO,GAAG,OAAO,CAAC,IAAI,EAAE,CAAC;IAEzB,+CAA+C;IAC/C,IAAI,OAAO,GAAG,EAAE,CAAC;IACjB,IAAI,KAAK,GAAG,CAAC,CAAC;IACd,IAAI,QAAQ,GAAG,KAAK,CAAC;IACrB,IAAI,UAAU,GAAG,EAAE,CAAC;IAEpB,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,OAAO,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;QACxC,MAAM,IAAI,GAAG,OAAO,CAAC,CAAC,CAAC,CAAC;QACxB,MAAM,QAAQ,GAAG,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;QAE7C,2BAA2B;QAC3B,IAAI,CAAC,IAAI,KAAK,GAAG,IAAI,IAAI,KAAK,GAAG,IAAI,IAAI,KAAK,GAAG,CAAC,IAAI,QAAQ,KAAK,IAAI,EAAE,CAAC;YACxE,IAAI,CAAC,QAAQ,EAAE,CAAC;gBACd,QAAQ,GAAG,IAAI,CAAC;gBAChB,UAAU,GAAG,IAAI,CAAC;YACpB,CAAC;iBAAM,IAAI,IAAI,KAAK,UAAU,EAAE,CAAC;gBAC/B,QAAQ,GAAG,KAAK,CAAC;YACnB,CAAC;QACH,CAAC;QAED,IAAI,CAAC,QAAQ,EAAE,CAAC;YACd,IAAI,IAAI,KAAK,GAAG,IAAI,IAAI,KAAK,GAAG,IAAI,IAAI,KAAK,GAAG,EAAE,CAAC;gBACjD,KAAK,EAAE,CAAC;YACV,CAAC;iBAAM,IAAI,IAAI,KAAK,GAAG,IAAI,IAAI,KAAK,GAAG,IAAI,IAAI,KAAK,GAAG,EAAE,CAAC;gBACxD,KAAK,EAAE,CAAC;YACV,CAAC;iBAAM,IAAI,IAAI,KAAK,GAAG,IAAI,KAAK,KAAK,CAAC,EAAE,CAAC;gBACvC,kBAAkB;gBAClB,MAAM,IAAI,GAAG,aAAa,CAAC,OAAO,CAAC,IAAI,EAAE,CAAC,CAAC;gBAC3C,IAAI,IAAI;oBAAE,UAAU,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;gBAChC,OAAO,GAAG,EAAE,CAAC;gBACb,SAAS;YACX,CAAC;QACH,CAAC;QAED,OAAO,IAAI,IAAI,CAAC;IAClB,CAAC;IAED,iCAAiC;IACjC,IAAI,OAAO,CAAC,IAAI,EAAE,EAAE,CAAC;QACnB,MAAM,IAAI,GAAG,aAAa,CAAC,OAAO,CAAC,IAAI,EAAE,CAAC,CAAC;QAC3C,IAAI,IAAI;YAAE,UAAU,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IAClC,CAAC;IAED,OAAO,UAAU,CAAC;AACpB,CAAC;AAED;;GAEG;AACH,SAAS,aAAa,CACpB,OAAe;IAEf,4CAA4C;IAC5C,MAAM,cAAc,GAAG,OAAO,CAAC,KAAK,CAAC,4BAA4B,CAAC,CAAC;IACnE,IAAI,cAAc,EAAE,CAAC;QACnB,OAAO,EAAE,IAAI,EAAE,cAAc,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,cAAc,CAAC,CAAC,CAAC,EAAE,CAAC;IAC/D,CAAC;IAED,qBAAqB;IACrB,MAAM,UAAU,GAAG,iBAAiB,CAAC,OAAO,CAAC,CAAC;IAC9C,IAAI,UAAU,KAAK,CAAC,CAAC,EAAE,CAAC;QACtB,OAAO,IAAI,CAAC;IACd,CAAC;IAED,MAAM,IAAI,GAAG,OAAO,CAAC,KAAK,CAAC,CAAC,EAAE,UAAU,CAAC,CAAC,IAAI,EAAE,CAAC;IACjD,MAAM,KAAK,GAAG,OAAO,CAAC,KAAK,CAAC,UAAU,GAAG,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC;IAEnD,oDAAoD;IACpD,MAAM,SAAS,GAAG,IAAI,CAAC,OAAO,CAAC,cAAc,EAAE,EAAE,CAAC,CAAC;IAEnD,OAAO,EAAE,IAAI,EAAE,SAAS,EAAE,KAAK,EAAE,CAAC;AACpC,CAAC;AAED;;;GAGG;AACH,SAAS,iBAAiB,CAAC,OAAe;IACxC,IAAI,KAAK,GAAG,CAAC,CAAC;IACd,IAAI,QAAQ,GAAG,KAAK,CAAC;IACrB,IAAI,UAAU,GAAG,EAAE,CAAC;IAEpB,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,OAAO,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;QACxC,MAAM,IAAI,GAAG,OAAO,CAAC,CAAC,CAAC,CAAC;QACxB,MAAM,QAAQ,GAAG,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;QAE7C,IAAI,CAAC,IAAI,KAAK,GAAG,IAAI,IAAI,KAAK,GAAG,IAAI,IAAI,KAAK,GAAG,CAAC,IAAI,QAAQ,KAAK,IAAI,EAAE,CAAC;YACxE,IAAI,CAAC,QAAQ,EAAE,CAAC;gBACd,QAAQ,GAAG,IAAI,CAAC;gBAChB,UAAU,GAAG,IAAI,CAAC;YACpB,CAAC;iBAAM,IAAI,IAAI,KAAK,UAAU,EAAE,CAAC;gBAC/B,QAAQ,GAAG,KAAK,CAAC;YACnB,CAAC;QACH,CAAC;QAED,IAAI,CAAC,QAAQ,EAAE,CAAC;YACd,IAAI,IAAI,KAAK,GAAG,IAAI,IAAI,KAAK,GAAG,IAAI,IAAI,KAAK,GAAG,EAAE,CAAC;gBACjD,KAAK,EAAE,CAAC;YACV,CAAC;iBAAM,IAAI,IAAI,KAAK,GAAG,IAAI,IAAI,KAAK,GAAG,IAAI,IAAI,KAAK,GAAG,EAAE,CAAC;gBACxD,KAAK,EAAE,CAAC;YACV,CAAC;iBAAM,IAAI,IAAI,KAAK,GAAG,IAAI,KAAK,KAAK,CAAC,EAAE,CAAC;gBACvC,OAAO,CAAC,CAAC;YACX,CAAC;QACH,CAAC;IACH,CAAC;IAED,OAAO,CAAC,CAAC,CAAC;AACZ,CAAC;AAED;;GAEG;AACH,SAAS,qBAAqB,CAAC,KAAa;IAC1C,MAAM,OAAO,GAAG,KAAK,CAAC,IAAI,EAAE,CAAC;IAE7B,4CAA4C;IAC5C,IAAI,0BAA0B,CAAC,IAAI,CAAC,OAAO,CAAC,EAAE,CAAC;QAC7C,OAAO,OAAO,CAAC;IACjB,CAAC;IAED,4EAA4E;IAC5E,IAAI,2CAA2C,CAAC,IAAI,CAAC,OAAO,CAAC,EAAE,CAAC;QAC9D,OAAO,OAAO,CAAC;IACjB,CAAC;IAED,+CAA+C;IAC/C,IACE,CAAC,OAAO,CAAC,UAAU,CAAC,GAAG,CAAC,IAAI,OAAO,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC;QACpD,OAAO,CAAC,MAAM,GAAG,EAAE,EACnB,CAAC;QACD,OAAO,OAAO,CAAC;IACjB,CAAC;IAED,uDAAuD;IACvD,8EAA8E;IAC9E,IACE,OAAO,CAAC,UAAU,CAAC,UAAU,CAAC;QAC9B,OAAO,CAAC,UAAU,CAAC,gBAAgB,CAAC;QACpC,OAAO,CAAC,QAAQ,CAAC,IAAI,CAAC;QACtB,CAAC,OAAO,CAAC,UAAU,CAAC,GAAG,CAAC,IAAI,OAAO,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,EACnD,CAAC;QACD,OAAO,mBAAmB,CAAC;IAC7B,CAAC;IAED,kDAAkD;IAClD,IAAI,OAAO,CAAC,UAAU,CAAC,GAAG,CAAC,EAAE,CAAC;QAC5B,MAAM,KAAK,GAAG,qBAAqB,CAAC,OAAO,CAAC,CAAC;QAC7C,IAAI,KAAK,CAAC,MAAM,GAAG,EAAE,EAAE,CAAC;YACtB,OAAO,iBAAiB,CAAC;QAC3B,CAAC;QACD,OAAO,KAAK,CAAC;IACf,CAAC;IAED,6BAA6B;IAC7B,8EAA8E;IAC9E,IAAI,OAAO,CAAC,UAAU,CAAC,GAAG,CAAC,EAAE,CAAC;QAC5B,OAAO,eAAe,CAAC;IACzB,CAAC;IAED,oEAAoE;IACpE,8EAA8E;IAC9E,MAAM,UAAU,GAAG,OAAO,CAAC,KAAK,CAAC,2BAA2B,CAAC,CAAC;IAC9D,IAAI,UAAU,IAAI,OAAO,CAAC,MAAM,GAAG,EAAE,EAAE,CAAC;QACtC,OAAO,UAAU,CAAC,CAAC,CAAC,GAAG,aAAa,CAAC;IACvC,CAAC;IAED,uBAAuB;IACvB,IAAI,OAAO,CAAC,MAAM,GAAG,EAAE,EAAE,CAAC;QACxB,OAAO,OAAO,CAAC,KAAK,CAAC,CAAC,EAAE,EAAE,CAAC,GAAG,aAAa,CAAC;IAC9C,CAAC;IAED,OAAO,OAAO,CAAC;AACjB,CAAC"}

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

@@ -316,14 +316,6 @@ function findControllablePrefix(path, attributesMap, fullToShortPathMap) {
  * @returns The controllable base path if found, null otherwise
  */
 function findControllableBaseForDerivedPath(path, attributesMap) {
-    // Use cleanPathOfNonTransformingFunctions to strip non-transforming methods
-    // like .filter(), .sort(), .slice() etc. This handles all method calls generically
-    // without hardcoding specific method names.
-    const cleanedPath = cleanPathOfNonTransformingFunctions(path);
-    // If the path wasn't changed, there are no non-transforming functions to strip
-    if (cleanedPath === path) {
-        return null;
-    }
     // Helper to check if a path is in attributesMap
     const checkInAttributesMap = (pathToCheck) => {
         // Direct match
@@ -337,18 +329,104 @@ function findControllableBaseForDerivedPath(path, attributesMap) {
         }
         return null;
     };
-    // Check the cleaned path directly
+    // Use cleanPathOfNonTransformingFunctions to strip non-transforming methods
+    // like .filter(), .sort(), .slice() etc. This handles all method calls generically
+    // without hardcoding specific method names.
+    const cleanedPath = cleanPathOfNonTransformingFunctions(path);
+    // Check the cleaned path directly (even if unchanged, later steps may find something)
     const directResult = checkInAttributesMap(cleanedPath);
     if (directResult) {
         return directResult;
     }
-    // Extract the short path (after functionCallReturnValue) for attributesMap lookup
-    const functionReturnMatch = cleanedPath.match(/\.functionCallReturnValue\.(.+)$/);
-    if (functionReturnMatch) {
-        const shortPath = functionReturnMatch[1];
+    // Fix 38: For controllability purposes, strip ALL derivation operations
+    // including transforming methods like .split(...) and array indices.
+    // The key insight is: if we control the base, we control all derived values.
+    //
+    // Pattern: base.split('...').filter(...)[N] → find base
+    // We progressively strip: [N] → filter(...) → split(...) → until we find something controllable
+    // Methods that derive new values (should be stripped for controllability)
+    const derivingMethods = [
+        'split',
+        'match',
+        'filter',
+        'map',
+        'find',
+        'findIndex',
+        'slice',
+        'substring',
+        'substr',
+        'trim',
+        'trimStart',
+        'trimEnd',
+        'toLowerCase',
+        'toUpperCase',
+        'replace',
+        'replaceAll',
+        'sort',
+        'reverse',
+        'flat',
+        'flatMap',
+    ];
+    // Build regex to match method calls: .methodName(...)
+    // Use a simpler approach: progressively strip trailing derivations
+    let workingPath = cleanedPath;
+    // Keep stripping derivations until we find something controllable or can't strip anymore
+    // Max 20 iterations to prevent infinite loops
+    for (let iteration = 0; iteration < 20; iteration++) {
+        // Try current path
+        const result = checkInAttributesMap(workingPath);
+        if (result) {
+            return result;
+        }
+        // Also check via functionCallReturnValue extraction
+        // Note: We return workingPath (the FULL path) if the short path is found,
+        // because the full path is more meaningful for execution flows
+        const functionReturnMatch = workingPath.match(/\.functionCallReturnValue\.(.+)$/);
+        if (functionReturnMatch) {
+            const shortPath = functionReturnMatch[1];
+            const shortResult = checkInAttributesMap(shortPath);
+            if (shortResult) {
+                // Return the full path, not just the short result
+                // This ensures execution flows have meaningful paths like
+                // "useParams().functionCallReturnValue.*" instead of just "*"
+                return workingPath;
+            }
+        }
+        // Try to strip trailing array index: path[0] → path
+        const arrayIndexMatch = workingPath.match(/^(.+)\[\d+\]$/);
+        if (arrayIndexMatch) {
+            workingPath = arrayIndexMatch[1];
+            continue;
+        }
+        // Try to strip trailing method call: path.method(...) → path
+        let strippedMethod = false;
+        for (const method of derivingMethods) {
+            // Match .method(...) at the end, handling nested parentheses
+            const methodRegex = new RegExp(`\\.${method}\\([^()]*(?:\\([^()]*\\)[^()]*)*\\)$`);
+            if (methodRegex.test(workingPath)) {
+                workingPath = workingPath.replace(methodRegex, '');
+                strippedMethod = true;
+                break;
+            }
+        }
+        if (!strippedMethod) {
+            // Can't strip anymore
+            break;
+        }
+    }
+    // Final check after stripping
+    const finalResult = checkInAttributesMap(workingPath);
+    if (finalResult) {
+        return finalResult;
+    }
+    // Also check via functionCallReturnValue extraction for final path
+    const finalFunctionReturnMatch = workingPath.match(/\.functionCallReturnValue\.(.+)$/);
+    if (finalFunctionReturnMatch) {
+        const shortPath = finalFunctionReturnMatch[1];
         const shortResult = checkInAttributesMap(shortPath);
         if (shortResult) {
-            return shortResult;
+            // Return the full path, not just the short result
+            return workingPath;
         }
     }
     return null;
@@ -423,7 +501,11 @@ export default function resolvePathToControllable(localPath, attributesMap, equi
     }
     // 4. Equivalent variable resolution
     const localVarName = extractLocalVariableName(localPath);
-    const dataSourceBase = equivalentSignatureVariables[localVarName];
+    // Handle array case (OR expressions) - use first element if array
+    const rawDataSourceBase = equivalentSignatureVariables[localVarName];
+    const dataSourceBase = Array.isArray(rawDataSourceBase)
+        ? rawDataSourceBase[0]
+        : rawDataSourceBase;
     if (dataSourceBase) {
         chain.push(`equivalent var: ${localVarName} → ${dataSourceBase}`);
         // Build the full resolved path
@@ -438,20 +520,26 @@ export default function resolvePathToControllable(localPath, attributesMap, equi
             const accessSuffix = arrayMatch[2]; // e.g., "[0]" or "[0].property"
             if (baseVarName in equivalentSignatureVariables &&
                 baseVarName !== localVarName) {
-                const baseDataSource = equivalentSignatureVariables[baseVarName];
-                chain.push(`transitive resolution: ${baseVarName} → ${baseDataSource}`);
-                // Append the array access suffix to the resolved base
-                if (baseDataSource.endsWith('()')) {
-                    fullResolvedPath =
-                        baseDataSource + '.functionCallReturnValue' + accessSuffix;
-                }
-                else if (baseDataSource.endsWith('.functionCallReturnValue')) {
-                    fullResolvedPath = baseDataSource + accessSuffix;
+                // Handle array case (OR expressions) - use first element if array
+                const rawBaseDataSource = equivalentSignatureVariables[baseVarName];
+                const baseDataSource = Array.isArray(rawBaseDataSource)
+                    ? rawBaseDataSource[0]
+                    : rawBaseDataSource;
+                if (baseDataSource) {
+                    chain.push(`transitive resolution: ${baseVarName} → ${baseDataSource}`);
+                    // Append the array access suffix to the resolved base
+                    if (baseDataSource.endsWith('()')) {
+                        fullResolvedPath =
+                            baseDataSource + '.functionCallReturnValue' + accessSuffix;
+                    }
+                    else if (baseDataSource.endsWith('.functionCallReturnValue')) {
+                        fullResolvedPath = baseDataSource + accessSuffix;
+                    }
+                    else {
+                        fullResolvedPath = baseDataSource + accessSuffix;
+                    }
+                    chain.push(`transitively resolved: ${fullResolvedPath}`);
                 }
-                else {
-                    fullResolvedPath = baseDataSource + accessSuffix;
-                }
-                chain.push(`transitively resolved: ${fullResolvedPath}`);
             }
         }
         // Check if resolved path is controllable
@@ -490,15 +578,18 @@ export default function resolvePathToControllable(localPath, attributesMap, equi
                 isControllable: true,
             };
         }
-        // Fix 34: Handle derived paths from controllable bases
+        // Fix 34/38: Handle derived paths from controllable bases
         // Paths like "useParams().functionCallReturnValue.*.split('/').filter(Boolean)[0]"
         // are derived from "useParams().functionCallReturnValue.*" via string methods.
-        // If the base path is controllable, the derived path should also be controllable.
+        // If the base path is controllable, return the BASE path (not the derived path)
+        // because that's what we need to mock in the data structure.
         const derivedBaseResult = findControllableBaseForDerivedPath(fullResolvedPath, attributesMap);
         if (derivedBaseResult) {
             chain.push(`derived from controllable base: ${derivedBaseResult}`);
             return {
-                resolvedPath: fullResolvedPath,
+                // Return the controllable BASE path, not the derived path
+                // This ensures execution flows contain paths that exist in attributesMap
+                resolvedPath: derivedBaseResult,
                 resolutionChain: chain,
                 isControllable: true,
             };
@@ -520,23 +611,46 @@ export default function resolvePathToControllable(localPath, attributesMap, equi
     // 5. Fix 35: Check for controllable prefix on the local path directly
     // This handles cases where the localPath itself (before any resolution)
     // has a controllable prefix in attributesMap
-    const localPrefixResult = findControllablePrefix(localPath, attributesMap, fullToShortPathMap);
-    if (localPrefixResult) {
-        chain.push(`nested property of controllable prefix (direct)`);
-        return {
-            resolvedPath: localPath,
-            resolutionChain: chain,
-            isControllable: true,
-        };
+    //
+    // IMPORTANT: Only do prefix matching if:
+    // a) The root variable is known in equivalentSignatureVariables, OR
+    // b) The path looks like an already-translated path (contains .functionCallReturnValue.)
+    //
+    // If the root variable is NOT known AND it's not an already-translated path, it's likely
+    // from an internal scope (callback, async function) and shouldn't be matched
+    // against similarly-named variables from different scopes.
+    // For example, "data.status" from inside `await response.json()` should NOT
+    // match the "data" from a useFetcher hook.
+    const localVarNameForPrefix = extractLocalVariableName(localPath);
+    const isKnownVariable = localVarNameForPrefix in equivalentSignatureVariables;
+    // Fix 35b: Also allow prefix matching for already-translated paths from child components
+    // These paths start with hook patterns like useLoaderData<...>().functionCallReturnValue
+    // and should be checked for controllable prefixes even if the hook itself isn't
+    // in equivalentSignatureVariables (because it's a fully resolved path, not a local variable)
+    const isAlreadyTranslatedPath = localPath.includes('.functionCallReturnValue.');
+    if (isKnownVariable || isAlreadyTranslatedPath) {
+        const localPrefixResult = findControllablePrefix(localPath, attributesMap, fullToShortPathMap);
+        if (localPrefixResult) {
+            chain.push(`nested property of controllable prefix (direct)`);
+            return {
+                resolvedPath: localPath,
+                resolutionChain: chain,
+                isControllable: true,
+            };
+        }
     }
-    // 5b. Fix 36: Check for derived paths on localPath directly
+    else {
+        chain.push(`root variable '${localVarNameForPrefix}' not in equivalentSignatureVariables - skipping prefix match`);
+    }
+    // 5b. Fix 36/38: Check for derived paths on localPath directly
     // When localPath contains derivation methods like .filter(cyScope()), check if
     // the base path (before .filter()) is controllable
     const localDerivedResult = findControllableBaseForDerivedPath(localPath, attributesMap);
     if (localDerivedResult) {
         chain.push(`derived from controllable base (direct): ${localDerivedResult}`);
         return {
-            resolvedPath: localPath,
+            // Return the controllable BASE path, not the derived path
+            resolvedPath: localDerivedResult,
             resolutionChain: chain,
             isControllable: true,
         };