@ai-dev-tools/csharp-copilot-core 0.0.19 → 0.0.29

package/out/command/utGenWrapper.js

@@ -2,6 +2,7 @@
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.generateUtCode = generateUtCode;
 exports.generateBatchUtCode = generateBatchUtCode;
+exports.generateUtBenchmark = generateUtBenchmark;
 exports.generateUtWithVsPlugin = generateUtWithVsPlugin;
 const stdListener_1 = require("../vsPlugin/stdListener");
 const constants_1 = require("../types/constants");
@@ -10,13 +11,18 @@ const csharpUtGen_1 = require("../gen/csharpUtGen");
 async function generateUtCode(c) {
     const autofix = c.autofix === 1;
     const keepFailedTest = c.keepFailedTest === 1;
-    await (0, csharpUtGen_1.generateCsharpUtCode)(c.sourceCodePath, c.testCodePath, autofix, keepFailedTest, constants_1.CSVSPLUGIN);
+    await (0, csharpUtGen_1.generateCsharpUtCode)(c.sourceCodePath, c.testCodePath, autofix, keepFailedTest, constants_1.CSHARPUTGEN, false, c.nugetPackagePath);
 }
 async function generateBatchUtCode(c) {
     const autofix = c.autofix === 1;
     const keepFailedTest = c.keepFailedTest === 1;
     await (0, generateCodeTests_1.generateBatchTests)(c.sourceCodePath, autofix, keepFailedTest);
 }
+async function generateUtBenchmark(c) {
+    const autofix = c.autofix === 1;
+    const keepFailedTest = c.keepFailedTest === 1;
+    await (0, generateCodeTests_1.generateBatchTests)(c.sourceCodePath, autofix, keepFailedTest, true);
+}
 async function generateUtWithVsPlugin(c) {
     try {
         const autofix = c.autofix === 1;
@@ -24,7 +30,7 @@ async function generateUtWithVsPlugin(c) {
             (0, stdListener_1.startStdListener)();
         }
         const keepFailedTest = c.keepFailedTest === 1;
-        await (0, csharpUtGen_1.generateCsharpUtCode)(c.sourceCodePath, c.testCodePath, autofix, keepFailedTest, constants_1.CSVSPLUGIN, true);
+        await (0, csharpUtGen_1.generateCsharpUtCode)(c.sourceCodePath, c.testCodePath, autofix, keepFailedTest, constants_1.CSVSPLUGIN, true, c.nugetPackagePath);
     }
     catch (error) {
         console.error("Error generateUtWithVsPlugin:", error);

package/out/command/utGenWrapper.js.map

@@ -1 +1 @@
-{"version":3,"file":"utGenWrapper.js","sourceRoot":"","sources":["../../src/command/utGenWrapper.ts"],"names":[],"mappings":";;AAMA,wCAIC;AAED,kDAIC;AAED,wDAgBC;AAlCD,yDAA8E;AAE9E,kDAAgD;AAChD,kEAAgE;AAChE,oDAA0D;AAEnD,KAAK,UAAU,cAAc,CAAC,CAAC;IAClC,MAAM,OAAO,GAAG,CAAC,CAAC,OAAO,KAAK,CAAC,CAAC;IAChC,MAAM,cAAc,GAAG,CAAC,CAAC,cAAc,KAAK,CAAC,CAAC;IAC9C,MAAM,IAAA,kCAAoB,EAAC,CAAC,CAAC,cAAc,EAAE,CAAC,CAAC,YAAY,EAAE,OAAO,EAAE,cAAc,EAAE,sBAAU,CAAC,CAAC;AACtG,CAAC;AAEM,KAAK,UAAU,mBAAmB,CAAC,CAAC;IACvC,MAAM,OAAO,GAAG,CAAC,CAAC,OAAO,KAAK,CAAC,CAAC;IAChC,MAAM,cAAc,GAAG,CAAC,CAAC,cAAc,KAAK,CAAC,CAAC;IAC9C,MAAM,IAAA,sCAAkB,EAAC,CAAC,CAAC,cAAc,EAAE,OAAO,EAAE,cAAc,CAAC,CAAC;AACxE,CAAC;AAEM,KAAK,UAAU,sBAAsB,CAAC,CAAC;IAC1C,IAAI,CAAC;QACD,MAAM,OAAO,GAAG,CAAC,CAAC,OAAO,KAAK,CAAC,CAAC;QAChC,IAAI,OAAO,EAAE,CAAC;YACV,IAAA,8BAAgB,GAAE,CAAC;QACvB,CAAC;QACD,MAAM,cAAc,GAAG,CAAC,CAAC,cAAc,KAAK,CAAC,CAAC;QAC9C,MAAM,IAAA,kCAAoB,EAAC,CAAC,CAAC,cAAc,EAAE,CAAC,CAAC,YAAY,EAAE,OAAO,EAAE,cAAc,EAAE,sBAAU,EAAE,IAAI,CAAC,CAAC;IAC5G,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACb,OAAO,CAAC,KAAK,CAAC,+BAA+B,EAAE,KAAK,CAAC,CAAC;IAC1D,CAAC;YAAS,CAAC;QACP,iDAAiD;QACjD,IAAI,CAAC,CAAC,OAAO,KAAK,CAAC,IAAI,+BAAiB,EAAE,CAAC;YACvC,+BAAiB,CAAC,KAAK,EAAE,CAAC;QAC9B,CAAC;IACL,CAAC;AACL,CAAC;AAGD,iNAAiN","sourcesContent":["import { readLineInterface, startStdListener } from \"../vsPlugin/stdListener\";\r\n\r\nimport { CSVSPLUGIN } from \"../types/constants\";\r\nimport { generateBatchTests } from \"../batch/generateCodeTests\";\r\nimport { generateCsharpUtCode } from \"../gen/csharpUtGen\";\r\n\r\nexport async function generateUtCode(c) {\r\n    const autofix = c.autofix === 1;\r\n    const keepFailedTest = c.keepFailedTest === 1;\r\n    await generateCsharpUtCode(c.sourceCodePath, c.testCodePath, autofix, keepFailedTest, CSVSPLUGIN);\r\n}\r\n\r\nexport async function generateBatchUtCode(c) {\r\n    const autofix = c.autofix === 1;\r\n    const keepFailedTest = c.keepFailedTest === 1;\r\n    await generateBatchTests(c.sourceCodePath, autofix, keepFailedTest);\r\n}\r\n\r\nexport async function generateUtWithVsPlugin(c) {\r\n    try {\r\n        const autofix = c.autofix === 1;\r\n        if (autofix) {\r\n            startStdListener();\r\n        }\r\n        const keepFailedTest = c.keepFailedTest === 1;\r\n        await generateCsharpUtCode(c.sourceCodePath, c.testCodePath, autofix, keepFailedTest, CSVSPLUGIN, true);\r\n    } catch (error) {\r\n        console.error(\"Error generateUtWithVsPlugin:\", error);\r\n    } finally {\r\n        // Ensure the listener is closed after processing\r\n        if (c.autofix === 1 && readLineInterface) {\r\n            readLineInterface.close();\r\n        }\r\n    }\r\n}\r\n\r\n\r\n// generateUtCode({sourceCodePath: \"D:\\\\code\\\\CS.Service.Fundamental\\\\SharedSegments\\\\SharedSegments\\\\SharedSegments.Plugins\\\\Workflows\\\\Overview\\\\HttpRequestParser\\\\Utils\\\\SqmidHelper.cs\", keepFailedTest: 1})"]}
+{"version":3,"file":"utGenWrapper.js","sourceRoot":"","sources":["../../src/command/utGenWrapper.ts"],"names":[],"mappings":";;AAMA,wCAIC;AAED,kDAIC;AAED,kDAIC;AAED,wDAgBC;AAxCD,yDAA8E;AAE9E,kDAA6D;AAC7D,kEAAgE;AAChE,oDAA0D;AAEnD,KAAK,UAAU,cAAc,CAAC,CAAC;IAClC,MAAM,OAAO,GAAG,CAAC,CAAC,OAAO,KAAK,CAAC,CAAC;IAChC,MAAM,cAAc,GAAG,CAAC,CAAC,cAAc,KAAK,CAAC,CAAC;IAC9C,MAAM,IAAA,kCAAoB,EAAC,CAAC,CAAC,cAAc,EAAE,CAAC,CAAC,YAAY,EAAE,OAAO,EAAE,cAAc,EAAE,uBAAW,EAAE,KAAK,EAAE,CAAC,CAAC,gBAAgB,CAAC,CAAC;AAClI,CAAC;AAEM,KAAK,UAAU,mBAAmB,CAAC,CAAC;IACvC,MAAM,OAAO,GAAG,CAAC,CAAC,OAAO,KAAK,CAAC,CAAC;IAChC,MAAM,cAAc,GAAG,CAAC,CAAC,cAAc,KAAK,CAAC,CAAC;IAC9C,MAAM,IAAA,sCAAkB,EAAC,CAAC,CAAC,cAAc,EAAE,OAAO,EAAE,cAAc,CAAC,CAAC;AACxE,CAAC;AAEM,KAAK,UAAU,mBAAmB,CAAC,CAAC;IACvC,MAAM,OAAO,GAAG,CAAC,CAAC,OAAO,KAAK,CAAC,CAAC;IAChC,MAAM,cAAc,GAAG,CAAC,CAAC,cAAc,KAAK,CAAC,CAAC;IAC9C,MAAM,IAAA,sCAAkB,EAAC,CAAC,CAAC,cAAc,EAAE,OAAO,EAAE,cAAc,EAAE,IAAI,CAAC,CAAC;AAC9E,CAAC;AAEM,KAAK,UAAU,sBAAsB,CAAC,CAAC;IAC1C,IAAI,CAAC;QACD,MAAM,OAAO,GAAG,CAAC,CAAC,OAAO,KAAK,CAAC,CAAC;QAChC,IAAI,OAAO,EAAE,CAAC;YACV,IAAA,8BAAgB,GAAE,CAAC;QACvB,CAAC;QACD,MAAM,cAAc,GAAG,CAAC,CAAC,cAAc,KAAK,CAAC,CAAC;QAC9C,MAAM,IAAA,kCAAoB,EAAC,CAAC,CAAC,cAAc,EAAE,CAAC,CAAC,YAAY,EAAE,OAAO,EAAE,cAAc,EAAE,sBAAU,EAAE,IAAI,EAAE,CAAC,CAAC,gBAAgB,CAAC,CAAC;IAChI,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACb,OAAO,CAAC,KAAK,CAAC,+BAA+B,EAAE,KAAK,CAAC,CAAC;IAC1D,CAAC;YAAS,CAAC;QACP,iDAAiD;QACjD,IAAI,CAAC,CAAC,OAAO,KAAK,CAAC,IAAI,+BAAiB,EAAE,CAAC;YACvC,+BAAiB,CAAC,KAAK,EAAE,CAAC;QAC9B,CAAC;IACL,CAAC;AACL,CAAC;AAGD,iNAAiN","sourcesContent":["import { readLineInterface, startStdListener } from \"../vsPlugin/stdListener\";\r\n\r\nimport { CSHARPUTGEN, CSVSPLUGIN } from \"../types/constants\";\r\nimport { generateBatchTests } from \"../batch/generateCodeTests\";\r\nimport { generateCsharpUtCode } from \"../gen/csharpUtGen\";\r\n\r\nexport async function generateUtCode(c) {\r\n    const autofix = c.autofix === 1;\r\n    const keepFailedTest = c.keepFailedTest === 1;\r\n    await generateCsharpUtCode(c.sourceCodePath, c.testCodePath, autofix, keepFailedTest, CSHARPUTGEN, false, c.nugetPackagePath);\r\n}\r\n\r\nexport async function generateBatchUtCode(c) {\r\n    const autofix = c.autofix === 1;\r\n    const keepFailedTest = c.keepFailedTest === 1;\r\n    await generateBatchTests(c.sourceCodePath, autofix, keepFailedTest);\r\n}\r\n\r\nexport async function generateUtBenchmark(c) {\r\n    const autofix = c.autofix === 1;\r\n    const keepFailedTest = c.keepFailedTest === 1;\r\n    await generateBatchTests(c.sourceCodePath, autofix, keepFailedTest, true);\r\n}\r\n\r\nexport async function generateUtWithVsPlugin(c) {\r\n    try {\r\n        const autofix = c.autofix === 1;\r\n        if (autofix) {\r\n            startStdListener();\r\n        }\r\n        const keepFailedTest = c.keepFailedTest === 1;\r\n        await generateCsharpUtCode(c.sourceCodePath, c.testCodePath, autofix, keepFailedTest, CSVSPLUGIN, true, c.nugetPackagePath);\r\n    } catch (error) {\r\n        console.error(\"Error generateUtWithVsPlugin:\", error);\r\n    } finally {\r\n        // Ensure the listener is closed after processing\r\n        if (c.autofix === 1 && readLineInterface) {\r\n            readLineInterface.close();\r\n        }\r\n    }\r\n}\r\n\r\n\r\n// generateUtCode({sourceCodePath: \"D:\\\\code\\\\CS.Service.Fundamental\\\\SharedSegments\\\\SharedSegments\\\\SharedSegments.Plugins\\\\Workflows\\\\Overview\\\\HttpRequestParser\\\\Utils\\\\SqmidHelper.cs\", keepFailedTest: 1})"]}

package/out/gen/autoFix.d.ts

@@ -1,4 +1,4 @@
 import { GenResult } from "../types/genResult";
-export declare function autoFixTestCode(prompts: any[], sourceCodePath: string, testCode: string, testProjectPath: string, testFilePath: string, isXapCode: boolean, isVsPlugin: boolean, retryTimes?: number, isBenchmark?: boolean): Promise<GenResult>;
-export declare function generateAutoFixCode(prompts: any[], generatedCode: string, errorMessage: string, isBuildFailure: boolean, testFilePath: string, isXapCode: boolean): Promise<string>;
-export declare function autoRetryTestCode(prompts: any[], sourceCodePath: string, testCode: string, testProjectPath: string, testFilePath: string, isXapCode: boolean, retryTimes?: number): Promise<GenResult>;
+export declare function autoFixTestCode(prompts: any[], sourceCodePath: string, testCode: string, testProjectPath: string, testFilePath: string, testFramework: string, isXapCode: boolean, isVsPlugin: boolean, retryTimes?: number, isBenchmark?: boolean): Promise<GenResult>;
+export declare function generateAutoFixCode(prompts: any[], generatedCode: string, errorMessage: string, isBuildFailure: boolean, testFilePath: string, testFramework: string, isXapCode: boolean): Promise<string>;
+export declare function autoRetryTestCode(prompts: any[], sourceCodePath: string, testCode: string, testProjectPath: string, testFilePath: string, testFramework: string, isXapCode: boolean, retryTimes?: number): Promise<GenResult>;

package/out/gen/autoFix.js

@@ -12,7 +12,7 @@ const verifyTestCode_1 = require("../utils/verifyTestCode");
 const getVsPluginVerifyResult_1 = require("../vsPlugin/getVsPluginVerifyResult");
 const writeGenCode_1 = require("../utils/writeGenCode");
 const fileUtils_1 = require("../utils/fileUtils");
-async function autoFixTestCode(prompts, sourceCodePath, testCode, testProjectPath, testFilePath, isXapCode, isVsPlugin, retryTimes = 3, isBenchmark = false) {
+async function autoFixTestCode(prompts, sourceCodePath, testCode, testProjectPath, testFilePath, testFramework, isXapCode, isVsPlugin, retryTimes = 3, isBenchmark = false) {
     const startTime = Date.now();
     let generatedCode = testCode;
     let testResult;
@@ -26,9 +26,9 @@ async function autoFixTestCode(prompts, sourceCodePath, testCode, testProjectPat
             // for case: generatedCode is invalid, just retry instead of auto fix process
             if (!generatedCode) {
                 console.error(`Generated code is invalid, try to regenerate. \nGenerated code: \n${generatedCode}`);
-                generatedCode = await (0, ensureValidLLMResponse_1.ensureValidLLMResponse)(prompts, isXapCode);
+                generatedCode = await (0, ensureValidLLMResponse_1.ensureValidLLMResponse)(prompts, testFramework, isXapCode);
                 // extract valid code from LLM response
-                generatedCode = (0, postGenProcess_1.postGenProcess)(generatedCode, isXapCode);
+                generatedCode = (0, postGenProcess_1.postGenProcess)(generatedCode, testFramework, isXapCode);
                 continue;
             }
             // analyze generated code issue before verify: build, because build will cost much time.
@@ -73,7 +73,7 @@ async function autoFixTestCode(prompts, sourceCodePath, testCode, testProjectPat
                 console.error("!!!Verify process failed, retrying...");
                 continue;
             }
-            generatedCode = await generateAutoFixCode(prompts, generatedCode, errorMessage, isBuildFailure, testFilePath, isXapCode);
+            generatedCode = await generateAutoFixCode(prompts, generatedCode, errorMessage, isBuildFailure, testFilePath, testFramework, isXapCode);
         }
         console.log(`Autofix time: ${(Date.now() - startTime) / 1000} seconds`);
         return { success: false, testFilePath, testProjectPath, codeFilePath: sourceCodePath, buildResult, testResult };
@@ -94,17 +94,17 @@ async function verifyGeneratedCode(generatedCode, testProjectPath, testFilePath,
         return verifyResult;
     }
 }
-async function generateAutoFixCode(prompts, generatedCode, errorMessage, isBuildFailure, testFilePath, isXapCode) {
+async function generateAutoFixCode(prompts, generatedCode, errorMessage, isBuildFailure, testFilePath, testFramework, isXapCode) {
     console.log("start auto-fix process...");
     const autofixPrompt = (0, preparePrompt_1.prepareUtAutoFixPrompt)(prompts, generatedCode, errorMessage, isBuildFailure);
     // get valid code from LLM response with retry
-    const newGeneratedCode = await (0, ensureValidLLMResponse_1.ensureValidLLMResponse)(autofixPrompt, isXapCode);
+    const newGeneratedCode = await (0, ensureValidLLMResponse_1.ensureValidLLMResponse)(autofixPrompt, testFramework, isXapCode);
     // if generate code is valid, write test code to test project
     !!newGeneratedCode && (0, fileUtils_1.writeTestCode)(newGeneratedCode, testFilePath);
     return newGeneratedCode;
 }
 // use for quick dev test
-async function autoRetryTestCode(prompts, sourceCodePath, testCode, testProjectPath, testFilePath, isXapCode, retryTimes = 3) {
+async function autoRetryTestCode(prompts, sourceCodePath, testCode, testProjectPath, testFilePath, testFramework, isXapCode, retryTimes = 3) {
     const startTime = Date.now();
     let generatedCode = testCode;
     let vsTestRes;
@@ -115,7 +115,7 @@ async function autoRetryTestCode(prompts, sourceCodePath, testCode, testProjectP
         }
         const autofixPrompt = prompts;
         // get valid code from LLM response
-        generatedCode = await (0, ensureValidLLMResponse_1.ensureValidLLMResponse)(autofixPrompt, isXapCode);
+        generatedCode = await (0, ensureValidLLMResponse_1.ensureValidLLMResponse)(autofixPrompt, testFramework, isXapCode);
         // write test code to test project
         (0, fileUtils_1.writeTestCode)(generatedCode, testFilePath);
     }

package/out/gen/autoFix.js.map

@@ -1 +1 @@
-{"version":3,"file":"autoFix.js","sourceRoot":"","sources":["../../src/gen/autoFix.ts"],"names":[],"mappings":";;AAcA,0CA2FC;AAkBD,kDAkBC;AAID,8CA6BC;AA5KD,gEAA6D;AAC7D,qEAAkE;AAElE,6DAA0D;AAC1D,wDAA8D;AAE9D,oFAAiF;AACjF,4DAAyD;AACzD,iFAA+E;AAC/E,wDAAqD;AACrD,kDAAmD;AAE5C,KAAK,UAAU,eAAe,CACjC,OAAc,EACd,cAAsB,EACtB,QAAgB,EAChB,eAAuB,EACvB,YAAoB,EACpB,SAAkB,EAClB,UAAmB,EACnB,aAAqB,CAAC,EACtB,cAAuB,KAAK;IAE5B,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;IAE7B,IAAI,aAAa,GAAG,QAAQ,CAAC;IAC7B,IAAI,UAAU,CAAC;IACf,IAAI,WAAW,CAAC;IAChB,IAAI,YAAY,CAAC;IACjB,IAAI,cAAc,GAAG,IAAI,CAAC;IAC1B,IAAI,CAAC;QACD,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,IAAI,UAAU,EAAE,CAAC,EAAE,EAAE,CAAC;YACnC,2DAA2D;YAC3D,WAAW,IAAI,IAAA,2BAAY,EAAC,cAAc,EAAE,aAAa,EAAE,CAAC,CAAC,CAAC;YAE9D,6EAA6E;YAC7E,IAAI,CAAC,aAAa,EAAE,CAAC;gBACjB,OAAO,CAAC,KAAK,CAAC,qEAAqE,aAAa,EAAE,CAAC,CAAC;gBACpG,aAAa,GAAG,MAAM,IAAA,+CAAsB,EAAC,OAAO,EAAE,SAAS,CAAC,CAAC;gBACjE,uCAAuC;gBACvC,aAAa,GAAG,IAAA,+BAAc,EAAC,aAAa,EAAE,SAAS,CAAC,CAAC;gBACzD,SAAS;YACb,CAAC;YAED,wFAAwF;YACxF,MAAM,YAAY,GAAG,MAAM,IAAA,iCAAe,EAAC,YAAY,CAAC,CAAC;YACzD,IAAI,YAAY,CAAC,OAAO,EAAE,CAAC;gBACvB,4EAA4E;gBAC5E,gHAAgH;gBAChH,MAAM,YAAY,GAAG,MAAM,mBAAmB,CAAC,aAAa,EAAE,eAAe,EAAE,YAAY,EAAE,UAAU,CAAC,CAAC;gBACzG,YAAY,GAAG,YAAY,CAAC,OAAO,CAAC;gBACpC,UAAU,GAAG,YAAY,CAAC,UAAU,CAAC;gBACrC,WAAW,GAAG,YAAY,CAAC,WAAW,CAAC;gBACvC,cAAc,GAAG,CAAC,UAAU,CAAC;gBAE7B,6FAA6F;gBAC7F,MAAM,eAAe,GAAG,IAAA,uDAA0B,EAAC,YAAY,EAAE,WAAW,CAAC,CAAC;gBAC9E,IAAI,CAAC,eAAe,EAAE,CAAC;oBACnB,OAAO,CAAC,KAAK,CAAC,wFAAwF,CAAC,CAAC;oBACxG,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,YAAY,EAAE,eAAe,EAAE,YAAY,EAAE,cAAc,EAAE,WAAW,EAAE,UAAU,EAAE,CAAC;gBACpH,CAAC;gBAED,IAAI,YAAY,CAAC,OAAO,EAAE,CAAC;oBACvB,OAAO,CAAC,GAAG,CAAC,mCAAmC,CAAC,CAAC;oBACjD,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,YAAY,EAAE,eAAe,EAAE,YAAY,EAAE,cAAc,EAAE,WAAW,EAAE,UAAU,EAAE,CAAC;gBACnH,CAAC;gBAED,OAAO,CAAC,KAAK,CAAC,kCAAkC,EAAE,YAAY,CAAC,CAAC;YACpE,CAAC;iBAAM,CAAC;gBACJ,WAAW,GAAG,YAAY,CAAC,WAAW,CAAC;gBACvC,YAAY,GAAG,YAAY,CAAC,OAAO,CAAC;gBACpC,OAAO,CAAC,KAAK,CAAC,sDAAsD,EAAE,YAAY,CAAC,CAAC;YACxF,CAAC;YAED,yDAAyD;YACzD,mDAAmD;YACnD,4DAA4D;YAC5D,gEAAgE;YAChE,yDAAyD;YACzD,IAAI;YAEJ,IAAI,CAAC,KAAK,UAAU,EAAE,CAAC;gBACnB,OAAO,CAAC,KAAK,CAAC,oCAAoC,UAAU,YAAY,CAAC,CAAC;gBAC1E,MAAM;YACV,CAAC;YAED,wDAAwD;YACxD,IAAI,CAAC,WAAW,IAAI,CAAC,UAAU,EAAE,CAAC;gBAC9B,OAAO,CAAC,KAAK,CAAC,uCAAuC,CAAC,CAAC;gBACvD,SAAS;YACb,CAAC;YAED,aAAa,GAAG,MAAM,mBAAmB,CAAC,OAAO,EAAE,aAAa,EAAE,YAAY,EAAE,cAAc,EAAE,YAAY,EAAE,SAAS,CAAC,CAAC;QAC7H,CAAC;QAED,OAAO,CAAC,GAAG,CAAC,iBAAiB,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,CAAC,GAAG,IAAI,UAAU,CAAC,CAAC;QAExE,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,YAAY,EAAE,eAAe,EAAE,YAAY,EAAE,cAAc,EAAE,WAAW,EAAE,UAAU,EAAE,CAAC;IACpH,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACb,OAAO,CAAC,KAAK,CAAC,gCAAgC,EAAE,KAAK,CAAC,CAAC;QACvD,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,YAAY,EAAE,eAAe,EAAE,YAAY,EAAE,cAAc,EAAE,WAAW,EAAE,IAAI,EAAE,UAAU,EAAE,IAAI,EAAE,CAAC;IAEhI,CAAC;AACL,CAAC;AAED,KAAK,UAAU,mBAAmB,CAC9B,aAAqB,EACrB,eAAuB,EACvB,YAAoB,EACpB,aAAsB,KAAK;IAE3B,IAAI,UAAU,EAAE,CAAC;QACb,MAAM,cAAc,GAAG,MAAM,IAAA,kDAAwB,EAAC,YAAY,CAAC,CAAC;QACpE,OAAO,cAAc,CAAC;IAC1B,CAAC;SAAM,CAAC;QACJ,gEAAgE;QAChE,MAAM,YAAY,GAAG,MAAM,IAAA,+BAAc,EAAC,aAAa,EAAE,eAAe,CAAC,CAAC;QAC1E,OAAO,YAAY,CAAC;IACxB,CAAC;AACL,CAAC;AAEM,KAAK,UAAU,mBAAmB,CACrC,OAAc,EACd,aAAqB,EACrB,YAAoB,EACpB,cAAuB,EACvB,YAAoB,EACpB,SAAkB;IAElB,OAAO,CAAC,GAAG,CAAC,2BAA2B,CAAC,CAAC;IACzC,MAAM,aAAa,GAAG,IAAA,sCAAsB,EAAC,OAAO,EAAE,aAAa,EAAE,YAAY,EAAE,cAAc,CAAC,CAAC;IAEnG,8CAA8C;IAC9C,MAAM,gBAAgB,GAAG,MAAM,IAAA,+CAAsB,EAAC,aAAa,EAAE,SAAS,CAAC,CAAC;IAEhF,6DAA6D;IAC7D,CAAC,CAAC,gBAAgB,IAAI,IAAA,yBAAa,EAAC,gBAAgB,EAAE,YAAY,CAAC,CAAC;IAEpE,OAAO,gBAAgB,CAAC;AAC5B,CAAC;AAGD,yBAAyB;AAClB,KAAK,UAAU,iBAAiB,CACnC,OAAc,EACd,cAAsB,EACtB,QAAgB,EAChB,eAAuB,EACvB,YAAoB,EACpB,SAAkB,EAClB,aAAqB,CAAC;IAEtB,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;IAE7B,IAAI,aAAa,GAAG,QAAQ,CAAC;IAC7B,IAAI,SAAS,CAAC;IACd,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,IAAI,UAAU,EAAE,CAAC,EAAE,EAAE,CAAC;QACnC,IAAI,CAAC,KAAK,UAAU,EAAE,CAAC;YACnB,OAAO,CAAC,KAAK,CAAC,uCAAuC,UAAU,YAAY,CAAC,CAAC;YAC7E,MAAM;QACV,CAAC;QAED,MAAM,aAAa,GAAG,OAAO,CAAC;QAC9B,mCAAmC;QACnC,aAAa,GAAG,MAAM,IAAA,+CAAsB,EAAC,aAAa,EAAE,SAAS,CAAC,CAAC;QACvE,kCAAkC;QAClC,IAAA,yBAAa,EAAC,aAAa,EAAE,YAAY,CAAC,CAAC;IAC/C,CAAC;IAED,OAAO,CAAC,GAAG,CAAC,iBAAiB,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,CAAC,GAAG,IAAI,UAAU,CAAC,CAAC;IAExE,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,YAAY,EAAE,YAAY,EAAE,eAAe,EAAE,YAAY,EAAE,cAAc,EAAE,WAAW,EAAE,IAAI,EAAE,UAAU,EAAE,SAAS,EAAE,CAAC;AAClJ,CAAC","sourcesContent":["import { GenResult } from \"../types/genResult\";\r\nimport { VerifyResult } from \"../types/verifyResult\";\r\nimport { checkCodeSyntax } from \"../analyze/checkCodeSyntax\";\r\nimport { ensureValidLLMResponse } from \"./ensureValidLLMResponse\";\r\nimport { parseSyntaxCheckErrors } from \"../utils/parseSyntaxCheckResult\";\r\nimport { postGenProcess } from \"./postGen/postGenProcess\";\r\nimport { prepareUtAutoFixPrompt } from \"../llm/preparePrompt\";\r\nimport { sendRequestToLLM } from \"../llm/sendRequestToLLM\";\r\nimport { verifyBuildErrorCausedFile } from \"../utils/verifyBuildErrorCausedFile\";\r\nimport { verifyTestCode } from \"../utils/verifyTestCode\";\r\nimport { verifyTestCodeByVsPlugin } from \"../vsPlugin/getVsPluginVerifyResult\";\r\nimport { writeGenCode } from \"../utils/writeGenCode\";\r\nimport { writeTestCode } from \"../utils/fileUtils\";\r\n\r\nexport async function autoFixTestCode(\r\n    prompts: any[],\r\n    sourceCodePath: string,\r\n    testCode: string,\r\n    testProjectPath: string,\r\n    testFilePath: string,\r\n    isXapCode: boolean,\r\n    isVsPlugin: boolean,\r\n    retryTimes: number = 3,\r\n    isBenchmark: boolean = false\r\n): Promise<GenResult> {\r\n    const startTime = Date.now();\r\n\r\n    let generatedCode = testCode;\r\n    let testResult;\r\n    let buildResult;\r\n    let errorMessage;\r\n    let isBuildFailure = true;\r\n    try {\r\n        for (let i = 0; i <= retryTimes; i++) {\r\n            // write generated code, to review and improve code quality\r\n            isBenchmark && writeGenCode(sourceCodePath, generatedCode, i);\r\n\r\n            // for case: generatedCode is invalid, just retry instead of auto fix process\r\n            if (!generatedCode) {\r\n                console.error(`Generated code is invalid, try to regenerate. \\nGenerated code: \\n${generatedCode}`);\r\n                generatedCode = await ensureValidLLMResponse(prompts, isXapCode);\r\n                // extract valid code from LLM response\r\n                generatedCode = postGenProcess(generatedCode, isXapCode);\r\n                continue;\r\n            }\r\n\r\n            // analyze generated code issue before verify: build, because build will cost much time.\r\n            const syntaxResult = await checkCodeSyntax(testFilePath);\r\n            if (syntaxResult.success) {\r\n                // if generated code passed the syntax check, verify the generated test code\r\n                // const { success, message, buildErrors, vsTestResult } = await verifyTestCode(generatedCode, testProjectPath);\r\n                const verifyResult = await verifyGeneratedCode(generatedCode, testProjectPath, testFilePath, isVsPlugin);\r\n                errorMessage = verifyResult.message;\r\n                testResult = verifyResult.testResult;\r\n                buildResult = verifyResult.buildErrors;\r\n                isBuildFailure = !testResult;\r\n                \r\n                // if build errors caused by other code files instead of generate test code, abort generation\r\n                const validBuildError = verifyBuildErrorCausedFile(testFilePath, buildResult);\r\n                if (!validBuildError) {\r\n                    console.error(\"Due to build errors caused by other code files, abort generation, please fix it first.\");\r\n                    return { success: false, testFilePath, testProjectPath, codeFilePath: sourceCodePath, buildResult, testResult };\r\n                }\r\n                \r\n                if (verifyResult.success) {\r\n                    console.log(\"XapUt code generated successfully\");\r\n                    return { success: true, testFilePath, testProjectPath, codeFilePath: sourceCodePath, buildResult, testResult };\r\n                }\r\n\r\n                console.error(\"Failed to verifyTestCode, error:\", errorMessage);\r\n            } else {\r\n                buildResult = syntaxResult.buildErrors;\r\n                errorMessage = syntaxResult.message;\r\n                console.error(\"Failed to check the generated UT code syntax, error:\", errorMessage);\r\n            }\r\n\r\n            // keep test failed tast cases in test file automatically\r\n            // if (!isBuildFailure && vsTestRes.passed !== 0) {\r\n            //     console.log(\"remove failed test cases in test file\");\r\n            //     removeFailedTestCasesInTestFile(vsTestRes, testCodePath);\r\n            //     return { success: true, testCode: generatedCode };\r\n            // }\r\n\r\n            if (i === retryTimes) {\r\n                console.error(`Failed to generate ut code after ${retryTimes} attempts.`);\r\n                break;\r\n            }\r\n\r\n            // case: vs plugin no response or msbuild/vstest failed.\r\n            if (!buildResult && !testResult) {\r\n                console.error(\"!!!Verify process failed, retrying...\");\r\n                continue;\r\n            }\r\n\r\n            generatedCode = await generateAutoFixCode(prompts, generatedCode, errorMessage, isBuildFailure, testFilePath, isXapCode);\r\n        }\r\n\r\n        console.log(`Autofix time: ${(Date.now() - startTime) / 1000} seconds`);\r\n\r\n        return { success: false, testFilePath, testProjectPath, codeFilePath: sourceCodePath, buildResult, testResult };\r\n    } catch (error) {\r\n        console.error(\"Error during auto-fix process:\", error);\r\n        return { success: false, testFilePath, testProjectPath, codeFilePath: sourceCodePath, buildResult: null, testResult: null };\r\n\r\n    }\r\n}\r\n\r\nasync function verifyGeneratedCode(\r\n    generatedCode: string,\r\n    testProjectPath: string,\r\n    testFilePath: string,\r\n    isVsPlugin: boolean = false\r\n): Promise<VerifyResult> {\r\n    if (isVsPlugin) {\r\n        const vsVerifyResult = await verifyTestCodeByVsPlugin(testFilePath);\r\n        return vsVerifyResult;\r\n    } else {\r\n        // verifyResult: { success, message, buildErrors, vsTestResult }\r\n        const verifyResult = await verifyTestCode(generatedCode, testProjectPath);\r\n        return verifyResult;\r\n    }\r\n}\r\n\r\nexport async function generateAutoFixCode(\r\n    prompts: any[],\r\n    generatedCode: string,\r\n    errorMessage: string,\r\n    isBuildFailure: boolean,\r\n    testFilePath: string,\r\n    isXapCode: boolean\r\n): Promise<string> {\r\n    console.log(\"start auto-fix process...\");\r\n    const autofixPrompt = prepareUtAutoFixPrompt(prompts, generatedCode, errorMessage, isBuildFailure);\r\n\r\n    // get valid code from LLM response with retry\r\n    const newGeneratedCode = await ensureValidLLMResponse(autofixPrompt, isXapCode);\r\n\r\n    // if generate code is valid, write test code to test project\r\n    !!newGeneratedCode && writeTestCode(newGeneratedCode, testFilePath);\r\n\r\n    return newGeneratedCode;\r\n}\r\n\r\n\r\n// use for quick dev test\r\nexport async function autoRetryTestCode(\r\n    prompts: any[],\r\n    sourceCodePath: string,\r\n    testCode: string,\r\n    testProjectPath: string,\r\n    testFilePath: string,\r\n    isXapCode: boolean,\r\n    retryTimes: number = 3\r\n): Promise<GenResult> {\r\n    const startTime = Date.now();\r\n\r\n    let generatedCode = testCode;\r\n    let vsTestRes;\r\n    for (let i = 0; i <= retryTimes; i++) {\r\n        if (i === retryTimes) {\r\n            console.error(`Failed to generate XapUt code after ${retryTimes} attempts.`);\r\n            break;\r\n        }\r\n\r\n        const autofixPrompt = prompts;\r\n        // get valid code from LLM response\r\n        generatedCode = await ensureValidLLMResponse(autofixPrompt, isXapCode);\r\n        // write test code to test project\r\n        writeTestCode(generatedCode, testFilePath);\r\n    }\r\n\r\n    console.log(`Autofix time: ${(Date.now() - startTime) / 1000} seconds`);\r\n\r\n    return { success: true, testFilePath: testFilePath, testProjectPath, codeFilePath: sourceCodePath, buildResult: null, testResult: vsTestRes };\r\n}"]}
+{"version":3,"file":"autoFix.js","sourceRoot":"","sources":["../../src/gen/autoFix.ts"],"names":[],"mappings":";;AAYA,0CA4FC;AAkBD,kDAmBC;AAID,8CA8BC;AA7KD,gEAA6D;AAC7D,qEAAkE;AAClE,6DAA0D;AAC1D,wDAA8D;AAC9D,oFAAiF;AACjF,4DAAyD;AACzD,iFAA+E;AAC/E,wDAAqD;AACrD,kDAAmD;AAE5C,KAAK,UAAU,eAAe,CACjC,OAAc,EACd,cAAsB,EACtB,QAAgB,EAChB,eAAuB,EACvB,YAAoB,EACpB,aAAqB,EACrB,SAAkB,EAClB,UAAmB,EACnB,aAAqB,CAAC,EACtB,cAAuB,KAAK;IAE5B,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;IAE7B,IAAI,aAAa,GAAG,QAAQ,CAAC;IAC7B,IAAI,UAAU,CAAC;IACf,IAAI,WAAW,CAAC;IAChB,IAAI,YAAY,CAAC;IACjB,IAAI,cAAc,GAAG,IAAI,CAAC;IAC1B,IAAI,CAAC;QACD,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,IAAI,UAAU,EAAE,CAAC,EAAE,EAAE,CAAC;YACnC,2DAA2D;YAC3D,WAAW,IAAI,IAAA,2BAAY,EAAC,cAAc,EAAE,aAAa,EAAE,CAAC,CAAC,CAAC;YAE9D,6EAA6E;YAC7E,IAAI,CAAC,aAAa,EAAE,CAAC;gBACjB,OAAO,CAAC,KAAK,CAAC,qEAAqE,aAAa,EAAE,CAAC,CAAC;gBACpG,aAAa,GAAG,MAAM,IAAA,+CAAsB,EAAC,OAAO,EAAE,aAAa,EAAE,SAAS,CAAC,CAAC;gBAChF,uCAAuC;gBACvC,aAAa,GAAG,IAAA,+BAAc,EAAC,aAAa,EAAE,aAAa,EAAE,SAAS,CAAC,CAAC;gBACxE,SAAS;YACb,CAAC;YAED,wFAAwF;YACxF,MAAM,YAAY,GAAG,MAAM,IAAA,iCAAe,EAAC,YAAY,CAAC,CAAC;YACzD,IAAI,YAAY,CAAC,OAAO,EAAE,CAAC;gBACvB,4EAA4E;gBAC5E,gHAAgH;gBAChH,MAAM,YAAY,GAAG,MAAM,mBAAmB,CAAC,aAAa,EAAE,eAAe,EAAE,YAAY,EAAE,UAAU,CAAC,CAAC;gBACzG,YAAY,GAAG,YAAY,CAAC,OAAO,CAAC;gBACpC,UAAU,GAAG,YAAY,CAAC,UAAU,CAAC;gBACrC,WAAW,GAAG,YAAY,CAAC,WAAW,CAAC;gBACvC,cAAc,GAAG,CAAC,UAAU,CAAC;gBAE7B,6FAA6F;gBAC7F,MAAM,eAAe,GAAG,IAAA,uDAA0B,EAAC,YAAY,EAAE,WAAW,CAAC,CAAC;gBAC9E,IAAI,CAAC,eAAe,EAAE,CAAC;oBACnB,OAAO,CAAC,KAAK,CAAC,wFAAwF,CAAC,CAAC;oBACxG,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,YAAY,EAAE,eAAe,EAAE,YAAY,EAAE,cAAc,EAAE,WAAW,EAAE,UAAU,EAAE,CAAC;gBACpH,CAAC;gBAED,IAAI,YAAY,CAAC,OAAO,EAAE,CAAC;oBACvB,OAAO,CAAC,GAAG,CAAC,mCAAmC,CAAC,CAAC;oBACjD,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,YAAY,EAAE,eAAe,EAAE,YAAY,EAAE,cAAc,EAAE,WAAW,EAAE,UAAU,EAAE,CAAC;gBACnH,CAAC;gBAED,OAAO,CAAC,KAAK,CAAC,kCAAkC,EAAE,YAAY,CAAC,CAAC;YACpE,CAAC;iBAAM,CAAC;gBACJ,WAAW,GAAG,YAAY,CAAC,WAAW,CAAC;gBACvC,YAAY,GAAG,YAAY,CAAC,OAAO,CAAC;gBACpC,OAAO,CAAC,KAAK,CAAC,sDAAsD,EAAE,YAAY,CAAC,CAAC;YACxF,CAAC;YAED,yDAAyD;YACzD,mDAAmD;YACnD,4DAA4D;YAC5D,gEAAgE;YAChE,yDAAyD;YACzD,IAAI;YAEJ,IAAI,CAAC,KAAK,UAAU,EAAE,CAAC;gBACnB,OAAO,CAAC,KAAK,CAAC,oCAAoC,UAAU,YAAY,CAAC,CAAC;gBAC1E,MAAM;YACV,CAAC;YAED,wDAAwD;YACxD,IAAI,CAAC,WAAW,IAAI,CAAC,UAAU,EAAE,CAAC;gBAC9B,OAAO,CAAC,KAAK,CAAC,uCAAuC,CAAC,CAAC;gBACvD,SAAS;YACb,CAAC;YAED,aAAa,GAAG,MAAM,mBAAmB,CAAC,OAAO,EAAE,aAAa,EAAE,YAAY,EAAE,cAAc,EAAE,YAAY,EAAE,aAAa,EAAE,SAAS,CAAC,CAAC;QAC5I,CAAC;QAED,OAAO,CAAC,GAAG,CAAC,iBAAiB,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,CAAC,GAAG,IAAI,UAAU,CAAC,CAAC;QAExE,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,YAAY,EAAE,eAAe,EAAE,YAAY,EAAE,cAAc,EAAE,WAAW,EAAE,UAAU,EAAE,CAAC;IACpH,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACb,OAAO,CAAC,KAAK,CAAC,gCAAgC,EAAE,KAAK,CAAC,CAAC;QACvD,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,YAAY,EAAE,eAAe,EAAE,YAAY,EAAE,cAAc,EAAE,WAAW,EAAE,IAAI,EAAE,UAAU,EAAE,IAAI,EAAE,CAAC;IAEhI,CAAC;AACL,CAAC;AAED,KAAK,UAAU,mBAAmB,CAC9B,aAAqB,EACrB,eAAuB,EACvB,YAAoB,EACpB,aAAsB,KAAK;IAE3B,IAAI,UAAU,EAAE,CAAC;QACb,MAAM,cAAc,GAAG,MAAM,IAAA,kDAAwB,EAAC,YAAY,CAAC,CAAC;QACpE,OAAO,cAAc,CAAC;IAC1B,CAAC;SAAM,CAAC;QACJ,gEAAgE;QAChE,MAAM,YAAY,GAAG,MAAM,IAAA,+BAAc,EAAC,aAAa,EAAE,eAAe,CAAC,CAAC;QAC1E,OAAO,YAAY,CAAC;IACxB,CAAC;AACL,CAAC;AAEM,KAAK,UAAU,mBAAmB,CACrC,OAAc,EACd,aAAqB,EACrB,YAAoB,EACpB,cAAuB,EACvB,YAAoB,EACpB,aAAqB,EACrB,SAAkB;IAElB,OAAO,CAAC,GAAG,CAAC,2BAA2B,CAAC,CAAC;IACzC,MAAM,aAAa,GAAG,IAAA,sCAAsB,EAAC,OAAO,EAAE,aAAa,EAAE,YAAY,EAAE,cAAc,CAAC,CAAC;IAEnG,8CAA8C;IAC9C,MAAM,gBAAgB,GAAG,MAAM,IAAA,+CAAsB,EAAC,aAAa,EAAE,aAAa,EAAE,SAAS,CAAC,CAAC;IAE/F,6DAA6D;IAC7D,CAAC,CAAC,gBAAgB,IAAI,IAAA,yBAAa,EAAC,gBAAgB,EAAE,YAAY,CAAC,CAAC;IAEpE,OAAO,gBAAgB,CAAC;AAC5B,CAAC;AAGD,yBAAyB;AAClB,KAAK,UAAU,iBAAiB,CACnC,OAAc,EACd,cAAsB,EACtB,QAAgB,EAChB,eAAuB,EACvB,YAAoB,EACpB,aAAqB,EACrB,SAAkB,EAClB,aAAqB,CAAC;IAEtB,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;IAE7B,IAAI,aAAa,GAAG,QAAQ,CAAC;IAC7B,IAAI,SAAS,CAAC;IACd,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,IAAI,UAAU,EAAE,CAAC,EAAE,EAAE,CAAC;QACnC,IAAI,CAAC,KAAK,UAAU,EAAE,CAAC;YACnB,OAAO,CAAC,KAAK,CAAC,uCAAuC,UAAU,YAAY,CAAC,CAAC;YAC7E,MAAM;QACV,CAAC;QAED,MAAM,aAAa,GAAG,OAAO,CAAC;QAC9B,mCAAmC;QACnC,aAAa,GAAG,MAAM,IAAA,+CAAsB,EAAC,aAAa,EAAE,aAAa,EAAE,SAAS,CAAC,CAAC;QACtF,kCAAkC;QAClC,IAAA,yBAAa,EAAC,aAAa,EAAE,YAAY,CAAC,CAAC;IAC/C,CAAC;IAED,OAAO,CAAC,GAAG,CAAC,iBAAiB,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,CAAC,GAAG,IAAI,UAAU,CAAC,CAAC;IAExE,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,YAAY,EAAE,YAAY,EAAE,eAAe,EAAE,YAAY,EAAE,cAAc,EAAE,WAAW,EAAE,IAAI,EAAE,UAAU,EAAE,SAAS,EAAE,CAAC;AAClJ,CAAC","sourcesContent":["import { GenResult } from \"../types/genResult\";\r\nimport { VerifyResult } from \"../types/verifyResult\";\r\nimport { checkCodeSyntax } from \"../analyze/checkCodeSyntax\";\r\nimport { ensureValidLLMResponse } from \"./ensureValidLLMResponse\";\r\nimport { postGenProcess } from \"./postGen/postGenProcess\";\r\nimport { prepareUtAutoFixPrompt } from \"../llm/preparePrompt\";\r\nimport { verifyBuildErrorCausedFile } from \"../utils/verifyBuildErrorCausedFile\";\r\nimport { verifyTestCode } from \"../utils/verifyTestCode\";\r\nimport { verifyTestCodeByVsPlugin } from \"../vsPlugin/getVsPluginVerifyResult\";\r\nimport { writeGenCode } from \"../utils/writeGenCode\";\r\nimport { writeTestCode } from \"../utils/fileUtils\";\r\n\r\nexport async function autoFixTestCode(\r\n    prompts: any[],\r\n    sourceCodePath: string,\r\n    testCode: string,\r\n    testProjectPath: string,\r\n    testFilePath: string,\r\n    testFramework: string,\r\n    isXapCode: boolean,\r\n    isVsPlugin: boolean,\r\n    retryTimes: number = 3,\r\n    isBenchmark: boolean = false\r\n): Promise<GenResult> {\r\n    const startTime = Date.now();\r\n\r\n    let generatedCode = testCode;\r\n    let testResult;\r\n    let buildResult;\r\n    let errorMessage;\r\n    let isBuildFailure = true;\r\n    try {\r\n        for (let i = 0; i <= retryTimes; i++) {\r\n            // write generated code, to review and improve code quality\r\n            isBenchmark && writeGenCode(sourceCodePath, generatedCode, i);\r\n\r\n            // for case: generatedCode is invalid, just retry instead of auto fix process\r\n            if (!generatedCode) {\r\n                console.error(`Generated code is invalid, try to regenerate. \\nGenerated code: \\n${generatedCode}`);\r\n                generatedCode = await ensureValidLLMResponse(prompts, testFramework, isXapCode);\r\n                // extract valid code from LLM response\r\n                generatedCode = postGenProcess(generatedCode, testFramework, isXapCode);\r\n                continue;\r\n            }\r\n\r\n            // analyze generated code issue before verify: build, because build will cost much time.\r\n            const syntaxResult = await checkCodeSyntax(testFilePath);\r\n            if (syntaxResult.success) {\r\n                // if generated code passed the syntax check, verify the generated test code\r\n                // const { success, message, buildErrors, vsTestResult } = await verifyTestCode(generatedCode, testProjectPath);\r\n                const verifyResult = await verifyGeneratedCode(generatedCode, testProjectPath, testFilePath, isVsPlugin);\r\n                errorMessage = verifyResult.message;\r\n                testResult = verifyResult.testResult;\r\n                buildResult = verifyResult.buildErrors;\r\n                isBuildFailure = !testResult;\r\n\r\n                // if build errors caused by other code files instead of generate test code, abort generation\r\n                const validBuildError = verifyBuildErrorCausedFile(testFilePath, buildResult);\r\n                if (!validBuildError) {\r\n                    console.error(\"Due to build errors caused by other code files, abort generation, please fix it first.\");\r\n                    return { success: false, testFilePath, testProjectPath, codeFilePath: sourceCodePath, buildResult, testResult };\r\n                }\r\n\r\n                if (verifyResult.success) {\r\n                    console.log(\"XapUt code generated successfully\");\r\n                    return { success: true, testFilePath, testProjectPath, codeFilePath: sourceCodePath, buildResult, testResult };\r\n                }\r\n\r\n                console.error(\"Failed to verifyTestCode, error:\", errorMessage);\r\n            } else {\r\n                buildResult = syntaxResult.buildErrors;\r\n                errorMessage = syntaxResult.message;\r\n                console.error(\"Failed to check the generated UT code syntax, error:\", errorMessage);\r\n            }\r\n\r\n            // keep test failed tast cases in test file automatically\r\n            // if (!isBuildFailure && vsTestRes.passed !== 0) {\r\n            //     console.log(\"remove failed test cases in test file\");\r\n            //     removeFailedTestCasesInTestFile(vsTestRes, testCodePath);\r\n            //     return { success: true, testCode: generatedCode };\r\n            // }\r\n\r\n            if (i === retryTimes) {\r\n                console.error(`Failed to generate ut code after ${retryTimes} attempts.`);\r\n                break;\r\n            }\r\n\r\n            // case: vs plugin no response or msbuild/vstest failed.\r\n            if (!buildResult && !testResult) {\r\n                console.error(\"!!!Verify process failed, retrying...\");\r\n                continue;\r\n            }\r\n\r\n            generatedCode = await generateAutoFixCode(prompts, generatedCode, errorMessage, isBuildFailure, testFilePath, testFramework, isXapCode);\r\n        }\r\n\r\n        console.log(`Autofix time: ${(Date.now() - startTime) / 1000} seconds`);\r\n\r\n        return { success: false, testFilePath, testProjectPath, codeFilePath: sourceCodePath, buildResult, testResult };\r\n    } catch (error) {\r\n        console.error(\"Error during auto-fix process:\", error);\r\n        return { success: false, testFilePath, testProjectPath, codeFilePath: sourceCodePath, buildResult: null, testResult: null };\r\n\r\n    }\r\n}\r\n\r\nasync function verifyGeneratedCode(\r\n    generatedCode: string,\r\n    testProjectPath: string,\r\n    testFilePath: string,\r\n    isVsPlugin: boolean = false\r\n): Promise<VerifyResult> {\r\n    if (isVsPlugin) {\r\n        const vsVerifyResult = await verifyTestCodeByVsPlugin(testFilePath);\r\n        return vsVerifyResult;\r\n    } else {\r\n        // verifyResult: { success, message, buildErrors, vsTestResult }\r\n        const verifyResult = await verifyTestCode(generatedCode, testProjectPath);\r\n        return verifyResult;\r\n    }\r\n}\r\n\r\nexport async function generateAutoFixCode(\r\n    prompts: any[],\r\n    generatedCode: string,\r\n    errorMessage: string,\r\n    isBuildFailure: boolean,\r\n    testFilePath: string,\r\n    testFramework: string,\r\n    isXapCode: boolean\r\n): Promise<string> {\r\n    console.log(\"start auto-fix process...\");\r\n    const autofixPrompt = prepareUtAutoFixPrompt(prompts, generatedCode, errorMessage, isBuildFailure);\r\n\r\n    // get valid code from LLM response with retry\r\n    const newGeneratedCode = await ensureValidLLMResponse(autofixPrompt, testFramework, isXapCode);\r\n\r\n    // if generate code is valid, write test code to test project\r\n    !!newGeneratedCode && writeTestCode(newGeneratedCode, testFilePath);\r\n\r\n    return newGeneratedCode;\r\n}\r\n\r\n\r\n// use for quick dev test\r\nexport async function autoRetryTestCode(\r\n    prompts: any[],\r\n    sourceCodePath: string,\r\n    testCode: string,\r\n    testProjectPath: string,\r\n    testFilePath: string,\r\n    testFramework: string,\r\n    isXapCode: boolean,\r\n    retryTimes: number = 3\r\n): Promise<GenResult> {\r\n    const startTime = Date.now();\r\n\r\n    let generatedCode = testCode;\r\n    let vsTestRes;\r\n    for (let i = 0; i <= retryTimes; i++) {\r\n        if (i === retryTimes) {\r\n            console.error(`Failed to generate XapUt code after ${retryTimes} attempts.`);\r\n            break;\r\n        }\r\n\r\n        const autofixPrompt = prompts;\r\n        // get valid code from LLM response\r\n        generatedCode = await ensureValidLLMResponse(autofixPrompt, testFramework, isXapCode);\r\n        // write test code to test project\r\n        writeTestCode(generatedCode, testFilePath);\r\n    }\r\n\r\n    console.log(`Autofix time: ${(Date.now() - startTime) / 1000} seconds`);\r\n\r\n    return { success: true, testFilePath: testFilePath, testProjectPath, codeFilePath: sourceCodePath, buildResult: null, testResult: vsTestRes };\r\n}"]}

package/out/gen/csharpUtGen.d.ts

@@ -1,2 +1,2 @@
 import { GenResult } from '../types/genResult';
-export declare function generateCsharpUtCode(sourceCodePath: string, testCodePath: string, enableAutoFix: boolean, keepFailedTests: boolean, source?: string, isBenchmark?: boolean): Promise<GenResult>;
+export declare function generateCsharpUtCode(sourceCodePath: string, testCodePath: string, enableAutoFix: boolean, keepFailedTests: boolean, source?: string, isBenchmark?: boolean, nugetPackagePath?: string): Promise<GenResult>;

package/out/gen/csharpUtGen.js

@@ -50,7 +50,7 @@ const preparePrompt_1 = require("../llm/preparePrompt");
 const removeFailedTestMethods_1 = require("../utils/removeFailedTestMethods");
 const appInsightTelemetry_1 = require("../metrics/appInsightTelemetry");
 const RETRYTIMES = 3;
-async function generateCsharpUtCode(sourceCodePath, testCodePath, enableAutoFix, keepFailedTests, source = constants_1.CSHARPUTGEN, isBenchmark = false) {
+async function generateCsharpUtCode(sourceCodePath, testCodePath, enableAutoFix, keepFailedTests, source = constants_1.CSHARPUTGEN, isBenchmark = false, nugetPackagePath = null) {
     const startTime = Date.now();
     console.log(`*****************Generating C# unit test code for: ${sourceCodePath}*****************`);
     console.log(`Current work dir: ${process.cwd()}`);
@@ -72,20 +72,19 @@ async function generateCsharpUtCode(sourceCodePath, testCodePath, enableAutoFix,
         return { success: false, testFilePath: undefined, testProjectPath: undefined, codeFilePath: sourceCodePath, buildResult: null, testResult: null };
     }
     // get code dependencies
-    let codeDependency = await (0, generateCodeDependency_1.generateCodeDependency)(sourceCodePath);
+    let codeDependency = await (0, generateCodeDependency_1.generateCodeDependency)(sourceCodePath, nugetPackagePath);
     // detect existing test file or create a new one by source code path
     const { testFilePath, testProjectPath, testFileExist } = (0, getTestFile_1.getOrCreatTestFile)(sourceCodePath, testCodePath, isXapCode);
     if (!testFilePath || !testProjectPath) {
-        console.error(`Failed to get or create test file for source code: ${sourceCodePath}`);
+        console.error(`Failed to get or create test file for source code: ${sourceCodePath}. Please check if the test project exists.`);
         return { success: false, testFilePath: undefined, testProjectPath: undefined, codeFilePath: sourceCodePath, buildResult: null, testResult: null };
     }
     // get test framework
     const testFramework = (0, detectTestFramework_1.detectTestFramework)(testProjectPath);
     // prepare prompt
     let prompt = (0, preparePrompt_1.prepareUtGenPrompt)(sourceCodePath, sourceCode, codeDependency, isXapCode, testFramework);
-    // let prompt = getPrompts(sourceCodePath, testFramework, codeDependency);
     // call LLM service to gen code with retry
-    const testCode = await (0, ensureValidLLMResponse_1.ensureValidLLMResponse)(prompt, isXapCode);
+    const testCode = await (0, ensureValidLLMResponse_1.ensureValidLLMResponse)(prompt, testFramework, isXapCode);
     // write test code to test project
     // To run vstest, need to write the test code to test project first, then build the test project and run the tests
     (0, fileUtils_1.writeTestCode)(testCode, testFilePath);
@@ -93,11 +92,11 @@ async function generateCsharpUtCode(sourceCodePath, testCodePath, enableAutoFix,
     let autoFixResult = { success: true, testFilePath, testProjectPath, codeFilePath: sourceCodePath, buildResult: null, testResult: null };
     if (enableAutoFix) {
         const isVsPlugin = source === constants_1.CSVSPLUGIN;
-        autoFixResult = await (0, autoFix_1.autoFixTestCode)(prompt, sourceCodePath, testCode, testProjectPath, testFilePath, isXapCode, isVsPlugin, RETRYTIMES, isBenchmark);
+        autoFixResult = await (0, autoFix_1.autoFixTestCode)(prompt, sourceCodePath, testCode, testProjectPath, testFilePath, testFramework, isXapCode, isVsPlugin, RETRYTIMES, isBenchmark);
         // autoFixResult = await autoRetryTestCode(prompt, sourceCode, testCode, testProjectPath, testFilePath, isXapTest);
     }
     if (autoFixResult.success) {
-        console.log(`Gerate UT success time: ${(Date.now() - startTime) / 1000} seconds`);
+        console.log(`Generate UT success time: ${(Date.now() - startTime) / 1000} seconds`);
         return autoFixResult;
     }
     // if retry max times or test generation failed, and keepFailedTests is false, delete failed test file or remove failed test methods
@@ -115,7 +114,7 @@ async function generateCsharpUtCode(sourceCodePath, testCodePath, enableAutoFix,
     else {
         console.log(`keep failed test file: ${testFilePath}`);
     }
-    console.log(`Gerate UT failed time: ${(Date.now() - startTime) / 1000} seconds`);
+    console.log(`Generate UT failed time: ${(Date.now() - startTime) / 1000} seconds`);
     return autoFixResult;
 }
 // generateCsharpUtCode("D:\\code\\IndexServe\\private\\indexserve\\RecommendationEngine\\RecoPS\\PSTableClient\\PSTableClient.cs", undefined, false, true, false);
@@ -123,29 +122,4 @@ async function generateCsharpUtCode(sourceCodePath, testCodePath, enableAutoFix,
 // generateCsharpUtCode("D:\\code\\CS.Service.Fundamental\\SharedSegments\\SharedSegments\\SharedSegments.Plugins\\Workflows\\Eplant\\ResponseBuilders\\EplantMigrateResponseBuilder.cs", undefined, true, true, true);
 // generateCsharpUtCode("D:\\code\\UCMDynamics\\private\\DynamicCode\\Plugin\\UCMDynamics.AppointmentPlugin\\AppointmentOperation.cs", undefined, false, true, false);
 // generateCsharpUtCode("D:\\code\\AIUnitTestExample\\src\\ToListinator.Analyzers\\ToListCountAnalyzer.cs", undefined, true, true, false);
-// function getPrompts(sourceCodePath, testFramework, codeDependency) {
-//     const code = fs.readFileSync(sourceCodePath, 'utf8');
-//     let guidancePromt = fs.readFileSync("D:\\atmp\\promptiteration.md", 'utf8');
-//     guidancePromt = guidancePromt.replace("xUnit", testFramework);
-//     const prompts = [
-//         // {
-//         //     "role": "system",
-//         //     "content": "Keep your answers short and impersonal.\nUse Markdown formatting in your answers.\nMake sure to include the programming language name at the start of the Markdown code blocks.\nAvoid wrapping the whole response in triple backticks.\nYou can only give one reply for each conversation turn."
-//         // },
-//         {
-//             "role": "user",
-//             // "content": "You are Roo, a highly skilled software engineer with extensive knowledge in many programming languages, frameworks, design patterns, and best practices.\n\n====\n\nRULES\n\n- MUST use xUnit test framework to generate test code.\n- **MUST NOT** contain any comments or explanations in test code.\n- Ensure the test code MUST has no syntax errors.\n- Test public methods of class in source code only.\n- **MUST NOT** access private members (fields or methods) of the source class in test code.\n- Test method names must follow the pattern: [MethodName]_[Scenario]_[ExpectedResult].\n- Return the test code in the format: ```C#\n{Test code}\n```.\n- Mock all dependent methods, classes, and values not included in the user input using the Moq library.\n- MUST include Test code in a namespace declaration and place all `using` statements above the namespace declaration.\n- MUST import source code in the test code.\n- MUST import all required .Net namespaces if any .Net class or interface are used in the test code. Like:\n    - .NET built-in types (e.g., `List<T>`, `IEnumerable<T>`, `Task`, `Assert`, `Moq.Mock`, etc.).\n    - Any attribute classes (e.g., `[TestMethod]`, `[TestClass]`, etc.).\n    - Any LINQ or threading related classes (e.g., `System.Linq`, `System.Threading.Tasks`).\n- All assignments and method arguments in test code MUST strictly match the declared types.\n- When generating code, always consider the context in which the code is being used. Ensure that your changes are compatible with the existing codebase and that they follow the project's coding standards and best practices.\n- Do not ask for more information than necessary. Use your knowledge to accomplish the user's request efficiently and effectively.\n- The user will provide the source code directly in their message.\n- Your goal is to try to accomplish the user's task, NOT engage in a back and forth conversation.\n\n====\n\nOBJECTIVE\n\nYou accomplish a given task iteratively, breaking it down into clear steps and working through them methodically.\n\n- **MUST** analyze the provided code and verify its logic by following these steps:\n    1. Describe the functionality of each public method, including its inputs, outputs, and key logic.\n    2. Identify all possible execution paths for public methods, including normal cases, edge cases, and error conditions.\n    3. Explain any conditional branches, loops, or exception handling for public methods in the code.\n    4. Based on this analysis, generate unit test cases that cover all identified execution paths, ensuring that mock data and assertions align with the code's logic.\n- **MUST** remove all code analysis information in responsed test code.\n- Absolutely forbidden to contains any other information in response except the test code.\n- **MUST NOT** generate tests for the methods that are attributed with **[ExcludeFromCodeCoverage]**.\n- **MUST NOT** generate tests for **private** methods.\n- Analyze the user's task and set clear, achievable goals to accomplish it. Prioritize these goals in a logical order.\n- Work through these goals sequentially, each goal should correspond to a distinct step in your problem-solving process.\n\n====\n\nUSER'S CUSTOM INSTRUCTIONS\n\nThe following additional instructions are provided by the user, and should be followed to the best of your ability without interfering with the TOOL USE guidelines.\n\nLanguage Preference:\nYou should always speak and think in the English language."
-//             "content": guidancePromt
-//         },
-//         {
-//             "role": "user",
-//             "content": `generate unit test for the following csharp code:\n\`\`\`${code}\`\`\``
-//         },
-//         // {
-//         //     "role": "user",
-//         //     "content": `following are the code dependencies of the source code:\n${codeDependency}`
-//         // }
-//     ];
-//     return prompts;
-// }
 //# sourceMappingURL=csharpUtGen.js.map

package/out/gen/csharpUtGen.js.map

@@ -1 +1 @@
-{"version":3,"file":"csharpUtGen.js","sourceRoot":"","sources":["../../src/gen/csharpUtGen.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAqBA,oDAwFC;AA7GD,uCAAyB;AACzB,2CAA6B;AAE7B,kDAA6D;AAC7D,kDAA2E;AAG3E,uCAA4C;AAC5C,wDAAqD;AACrD,sEAAmE;AACnE,qEAAkE;AAClE,8EAA2E;AAC3E,sDAA0D;AAC1D,wEAA6E;AAC7E,0DAAuD;AACvD,wDAA0D;AAC1D,8EAAmF;AACnF,wEAAoE;AAEpE,MAAM,UAAU,GAAG,CAAC,CAAC;AAEd,KAAK,UAAU,oBAAoB,CACtC,cAAsB,EACtB,YAAoB,EACpB,aAAsB,EACtB,eAAwB,EACxB,SAAiB,uBAAW,EAC5B,cAAuB,KAAK;IAE5B,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;IAE7B,OAAO,CAAC,GAAG,CAAC,sDAAsD,cAAc,mBAAmB,CAAC,CAAC;IACrG,OAAO,CAAC,GAAG,CAAC,qBAAqB,OAAO,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC;IAElD,qBAAqB;IACrB,IAAA,wCAAkB,EAAC,MAAM,CAAC,CAAC;IAE3B,cAAc,GAAG,IAAI,CAAC,SAAS,CAAC,cAAc,CAAC,CAAC;IAChD,IAAI,CAAC,EAAE,CAAC,UAAU,CAAC,cAAc,CAAC,EAAE,CAAC;QACjC,OAAO,CAAC,KAAK,CAAC,+BAA+B,cAAc,EAAE,CAAC,CAAC;QAC/D,OAAO;IACX,CAAC;IAED,yCAAyC;IACzC,IAAA,iDAA2B,GAAE,CAAC;IAE9B,IAAI,UAAU,GAAG,EAAE,CAAC,YAAY,CAAC,cAAc,EAAE,OAAO,CAAC,CAAC;IAE1D,0EAA0E;IAC1E,MAAM,SAAS,GAAG,IAAA,2BAAY,EAAC,UAAU,CAAC,CAAC;IAE3C,UAAU,GAAG,MAAM,IAAA,6BAAa,EAAC,UAAU,EAAE,cAAc,EAAE,SAAS,CAAC,CAAC;IACxE,IAAI,CAAC,UAAU,EAAE,CAAC;QACd,OAAO,CAAC,KAAK,CAAC,uCAAuC,cAAc,EAAE,CAAC,CAAC;QACvE,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,YAAY,EAAE,SAAS,EAAE,eAAe,EAAE,SAAS,EAAE,YAAY,EAAE,cAAc,EAAE,WAAW,EAAE,IAAI,EAAE,UAAU,EAAE,IAAI,EAAE,CAAC;IACtJ,CAAC;IAED,wBAAwB;IACxB,IAAI,cAAc,GAAG,MAAM,IAAA,+CAAsB,EAAC,cAAc,CAAC,CAAC;IAElE,oEAAoE;IACpE,MAAM,EAAE,YAAY,EAAE,eAAe,EAAE,aAAa,EAAE,GAAG,IAAA,gCAAkB,EAAC,cAAc,EAAE,YAAY,EAAE,SAAS,CAAC,CAAC;IACrH,IAAI,CAAC,YAAY,IAAI,CAAC,eAAe,EAAE,CAAC;QACpC,OAAO,CAAC,KAAK,CAAC,sDAAsD,cAAc,EAAE,CAAC,CAAC;QACtF,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,YAAY,EAAE,SAAS,EAAE,eAAe,EAAE,SAAS,EAAE,YAAY,EAAE,cAAc,EAAE,WAAW,EAAE,IAAI,EAAE,UAAU,EAAE,IAAI,EAAE,CAAC;IACtJ,CAAC;IAED,qBAAqB;IACrB,MAAM,aAAa,GAAG,IAAA,yCAAmB,EAAC,eAAe,CAAC,CAAC;IAE3D,iBAAiB;IACjB,IAAI,MAAM,GAAG,IAAA,kCAAkB,EAAC,cAAc,EAAE,UAAU,EAAE,cAAc,EAAE,SAAS,EAAE,aAAa,CAAC,CAAC;IACtG,0EAA0E;IAE1E,0CAA0C;IAC1C,MAAM,QAAQ,GAAG,MAAM,IAAA,+CAAsB,EAAC,MAAM,EAAE,SAAS,CAAC,CAAC;IAEjE,kCAAkC;IAClC,kHAAkH;IAClH,IAAA,yBAAa,EAAC,QAAQ,EAAE,YAAY,CAAC,CAAC;IAEtC,UAAU;IACV,IAAI,aAAa,GAAc,EAAE,OAAO,EAAE,IAAI,EAAE,YAAY,EAAE,eAAe,EAAE,YAAY,EAAE,cAAc,EAAE,WAAW,EAAE,IAAI,EAAE,UAAU,EAAE,IAAI,EAAE,CAAC;IACnJ,IAAI,aAAa,EAAE,CAAC;QAChB,MAAM,UAAU,GAAG,MAAM,KAAK,sBAAU,CAAC;QACzC,aAAa,GAAG,MAAM,IAAA,yBAAe,EAAC,MAAM,EAAE,cAAc,EAAE,QAAQ,EAAE,eAAe,EAAE,YAAY,EAAE,SAAS,EAAE,UAAU,EAAE,UAAU,EAAE,WAAW,CAAC,CAAC;QACvJ,mHAAmH;IACvH,CAAC;IAED,IAAI,aAAa,CAAC,OAAO,EAAE,CAAC;QACxB,OAAO,CAAC,GAAG,CAAC,2BAA2B,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,CAAC,GAAG,IAAI,UAAU,CAAC,CAAC;QAClF,OAAO,aAAa,CAAC;IACzB,CAAC;IAED,oIAAoI;IACpI,IAAI,CAAC,eAAe,EAAE,CAAC;QACnB,8CAA8C;QAC9C,IAAI,CAAC,aAAa,CAAC,UAAU,IAAI,aAAa,CAAC,UAAU,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACrE,CAAC,aAAa,IAAI,IAAA,kCAAsB,EAAC,YAAY,CAAC,CAAC;YACvD,OAAO,CAAC,GAAG,CAAC,2DAA2D,YAAY,EAAE,CAAC,CAAC;QAC3F,CAAC;aAAM,CAAC;YACJ,IAAA,yDAA+B,EAAC,aAAa,CAAC,UAAU,EAAE,YAAY,CAAC,CAAC;YACxE,OAAO,CAAC,GAAG,CAAC,oCAAoC,YAAY,EAAE,CAAC,CAAC;QACpE,CAAC;IACL,CAAC;SAAM,CAAC;QACJ,OAAO,CAAC,GAAG,CAAC,0BAA0B,YAAY,EAAE,CAAC,CAAC;IAC1D,CAAC;IACD,OAAO,CAAC,GAAG,CAAC,0BAA0B,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,CAAC,GAAG,IAAI,UAAU,CAAC,CAAC;IACjF,OAAO,aAAa,CAAC;AACzB,CAAC;AAED,mKAAmK;AACnK,iNAAiN;AACjN,uNAAuN;AACvN,sKAAsK;AACtK,0IAA0I;AAE1I,uEAAuE;AACvE,4DAA4D;AAC5D,mFAAmF;AACnF,qEAAqE;AACrE,wBAAwB;AAExB,eAAe;AACf,mCAAmC;AACnC,+TAA+T;AAC/T,gBAAgB;AAChB,YAAY;AACZ,8BAA8B;AAC9B,47GAA47G;AAC57G,uCAAuC;AACvC,aAAa;AACb,YAAY;AACZ,8BAA8B;AAC9B,kGAAkG;AAClG,aAAa;AACb,eAAe;AACf,iCAAiC;AACjC,yGAAyG;AACzG,eAAe;AACf,SAAS;AACT,sBAAsB;AACtB,IAAI","sourcesContent":["import * as fs from 'fs';\r\nimport * as path from 'path';\r\n\r\nimport { CSHARPUTGEN, CSVSPLUGIN } from '../types/constants';\r\nimport { deleteTestFileIfExists, writeTestCode } from \"../utils/fileUtils\";\r\n\r\nimport { GenResult } from '../types/genResult';\r\nimport { autoFixTestCode } from './autoFix';\r\nimport { checkXapCode } from '../utils/checkXapCode';\r\nimport { detectTestFramework } from '../utils/detectTestFramework';\r\nimport { ensureValidLLMResponse } from './ensureValidLLMResponse';\r\nimport { generateCodeDependency } from '../analyze/generateCodeDependency';\r\nimport { getOrCreatTestFile } from '../utils/getTestFile';\r\nimport { installOrUpdateAnalyzerTool } from '../analyze/installAnalyzerTool';\r\nimport { preGenProcess } from './preGen/preGenProcess';\r\nimport { prepareUtGenPrompt } from '../llm/preparePrompt';\r\nimport { removeFailedTestCasesInTestFile } from '../utils/removeFailedTestMethods';\r\nimport { sendUserUsageEvent } from '../metrics/appInsightTelemetry';\r\n\r\nconst RETRYTIMES = 3;\r\n\r\nexport async function generateCsharpUtCode(\r\n    sourceCodePath: string,\r\n    testCodePath: string,\r\n    enableAutoFix: boolean,\r\n    keepFailedTests: boolean,\r\n    source: string = CSHARPUTGEN,\r\n    isBenchmark: boolean = false\r\n): Promise<GenResult> {\r\n    const startTime = Date.now();\r\n\r\n    console.log(`*****************Generating C# unit test code for: ${sourceCodePath}*****************`);\r\n    console.log(`Current work dir: ${process.cwd()}`);\r\n\r\n    // send usage metrics\r\n    sendUserUsageEvent(source);\r\n\r\n    sourceCodePath = path.normalize(sourceCodePath);\r\n    if (!fs.existsSync(sourceCodePath)) {\r\n        console.error(`Source code file not found: ${sourceCodePath}`);\r\n        return;\r\n    }\r\n\r\n    // install nuget cli tool to analyze code\r\n    installOrUpdateAnalyzerTool();\r\n\r\n    let sourceCode = fs.readFileSync(sourceCodePath, 'utf-8');\r\n\r\n    // check if source code is xap code, if so, use xap code generation prompt\r\n    const isXapCode = checkXapCode(sourceCode);\r\n\r\n    sourceCode = await preGenProcess(sourceCode, sourceCodePath, isXapCode);\r\n    if (!sourceCode) {\r\n        console.error(`Skip generate test for source code: ${sourceCodePath}`);\r\n        return { success: false, testFilePath: undefined, testProjectPath: undefined, codeFilePath: sourceCodePath, buildResult: null, testResult: null };\r\n    }\r\n\r\n    // get code dependencies\r\n    let codeDependency = await generateCodeDependency(sourceCodePath);\r\n\r\n    // detect existing test file or create a new one by source code path\r\n    const { testFilePath, testProjectPath, testFileExist } = getOrCreatTestFile(sourceCodePath, testCodePath, isXapCode);\r\n    if (!testFilePath || !testProjectPath) {\r\n        console.error(`Failed to get or create test file for source code: ${sourceCodePath}`);\r\n        return { success: false, testFilePath: undefined, testProjectPath: undefined, codeFilePath: sourceCodePath, buildResult: null, testResult: null };\r\n    }\r\n\r\n    // get test framework\r\n    const testFramework = detectTestFramework(testProjectPath);\r\n\r\n    // prepare prompt\r\n    let prompt = prepareUtGenPrompt(sourceCodePath, sourceCode, codeDependency, isXapCode, testFramework);\r\n    // let prompt = getPrompts(sourceCodePath, testFramework, codeDependency);\r\n\r\n    // call LLM service to gen code with retry\r\n    const testCode = await ensureValidLLMResponse(prompt, isXapCode);\r\n\r\n    // write test code to test project\r\n    // To run vstest, need to write the test code to test project first, then build the test project and run the tests\r\n    writeTestCode(testCode, testFilePath);\r\n\r\n    // autofix\r\n    let autoFixResult: GenResult = { success: true, testFilePath, testProjectPath, codeFilePath: sourceCodePath, buildResult: null, testResult: null };\r\n    if (enableAutoFix) {\r\n        const isVsPlugin = source === CSVSPLUGIN;\r\n        autoFixResult = await autoFixTestCode(prompt, sourceCodePath, testCode, testProjectPath, testFilePath, isXapCode, isVsPlugin, RETRYTIMES, isBenchmark);\r\n        // autoFixResult = await autoRetryTestCode(prompt, sourceCode, testCode, testProjectPath, testFilePath, isXapTest);\r\n    }\r\n\r\n    if (autoFixResult.success) {\r\n        console.log(`Gerate UT success time: ${(Date.now() - startTime) / 1000} seconds`);\r\n        return autoFixResult;\r\n    }\r\n\r\n    // if retry max times or test generation failed, and keepFailedTests is false, delete failed test file or remove failed test methods\r\n    if (!keepFailedTests) {\r\n        // keep succeed test code, or delete test file\r\n        if (!autoFixResult.testResult || autoFixResult.testResult.passed === 0) {\r\n            !testFileExist && deleteTestFileIfExists(testFilePath);\r\n            console.log(`Failed to generate csharp test code, deleted test file: ${testFilePath}`);\r\n        } else {\r\n            removeFailedTestCasesInTestFile(autoFixResult.testResult, testFilePath);\r\n            console.log(`write succeed test code to file: ${testFilePath}`);\r\n        }\r\n    } else {\r\n        console.log(`keep failed test file: ${testFilePath}`);\r\n    }\r\n    console.log(`Gerate UT failed time: ${(Date.now() - startTime) / 1000} seconds`);\r\n    return autoFixResult;\r\n}\r\n\r\n// generateCsharpUtCode(\"D:\\\\code\\\\IndexServe\\\\private\\\\indexserve\\\\RecommendationEngine\\\\RecoPS\\\\PSTableClient\\\\PSTableClient.cs\", undefined, false, true, false);\r\n// generateCsharpUtCode(\"D:\\\\code\\\\CS.Service.Fundamental\\\\SharedSegments\\\\SharedSegments\\\\SharedSegments.Plugins\\\\Workflows\\\\Overview\\\\HttpRequestParser\\\\Utils\\\\SqmidHelper.cs\", undefined, true, true, false);\r\n// generateCsharpUtCode(\"D:\\\\code\\\\CS.Service.Fundamental\\\\SharedSegments\\\\SharedSegments\\\\SharedSegments.Plugins\\\\Workflows\\\\Eplant\\\\ResponseBuilders\\\\EplantMigrateResponseBuilder.cs\", undefined, true, true, true);\r\n// generateCsharpUtCode(\"D:\\\\code\\\\UCMDynamics\\\\private\\\\DynamicCode\\\\Plugin\\\\UCMDynamics.AppointmentPlugin\\\\AppointmentOperation.cs\", undefined, false, true, false);\r\n// generateCsharpUtCode(\"D:\\\\code\\\\AIUnitTestExample\\\\src\\\\ToListinator.Analyzers\\\\ToListCountAnalyzer.cs\", undefined, true, true, false);\r\n\r\n// function getPrompts(sourceCodePath, testFramework, codeDependency) {\r\n//     const code = fs.readFileSync(sourceCodePath, 'utf8');\r\n//     let guidancePromt = fs.readFileSync(\"D:\\\\atmp\\\\promptiteration.md\", 'utf8');\r\n//     guidancePromt = guidancePromt.replace(\"xUnit\", testFramework);\r\n//     const prompts = [\r\n\r\n//         // {\r\n//         //     \"role\": \"system\",\r\n//         //     \"content\": \"Keep your answers short and impersonal.\\nUse Markdown formatting in your answers.\\nMake sure to include the programming language name at the start of the Markdown code blocks.\\nAvoid wrapping the whole response in triple backticks.\\nYou can only give one reply for each conversation turn.\"\r\n//         // },\r\n//         {\r\n//             \"role\": \"user\",\r\n//             // \"content\": \"You are Roo, a highly skilled software engineer with extensive knowledge in many programming languages, frameworks, design patterns, and best practices.\\n\\n====\\n\\nRULES\\n\\n- MUST use xUnit test framework to generate test code.\\n- **MUST NOT** contain any comments or explanations in test code.\\n- Ensure the test code MUST has no syntax errors.\\n- Test public methods of class in source code only.\\n- **MUST NOT** access private members (fields or methods) of the source class in test code.\\n- Test method names must follow the pattern: [MethodName]_[Scenario]_[ExpectedResult].\\n- Return the test code in the format: ```C#\\n{Test code}\\n```.\\n- Mock all dependent methods, classes, and values not included in the user input using the Moq library.\\n- MUST include Test code in a namespace declaration and place all `using` statements above the namespace declaration.\\n- MUST import source code in the test code.\\n- MUST import all required .Net namespaces if any .Net class or interface are used in the test code. Like:\\n    - .NET built-in types (e.g., `List<T>`, `IEnumerable<T>`, `Task`, `Assert`, `Moq.Mock`, etc.).\\n    - Any attribute classes (e.g., `[TestMethod]`, `[TestClass]`, etc.).\\n    - Any LINQ or threading related classes (e.g., `System.Linq`, `System.Threading.Tasks`).\\n- All assignments and method arguments in test code MUST strictly match the declared types.\\n- When generating code, always consider the context in which the code is being used. Ensure that your changes are compatible with the existing codebase and that they follow the project's coding standards and best practices.\\n- Do not ask for more information than necessary. Use your knowledge to accomplish the user's request efficiently and effectively.\\n- The user will provide the source code directly in their message.\\n- Your goal is to try to accomplish the user's task, NOT engage in a back and forth conversation.\\n\\n====\\n\\nOBJECTIVE\\n\\nYou accomplish a given task iteratively, breaking it down into clear steps and working through them methodically.\\n\\n- **MUST** analyze the provided code and verify its logic by following these steps:\\n    1. Describe the functionality of each public method, including its inputs, outputs, and key logic.\\n    2. Identify all possible execution paths for public methods, including normal cases, edge cases, and error conditions.\\n    3. Explain any conditional branches, loops, or exception handling for public methods in the code.\\n    4. Based on this analysis, generate unit test cases that cover all identified execution paths, ensuring that mock data and assertions align with the code's logic.\\n- **MUST** remove all code analysis information in responsed test code.\\n- Absolutely forbidden to contains any other information in response except the test code.\\n- **MUST NOT** generate tests for the methods that are attributed with **[ExcludeFromCodeCoverage]**.\\n- **MUST NOT** generate tests for **private** methods.\\n- Analyze the user's task and set clear, achievable goals to accomplish it. Prioritize these goals in a logical order.\\n- Work through these goals sequentially, each goal should correspond to a distinct step in your problem-solving process.\\n\\n====\\n\\nUSER'S CUSTOM INSTRUCTIONS\\n\\nThe following additional instructions are provided by the user, and should be followed to the best of your ability without interfering with the TOOL USE guidelines.\\n\\nLanguage Preference:\\nYou should always speak and think in the English language.\"\r\n//             \"content\": guidancePromt\r\n//         },\r\n//         {\r\n//             \"role\": \"user\",\r\n//             \"content\": `generate unit test for the following csharp code:\\n\\`\\`\\`${code}\\`\\`\\``\r\n//         },\r\n//         // {\r\n//         //     \"role\": \"user\",\r\n//         //     \"content\": `following are the code dependencies of the source code:\\n${codeDependency}`\r\n//         // }\r\n//     ];\r\n//     return prompts;\r\n// }\r\n\r\n\r\n"]}
+{"version":3,"file":"csharpUtGen.js","sourceRoot":"","sources":["../../src/gen/csharpUtGen.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAqBA,oDAwFC;AA7GD,uCAAyB;AACzB,2CAA6B;AAE7B,kDAA6D;AAC7D,kDAA2E;AAG3E,uCAA4C;AAC5C,wDAAqD;AACrD,sEAAmE;AACnE,qEAAkE;AAClE,8EAA2E;AAC3E,sDAA0D;AAC1D,wEAA6E;AAC7E,0DAAuD;AACvD,wDAA0D;AAC1D,8EAAmF;AACnF,wEAAoE;AAEpE,MAAM,UAAU,GAAG,CAAC,CAAC;AAEd,KAAK,UAAU,oBAAoB,CACtC,cAAsB,EACtB,YAAoB,EACpB,aAAsB,EACtB,eAAwB,EACxB,SAAiB,uBAAW,EAC5B,cAAuB,KAAK,EAC5B,mBAA2B,IAAI;IAE/B,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;IAE7B,OAAO,CAAC,GAAG,CAAC,sDAAsD,cAAc,mBAAmB,CAAC,CAAC;IACrG,OAAO,CAAC,GAAG,CAAC,qBAAqB,OAAO,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC;IAElD,qBAAqB;IACrB,IAAA,wCAAkB,EAAC,MAAM,CAAC,CAAC;IAE3B,cAAc,GAAG,IAAI,CAAC,SAAS,CAAC,cAAc,CAAC,CAAC;IAChD,IAAI,CAAC,EAAE,CAAC,UAAU,CAAC,cAAc,CAAC,EAAE,CAAC;QACjC,OAAO,CAAC,KAAK,CAAC,+BAA+B,cAAc,EAAE,CAAC,CAAC;QAC/D,OAAO;IACX,CAAC;IAED,yCAAyC;IACzC,IAAA,iDAA2B,GAAE,CAAC;IAE9B,IAAI,UAAU,GAAG,EAAE,CAAC,YAAY,CAAC,cAAc,EAAE,OAAO,CAAC,CAAC;IAE1D,0EAA0E;IAC1E,MAAM,SAAS,GAAG,IAAA,2BAAY,EAAC,UAAU,CAAC,CAAC;IAE3C,UAAU,GAAG,MAAM,IAAA,6BAAa,EAAC,UAAU,EAAE,cAAc,EAAE,SAAS,CAAC,CAAC;IACxE,IAAI,CAAC,UAAU,EAAE,CAAC;QACd,OAAO,CAAC,KAAK,CAAC,uCAAuC,cAAc,EAAE,CAAC,CAAC;QACvE,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,YAAY,EAAE,SAAS,EAAE,eAAe,EAAE,SAAS,EAAE,YAAY,EAAE,cAAc,EAAE,WAAW,EAAE,IAAI,EAAE,UAAU,EAAE,IAAI,EAAE,CAAC;IACtJ,CAAC;IAED,wBAAwB;IACxB,IAAI,cAAc,GAAG,MAAM,IAAA,+CAAsB,EAAC,cAAc,EAAE,gBAAgB,CAAC,CAAC;IAEpF,oEAAoE;IACpE,MAAM,EAAE,YAAY,EAAE,eAAe,EAAE,aAAa,EAAE,GAAG,IAAA,gCAAkB,EAAC,cAAc,EAAE,YAAY,EAAE,SAAS,CAAC,CAAC;IACrH,IAAI,CAAC,YAAY,IAAI,CAAC,eAAe,EAAE,CAAC;QACpC,OAAO,CAAC,KAAK,CAAC,sDAAsD,cAAc,4CAA4C,CAAC,CAAC;QAChI,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,YAAY,EAAE,SAAS,EAAE,eAAe,EAAE,SAAS,EAAE,YAAY,EAAE,cAAc,EAAE,WAAW,EAAE,IAAI,EAAE,UAAU,EAAE,IAAI,EAAE,CAAC;IACtJ,CAAC;IAED,qBAAqB;IACrB,MAAM,aAAa,GAAG,IAAA,yCAAmB,EAAC,eAAe,CAAC,CAAC;IAE3D,iBAAiB;IACjB,IAAI,MAAM,GAAG,IAAA,kCAAkB,EAAC,cAAc,EAAE,UAAU,EAAE,cAAc,EAAE,SAAS,EAAE,aAAa,CAAC,CAAC;IAEtG,0CAA0C;IAC1C,MAAM,QAAQ,GAAG,MAAM,IAAA,+CAAsB,EAAC,MAAM,EAAE,aAAa,EAAE,SAAS,CAAC,CAAC;IAEhF,kCAAkC;IAClC,kHAAkH;IAClH,IAAA,yBAAa,EAAC,QAAQ,EAAE,YAAY,CAAC,CAAC;IAEtC,UAAU;IACV,IAAI,aAAa,GAAc,EAAE,OAAO,EAAE,IAAI,EAAE,YAAY,EAAE,eAAe,EAAE,YAAY,EAAE,cAAc,EAAE,WAAW,EAAE,IAAI,EAAE,UAAU,EAAE,IAAI,EAAE,CAAC;IACnJ,IAAI,aAAa,EAAE,CAAC;QAChB,MAAM,UAAU,GAAG,MAAM,KAAK,sBAAU,CAAC;QACzC,aAAa,GAAG,MAAM,IAAA,yBAAe,EAAC,MAAM,EAAE,cAAc,EAAE,QAAQ,EAAE,eAAe,EAAE,YAAY,EAAE,aAAa,EAAE,SAAS,EAAE,UAAU,EAAE,UAAU,EAAE,WAAW,CAAC,CAAC;QACtK,mHAAmH;IACvH,CAAC;IAED,IAAI,aAAa,CAAC,OAAO,EAAE,CAAC;QACxB,OAAO,CAAC,GAAG,CAAC,6BAA6B,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,CAAC,GAAG,IAAI,UAAU,CAAC,CAAC;QACpF,OAAO,aAAa,CAAC;IACzB,CAAC;IAED,oIAAoI;IACpI,IAAI,CAAC,eAAe,EAAE,CAAC;QACnB,8CAA8C;QAC9C,IAAI,CAAC,aAAa,CAAC,UAAU,IAAI,aAAa,CAAC,UAAU,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACrE,CAAC,aAAa,IAAI,IAAA,kCAAsB,EAAC,YAAY,CAAC,CAAC;YACvD,OAAO,CAAC,GAAG,CAAC,2DAA2D,YAAY,EAAE,CAAC,CAAC;QAC3F,CAAC;aAAM,CAAC;YACJ,IAAA,yDAA+B,EAAC,aAAa,CAAC,UAAU,EAAE,YAAY,CAAC,CAAC;YACxE,OAAO,CAAC,GAAG,CAAC,oCAAoC,YAAY,EAAE,CAAC,CAAC;QACpE,CAAC;IACL,CAAC;SAAM,CAAC;QACJ,OAAO,CAAC,GAAG,CAAC,0BAA0B,YAAY,EAAE,CAAC,CAAC;IAC1D,CAAC;IACD,OAAO,CAAC,GAAG,CAAC,4BAA4B,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,CAAC,GAAG,IAAI,UAAU,CAAC,CAAC;IACnF,OAAO,aAAa,CAAC;AACzB,CAAC;AAED,mKAAmK;AACnK,iNAAiN;AACjN,uNAAuN;AACvN,sKAAsK;AACtK,0IAA0I","sourcesContent":["import * as fs from 'fs';\r\nimport * as path from 'path';\r\n\r\nimport { CSHARPUTGEN, CSVSPLUGIN } from '../types/constants';\r\nimport { deleteTestFileIfExists, writeTestCode } from \"../utils/fileUtils\";\r\n\r\nimport { GenResult } from '../types/genResult';\r\nimport { autoFixTestCode } from './autoFix';\r\nimport { checkXapCode } from '../utils/checkXapCode';\r\nimport { detectTestFramework } from '../utils/detectTestFramework';\r\nimport { ensureValidLLMResponse } from './ensureValidLLMResponse';\r\nimport { generateCodeDependency } from '../analyze/generateCodeDependency';\r\nimport { getOrCreatTestFile } from '../utils/getTestFile';\r\nimport { installOrUpdateAnalyzerTool } from '../analyze/installAnalyzerTool';\r\nimport { preGenProcess } from './preGen/preGenProcess';\r\nimport { prepareUtGenPrompt } from '../llm/preparePrompt';\r\nimport { removeFailedTestCasesInTestFile } from '../utils/removeFailedTestMethods';\r\nimport { sendUserUsageEvent } from '../metrics/appInsightTelemetry';\r\n\r\nconst RETRYTIMES = 3;\r\n\r\nexport async function generateCsharpUtCode(\r\n    sourceCodePath: string,\r\n    testCodePath: string,\r\n    enableAutoFix: boolean,\r\n    keepFailedTests: boolean,\r\n    source: string = CSHARPUTGEN,\r\n    isBenchmark: boolean = false,\r\n    nugetPackagePath: string = null\r\n): Promise<GenResult> {\r\n    const startTime = Date.now();\r\n\r\n    console.log(`*****************Generating C# unit test code for: ${sourceCodePath}*****************`);\r\n    console.log(`Current work dir: ${process.cwd()}`);\r\n\r\n    // send usage metrics\r\n    sendUserUsageEvent(source);\r\n\r\n    sourceCodePath = path.normalize(sourceCodePath);\r\n    if (!fs.existsSync(sourceCodePath)) {\r\n        console.error(`Source code file not found: ${sourceCodePath}`);\r\n        return;\r\n    }\r\n\r\n    // install nuget cli tool to analyze code\r\n    installOrUpdateAnalyzerTool();\r\n\r\n    let sourceCode = fs.readFileSync(sourceCodePath, 'utf-8');\r\n\r\n    // check if source code is xap code, if so, use xap code generation prompt\r\n    const isXapCode = checkXapCode(sourceCode);\r\n\r\n    sourceCode = await preGenProcess(sourceCode, sourceCodePath, isXapCode);\r\n    if (!sourceCode) {\r\n        console.error(`Skip generate test for source code: ${sourceCodePath}`);\r\n        return { success: false, testFilePath: undefined, testProjectPath: undefined, codeFilePath: sourceCodePath, buildResult: null, testResult: null };\r\n    }\r\n\r\n    // get code dependencies\r\n    let codeDependency = await generateCodeDependency(sourceCodePath, nugetPackagePath);\r\n\r\n    // detect existing test file or create a new one by source code path\r\n    const { testFilePath, testProjectPath, testFileExist } = getOrCreatTestFile(sourceCodePath, testCodePath, isXapCode);\r\n    if (!testFilePath || !testProjectPath) {\r\n        console.error(`Failed to get or create test file for source code: ${sourceCodePath}. Please check if the test project exists.`);\r\n        return { success: false, testFilePath: undefined, testProjectPath: undefined, codeFilePath: sourceCodePath, buildResult: null, testResult: null };\r\n    }\r\n\r\n    // get test framework\r\n    const testFramework = detectTestFramework(testProjectPath);\r\n\r\n    // prepare prompt\r\n    let prompt = prepareUtGenPrompt(sourceCodePath, sourceCode, codeDependency, isXapCode, testFramework);\r\n\r\n    // call LLM service to gen code with retry\r\n    const testCode = await ensureValidLLMResponse(prompt, testFramework, isXapCode);\r\n\r\n    // write test code to test project\r\n    // To run vstest, need to write the test code to test project first, then build the test project and run the tests\r\n    writeTestCode(testCode, testFilePath);\r\n\r\n    // autofix\r\n    let autoFixResult: GenResult = { success: true, testFilePath, testProjectPath, codeFilePath: sourceCodePath, buildResult: null, testResult: null };\r\n    if (enableAutoFix) {\r\n        const isVsPlugin = source === CSVSPLUGIN;\r\n        autoFixResult = await autoFixTestCode(prompt, sourceCodePath, testCode, testProjectPath, testFilePath, testFramework, isXapCode, isVsPlugin, RETRYTIMES, isBenchmark);\r\n        // autoFixResult = await autoRetryTestCode(prompt, sourceCode, testCode, testProjectPath, testFilePath, isXapTest);\r\n    }\r\n\r\n    if (autoFixResult.success) {\r\n        console.log(`Generate UT success time: ${(Date.now() - startTime) / 1000} seconds`);\r\n        return autoFixResult;\r\n    }\r\n\r\n    // if retry max times or test generation failed, and keepFailedTests is false, delete failed test file or remove failed test methods\r\n    if (!keepFailedTests) {\r\n        // keep succeed test code, or delete test file\r\n        if (!autoFixResult.testResult || autoFixResult.testResult.passed === 0) {\r\n            !testFileExist && deleteTestFileIfExists(testFilePath);\r\n            console.log(`Failed to generate csharp test code, deleted test file: ${testFilePath}`);\r\n        } else {\r\n            removeFailedTestCasesInTestFile(autoFixResult.testResult, testFilePath);\r\n            console.log(`write succeed test code to file: ${testFilePath}`);\r\n        }\r\n    } else {\r\n        console.log(`keep failed test file: ${testFilePath}`);\r\n    }\r\n    console.log(`Generate UT failed time: ${(Date.now() - startTime) / 1000} seconds`);\r\n    return autoFixResult;\r\n}\r\n\r\n// generateCsharpUtCode(\"D:\\\\code\\\\IndexServe\\\\private\\\\indexserve\\\\RecommendationEngine\\\\RecoPS\\\\PSTableClient\\\\PSTableClient.cs\", undefined, false, true, false);\r\n// generateCsharpUtCode(\"D:\\\\code\\\\CS.Service.Fundamental\\\\SharedSegments\\\\SharedSegments\\\\SharedSegments.Plugins\\\\Workflows\\\\Overview\\\\HttpRequestParser\\\\Utils\\\\SqmidHelper.cs\", undefined, true, true, false);\r\n// generateCsharpUtCode(\"D:\\\\code\\\\CS.Service.Fundamental\\\\SharedSegments\\\\SharedSegments\\\\SharedSegments.Plugins\\\\Workflows\\\\Eplant\\\\ResponseBuilders\\\\EplantMigrateResponseBuilder.cs\", undefined, true, true, true);\r\n// generateCsharpUtCode(\"D:\\\\code\\\\UCMDynamics\\\\private\\\\DynamicCode\\\\Plugin\\\\UCMDynamics.AppointmentPlugin\\\\AppointmentOperation.cs\", undefined, false, true, false);\r\n// generateCsharpUtCode(\"D:\\\\code\\\\AIUnitTestExample\\\\src\\\\ToListinator.Analyzers\\\\ToListCountAnalyzer.cs\", undefined, true, true, false);\r\n"]}

package/out/gen/ensureValidLLMResponse.d.ts

@@ -1 +1 @@
-export declare function ensureValidLLMResponse(prompt: any[], isXapCode: boolean, retryTimes?: number): Promise<string>;
+export declare function ensureValidLLMResponse(prompt: any[], testFramework: string, isXapCode: boolean, retryTimes?: number): Promise<string>;

package/out/gen/ensureValidLLMResponse.js

@@ -3,12 +3,12 @@ Object.defineProperty(exports, "__esModule", { value: true });
 exports.ensureValidLLMResponse = ensureValidLLMResponse;
 const postGenProcess_1 = require("./postGen/postGenProcess");
 const sendRequestToLLM_1 = require("../llm/sendRequestToLLM");
-async function ensureValidLLMResponse(prompt, isXapCode, retryTimes = 3) {
+async function ensureValidLLMResponse(prompt, testFramework, isXapCode, retryTimes = 3) {
     for (let i = 0; i < retryTimes; i++) {
         // call LLM service to gen code
         const { content: generatedCode } = await (0, sendRequestToLLM_1.sendRequestToLLM)(prompt);
         // extract valid code from LLM response
-        const testCode = (0, postGenProcess_1.postGenProcess)(generatedCode, isXapCode);
+        const testCode = (0, postGenProcess_1.postGenProcess)(generatedCode, testFramework, isXapCode);
         if (!!testCode) {
             return testCode;
         }

package/out/gen/ensureValidLLMResponse.js.map

@@ -1 +1 @@
-{"version":3,"file":"ensureValidLLMResponse.js","sourceRoot":"","sources":["../../src/gen/ensureValidLLMResponse.ts"],"names":[],"mappings":";;AAGA,wDAcC;AAjBD,6DAA0D;AAC1D,8DAA2D;AAEpD,KAAK,UAAU,sBAAsB,CAAC,MAAa,EAAE,SAAkB,EAAE,aAAqB,CAAC;IAClG,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,UAAU,EAAE,CAAC,EAAE,EAAE,CAAC;QAClC,+BAA+B;QAC/B,MAAM,EAAE,OAAO,EAAE,aAAa,EAAE,GAAG,MAAM,IAAA,mCAAgB,EAAC,MAAM,CAAC,CAAC;QAElE,uCAAuC;QACvC,MAAM,QAAQ,GAAG,IAAA,+BAAc,EAAC,aAAa,EAAE,SAAS,CAAC,CAAC;QAC1D,IAAI,CAAC,CAAC,QAAQ,EAAE,CAAC;YACb,OAAO,QAAQ,CAAC;QACpB,CAAC;QACD,OAAO,CAAC,GAAG,CAAC,oDAAoD,CAAC,GAAG,CAAC,IAAI,UAAU,GAAG,CAAC,CAAC;IAC5F,CAAC;IAED,OAAO,SAAS,CAAC;AACrB,CAAC","sourcesContent":["import { postGenProcess } from \"./postGen/postGenProcess\";\r\nimport { sendRequestToLLM } from \"../llm/sendRequestToLLM\";\r\n\r\nexport async function ensureValidLLMResponse(prompt: any[], isXapCode: boolean, retryTimes: number = 3): Promise<string> {\r\n    for (let i = 0; i < retryTimes; i++) {\r\n        // call LLM service to gen code\r\n        const { content: generatedCode } = await sendRequestToLLM(prompt);\r\n    \r\n        // extract valid code from LLM response\r\n        const testCode = postGenProcess(generatedCode, isXapCode);\r\n        if (!!testCode) {\r\n            return testCode;\r\n        }\r\n        console.log(`Failed to generate valid test code, retrying... (${i + 1}/${retryTimes})`);\r\n    }\r\n\r\n    return undefined;\r\n}"]}
+{"version":3,"file":"ensureValidLLMResponse.js","sourceRoot":"","sources":["../../src/gen/ensureValidLLMResponse.ts"],"names":[],"mappings":";;AAGA,wDAcC;AAjBD,6DAA0D;AAC1D,8DAA2D;AAEpD,KAAK,UAAU,sBAAsB,CAAC,MAAa,EAAE,aAAqB,EAAE,SAAkB,EAAE,aAAqB,CAAC;IACzH,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,UAAU,EAAE,CAAC,EAAE,EAAE,CAAC;QAClC,+BAA+B;QAC/B,MAAM,EAAE,OAAO,EAAE,aAAa,EAAE,GAAG,MAAM,IAAA,mCAAgB,EAAC,MAAM,CAAC,CAAC;QAElE,uCAAuC;QACvC,MAAM,QAAQ,GAAG,IAAA,+BAAc,EAAC,aAAa,EAAE,aAAa,EAAE,SAAS,CAAC,CAAC;QACzE,IAAI,CAAC,CAAC,QAAQ,EAAE,CAAC;YACb,OAAO,QAAQ,CAAC;QACpB,CAAC;QACD,OAAO,CAAC,GAAG,CAAC,oDAAoD,CAAC,GAAG,CAAC,IAAI,UAAU,GAAG,CAAC,CAAC;IAC5F,CAAC;IAED,OAAO,SAAS,CAAC;AACrB,CAAC","sourcesContent":["import { postGenProcess } from \"./postGen/postGenProcess\";\r\nimport { sendRequestToLLM } from \"../llm/sendRequestToLLM\";\r\n\r\nexport async function ensureValidLLMResponse(prompt: any[], testFramework: string, isXapCode: boolean, retryTimes: number = 3): Promise<string> {\r\n    for (let i = 0; i < retryTimes; i++) {\r\n        // call LLM service to gen code\r\n        const { content: generatedCode } = await sendRequestToLLM(prompt);\r\n    \r\n        // extract valid code from LLM response\r\n        const testCode = postGenProcess(generatedCode, testFramework, isXapCode);\r\n        if (!!testCode) {\r\n            return testCode;\r\n        }\r\n        console.log(`Failed to generate valid test code, retrying... (${i + 1}/${retryTimes})`);\r\n    }\r\n\r\n    return undefined;\r\n}"]}

package/out/gen/postGen/postGenProcess.d.ts

@@ -1 +1 @@
-export declare function postGenProcess(generatedCode: string, isXapCode: boolean): string;
+export declare function postGenProcess(generatedCode: string, testFramework: string, isXapCode: boolean): string;

package/out/gen/postGen/postGenProcess.js

@@ -6,13 +6,14 @@ const removeSingleLines_1 = require("./removeSingleLines");
 const removeComments_1 = require("./removeComments");
 const repairRequiredNameSpaces_1 = require("./repairRequiredNameSpaces");
 const validateTestCode_1 = require("./validateTestCode");
-function postGenProcess(generatedCode, isXapCode) {
+function postGenProcess(generatedCode, testFramework, isXapCode) {
     if (!isValidTestCode(generatedCode)) {
         return undefined;
     }
     let testCode = (0, extractCodeFromResponse_1.extractCodeFromResponse)(generatedCode);
     testCode = (0, removeComments_1.removeTestCodeComments)(testCode);
     testCode = (0, removeSingleLines_1.removeCSharpOnlyLines)(testCode);
+    testCode = (0, repairRequiredNameSpaces_1.repairRequiredNameSpaces)(testCode, testFramework);
     if (isXapCode) {
         testCode = xapCodePostGenProcess(testCode);
     }
@@ -20,7 +21,7 @@ function postGenProcess(generatedCode, isXapCode) {
     return testCode;
 }
 function xapCodePostGenProcess(testCode) {
-    const processedCode = (0, repairRequiredNameSpaces_1.repairRequiredNameSpaces)(testCode);
+    const processedCode = (0, repairRequiredNameSpaces_1.repairXapRequiredNameSpaces)(testCode);
     return processedCode;
 }
 function isValidTestCode(testCode) {

package/out/gen/postGen/postGenProcess.js.map

@@ -1 +1 @@
-{"version":3,"file":"postGenProcess.js","sourceRoot":"","sources":["../../../src/gen/postGen/postGenProcess.ts"],"names":[],"mappings":";;AAQA,wCAaC;AAnBD,uEAAoE;AACpE,2DAA4D;AAC5D,qDAA0D;AAC1D,yEAAsE;AACtE,yDAAsD;AAEtD,SAAgB,cAAc,CAAC,aAAqB,EAAE,SAAkB;IACpE,IAAI,CAAC,eAAe,CAAC,aAAa,CAAC,EAAE,CAAC;QAClC,OAAO,SAAS,CAAC;IACrB,CAAC;IAED,IAAI,QAAQ,GAAG,IAAA,iDAAuB,EAAC,aAAa,CAAC,CAAC;IACtD,QAAQ,GAAG,IAAA,uCAAsB,EAAC,QAAQ,CAAC,CAAC;IAC5C,QAAQ,GAAG,IAAA,yCAAqB,EAAC,QAAQ,CAAC,CAAC;IAC3C,IAAI,SAAS,EAAE,CAAC;QACZ,QAAQ,GAAG,qBAAqB,CAAC,QAAQ,CAAC,CAAC;IAC/C,CAAC;IACD,OAAO,CAAC,GAAG,CAAC,mDAAmD,QAAQ,CAAC,MAAM,EAAE,CAAC,CAAC;IAClF,OAAO,QAAQ,CAAC;AACpB,CAAC;AAGD,SAAS,qBAAqB,CAAC,QAAgB;IAC3C,MAAM,aAAa,GAAG,IAAA,mDAAwB,EAAC,QAAQ,CAAC,CAAC;IACzD,OAAO,aAAa,CAAC;AACzB,CAAC;AAED,SAAS,eAAe,CAAC,QAAgB;IACrC,IAAI,CAAC,QAAQ,IAAI,QAAQ,CAAC,IAAI,EAAE,KAAK,EAAE,EAAE,CAAC;QACtC,OAAO,KAAK,CAAC,CAAC,0BAA0B;IAC5C,CAAC;IACD,0EAA0E;IAC1E,MAAM,OAAO,GAAG,IAAA,mCAAgB,EAAC,QAAQ,CAAC,CAAC;IAC3C,OAAO,OAAO,CAAC;AACnB,CAAC;AAED,uMAAuM;AACvM,+BAA+B","sourcesContent":["import * as fs from 'fs';\r\n\r\nimport { extractCodeFromResponse } from \"./extractCodeFromResponse\";\r\nimport { removeCSharpOnlyLines } from \"./removeSingleLines\";\r\nimport { removeTestCodeComments } from \"./removeComments\";\r\nimport { repairRequiredNameSpaces } from \"./repairRequiredNameSpaces\";\r\nimport { validateTestCode } from \"./validateTestCode\";\r\n\r\nexport function postGenProcess(generatedCode: string, isXapCode: boolean): string {\r\n    if (!isValidTestCode(generatedCode)) {\r\n        return undefined;\r\n    }\r\n    \r\n    let testCode = extractCodeFromResponse(generatedCode);\r\n    testCode = removeTestCodeComments(testCode);\r\n    testCode = removeCSharpOnlyLines(testCode);\r\n    if (isXapCode) {\r\n        testCode = xapCodePostGenProcess(testCode);\r\n    }\r\n    console.log(`Post generation process completed, code length: ${testCode.length}`);\r\n    return testCode;\r\n}\r\n\r\n\r\nfunction xapCodePostGenProcess(testCode: string): string {\r\n    const processedCode = repairRequiredNameSpaces(testCode);\r\n    return processedCode;\r\n}\r\n\r\nfunction isValidTestCode(testCode: string): boolean {\r\n    if (!testCode || testCode.trim() === '') {\r\n        return false; // Empty code is not valid\r\n    }\r\n    // Check if the test code contains any test attributes or using statements\r\n    const isValid = validateTestCode(testCode);\r\n    return isValid;\r\n}\r\n\r\n// const code = fs.readFileSync(\"D:\\\\code\\\\CS.Service.Fundamental\\\\SharedSegments\\\\SharedSegments\\\\SharedSegments.Tests\\\\Workflows\\\\Overview\\\\HttpRequestParser\\\\Utils\\\\SqmidHelperTests.cs\", 'utf-8');\r\n// postGenProcess(code, false);\r\n"]}
+{"version":3,"file":"postGenProcess.js","sourceRoot":"","sources":["../../../src/gen/postGen/postGenProcess.ts"],"names":[],"mappings":";;AAMA,wCAcC;AApBD,uEAAoE;AACpE,2DAA4D;AAC5D,qDAA0D;AAC1D,yEAAmG;AACnG,yDAAsD;AAEtD,SAAgB,cAAc,CAAC,aAAqB,EAAE,aAAqB,EAAE,SAAkB;IAC3F,IAAI,CAAC,eAAe,CAAC,aAAa,CAAC,EAAE,CAAC;QAClC,OAAO,SAAS,CAAC;IACrB,CAAC;IAED,IAAI,QAAQ,GAAG,IAAA,iDAAuB,EAAC,aAAa,CAAC,CAAC;IACtD,QAAQ,GAAG,IAAA,uCAAsB,EAAC,QAAQ,CAAC,CAAC;IAC5C,QAAQ,GAAG,IAAA,yCAAqB,EAAC,QAAQ,CAAC,CAAC;IAC3C,QAAQ,GAAG,IAAA,mDAAwB,EAAC,QAAQ,EAAE,aAAa,CAAC,CAAC;IAC7D,IAAI,SAAS,EAAE,CAAC;QACZ,QAAQ,GAAG,qBAAqB,CAAC,QAAQ,CAAC,CAAC;IAC/C,CAAC;IACD,OAAO,CAAC,GAAG,CAAC,mDAAmD,QAAQ,CAAC,MAAM,EAAE,CAAC,CAAC;IAClF,OAAO,QAAQ,CAAC;AACpB,CAAC;AAED,SAAS,qBAAqB,CAAC,QAAgB;IAC3C,MAAM,aAAa,GAAG,IAAA,sDAA2B,EAAC,QAAQ,CAAC,CAAC;IAC5D,OAAO,aAAa,CAAC;AACzB,CAAC;AAED,SAAS,eAAe,CAAC,QAAgB;IACrC,IAAI,CAAC,QAAQ,IAAI,QAAQ,CAAC,IAAI,EAAE,KAAK,EAAE,EAAE,CAAC;QACtC,OAAO,KAAK,CAAC,CAAC,0BAA0B;IAC5C,CAAC;IACD,0EAA0E;IAC1E,MAAM,OAAO,GAAG,IAAA,mCAAgB,EAAC,QAAQ,CAAC,CAAC;IAC3C,OAAO,OAAO,CAAC;AACnB,CAAC;AAED,uMAAuM;AACvM,+BAA+B","sourcesContent":["import { extractCodeFromResponse } from \"./extractCodeFromResponse\";\r\nimport { removeCSharpOnlyLines } from \"./removeSingleLines\";\r\nimport { removeTestCodeComments } from \"./removeComments\";\r\nimport { repairRequiredNameSpaces, repairXapRequiredNameSpaces } from \"./repairRequiredNameSpaces\";\r\nimport { validateTestCode } from \"./validateTestCode\";\r\n\r\nexport function postGenProcess(generatedCode: string, testFramework: string, isXapCode: boolean): string {\r\n    if (!isValidTestCode(generatedCode)) {\r\n        return undefined;\r\n    }\r\n\r\n    let testCode = extractCodeFromResponse(generatedCode);\r\n    testCode = removeTestCodeComments(testCode);\r\n    testCode = removeCSharpOnlyLines(testCode);\r\n    testCode = repairRequiredNameSpaces(testCode, testFramework);\r\n    if (isXapCode) {\r\n        testCode = xapCodePostGenProcess(testCode);\r\n    }\r\n    console.log(`Post generation process completed, code length: ${testCode.length}`);\r\n    return testCode;\r\n}\r\n\r\nfunction xapCodePostGenProcess(testCode: string): string {\r\n    const processedCode = repairXapRequiredNameSpaces(testCode);\r\n    return processedCode;\r\n}\r\n\r\nfunction isValidTestCode(testCode: string): boolean {\r\n    if (!testCode || testCode.trim() === '') {\r\n        return false; // Empty code is not valid\r\n    }\r\n    // Check if the test code contains any test attributes or using statements\r\n    const isValid = validateTestCode(testCode);\r\n    return isValid;\r\n}\r\n\r\n// const code = fs.readFileSync(\"D:\\\\code\\\\CS.Service.Fundamental\\\\SharedSegments\\\\SharedSegments\\\\SharedSegments.Tests\\\\Workflows\\\\Overview\\\\HttpRequestParser\\\\Utils\\\\SqmidHelperTests.cs\", 'utf-8');\r\n// postGenProcess(code, false);\r\n"]}

package/out/gen/postGen/repairRequiredNameSpaces.d.ts

@@ -1 +1,2 @@
-export declare function repairRequiredNameSpaces(code: string): string;
+export declare function repairRequiredNameSpaces(code: string, testFramework: string): string;
+export declare function repairXapRequiredNameSpaces(code: string): string;

package/out/gen/postGen/repairRequiredNameSpaces.js

@@ -1,16 +1,29 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.repairRequiredNameSpaces = repairRequiredNameSpaces;
-function repairRequiredNameSpaces(code) {
+exports.repairXapRequiredNameSpaces = repairXapRequiredNameSpaces;
+const constants_1 = require("../../types/constants");
+function repairRequiredNameSpaces(code, testFramework) {
+    const testFrameworkUsing = getTestFrameworkNameSpaces(testFramework);
     const requiredUsings = [
-        "using Xap.ExecutionFramework;",
-        "using Xap.PluginFramework;",
+        testFrameworkUsing,
+    ];
+    code = repairMissingUsings(code, requiredUsings);
+    return code;
+}
+function repairXapRequiredNameSpaces(code) {
+    const xapUsings = [
         "using System;",
         "using System.Threading.Tasks;",
         "using System.Collections.Generic;",
+        "using Xap.ExecutionFramework;",
+        "using Xap.PluginFramework;",
         "using EMPTY = Microsoft.Bing.Xap.Plugins.Void;",
-        "using Microsoft.VisualStudio.TestTools.UnitTesting;"
     ];
+    code = repairMissingUsings(code, xapUsings);
+    return code;
+}
+function repairMissingUsings(code, requiredUsings) {
     // get all using lines
     const usingRegex = /^\s*using\s+[^\r\n;]+;/gm;
     const existingUsings = Array.from(code.matchAll(usingRegex)).map(match => match[0].trim());
@@ -37,6 +50,18 @@ function repairRequiredNameSpaces(code) {
     console.log("repaired namespaces:", missingUsings);
     return newLines.join('\n');
 }
+function getTestFrameworkNameSpaces(testFramework) {
+    const msTestUsing = "using Microsoft.VisualStudio.TestTools.UnitTesting;";
+    if (!testFramework) {
+        return msTestUsing;
+    }
+    const FRAMEWORK_USINGS = {
+        [constants_1.MSTESTFRAMEWORK]: msTestUsing,
+        [constants_1.NUNITTESTFRAMEWORK]: "using NUnit.Framework;",
+        [constants_1.XUNITTESTFRAMEWORK]: "using Xunit;"
+    };
+    return FRAMEWORK_USINGS[testFramework] || msTestUsing;
+}
 // const code = `
 // using Microsoft.VisualStudio.TestTools.UnitTesting;
 // using SharedSegments.Plugins.Workflows.Overview.HttpRequestParser.Utils;

package/out/gen/postGen/repairRequiredNameSpaces.js.map

@@ -1 +1 @@
-{"version":3,"file":"repairRequiredNameSpaces.js","sourceRoot":"","sources":["../../../src/gen/postGen/repairRequiredNameSpaces.ts"],"names":[],"mappings":";;AAGA,4DA2CC;AA3CD,SAAgB,wBAAwB,CAAC,IAAY;IACjD,MAAM,cAAc,GAAG;QACnB,+BAA+B;QAC/B,4BAA4B;QAC5B,eAAe;QACf,+BAA+B;QAC/B,mCAAmC;QACnC,gDAAgD;QAChD,qDAAqD;KACxD,CAAC;IAEF,sBAAsB;IACtB,MAAM,UAAU,GAAG,0BAA0B,CAAC;IAC9C,MAAM,cAAc,GAAG,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,UAAU,CAAC,CAAC,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC;IAE3F,MAAM,aAAa,GAAG,cAAc,CAAC,MAAM,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC,cAAc,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC;IAElF,IAAI,aAAa,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QAC7B,OAAO,IAAI,CAAC;IAChB,CAAC;IAED,oFAAoF;IACpF,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;IAClC,IAAI,WAAW,GAAG,CAAC,CAAC;IAEpB,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,KAAK,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;QACpC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC,UAAU,CAAC,OAAO,CAAC,IAAI,KAAK,CAAC,CAAC,CAAC,CAAC,IAAI,EAAE,KAAK,EAAE,EAAE,CAAC;YACjE,WAAW,GAAG,CAAC,CAAC;YAChB,MAAM;QACV,CAAC;QACD,WAAW,GAAG,CAAC,GAAG,CAAC,CAAC;IACxB,CAAC;IAED,iBAAiB;IACjB,MAAM,QAAQ,GAAG;QACb,GAAG,KAAK,CAAC,KAAK,CAAC,CAAC,EAAE,WAAW,CAAC;QAC9B,GAAG,aAAa;QAChB,GAAG,KAAK,CAAC,KAAK,CAAC,WAAW,CAAC;KAC9B,CAAC;IAEF,OAAO,CAAC,GAAG,CAAC,sBAAsB,EAAE,aAAa,CAAC,CAAC;IAEnD,OAAO,QAAQ,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;AAC/B,CAAC;AAGD,iBAAiB;AACjB,sDAAsD;AACtD,2EAA2E;AAC3E,gBAAgB;AAChB,oCAAoC;AACpC,qBAAqB;AACrB,aAAa;AAEb,oFAAoF;AACpF,IAAI;AACJ,kBAAkB;AAClB,oCAAoC;AACpC,QAAQ;AACR,uBAAuB;AACvB,uFAAuF;AACvF,YAAY;AACZ,yBAAyB;AACzB,2DAA2D;AAC3D,gBAAgB;AAChB,mHAAmH;AACnH,iBAAiB;AAEjB,qBAAqB;AACrB,oEAAoE;AAEpE,wBAAwB;AACxB,wDAAwD;AACxD,YAAY;AAEZ,uBAAuB;AACvB,iFAAiF;AACjF,YAAY;AACZ,yBAAyB;AACzB,2DAA2D;AAC3D,gBAAgB;AAChB,qDAAqD;AACrD,iBAAiB;AAEjB,qBAAqB;AACrB,oEAAoE;AAEpE,wBAAwB;AACxB,qCAAqC;AACrC,YAAY;AAEZ,uBAAuB;AACvB,2EAA2E;AAC3E,YAAY;AACZ,yBAAyB;AACzB,8DAA8D;AAE9D,qBAAqB;AACrB,oEAAoE;AAEpE,wBAAwB;AACxB,qDAAqD;AACrD,YAAY;AAEZ,uBAAuB;AACvB,0EAA0E;AAC1E,YAAY;AACZ,yBAAyB;AACzB,0DAA0D;AAE1D,qBAAqB;AACrB,oEAAoE;AAEpE,wBAAwB;AACxB,qDAAqD;AACrD,YAAY;AAEZ,uBAAuB;AACvB,mFAAmF;AACnF,YAAY;AACZ,yBAAyB;AACzB,2DAA2D;AAC3D,gBAAgB;AAChB,mCAAmC;AACnC,iBAAiB;AAEjB,qBAAqB;AACrB,oEAAoE;AAEpE,wBAAwB;AACxB,qDAAqD;AACrD,YAAY;AACZ,QAAQ;AACR,IAAI;AACJ,KAAK;AAEL,sDAAsD;AACtD,4BAA4B","sourcesContent":["import fs from 'fs';\r\nimport path from 'path';\r\n\r\nexport function repairRequiredNameSpaces(code: string): string {\r\n    const requiredUsings = [\r\n        \"using Xap.ExecutionFramework;\",\r\n        \"using Xap.PluginFramework;\",\r\n        \"using System;\",\r\n        \"using System.Threading.Tasks;\",\r\n        \"using System.Collections.Generic;\",\r\n        \"using EMPTY = Microsoft.Bing.Xap.Plugins.Void;\",\r\n        \"using Microsoft.VisualStudio.TestTools.UnitTesting;\"\r\n    ];\r\n\r\n    // get all using lines\r\n    const usingRegex = /^\\s*using\\s+[^\\r\\n;]+;/gm;\r\n    const existingUsings = Array.from(code.matchAll(usingRegex)).map(match => match[0].trim());\r\n\r\n    const missingUsings = requiredUsings.filter(req => !existingUsings.includes(req));\r\n\r\n    if (missingUsings.length === 0) {\r\n        return code;\r\n    }\r\n\r\n    // find the end of the using block: the first non-using line, or the end of the file\r\n    const lines = code.split(/\\r?\\n/);\r\n    let insertIndex = 0;\r\n\r\n    for (let i = 0; i < lines.length; i++) {\r\n        if (!lines[i].trim().startsWith(\"using\") && lines[i].trim() !== \"\") {\r\n            insertIndex = i;\r\n            break;\r\n        }\r\n        insertIndex = i + 1;\r\n    }\r\n\r\n    // build new code\r\n    const newLines = [\r\n        ...lines.slice(0, insertIndex),\r\n        ...missingUsings,\r\n        ...lines.slice(insertIndex)\r\n    ];\r\n\r\n    console.log(\"repaired namespaces:\", missingUsings);\r\n\r\n    return newLines.join('\\n');\r\n}\r\n\r\n\r\n// const code = `\r\n// using Microsoft.VisualStudio.TestTools.UnitTesting;\r\n// using SharedSegments.Plugins.Workflows.Overview.HttpRequestParser.Utils;\r\n// using System;\r\n// using System.Collections.Generic;\r\n// using System.Text;\r\n// using Moq;\r\n\r\n// namespace SharedSegments.Plugins.Tests.Workflows.Overview.HttpRequestParser.Utils\r\n// {\r\n//     [TestClass]\r\n//     public class SqmidHelperTests\r\n//     {\r\n//         [TestMethod]\r\n//         public void GetSqmidFromCookie_ValidCookiesWithBlisId_ReturnsDecodedBlisId()\r\n//         {\r\n//             // Arrange\r\n//             var cookies = new Dictionary<string, string>\r\n//             {\r\n//                 { \"USRLOC\", \"BID=\" + Convert.ToBase64String(Encoding.UTF8.GetBytes(\"test_decodedBlisId_test\")) }\r\n//             };\r\n\r\n//             // Act\r\n//             var result = SqmidHelper.GetSqmidFromCookie(cookies);\r\n\r\n//             // Assert\r\n//             Assert.AreEqual(\"decodedBlisId\", result);\r\n//         }\r\n\r\n//         [TestMethod]\r\n//         public void GetSqmidFromCookie_ValidCookiesWithoutBlisId_ReturnsNull()\r\n//         {\r\n//             // Arrange\r\n//             var cookies = new Dictionary<string, string>\r\n//             {\r\n//                 { \"USRLOC\", \"SomeOtherValue=123\" }\r\n//             };\r\n\r\n//             // Act\r\n//             var result = SqmidHelper.GetSqmidFromCookie(cookies);\r\n\r\n//             // Assert\r\n//             Assert.IsNull(result);\r\n//         }\r\n\r\n//         [TestMethod]\r\n//         public void GetSqmidFromCookie_EmptyCookies_ReturnsEmptyString()\r\n//         {\r\n//             // Arrange\r\n//             var cookies = new Dictionary<string, string>();\r\n\r\n//             // Act\r\n//             var result = SqmidHelper.GetSqmidFromCookie(cookies);\r\n\r\n//             // Assert\r\n//             Assert.AreEqual(string.Empty, result);\r\n//         }\r\n\r\n//         [TestMethod]\r\n//         public void GetSqmidFromCookie_NullCookies_ReturnsEmptyString()\r\n//         {\r\n//             // Arrange\r\n//             IDictionary<string, string> cookies = null;\r\n\r\n//             // Act\r\n//             var result = SqmidHelper.GetSqmidFromCookie(cookies);\r\n\r\n//             // Assert\r\n//             Assert.AreEqual(string.Empty, result);\r\n//         }\r\n\r\n//         [TestMethod]\r\n//         public void GetSqmidFromCookie_UsrLocWithEmptyValue_ReturnsEmptyString()\r\n//         {\r\n//             // Arrange\r\n//             var cookies = new Dictionary<string, string>\r\n//             {\r\n//                 { \"USRLOC\", \"\" }\r\n//             };\r\n\r\n//             // Act\r\n//             var result = SqmidHelper.GetSqmidFromCookie(cookies);\r\n\r\n//             // Assert\r\n//             Assert.AreEqual(string.Empty, result);\r\n//         }\r\n//     }\r\n// }\r\n// `;\r\n\r\n// const cleanedCode = repairRequiredNameSpaces(code);\r\n// console.log(cleanedCode);"]}
+{"version":3,"file":"repairRequiredNameSpaces.js","sourceRoot":"","sources":["../../../src/gen/postGen/repairRequiredNameSpaces.ts"],"names":[],"mappings":";;AAEA,4DAQC;AAED,kEAYC;AAxBD,qDAAgG;AAEhG,SAAgB,wBAAwB,CAAC,IAAY,EAAE,aAAqB;IACxE,MAAM,kBAAkB,GAAG,0BAA0B,CAAC,aAAa,CAAC,CAAC;IACrE,MAAM,cAAc,GAAG;QACnB,kBAAkB;KACrB,CAAC;IAEF,IAAI,GAAG,mBAAmB,CAAC,IAAI,EAAE,cAAc,CAAC,CAAC;IACjD,OAAO,IAAI,CAAC;AAChB,CAAC;AAED,SAAgB,2BAA2B,CAAC,IAAY;IACpD,MAAM,SAAS,GAAG;QACd,eAAe;QACf,+BAA+B;QAC/B,mCAAmC;QACnC,+BAA+B;QAC/B,4BAA4B;QAC5B,gDAAgD;KACnD,CAAC;IAEF,IAAI,GAAG,mBAAmB,CAAC,IAAI,EAAE,SAAS,CAAC,CAAC;IAC5C,OAAO,IAAI,CAAC;AAChB,CAAC;AAED,SAAS,mBAAmB,CAAC,IAAY,EAAE,cAAwB;IAC/D,sBAAsB;IACtB,MAAM,UAAU,GAAG,0BAA0B,CAAC;IAC9C,MAAM,cAAc,GAAG,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,UAAU,CAAC,CAAC,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC;IAE3F,MAAM,aAAa,GAAG,cAAc,CAAC,MAAM,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC,cAAc,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC;IAElF,IAAI,aAAa,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QAC7B,OAAO,IAAI,CAAC;IAChB,CAAC;IAED,oFAAoF;IACpF,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;IAClC,IAAI,WAAW,GAAG,CAAC,CAAC;IAEpB,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,KAAK,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;QACpC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC,UAAU,CAAC,OAAO,CAAC,IAAI,KAAK,CAAC,CAAC,CAAC,CAAC,IAAI,EAAE,KAAK,EAAE,EAAE,CAAC;YACjE,WAAW,GAAG,CAAC,CAAC;YAChB,MAAM;QACV,CAAC;QACD,WAAW,GAAG,CAAC,GAAG,CAAC,CAAC;IACxB,CAAC;IAED,iBAAiB;IACjB,MAAM,QAAQ,GAAG;QACb,GAAG,KAAK,CAAC,KAAK,CAAC,CAAC,EAAE,WAAW,CAAC;QAC9B,GAAG,aAAa;QAChB,GAAG,KAAK,CAAC,KAAK,CAAC,WAAW,CAAC;KAC9B,CAAC;IAEF,OAAO,CAAC,GAAG,CAAC,sBAAsB,EAAE,aAAa,CAAC,CAAC;IAEnD,OAAO,QAAQ,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;AAC/B,CAAC;AAED,SAAS,0BAA0B,CAAC,aAAqB;IACrD,MAAM,WAAW,GAAG,qDAAqD,CAAC;IAE1E,IAAI,CAAC,aAAa,EAAE,CAAC;QACjB,OAAO,WAAW,CAAC;IACvB,CAAC;IAED,MAAM,gBAAgB,GAAG;QACrB,CAAC,2BAAe,CAAC,EAAE,WAAW;QAC9B,CAAC,8BAAkB,CAAC,EAAE,wBAAwB;QAC9C,CAAC,8BAAkB,CAAC,EAAE,cAAc;KAC9B,CAAC;IAEX,OAAO,gBAAgB,CAAC,aAAa,CAAC,IAAI,WAAW,CAAC;AAC1D,CAAC;AAID,iBAAiB;AACjB,sDAAsD;AACtD,2EAA2E;AAC3E,gBAAgB;AAChB,oCAAoC;AACpC,qBAAqB;AACrB,aAAa;AAEb,oFAAoF;AACpF,IAAI;AACJ,kBAAkB;AAClB,oCAAoC;AACpC,QAAQ;AACR,uBAAuB;AACvB,uFAAuF;AACvF,YAAY;AACZ,yBAAyB;AACzB,2DAA2D;AAC3D,gBAAgB;AAChB,mHAAmH;AACnH,iBAAiB;AAEjB,qBAAqB;AACrB,oEAAoE;AAEpE,wBAAwB;AACxB,wDAAwD;AACxD,YAAY;AAEZ,uBAAuB;AACvB,iFAAiF;AACjF,YAAY;AACZ,yBAAyB;AACzB,2DAA2D;AAC3D,gBAAgB;AAChB,qDAAqD;AACrD,iBAAiB;AAEjB,qBAAqB;AACrB,oEAAoE;AAEpE,wBAAwB;AACxB,qCAAqC;AACrC,YAAY;AAEZ,uBAAuB;AACvB,2EAA2E;AAC3E,YAAY;AACZ,yBAAyB;AACzB,8DAA8D;AAE9D,qBAAqB;AACrB,oEAAoE;AAEpE,wBAAwB;AACxB,qDAAqD;AACrD,YAAY;AAEZ,uBAAuB;AACvB,0EAA0E;AAC1E,YAAY;AACZ,yBAAyB;AACzB,0DAA0D;AAE1D,qBAAqB;AACrB,oEAAoE;AAEpE,wBAAwB;AACxB,qDAAqD;AACrD,YAAY;AAEZ,uBAAuB;AACvB,mFAAmF;AACnF,YAAY;AACZ,yBAAyB;AACzB,2DAA2D;AAC3D,gBAAgB;AAChB,mCAAmC;AACnC,iBAAiB;AAEjB,qBAAqB;AACrB,oEAAoE;AAEpE,wBAAwB;AACxB,qDAAqD;AACrD,YAAY;AACZ,QAAQ;AACR,IAAI;AACJ,KAAK;AAEL,sDAAsD;AACtD,4BAA4B","sourcesContent":["import { MSTESTFRAMEWORK, NUNITTESTFRAMEWORK, XUNITTESTFRAMEWORK } from '../../types/constants';\r\n\r\nexport function repairRequiredNameSpaces(code: string, testFramework: string): string {\r\n    const testFrameworkUsing = getTestFrameworkNameSpaces(testFramework);\r\n    const requiredUsings = [\r\n        testFrameworkUsing,\r\n    ];\r\n\r\n    code = repairMissingUsings(code, requiredUsings);\r\n    return code;\r\n}\r\n\r\nexport function repairXapRequiredNameSpaces(code: string): string {\r\n    const xapUsings = [\r\n        \"using System;\",\r\n        \"using System.Threading.Tasks;\",\r\n        \"using System.Collections.Generic;\",\r\n        \"using Xap.ExecutionFramework;\",\r\n        \"using Xap.PluginFramework;\",\r\n        \"using EMPTY = Microsoft.Bing.Xap.Plugins.Void;\",\r\n    ];\r\n\r\n    code = repairMissingUsings(code, xapUsings);\r\n    return code;\r\n}\r\n\r\nfunction repairMissingUsings(code: string, requiredUsings: string[]): string {\r\n    // get all using lines\r\n    const usingRegex = /^\\s*using\\s+[^\\r\\n;]+;/gm;\r\n    const existingUsings = Array.from(code.matchAll(usingRegex)).map(match => match[0].trim());\r\n\r\n    const missingUsings = requiredUsings.filter(req => !existingUsings.includes(req));\r\n\r\n    if (missingUsings.length === 0) {\r\n        return code;\r\n    }\r\n\r\n    // find the end of the using block: the first non-using line, or the end of the file\r\n    const lines = code.split(/\\r?\\n/);\r\n    let insertIndex = 0;\r\n\r\n    for (let i = 0; i < lines.length; i++) {\r\n        if (!lines[i].trim().startsWith(\"using\") && lines[i].trim() !== \"\") {\r\n            insertIndex = i;\r\n            break;\r\n        }\r\n        insertIndex = i + 1;\r\n    }\r\n\r\n    // build new code\r\n    const newLines = [\r\n        ...lines.slice(0, insertIndex),\r\n        ...missingUsings,\r\n        ...lines.slice(insertIndex)\r\n    ];\r\n\r\n    console.log(\"repaired namespaces:\", missingUsings);\r\n\r\n    return newLines.join('\\n');\r\n}\r\n\r\nfunction getTestFrameworkNameSpaces(testFramework: string): string {\r\n    const msTestUsing = \"using Microsoft.VisualStudio.TestTools.UnitTesting;\";\r\n\r\n    if (!testFramework) {\r\n        return msTestUsing;\r\n    }\r\n\r\n    const FRAMEWORK_USINGS = {\r\n        [MSTESTFRAMEWORK]: msTestUsing,\r\n        [NUNITTESTFRAMEWORK]: \"using NUnit.Framework;\",\r\n        [XUNITTESTFRAMEWORK]: \"using Xunit;\"\r\n    } as const;\r\n\r\n    return FRAMEWORK_USINGS[testFramework] || msTestUsing;\r\n}\r\n\r\n\r\n\r\n// const code = `\r\n// using Microsoft.VisualStudio.TestTools.UnitTesting;\r\n// using SharedSegments.Plugins.Workflows.Overview.HttpRequestParser.Utils;\r\n// using System;\r\n// using System.Collections.Generic;\r\n// using System.Text;\r\n// using Moq;\r\n\r\n// namespace SharedSegments.Plugins.Tests.Workflows.Overview.HttpRequestParser.Utils\r\n// {\r\n//     [TestClass]\r\n//     public class SqmidHelperTests\r\n//     {\r\n//         [TestMethod]\r\n//         public void GetSqmidFromCookie_ValidCookiesWithBlisId_ReturnsDecodedBlisId()\r\n//         {\r\n//             // Arrange\r\n//             var cookies = new Dictionary<string, string>\r\n//             {\r\n//                 { \"USRLOC\", \"BID=\" + Convert.ToBase64String(Encoding.UTF8.GetBytes(\"test_decodedBlisId_test\")) }\r\n//             };\r\n\r\n//             // Act\r\n//             var result = SqmidHelper.GetSqmidFromCookie(cookies);\r\n\r\n//             // Assert\r\n//             Assert.AreEqual(\"decodedBlisId\", result);\r\n//         }\r\n\r\n//         [TestMethod]\r\n//         public void GetSqmidFromCookie_ValidCookiesWithoutBlisId_ReturnsNull()\r\n//         {\r\n//             // Arrange\r\n//             var cookies = new Dictionary<string, string>\r\n//             {\r\n//                 { \"USRLOC\", \"SomeOtherValue=123\" }\r\n//             };\r\n\r\n//             // Act\r\n//             var result = SqmidHelper.GetSqmidFromCookie(cookies);\r\n\r\n//             // Assert\r\n//             Assert.IsNull(result);\r\n//         }\r\n\r\n//         [TestMethod]\r\n//         public void GetSqmidFromCookie_EmptyCookies_ReturnsEmptyString()\r\n//         {\r\n//             // Arrange\r\n//             var cookies = new Dictionary<string, string>();\r\n\r\n//             // Act\r\n//             var result = SqmidHelper.GetSqmidFromCookie(cookies);\r\n\r\n//             // Assert\r\n//             Assert.AreEqual(string.Empty, result);\r\n//         }\r\n\r\n//         [TestMethod]\r\n//         public void GetSqmidFromCookie_NullCookies_ReturnsEmptyString()\r\n//         {\r\n//             // Arrange\r\n//             IDictionary<string, string> cookies = null;\r\n\r\n//             // Act\r\n//             var result = SqmidHelper.GetSqmidFromCookie(cookies);\r\n\r\n//             // Assert\r\n//             Assert.AreEqual(string.Empty, result);\r\n//         }\r\n\r\n//         [TestMethod]\r\n//         public void GetSqmidFromCookie_UsrLocWithEmptyValue_ReturnsEmptyString()\r\n//         {\r\n//             // Arrange\r\n//             var cookies = new Dictionary<string, string>\r\n//             {\r\n//                 { \"USRLOC\", \"\" }\r\n//             };\r\n\r\n//             // Act\r\n//             var result = SqmidHelper.GetSqmidFromCookie(cookies);\r\n\r\n//             // Assert\r\n//             Assert.AreEqual(string.Empty, result);\r\n//         }\r\n//     }\r\n// }\r\n// `;\r\n\r\n// const cleanedCode = repairRequiredNameSpaces(code);\r\n// console.log(cleanedCode);"]}

package/out/index.d.ts

@@ -1,2 +1,3 @@
 export { generateCsharpUtCode } from './gen/csharpUtGen';
+export { generateBatchTests } from './batch/generateCodeTests';
 export { setup } from './setup/setup';

package/out/index.js

@@ -1,8 +1,10 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.setup = exports.generateCsharpUtCode = void 0;
+exports.setup = exports.generateBatchTests = exports.generateCsharpUtCode = void 0;
 var csharpUtGen_1 = require("./gen/csharpUtGen");
 Object.defineProperty(exports, "generateCsharpUtCode", { enumerable: true, get: function () { return csharpUtGen_1.generateCsharpUtCode; } });
+var generateCodeTests_1 = require("./batch/generateCodeTests");
+Object.defineProperty(exports, "generateBatchTests", { enumerable: true, get: function () { return generateCodeTests_1.generateBatchTests; } });
 var setup_1 = require("./setup/setup");
 Object.defineProperty(exports, "setup", { enumerable: true, get: function () { return setup_1.setup; } });
 //# sourceMappingURL=index.js.map

package/out/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":";;;AAAA,iDAAyD;AAAhD,mHAAA,oBAAoB,OAAA;AAC7B,uCAAsC;AAA7B,8FAAA,KAAK,OAAA","sourcesContent":["export { generateCsharpUtCode } from './gen/csharpUtGen';\r\nexport { setup } from './setup/setup';"]}
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":";;;AAAA,iDAAyD;AAAhD,mHAAA,oBAAoB,OAAA;AAC7B,+DAA+D;AAAtD,uHAAA,kBAAkB,OAAA;AAC3B,uCAAsC;AAA7B,8FAAA,KAAK,OAAA","sourcesContent":["export { generateCsharpUtCode } from './gen/csharpUtGen';\r\nexport { generateBatchTests } from './batch/generateCodeTests';\r\nexport { setup } from './setup/setup';"]}

package/out/llm/model/{Gpt4o.js → Gpt.js}

@@ -10,7 +10,8 @@ const runCommand_1 = require("../../utils/runCommand");
 let cachedToken = null;
 let tokenExpiration = null;
 // platform provided endpoint
-const gptEndpoint = "https://weatherut.openai.azure.com/openai/deployments/gpt-4o/chat/completions?api-version=2025-01-01-preview";
+const gptEndpoint = "https://weatherutswedencentral.cognitiveservices.azure.com/openai/deployments/gpt-5-chat/chat/completions?api-version=2025-01-01-preview";
+// const gptEndpoint = "https://weatherut.openai.azure.com/openai/deployments/gpt-4o/chat/completions?api-version=2025-01-01-preview";
 // const gptEndpoint = "https://autogenerate-unit-tests-draft-1.openai.azure.com/openai/deployments/gpt-4o/chat/completions?api-version=2025-01-01-preview";
 // const gptEndpoint = "https://weathergptv4.openai.azure.com/openai/deployments/gpt-4.1/chat/completions?api-version=2025-01-01-preview";
 // const gptEndpoint = "https://takht-m99407ib-swedencentral.cognitiveservices.azure.com/openai/deployments/gpt-4o/chat/completions?api-version=2025-01-01-preview";
@@ -21,7 +22,8 @@ async function getToken() {
         return cachedToken;
     }
     try {
-        const credential = new identity_1.ChainedTokenCredential(new identity_1.AzureCliCredential(), new identity_1.ManagedIdentityCredential(), new identity_1.DefaultAzureCredential());
+        const credential = new identity_1.ChainedTokenCredential(new identity_1.AzureCliCredential(), new identity_1.ManagedIdentityCredential(), new identity_1.DefaultAzureCredential() // CodeQL [SM05138] This is non-production testing code which is not deployed.
+        );
         const tokenResponse = await credential.getToken('https://cognitiveservices.azure.com/.default');
         if (tokenResponse) {
             cachedToken = tokenResponse.token;
@@ -88,4 +90,4 @@ async function getLLMResponse(prompt) {
 // getLLMResponse("Generate xUnit tests for the ToListCountAnalyzer class.").then((response) => {
 //     console.log(response.choices[0]?.message?.content)
 // });
-//# sourceMappingURL=Gpt4o.js.map
+//# sourceMappingURL=Gpt.js.map