genaicode 0.14.2 → 1.0.0

package/dist/prompt/steps/step-iterate/handlers/handle-remove-files-from-context.js

@@ -0,0 +1,100 @@
+import { ModelType } from '../../../../ai-service/common-types.js';
+import { putSystemMessage } from '../../../../main/common/content-bus.js';
+import { getFunctionDefs } from '../../../function-calling.js';
+import { registerActionHandler } from '../step-iterate-handlers.js';
+registerActionHandler('removeFilesFromContext', handleRemoveFilesFromContext);
+export async function handleRemoveFilesFromContext({ iterateCall, generateContentFn, prompt, options, }) {
+    const [removeFilesFromContextCall] = (await generateContentFn([
+        ...prompt,
+        {
+            type: 'assistant',
+            text: iterateCall.args?.message ?? '',
+        },
+        {
+            type: 'user',
+            text: 'Yes, you can request the removal of files from context.',
+        },
+    ], {
+        functionDefs: getFunctionDefs(),
+        requiredFunctionName: 'removeFilesFromContext',
+        temperature: 0.7,
+        modelType: ModelType.CHEAP,
+        expectedResponseType: {
+            text: false,
+            functionCall: true,
+            media: false,
+        },
+    }, options))
+        .filter((item) => item.type === 'functionCall')
+        .map((item) => item.functionCall);
+    if (!removeFilesFromContextCall) {
+        return {
+            breakLoop: false,
+            items: [],
+        };
+    }
+    const filesToRemove = removeFilesFromContextCall.args?.filePaths ?? [];
+    let userText = '';
+    if (filesToRemove.length > 0) {
+        removeFileContentsFromPrompt(prompt, filesToRemove);
+        putSystemMessage('Context reduction applied', filesToRemove);
+        userText = 'Context reduction applied';
+    }
+    else {
+        putSystemMessage('No specific files were provided for context reduction. The context remains unchanged.');
+        userText = 'No specific files were provided for context reduction. The context remains unchanged.';
+    }
+    prompt.push({
+        type: 'assistant',
+        text: iterateCall.args?.message ?? '',
+        functionCalls: [removeFilesFromContextCall],
+    }, {
+        type: 'user',
+        text: userText,
+        functionResponses: [
+            { name: 'removeFilesFromContext', call_id: removeFilesFromContextCall.id, content: undefined },
+        ],
+    });
+    return {
+        breakLoop: false,
+        items: [],
+    };
+}
+function removeFileContentsFromPrompt(prompt, filesToRemove) {
+    // Find all source code responses in the prompt history
+    const sourceCodeResponses = prompt.filter((item) => item.type === 'user' && item.functionResponses?.some((fr) => fr.name === 'getSourceCode'));
+    if (sourceCodeResponses.length === 0) {
+        console.warn('No source code responses found in prompt history');
+        return;
+    }
+    sourceCodeResponses.forEach((response) => {
+        const sourceCodeResponse = response.functionResponses?.find((fr) => fr.name === 'getSourceCode');
+        if (!sourceCodeResponse || !sourceCodeResponse.content) {
+            console.warn('Skipping invalid source code response');
+            return;
+        }
+        try {
+            const contentObj = JSON.parse(sourceCodeResponse.content);
+            let modifiedFiles = 0;
+            filesToRemove.forEach((file) => {
+                if (contentObj[file] && 'content' in contentObj[file]) {
+                    contentObj[file].content = null;
+                    modifiedFiles++;
+                }
+                else if (contentObj[file] && 'fragments' in contentObj[file]) {
+                    delete contentObj[file].fragments;
+                    modifiedFiles++;
+                }
+            });
+            // Only update if files were actually modified
+            if (modifiedFiles > 0) {
+                sourceCodeResponse.content = JSON.stringify(contentObj);
+                console.log(`Removed content from ${modifiedFiles} files in source code response`);
+            }
+        }
+        catch (error) {
+            console.error('Error processing source code response:', error);
+        }
+    });
+}
+//# sourceMappingURL=handle-remove-files-from-context.js.map

package/dist/prompt/steps/step-iterate/handlers/handle-remove-files-from-context.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"handle-remove-files-from-context.js","sourceRoot":"","sources":["../../../../../src/prompt/steps/step-iterate/handlers/handle-remove-files-from-context.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,SAAS,EAAE,MAAM,wCAAwC,CAAC;AACnE,OAAO,EAAE,gBAAgB,EAAE,MAAM,wCAAwC,CAAC;AAC1E,OAAO,EAAE,eAAe,EAAE,MAAM,8BAA8B,CAAC;AAC/D,OAAO,EAAE,qBAAqB,EAAE,MAAM,6BAA6B,CAAC;AAGpE,qBAAqB,CAAC,wBAAwB,EAAE,4BAA4B,CAAC,CAAC;AAE9E,MAAM,CAAC,KAAK,UAAU,4BAA4B,CAAC,EACjD,WAAW,EACX,iBAAiB,EACjB,MAAM,EACN,OAAO,GACY;IACnB,MAAM,CAAC,0BAA0B,CAAC,GAAG,CACnC,MAAM,iBAAiB,CACrB;QACE,GAAG,MAAM;QACT;YACE,IAAI,EAAE,WAAW;YACjB,IAAI,EAAE,WAAW,CAAC,IAAI,EAAE,OAAO,IAAI,EAAE;SACtC;QACD;YACE,IAAI,EAAE,MAAM;YACZ,IAAI,EAAE,yDAAyD;SAChE;KACF,EACD;QACE,YAAY,EAAE,eAAe,EAAE;QAC/B,oBAAoB,EAAE,wBAAwB;QAC9C,WAAW,EAAE,GAAG;QAChB,SAAS,EAAE,SAAS,CAAC,KAAK;QAC1B,oBAAoB,EAAE;YACpB,IAAI,EAAE,KAAK;YACX,YAAY,EAAE,IAAI;YAClB,KAAK,EAAE,KAAK;SACb;KACF,EACD,OAAO,CACR,CACF;SACE,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,KAAK,cAAc,CAAC;SAC9C,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,YAAY,CAA2D,CAAC;IAE9F,IAAI,CAAC,0BAA0B,EAAE,CAAC;QAChC,OAAO;YACL,SAAS,EAAE,KAAK;YAChB,KAAK,EAAE,EAAE;SACV,CAAC;IACJ,CAAC;IAED,MAAM,aAAa,GAAG,0BAA0B,CAAC,IAAI,EAAE,SAAS,IAAI,EAAE,CAAC;IACvE,IAAI,QAAQ,GAAG,EAAE,CAAC;IAElB,IAAI,aAAa,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QAC7B,4BAA4B,CAAC,MAAM,EAAE,aAAa,CAAC,CAAC;QACpD,gBAAgB,CAAC,2BAA2B,EAAE,aAAa,CAAC,CAAC;QAC7D,QAAQ,GAAG,2BAA2B,CAAC;IACzC,CAAC;SAAM,CAAC;QACN,gBAAgB,CAAC,uFAAuF,CAAC,CAAC;QAC1G,QAAQ,GAAG,uFAAuF,CAAC;IACrG,CAAC;IAED,MAAM,CAAC,IAAI,CACT;QACE,IAAI,EAAE,WAAW;QACjB,IAAI,EAAE,WAAW,CAAC,IAAI,EAAE,OAAO,IAAI,EAAE;QACrC,aAAa,EAAE,CAAC,0BAA0B,CAAC;KAC5C,EACD;QACE,IAAI,EAAE,MAAM;QACZ,IAAI,EAAE,QAAQ;QACd,iBAAiB,EAAE;YACjB,EAAE,IAAI,EAAE,wBAAwB,EAAE,OAAO,EAAE,0BAA0B,CAAC,EAAE,EAAE,OAAO,EAAE,SAAS,EAAE;SAC/F;KACF,CACF,CAAC;IAEF,OAAO;QACL,SAAS,EAAE,KAAK;QAChB,KAAK,EAAE,EAAE;KACV,CAAC;AACJ,CAAC;AAED,SAAS,4BAA4B,CAAC,MAAoB,EAAE,aAAuB;IACjF,uDAAuD;IACvD,MAAM,mBAAmB,GAAG,MAAM,CAAC,MAAM,CACvC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,KAAK,MAAM,IAAI,IAAI,CAAC,iBAAiB,EAAE,IAAI,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,EAAE,CAAC,IAAI,KAAK,eAAe,CAAC,CACpG,CAAC;IAEF,IAAI,mBAAmB,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QACrC,OAAO,CAAC,IAAI,CAAC,kDAAkD,CAAC,CAAC;QACjE,OAAO;IACT,CAAC;IAED,mBAAmB,CAAC,OAAO,CAAC,CAAC,QAAQ,EAAE,EAAE;QACvC,MAAM,kBAAkB,GAAG,QAAQ,CAAC,iBAAiB,EAAE,IAAI,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,EAAE,CAAC,IAAI,KAAK,eAAe,CAAC,CAAC;QAEjG,IAAI,CAAC,kBAAkB,IAAI,CAAC,kBAAkB,CAAC,OAAO,EAAE,CAAC;YACvD,OAAO,CAAC,IAAI,CAAC,uCAAuC,CAAC,CAAC;YACtD,OAAO;QACT,CAAC;QAED,IAAI,CAAC;YACH,MAAM,UAAU,GAAG,IAAI,CAAC,KAAK,CAAC,kBAAkB,CAAC,OAAO,CAAC,CAAC;YAE1D,IAAI,aAAa,GAAG,CAAC,CAAC;YACtB,aAAa,CAAC,OAAO,CAAC,CAAC,IAAI,EAAE,EAAE;gBAC7B,IAAI,UAAU,CAAC,IAAI,CAAC,IAAI,SAAS,IAAI,UAAU,CAAC,IAAI,CAAC,EAAE,CAAC;oBACtD,UAAU,CAAC,IAAI,CAAC,CAAC,OAAO,GAAG,IAAI,CAAC;oBAChC,aAAa,EAAE,CAAC;gBAClB,CAAC;qBAAM,IAAI,UAAU,CAAC,IAAI,CAAC,IAAI,WAAW,IAAI,UAAU,CAAC,IAAI,CAAC,EAAE,CAAC;oBAC/D,OAAO,UAAU,CAAC,IAAI,CAAC,CAAC,SAAS,CAAC;oBAClC,aAAa,EAAE,CAAC;gBAClB,CAAC;YACH,CAAC,CAAC,CAAC;YAEH,8CAA8C;YAC9C,IAAI,aAAa,GAAG,CAAC,EAAE,CAAC;gBACtB,kBAAkB,CAAC,OAAO,GAAG,IAAI,CAAC,SAAS,CAAC,UAAU,CAAC,CAAC;gBACxD,OAAO,CAAC,GAAG,CAAC,wBAAwB,aAAa,gCAAgC,CAAC,CAAC;YACrF,CAAC;QACH,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,CAAC,KAAK,CAAC,wCAAwC,EAAE,KAAK,CAAC,CAAC;QACjE,CAAC;IACH,CAAC,CAAC,CAAC;AACL,CAAC"}

package/dist/prompt/steps/step-iterate/handlers/handle-request-files-content.d.ts

@@ -0,0 +1,8 @@
+import { GenerateContentFunction } from '../../../../ai-service/common-types.js';
+import { PromptItem } from '../../../../ai-service/common-types.js';
+import { FunctionCall } from '../../../../ai-service/common-types.js';
+import { ModelType } from '../../../../ai-service/common-types.js';
+import { CodegenOptions } from '../../../../main/codegen-types.js';
+import { ActionHandlerProps, ActionResult, IterateCall, RequestFilesContentArgs } from '../step-iterate-types.js';
+export declare function handleRequestFilesContent({ iterateCall, options, prompt, generateContentFn, }: ActionHandlerProps): Promise<ActionResult>;
+export declare function generateRequestFilesContentCall(generateContentFn: GenerateContentFunction, prompt: PromptItem[], iterateCall: IterateCall, options: CodegenOptions, modelType: ModelType): Promise<FunctionCall<RequestFilesContentArgs> | undefined>;

package/dist/prompt/steps/step-iterate/handlers/handle-request-files-content.js

@@ -0,0 +1,146 @@
+import { ModelType } from '../../../../ai-service/common-types.js';
+import { refreshFiles } from '../../../../files/find-files.js';
+import { getSourceCode } from '../../../../files/read-files.js';
+import { getExpandedContextPaths } from '../../../../files/source-code-utils.js';
+import { putSystemMessage } from '../../../../main/common/content-bus.js';
+import { getFunctionDefs } from '../../../function-calling.js';
+import { registerActionHandler } from '../step-iterate-handlers.js';
+import { isFileContentAlreadyProvided, categorizeLegitimateFiles } from './file-request-utils.js';
+registerActionHandler('requestFilesContent', handleRequestFilesContent);
+export async function handleRequestFilesContent({ iterateCall, options, prompt, generateContentFn, }) {
+    let requestFilesContentCall = await generateRequestFilesContentCall(generateContentFn, prompt, iterateCall, options, ModelType.CHEAP);
+    if (!requestFilesContentCall) {
+        return {
+            breakLoop: false,
+            items: [],
+        };
+    }
+    let requestedFiles = getExpandedContextPaths(requestFilesContentCall.args?.filePaths ?? [], options);
+    putSystemMessage('Requesting files content', {
+        requestedFiles,
+        expandedRequestedFiles: requestedFiles.filter((file) => !requestFilesContentCall?.args?.filePaths?.includes(file)),
+    });
+    // Check which files are already provided in the conversation history
+    const alreadyProvidedFiles = requestedFiles.filter((file) => isFileContentAlreadyProvided(file, prompt));
+    if (alreadyProvidedFiles.length === requestedFiles.length) {
+        putSystemMessage('All requested files are already provided');
+        // All requested files are already provided
+        prompt.push({
+            type: 'assistant',
+            text: iterateCall.args?.message ?? '',
+            functionCalls: [requestFilesContentCall],
+        }, {
+            type: 'user',
+            text: 'The requested file contents are already provided in the conversation history.',
+            functionResponses: [
+                {
+                    name: 'requestFilesContent',
+                    call_id: requestFilesContentCall.id,
+                    content: JSON.stringify({ filePaths: requestedFiles }),
+                },
+            ],
+            cache: true,
+        });
+        return { breakLoop: false, items: [] };
+    }
+    // Filter out already provided files
+    requestedFiles = requestedFiles.filter((file) => !isFileContentAlreadyProvided(file, prompt));
+    // the request may be caused be an appearance of a new file, so lets refresh
+    refreshFiles();
+    let { legitimateFiles, illegitimateFiles } = categorizeLegitimateFiles(requestedFiles);
+    if (illegitimateFiles.length > 0) {
+        requestFilesContentCall = await generateRequestFilesContentCall(generateContentFn, prompt, iterateCall, options, 
+        // use non cheap mode, as maybe the cheap mode didn't provide correct files
+        ModelType.DEFAULT);
+        if (!requestFilesContentCall) {
+            return {
+                breakLoop: false,
+                items: [],
+            };
+        }
+        requestedFiles = requestFilesContentCall.args?.filePaths ?? [];
+        const categorized = categorizeLegitimateFiles(requestedFiles);
+        legitimateFiles = categorized.legitimateFiles;
+        illegitimateFiles = categorized.illegitimateFiles;
+    }
+    const sourceCallId = (iterateCall.id ?? iterateCall.name) + '_source';
+    const assistant = {
+        type: 'assistant',
+        text: iterateCall.args?.message ?? '',
+        functionCalls: [
+            requestFilesContentCall,
+            { name: 'getSourceCode', id: sourceCallId, args: { filePaths: legitimateFiles } },
+        ],
+    };
+    const sourceCode = getSourceCode({ filterPaths: legitimateFiles, forceAll: true, ignoreImportantFiles: true }, options);
+    const user = {
+        type: 'user',
+        text: (alreadyProvidedFiles.length > 0
+            ? // TODO: Maybe reference to some specific getSourceCode call id
+                `Some files were already provided in the conversation history:
+${alreadyProvidedFiles.map((path) => `- ${path}`).join('\n')}
+
+Providing content for the remaining files:
+${requestedFiles.map((path) => `- ${path}`).join('\n')}`
+            : `All requested file contents have been provided:
+${legitimateFiles.map((path) => `- ${path}`).join('\n')}`) +
+            (illegitimateFiles.length > 0
+                ? `\n\nSome files are not legitimate and their content cannot be provided:
+${illegitimateFiles.map((path) => `- ${path}`).join('\n')}`
+                : '') +
+            '\n\nPlease consider removing files that are no longer needed from the context using `removeFilesFromContext`.',
+        data: { legitimateFiles, illegitimateFiles },
+        functionResponses: [
+            {
+                name: 'requestFilesContent',
+                call_id: requestFilesContentCall.id,
+                content: JSON.stringify({ filePaths: legitimateFiles }),
+            },
+            {
+                name: 'getSourceCode',
+                call_id: sourceCallId,
+                content: JSON.stringify(sourceCode),
+            },
+        ],
+        cache: true,
+    };
+    putSystemMessage('Providing files content', {
+        requestedFiles,
+        legitimateFiles,
+        illegitimateFiles,
+    });
+    prompt.push(assistant, user);
+    return { breakLoop: false, items: [] };
+}
+export async function generateRequestFilesContentCall(generateContentFn, prompt, iterateCall, options, modelType) {
+    const req = [
+        [
+            ...prompt,
+            {
+                type: 'assistant',
+                text: iterateCall.args?.message ?? '',
+            },
+            {
+                type: 'user',
+                text: 'Yes, you can request the files contents.',
+            },
+        ],
+        {
+            functionDefs: getFunctionDefs(),
+            requiredFunctionName: 'requestFilesContent',
+            temperature: 0.7,
+            modelType,
+            expectedResponseType: {
+                text: false,
+                functionCall: true,
+                media: false,
+            },
+        },
+        options,
+    ];
+    const [requestFilesContentCall] = (await generateContentFn(...req))
+        .filter((item) => item.type === 'functionCall')
+        .map((item) => item.functionCall);
+    return requestFilesContentCall;
+}
+//# sourceMappingURL=handle-request-files-content.js.map

package/dist/prompt/steps/step-iterate/handlers/handle-request-files-content.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"handle-request-files-content.js","sourceRoot":"","sources":["../../../../../src/prompt/steps/step-iterate/handlers/handle-request-files-content.ts"],"names":[],"mappings":"AAGA,OAAO,EAAE,SAAS,EAAE,MAAM,wCAAwC,CAAC;AACnE,OAAO,EAAE,YAAY,EAAE,MAAM,iCAAiC,CAAC;AAC/D,OAAO,EAAE,aAAa,EAAE,MAAM,iCAAiC,CAAC;AAChE,OAAO,EAAE,uBAAuB,EAAE,MAAM,wCAAwC,CAAC;AAEjF,OAAO,EAAE,gBAAgB,EAAE,MAAM,wCAAwC,CAAC;AAC1E,OAAO,EAAE,eAAe,EAAE,MAAM,8BAA8B,CAAC;AAC/D,OAAO,EAAE,qBAAqB,EAAE,MAAM,6BAA6B,CAAC;AASpE,OAAO,EAAE,4BAA4B,EAAE,yBAAyB,EAAE,MAAM,yBAAyB,CAAC;AAElG,qBAAqB,CAAC,qBAAqB,EAAE,yBAAyB,CAAC,CAAC;AAExE,MAAM,CAAC,KAAK,UAAU,yBAAyB,CAAC,EAC9C,WAAW,EACX,OAAO,EACP,MAAM,EACN,iBAAiB,GACE;IACnB,IAAI,uBAAuB,GAAG,MAAM,+BAA+B,CACjE,iBAAiB,EACjB,MAAM,EACN,WAAW,EACX,OAAO,EACP,SAAS,CAAC,KAAK,CAChB,CAAC;IAEF,IAAI,CAAC,uBAAuB,EAAE,CAAC;QAC7B,OAAO;YACL,SAAS,EAAE,KAAK;YAChB,KAAK,EAAE,EAAE;SACV,CAAC;IACJ,CAAC;IAED,IAAI,cAAc,GAAG,uBAAuB,CAAC,uBAAuB,CAAC,IAAI,EAAE,SAAS,IAAI,EAAE,EAAE,OAAO,CAAC,CAAC;IAErG,gBAAgB,CAAC,0BAA0B,EAAE;QAC3C,cAAc;QACd,sBAAsB,EAAE,cAAc,CAAC,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,uBAAuB,EAAE,IAAI,EAAE,SAAS,EAAE,QAAQ,CAAC,IAAI,CAAC,CAAC;KACnH,CAAC,CAAC;IAEH,qEAAqE;IACrE,MAAM,oBAAoB,GAAG,cAAc,CAAC,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,4BAA4B,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC,CAAC;IAEzG,IAAI,oBAAoB,CAAC,MAAM,KAAK,cAAc,CAAC,MAAM,EAAE,CAAC;QAC1D,gBAAgB,CAAC,0CAA0C,CAAC,CAAC;QAC7D,2CAA2C;QAC3C,MAAM,CAAC,IAAI,CACT;YACE,IAAI,EAAE,WAAW;YACjB,IAAI,EAAE,WAAW,CAAC,IAAI,EAAE,OAAO,IAAI,EAAE;YACrC,aAAa,EAAE,CAAC,uBAAuB,CAAC;SACzC,EACD;YACE,IAAI,EAAE,MAAM;YACZ,IAAI,EAAE,+EAA+E;YACrF,iBAAiB,EAAE;gBACjB;oBACE,IAAI,EAAE,qBAAqB;oBAC3B,OAAO,EAAE,uBAAuB,CAAC,EAAE;oBACnC,OAAO,EAAE,IAAI,CAAC,SAAS,CAAC,EAAE,SAAS,EAAE,cAAc,EAAE,CAAC;iBACvD;aACF;YACD,KAAK,EAAE,IAAI;SACZ,CACF,CAAC;QAEF,OAAO,EAAE,SAAS,EAAE,KAAK,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC;IACzC,CAAC;IAED,oCAAoC;IACpC,cAAc,GAAG,cAAc,CAAC,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,4BAA4B,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC,CAAC;IAE9F,4EAA4E;IAC5E,YAAY,EAAE,CAAC;IAEf,IAAI,EAAE,eAAe,EAAE,iBAAiB,EAAE,GAAG,yBAAyB,CAAC,cAAc,CAAC,CAAC;IAEvF,IAAI,iBAAiB,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QACjC,uBAAuB,GAAG,MAAM,+BAA+B,CAC7D,iBAAiB,EACjB,MAAM,EACN,WAAW,EACX,OAAO;QACP,2EAA2E;QAC3E,SAAS,CAAC,OAAO,CAClB,CAAC;QAEF,IAAI,CAAC,uBAAuB,EAAE,CAAC;YAC7B,OAAO;gBACL,SAAS,EAAE,KAAK;gBAChB,KAAK,EAAE,EAAE;aACV,CAAC;QACJ,CAAC;QAED,cAAc,GAAG,uBAAuB,CAAC,IAAI,EAAE,SAAS,IAAI,EAAE,CAAC;QAC/D,MAAM,WAAW,GAAG,yBAAyB,CAAC,cAAc,CAAC,CAAC;QAC9D,eAAe,GAAG,WAAW,CAAC,eAAe,CAAC;QAC9C,iBAAiB,GAAG,WAAW,CAAC,iBAAiB,CAAC;IACpD,CAAC;IAED,MAAM,YAAY,GAAG,CAAC,WAAW,CAAC,EAAE,IAAI,WAAW,CAAC,IAAI,CAAC,GAAG,SAAS,CAAC;IACtE,MAAM,SAAS,GAAkB;QAC/B,IAAI,EAAE,WAAW;QACjB,IAAI,EAAE,WAAW,CAAC,IAAI,EAAE,OAAO,IAAI,EAAE;QACrC,aAAa,EAAE;YACb,uBAAuB;YACvB,EAAE,IAAI,EAAE,eAAe,EAAE,EAAE,EAAE,YAAY,EAAE,IAAI,EAAE,EAAE,SAAS,EAAE,eAAe,EAAE,EAAE;SAClF;KACF,CAAC;IAEF,MAAM,UAAU,GAAG,aAAa,CAC9B,EAAE,WAAW,EAAE,eAAe,EAAE,QAAQ,EAAE,IAAI,EAAE,oBAAoB,EAAE,IAAI,EAAE,EAC5E,OAAO,CACR,CAAC;IACF,MAAM,IAAI,GAAa;QACrB,IAAI,EAAE,MAAM;QACZ,IAAI,EACF,CAAC,oBAAoB,CAAC,MAAM,GAAG,CAAC;YAC9B,CAAC,CAAC,+DAA+D;gBAC/D;EACR,oBAAoB,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,KAAK,IAAI,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC;;;EAG1D,cAAc,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,KAAK,IAAI,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE;YAChD,CAAC,CAAC;EACR,eAAe,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,KAAK,IAAI,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC;YACpD,CAAC,iBAAiB,CAAC,MAAM,GAAG,CAAC;gBAC3B,CAAC,CAAC;EACR,iBAAiB,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,KAAK,IAAI,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE;gBACnD,CAAC,CAAC,EAAE,CAAC;YACP,+GAA+G;QACjH,IAAI,EAAE,EAAE,eAAe,EAAE,iBAAiB,EAAE;QAC5C,iBAAiB,EAAE;YACjB;gBACE,IAAI,EAAE,qBAAqB;gBAC3B,OAAO,EAAE,uBAAuB,CAAC,EAAE;gBACnC,OAAO,EAAE,IAAI,CAAC,SAAS,CAAC,EAAE,SAAS,EAAE,eAAe,EAAE,CAAC;aACxD;YACD;gBACE,IAAI,EAAE,eAAe;gBACrB,OAAO,EAAE,YAAY;gBACrB,OAAO,EAAE,IAAI,CAAC,SAAS,CAAC,UAAU,CAAC;aACpC;SACF;QACD,KAAK,EAAE,IAAI;KACZ,CAAC;IAEF,gBAAgB,CAAC,yBAAyB,EAAE;QAC1C,cAAc;QACd,eAAe;QACf,iBAAiB;KAClB,CAAC,CAAC;IAEH,MAAM,CAAC,IAAI,CAAC,SAAS,EAAE,IAAI,CAAC,CAAC;IAE7B,OAAO,EAAE,SAAS,EAAE,KAAK,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC;AACzC,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,+BAA+B,CACnD,iBAA0C,EAC1C,MAAoB,EACpB,WAAwB,EACxB,OAAuB,EACvB,SAAoB;IAEpB,MAAM,GAAG,GAAwB;QAC/B;YACE,GAAG,MAAM;YACT;gBACE,IAAI,EAAE,WAAW;gBACjB,IAAI,EAAE,WAAW,CAAC,IAAI,EAAE,OAAO,IAAI,EAAE;aACtC;YACD;gBACE,IAAI,EAAE,MAAM;gBACZ,IAAI,EAAE,0CAA0C;aACjD;SACF;QACD;YACE,YAAY,EAAE,eAAe,EAAE;YAC/B,oBAAoB,EAAE,qBAAqB;YAC3C,WAAW,EAAE,GAAG;YAChB,SAAS;YACT,oBAAoB,EAAE;gBACpB,IAAI,EAAE,KAAK;gBACX,YAAY,EAAE,IAAI;gBAClB,KAAK,EAAE,KAAK;aACb;SACF;QACD,OAAO;KACR,CAAC;IACF,MAAM,CAAC,uBAAuB,CAAC,GAAG,CAAC,MAAM,iBAAiB,CAAC,GAAG,GAAG,CAAC,CAAC;SAChE,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,KAAK,cAAc,CAAC;SAC9C,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,YAAY,CAAwD,CAAC;IAE3F,OAAO,uBAAuB,CAAC;AACjC,CAAC"}

package/dist/prompt/steps/step-iterate/handlers/handle-request-files-fragments.d.ts

@@ -0,0 +1,8 @@
+import { GenerateContentFunction } from '../../../../ai-service/common-types.js';
+import { PromptItem } from '../../../../ai-service/common-types.js';
+import { FunctionCall } from '../../../../ai-service/common-types.js';
+import { ModelType } from '../../../../ai-service/common-types.js';
+import { CodegenOptions } from '../../../../main/codegen-types.js';
+import { ActionHandlerProps, ActionResult, IterateCall, RequestFilesFragmentsArgs } from '../step-iterate-types.js';
+export declare function handleRequestFilesFragments({ iterateCall, options, prompt, generateContentFn, }: ActionHandlerProps): Promise<ActionResult>;
+export declare function generateRequestFilesFragmentsCall(generateContentFn: GenerateContentFunction, prompt: PromptItem[], iterateCall: IterateCall, options: CodegenOptions, modelType: ModelType): Promise<FunctionCall<RequestFilesFragmentsArgs> | undefined>;

package/dist/prompt/steps/step-iterate/handlers/handle-request-files-fragments.js

@@ -0,0 +1,197 @@
+import { ModelType } from '../../../../ai-service/common-types.js';
+import { generateFileId } from '../../../../files/file-id-utils.js';
+import { getSourceCode } from '../../../../files/read-files.js';
+import { putSystemMessage } from '../../../../main/common/content-bus.js';
+import { getFunctionDefs } from '../../../function-calling.js';
+import { registerActionHandler } from '../step-iterate-handlers.js';
+import { processFileRequests, generateFilesContentPrompt } from './file-request-utils.js';
+registerActionHandler('requestFilesFragments', handleRequestFilesFragments);
+export async function handleRequestFilesFragments({ iterateCall, options, prompt, generateContentFn, }) {
+    let requestFilesFragmentsCall = await generateRequestFilesFragmentsCall(generateContentFn, prompt, iterateCall, options, ModelType.CHEAP);
+    if (!requestFilesFragmentsCall) {
+        return {
+            breakLoop: false,
+            items: [],
+        };
+    }
+    const fragmentPrompt = requestFilesFragmentsCall.args?.fragmentPrompt;
+    if (!fragmentPrompt) {
+        putSystemMessage('No fragment prompt provided');
+        return {
+            breakLoop: false,
+            items: [],
+        };
+    }
+    // Process file requests using the utility function
+    let { requestedFiles, alreadyProvidedFiles, legitimateFiles, illegitimateFiles } = processFileRequests(requestFilesFragmentsCall.args?.filePaths ?? [], prompt, options);
+    if (alreadyProvidedFiles.length === requestedFiles.length) {
+        putSystemMessage('All requested files are already provided');
+        prompt.push({
+            type: 'assistant',
+            text: iterateCall.args?.message ?? '',
+            functionCalls: [requestFilesFragmentsCall],
+        }, {
+            type: 'user',
+            text: 'The requested file fragments are already provided in the conversation history.',
+            functionResponses: [
+                {
+                    name: 'requestFilesFragments',
+                    call_id: requestFilesFragmentsCall.id,
+                    content: JSON.stringify({ filePaths: requestedFiles }),
+                },
+            ],
+            cache: true,
+        });
+        return { breakLoop: false, items: [] };
+    }
+    // If there are illegitimate files, try again with non-cheap mode
+    if (illegitimateFiles.length > 0) {
+        requestFilesFragmentsCall = await generateRequestFilesFragmentsCall(generateContentFn, prompt, iterateCall, options, ModelType.DEFAULT);
+        if (!requestFilesFragmentsCall) {
+            return {
+                breakLoop: false,
+                items: [],
+            };
+        }
+        const processed = processFileRequests(requestFilesFragmentsCall.args?.filePaths ?? [], prompt, options);
+        requestedFiles = processed.requestedFiles;
+        alreadyProvidedFiles = processed.alreadyProvidedFiles;
+        legitimateFiles = processed.legitimateFiles;
+        illegitimateFiles = processed.illegitimateFiles;
+    }
+    // Get the content of legitimate files
+    const sourceCodeMap = await getSourceCode({ filterPaths: legitimateFiles, forceAll: true, ignoreImportantFiles: true }, options);
+    // Extract fragments from files using LLM
+    const extractedFragments = await extractFragments(sourceCodeMap, fragmentPrompt, generateContentFn, options);
+    const sourceCallId = (iterateCall.id ?? iterateCall.name) + '_source';
+    const assistant = {
+        type: 'assistant',
+        text: iterateCall.args?.message ?? '',
+        functionCalls: [
+            requestFilesFragmentsCall,
+            { name: 'getSourceCode', id: sourceCallId, args: { filePaths: legitimateFiles } },
+        ],
+    };
+    const fragmentsSourceCode = {};
+    for (const fragment of extractedFragments.fragments) {
+        fragmentsSourceCode[fragment.filePath] = {
+            fileId: generateFileId(fragment.filePath),
+            fragments: [fragment.content],
+        };
+    }
+    const user = {
+        type: 'user',
+        text: generateFilesContentPrompt(alreadyProvidedFiles, requestedFiles, legitimateFiles, illegitimateFiles, 'fragments'),
+        data: { legitimateFiles, illegitimateFiles },
+        functionResponses: [
+            {
+                name: 'requestFilesFragments',
+                call_id: requestFilesFragmentsCall.id,
+                content: JSON.stringify({ filePaths: legitimateFiles }),
+            },
+            {
+                name: 'getSourceCode',
+                call_id: sourceCallId,
+                content: JSON.stringify(fragmentsSourceCode),
+            },
+        ],
+        cache: true,
+    };
+    putSystemMessage('Providing files fragments', {
+        requestedFiles,
+        legitimateFiles,
+        illegitimateFiles,
+        fragmentCount: extractedFragments.fragments.length,
+    });
+    prompt.push(assistant, user);
+    return { breakLoop: false, items: [] };
+}
+export async function generateRequestFilesFragmentsCall(generateContentFn, prompt, iterateCall, options, modelType) {
+    const [requestFilesFragmentsCall] = (await generateContentFn([
+        ...prompt,
+        {
+            type: 'assistant',
+            text: iterateCall.args?.message ?? '',
+        },
+        {
+            type: 'user',
+            text: 'Yes, you can request the file fragments.',
+        },
+    ], {
+        functionDefs: getFunctionDefs(),
+        requiredFunctionName: 'requestFilesFragments',
+        temperature: 0.7,
+        modelType,
+        expectedResponseType: {
+            text: false,
+            functionCall: true,
+            media: false,
+        },
+    }, options))
+        .filter((item) => item.type === 'functionCall')
+        .map((item) => item.functionCall);
+    return requestFilesFragmentsCall;
+}
+/**
+ * Extract fragments from files using LLM with multi-file support
+ */
+async function extractFragments(sourceCodeMap, fragmentPrompt, generateContentFn, options) {
+    const filePaths = Object.keys(sourceCodeMap);
+    try {
+        // Process all files in a single LLM call
+        const [extractFragmentsCall] = (await generateContentFn([
+            {
+                type: 'systemPrompt',
+                systemPrompt: 'You are tasked with extracting relevant fragments from multiple source code files based on a given prompt. ' +
+                    'Analyze each file thoroughly, identify meaningful sections that match the prompt criteria, and provide clear ' +
+                    'explanations for your selections. Each fragment must be attributed to its source file for proper tracking.',
+            },
+            {
+                type: 'assistant',
+                text: 'I will analyze the provided files and extract relevant fragments based on your prompt.',
+                functionCalls: [{ name: 'getSourceCode', args: { filePaths } }],
+            },
+            {
+                type: 'user',
+                text: `Please extract relevant fragments from the provided files.\nExtraction prompt: ${fragmentPrompt}`,
+                functionResponses: [
+                    {
+                        name: 'getSourceCode',
+                        content: JSON.stringify(sourceCodeMap),
+                    },
+                ],
+            },
+        ], {
+            functionDefs: getFunctionDefs(),
+            requiredFunctionName: 'extractFileFragments',
+            temperature: 0.2,
+            modelType: ModelType.CHEAP,
+            expectedResponseType: {
+                text: false,
+                functionCall: true,
+                media: false,
+            },
+        }, options))
+            .filter((item) => item.type === 'functionCall')
+            .map((item) => item.functionCall);
+        if (!extractFragmentsCall?.args) {
+            putSystemMessage('Failed to extract fragments');
+            return {
+                filePaths,
+                fragments: [],
+                reasoning: 'Fragment extraction failed',
+            };
+        }
+        putSystemMessage(`Successfully extracted ${extractFragmentsCall.args.fragments.length} fragments from ${filePaths.length} files`);
+        return extractFragmentsCall.args;
+    }
+    catch (error) {
+        putSystemMessage('Error during fragment extraction', { error });
+        return {
+            filePaths,
+            fragments: [],
+            reasoning: `Error during fragment extraction: ${error}`,
+        };
+    }
+}
+//# sourceMappingURL=handle-request-files-fragments.js.map

package/dist/prompt/steps/step-iterate/handlers/handle-request-files-fragments.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"handle-request-files-fragments.js","sourceRoot":"","sources":["../../../../../src/prompt/steps/step-iterate/handlers/handle-request-files-fragments.ts"],"names":[],"mappings":"AAGA,OAAO,EAAE,SAAS,EAAE,MAAM,wCAAwC,CAAC;AACnE,OAAO,EAAE,cAAc,EAAE,MAAM,oCAAoC,CAAC;AACpE,OAAO,EAAE,aAAa,EAAE,MAAM,iCAAiC,CAAC;AAGhE,OAAO,EAAE,gBAAgB,EAAE,MAAM,wCAAwC,CAAC;AAC1E,OAAO,EAAE,eAAe,EAAE,MAAM,8BAA8B,CAAC;AAE/D,OAAO,EAAE,qBAAqB,EAAE,MAAM,6BAA6B,CAAC;AASpE,OAAO,EAAE,mBAAmB,EAAE,0BAA0B,EAAE,MAAM,yBAAyB,CAAC;AAE1F,qBAAqB,CAAC,uBAAuB,EAAE,2BAA2B,CAAC,CAAC;AAE5E,MAAM,CAAC,KAAK,UAAU,2BAA2B,CAAC,EAChD,WAAW,EACX,OAAO,EACP,MAAM,EACN,iBAAiB,GACE;IACnB,IAAI,yBAAyB,GAAG,MAAM,iCAAiC,CACrE,iBAAiB,EACjB,MAAM,EACN,WAAW,EACX,OAAO,EACP,SAAS,CAAC,KAAK,CAChB,CAAC;IAEF,IAAI,CAAC,yBAAyB,EAAE,CAAC;QAC/B,OAAO;YACL,SAAS,EAAE,KAAK;YAChB,KAAK,EAAE,EAAE;SACV,CAAC;IACJ,CAAC;IAED,MAAM,cAAc,GAAG,yBAAyB,CAAC,IAAI,EAAE,cAAc,CAAC;IACtE,IAAI,CAAC,cAAc,EAAE,CAAC;QACpB,gBAAgB,CAAC,6BAA6B,CAAC,CAAC;QAChD,OAAO;YACL,SAAS,EAAE,KAAK;YAChB,KAAK,EAAE,EAAE;SACV,CAAC;IACJ,CAAC;IAED,mDAAmD;IACnD,IAAI,EAAE,cAAc,EAAE,oBAAoB,EAAE,eAAe,EAAE,iBAAiB,EAAE,GAAG,mBAAmB,CACpG,yBAAyB,CAAC,IAAI,EAAE,SAAS,IAAI,EAAE,EAC/C,MAAM,EACN,OAAO,CACR,CAAC;IAEF,IAAI,oBAAoB,CAAC,MAAM,KAAK,cAAc,CAAC,MAAM,EAAE,CAAC;QAC1D,gBAAgB,CAAC,0CAA0C,CAAC,CAAC;QAC7D,MAAM,CAAC,IAAI,CACT;YACE,IAAI,EAAE,WAAW;YACjB,IAAI,EAAE,WAAW,CAAC,IAAI,EAAE,OAAO,IAAI,EAAE;YACrC,aAAa,EAAE,CAAC,yBAAyB,CAAC;SAC3C,EACD;YACE,IAAI,EAAE,MAAM;YACZ,IAAI,EAAE,gFAAgF;YACtF,iBAAiB,EAAE;gBACjB;oBACE,IAAI,EAAE,uBAAuB;oBAC7B,OAAO,EAAE,yBAAyB,CAAC,EAAE;oBACrC,OAAO,EAAE,IAAI,CAAC,SAAS,CAAC,EAAE,SAAS,EAAE,cAAc,EAAE,CAAC;iBACvD;aACF;YACD,KAAK,EAAE,IAAI;SACZ,CACF,CAAC;QAEF,OAAO,EAAE,SAAS,EAAE,KAAK,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC;IACzC,CAAC;IAED,iEAAiE;IACjE,IAAI,iBAAiB,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QACjC,yBAAyB,GAAG,MAAM,iCAAiC,CACjE,iBAAiB,EACjB,MAAM,EACN,WAAW,EACX,OAAO,EACP,SAAS,CAAC,OAAO,CAClB,CAAC;QAEF,IAAI,CAAC,yBAAyB,EAAE,CAAC;YAC/B,OAAO;gBACL,SAAS,EAAE,KAAK;gBAChB,KAAK,EAAE,EAAE;aACV,CAAC;QACJ,CAAC;QAED,MAAM,SAAS,GAAG,mBAAmB,CAAC,yBAAyB,CAAC,IAAI,EAAE,SAAS,IAAI,EAAE,EAAE,MAAM,EAAE,OAAO,CAAC,CAAC;QACxG,cAAc,GAAG,SAAS,CAAC,cAAc,CAAC;QAC1C,oBAAoB,GAAG,SAAS,CAAC,oBAAoB,CAAC;QACtD,eAAe,GAAG,SAAS,CAAC,eAAe,CAAC;QAC5C,iBAAiB,GAAG,SAAS,CAAC,iBAAiB,CAAC;IAClD,CAAC;IAED,sCAAsC;IACtC,MAAM,aAAa,GAAG,MAAM,aAAa,CACvC,EAAE,WAAW,EAAE,eAAe,EAAE,QAAQ,EAAE,IAAI,EAAE,oBAAoB,EAAE,IAAI,EAAE,EAC5E,OAAO,CACR,CAAC;IAEF,yCAAyC;IACzC,MAAM,kBAAkB,GAAG,MAAM,gBAAgB,CAAC,aAAa,EAAE,cAAc,EAAE,iBAAiB,EAAE,OAAO,CAAC,CAAC;IAE7G,MAAM,YAAY,GAAG,CAAC,WAAW,CAAC,EAAE,IAAI,WAAW,CAAC,IAAI,CAAC,GAAG,SAAS,CAAC;IACtE,MAAM,SAAS,GAAkB;QAC/B,IAAI,EAAE,WAAW;QACjB,IAAI,EAAE,WAAW,CAAC,IAAI,EAAE,OAAO,IAAI,EAAE;QACrC,aAAa,EAAE;YACb,yBAAyB;YACzB,EAAE,IAAI,EAAE,eAAe,EAAE,EAAE,EAAE,YAAY,EAAE,IAAI,EAAE,EAAE,SAAS,EAAE,eAAe,EAAE,EAAE;SAClF;KACF,CAAC;IAEF,MAAM,mBAAmB,GAAkB,EAAE,CAAC;IAC9C,KAAK,MAAM,QAAQ,IAAI,kBAAkB,CAAC,SAAS,EAAE,CAAC;QACpD,mBAAmB,CAAC,QAAQ,CAAC,QAAQ,CAAC,GAAG;YACvC,MAAM,EAAE,cAAc,CAAC,QAAQ,CAAC,QAAQ,CAAC;YACzC,SAAS,EAAE,CAAC,QAAQ,CAAC,OAAO,CAAC;SAC9B,CAAC;IACJ,CAAC;IAED,MAAM,IAAI,GAAa;QACrB,IAAI,EAAE,MAAM;QACZ,IAAI,EAAE,0BAA0B,CAC9B,oBAAoB,EACpB,cAAc,EACd,eAAe,EACf,iBAAiB,EACjB,WAAW,CACZ;QACD,IAAI,EAAE,EAAE,eAAe,EAAE,iBAAiB,EAAE;QAC5C,iBAAiB,EAAE;YACjB;gBACE,IAAI,EAAE,uBAAuB;gBAC7B,OAAO,EAAE,yBAAyB,CAAC,EAAE;gBACrC,OAAO,EAAE,IAAI,CAAC,SAAS,CAAC,EAAE,SAAS,EAAE,eAAe,EAAE,CAAC;aACxD;YACD;gBACE,IAAI,EAAE,eAAe;gBACrB,OAAO,EAAE,YAAY;gBACrB,OAAO,EAAE,IAAI,CAAC,SAAS,CAAC,mBAAmB,CAAC;aAC7C;SACF;QACD,KAAK,EAAE,IAAI;KACZ,CAAC;IAEF,gBAAgB,CAAC,2BAA2B,EAAE;QAC5C,cAAc;QACd,eAAe;QACf,iBAAiB;QACjB,aAAa,EAAE,kBAAkB,CAAC,SAAS,CAAC,MAAM;KACnD,CAAC,CAAC;IAEH,MAAM,CAAC,IAAI,CAAC,SAAS,EAAE,IAAI,CAAC,CAAC;IAE7B,OAAO,EAAE,SAAS,EAAE,KAAK,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC;AACzC,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,iCAAiC,CACrD,iBAA0C,EAC1C,MAAoB,EACpB,WAAwB,EACxB,OAAuB,EACvB,SAAoB;IAEpB,MAAM,CAAC,yBAAyB,CAAC,GAAG,CAClC,MAAM,iBAAiB,CACrB;QACE,GAAG,MAAM;QACT;YACE,IAAI,EAAE,WAAW;YACjB,IAAI,EAAE,WAAW,CAAC,IAAI,EAAE,OAAO,IAAI,EAAE;SACtC;QACD;YACE,IAAI,EAAE,MAAM;YACZ,IAAI,EAAE,0CAA0C;SACjD;KACF,EACD;QACE,YAAY,EAAE,eAAe,EAAE;QAC/B,oBAAoB,EAAE,uBAAuB;QAC7C,WAAW,EAAE,GAAG;QAChB,SAAS;QACT,oBAAoB,EAAE;YACpB,IAAI,EAAE,KAAK;YACX,YAAY,EAAE,IAAI;YAClB,KAAK,EAAE,KAAK;SACb;KACF,EACD,OAAO,CACR,CACF;SACE,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,KAAK,cAAc,CAAC;SAC9C,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,YAAY,CAA0D,CAAC;IAE7F,OAAO,yBAAyB,CAAC;AACnC,CAAC;AAED;;GAEG;AACH,KAAK,UAAU,gBAAgB,CAC7B,aAA4B,EAC5B,cAAsB,EACtB,iBAA0C,EAC1C,OAAuB;IAEvB,MAAM,SAAS,GAAG,MAAM,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;IAE7C,IAAI,CAAC;QACH,yCAAyC;QACzC,MAAM,CAAC,oBAAoB,CAAC,GAAG,CAC7B,MAAM,iBAAiB,CACrB;YACE;gBACE,IAAI,EAAE,cAAc;gBACpB,YAAY,EACV,6GAA6G;oBAC7G,+GAA+G;oBAC/G,4GAA4G;aAC/G;YACD;gBACE,IAAI,EAAE,WAAW;gBACjB,IAAI,EAAE,wFAAwF;gBAC9F,aAAa,EAAE,CAAC,EAAE,IAAI,EAAE,eAAe,EAAE,IAAI,EAAE,EAAE,SAAS,EAAE,EAAE,CAAC;aAChE;YACD;gBACE,IAAI,EAAE,MAAM;gBACZ,IAAI,EAAE,kFAAkF,cAAc,EAAE;gBACxG,iBAAiB,EAAE;oBACjB;wBACE,IAAI,EAAE,eAAe;wBACrB,OAAO,EAAE,IAAI,CAAC,SAAS,CAAC,aAAa,CAAC;qBACvC;iBACF;aACF;SACF,EACD;YACE,YAAY,EAAE,eAAe,EAAE;YAC/B,oBAAoB,EAAE,sBAAsB;YAC5C,WAAW,EAAE,GAAG;YAChB,SAAS,EAAE,SAAS,CAAC,KAAK;YAC1B,oBAAoB,EAAE;gBACpB,IAAI,EAAE,KAAK;gBACX,YAAY,EAAE,IAAI;gBAClB,KAAK,EAAE,KAAK;aACb;SACF,EACD,OAAO,CACR,CACF;aACE,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,KAAK,cAAc,CAAC;aAC9C,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,YAAY,CAAyD,CAAC;QAE5F,IAAI,CAAC,oBAAoB,EAAE,IAAI,EAAE,CAAC;YAChC,gBAAgB,CAAC,6BAA6B,CAAC,CAAC;YAChD,OAAO;gBACL,SAAS;gBACT,SAAS,EAAE,EAAE;gBACb,SAAS,EAAE,4BAA4B;aACxC,CAAC;QACJ,CAAC;QAED,gBAAgB,CACd,0BAA0B,oBAAoB,CAAC,IAAI,CAAC,SAAS,CAAC,MAAM,mBAAmB,SAAS,CAAC,MAAM,QAAQ,CAChH,CAAC;QACF,OAAO,oBAAoB,CAAC,IAAI,CAAC;IACnC,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,gBAAgB,CAAC,kCAAkC,EAAE,EAAE,KAAK,EAAE,CAAC,CAAC;QAChE,OAAO;YACL,SAAS;YACT,SAAS,EAAE,EAAE;YACb,SAAS,EAAE,qCAAqC,KAAK,EAAE;SACxD,CAAC;IACJ,CAAC;AACH,CAAC"}

package/dist/prompt/steps/step-iterate/handlers/handle-request-git-context.d.ts

@@ -0,0 +1,3 @@
+import { ActionHandler } from '../step-iterate-types.js';
+export declare const GIT_CONTEXT_INSTRUCTION_PROMPT = "\n    You need to call the 'requestGitContext' function. Please determine the appropriate arguments based on the user's request.\n    Parameters:\n    - requestType (required): Choose one of 'commits', 'fileChanges', 'blame', 'fileDiff', 'workingChanges', or 'workingDiff'.\n      - 'commits': Get recent commit history for the entire repository.\n      - 'fileChanges': Get commit history specifically for the given 'filePath'.\n      - 'blame': Get line-by-line authorship information (git blame) for the given 'filePath'.\n      - 'fileDiff': Get the changes made to a specific 'filePath' in a given 'commitHash'.\n      - 'workingChanges': List changed files in the current working copy (staged, unstaged, untracked).\n      - 'workingDiff': Show the diff for a specific file in the working copy.\n    - filePath (absolute, required for 'fileChanges', 'blame', 'fileDiff', and 'workingDiff'): The absolute path to the file within the project. IMPORTANT: Ensure the provided filePath is absolute and is inside of the project root and exists within the project directory structure.\n    - commitHash (required for 'blame' and 'fileDiff'): A specific commit hash to focus the request.\n    - count (required for 'commits' and 'fileChanges'): The maximum number of commits to retrieve.\n    - includeUntracked (optional, for 'workingChanges'): Set to false to exclude untracked files. Defaults to true.\n    - stagedOnly (optional, for 'workingChanges'): Set to true to only show staged changes. Mutually exclusive with unstagedOnly.\n    - unstagedOnly (optional, for 'workingChanges'): Set to true to only show unstaged changes. Mutually exclusive with stagedOnly.\n    - staged (optional, for 'workingDiff'): Set to true to show the diff for staged changes (--cached) instead of unstaged changes.\n\n    Analyze the user's request and provide the arguments accurately.\n  ";
+export declare const handleRequestGitContext: ActionHandler;