@kapeta/local-cluster-service 0.51.4 → 0.52.1

package/CHANGELOG.md

@@ -1,3 +1,19 @@
+## [0.52.1](https://github.com/kapetacom/local-cluster-service/compare/v0.52.0...v0.52.1) (2024-06-12)
+
+
+### Bug Fixes
+
+* adjust errorclassifier, add errordetails and adjust codefix ([1eb8000](https://github.com/kapetacom/local-cluster-service/commit/1eb80005f59c103d558219d07bed6f01b9ebc562))
+* look for similar files when file asked for does not exist ([abe75c4](https://github.com/kapetacom/local-cluster-service/commit/abe75c4fbcbd812565aa154d6437a288e58403ac))
+* review comments ([542f3dc](https://github.com/kapetacom/local-cluster-service/commit/542f3dca02fc006ee0bd7adf2ad76f28295ba6f2))
+
+# [0.52.0](https://github.com/kapetacom/local-cluster-service/compare/v0.51.4...v0.52.0) (2024-06-10)
+
+
+### Features
+
+* bump codegen to get codeGenerator.validateForTarget ([#171](https://github.com/kapetacom/local-cluster-service/issues/171)) ([2969f15](https://github.com/kapetacom/local-cluster-service/commit/2969f15a4c7909af58886310aa04140b8b890751))
+
 ## [0.51.4](https://github.com/kapetacom/local-cluster-service/compare/v0.51.3...v0.51.4) (2024-06-10)
 
 

package/dist/cjs/src/storm/codegen.d.ts

@@ -26,6 +26,11 @@ export declare class StormCodegen {
      */
     private processBlockCode;
     private verifyAndFixCode;
+    private tryToFixFile;
+    private classifyErrors;
+    private getErrorDetailsForFile;
+    private createFixRequestForFile;
+    private getErrorLine;
     removePrefix(prefix: string, str: string): string;
     /**
      * Sends the code to the AI for a fix

package/dist/cjs/src/storm/codegen.js

@@ -38,11 +38,13 @@ const event_parser_1 = require("./event-parser");
 const stream_1 = require("./stream");
 const nodejs_utils_1 = require("@kapeta/nodejs-utils");
 const promises_1 = require("fs/promises");
-const path_1 = __importStar(require("path"));
+const path_1 = __importDefault(require("path"));
+const path_2 = __importStar(require("path"));
 const node_os_1 = __importDefault(require("node:os"));
 const fs_1 = require("fs");
-const path_2 = __importDefault(require("path"));
 const yaml_1 = __importDefault(require("yaml"));
+const fs = __importStar(require("node:fs"));
+const uuid_1 = require("uuid");
 const SIMULATED_DELAY = 1000;
 class SimulatedFileDelay {
     file;
@@ -197,7 +199,7 @@ class StormCodegen {
                     ...data,
                     payload: {
                         ...data.payload,
-                        path: (0, path_1.join)(basePath, data.payload.filename),
+                        path: (0, path_2.join)(basePath, data.payload.filename),
                         blockName,
                         blockRef,
                         instanceId,
@@ -209,7 +211,7 @@ class StormCodegen {
                     ...data,
                     payload: {
                         ...data.payload,
-                        path: (0, path_1.join)(basePath, data.payload.filename),
+                        path: (0, path_2.join)(basePath, data.payload.filename),
                         blockName,
                         blockRef,
                         instanceId,
@@ -221,7 +223,7 @@ class StormCodegen {
                     ...data,
                     payload: {
                         ...data.payload,
-                        path: (0, path_1.join)(basePath, data.payload.filename),
+                        path: (0, path_2.join)(basePath, data.payload.filename),
                         blockName,
                         blockRef,
                         instanceId,
@@ -241,7 +243,7 @@ class StormCodegen {
                     created: Date.now(),
                     payload: {
                         filename: data.payload.filename,
-                        path: (0, path_1.join)(this.getBasePath(blockUri.fullName), data.payload.filename),
+                        path: (0, path_2.join)(this.getBasePath(blockUri.fullName), data.payload.filename),
                         content: data.payload.content,
                         blockRef: ref,
                         instanceId: event_parser_1.StormEventParser.toInstanceIdFromRef(ref),
@@ -297,7 +299,7 @@ class StormCodegen {
         }
         // Gather the context files for implementation. These will be all be passed to the AI
         const contextFiles = relevantFiles.filter((file) => ![codegen_1.AIFileTypes.SERVICE, codegen_1.AIFileTypes.WEB_SCREEN].includes(file.type));
-        // Send the service and UI templates to the AI. These will be send one-by-one in addition to the context files
+        // Send the service and UI templates to the AI. These will be sent one-by-one in addition to the context files
         const serviceFiles = allFiles.filter((file) => file.type === codegen_1.AIFileTypes.SERVICE);
         if (serviceFiles.length > 0) {
             await this.processTemplates((0, nodejs_utils_1.parseKapetaUri)(block.uri), block.aiName, stormClient_1.stormClient.createServiceImplementation.bind(stormClient_1.stormClient), serviceFiles, contextFiles);
@@ -307,17 +309,17 @@ class StormCodegen {
             return;
         }
         for (const serviceFile of serviceFiles) {
-            const filePath = (0, path_1.join)(basePath, serviceFile.filename);
+            const filePath = (0, path_2.join)(basePath, serviceFile.filename);
             await (0, promises_1.writeFile)(filePath, serviceFile.content);
         }
         for (const serviceFile of contextFiles) {
-            const filePath = (0, path_1.join)(basePath, serviceFile.filename);
+            const filePath = (0, path_2.join)(basePath, serviceFile.filename);
             await (0, promises_1.writeFile)(filePath, serviceFile.content);
         }
-        const kapetaYmlPath = (0, path_1.join)(basePath, 'kapeta.yml');
+        const kapetaYmlPath = (0, path_2.join)(basePath, 'kapeta.yml');
         await (0, promises_1.writeFile)(kapetaYmlPath, yaml_1.default.stringify(block.content));
         for (const screenFile of screenFiles) {
-            const filePath = (0, path_1.join)(basePath, screenFile.payload.filename);
+            const filePath = (0, path_2.join)(basePath, screenFile.payload.filename);
             await (0, promises_1.writeFile)(filePath, screenFile.payload.content);
         }
         const screenFilesConverted = screenFiles.map(screenFile => {
@@ -329,8 +331,8 @@ class StormCodegen {
                 type: codegen_1.AIFileTypes.WEB_SCREEN,
             };
         });
-        const filesToBeFixed = serviceFiles.concat(contextFiles).concat(screenFilesConverted);
         allFiles.push(...screenFilesConverted);
+        const filesToBeFixed = serviceFiles.concat(contextFiles).concat(screenFilesConverted);
         const codeGenerator = new codegen_1.BlockCodeGenerator(block.content);
         await this.verifyAndFixCode(codeGenerator, basePath, filesToBeFixed, allFiles);
         const blockRef = block.uri;
@@ -346,7 +348,7 @@ class StormCodegen {
             },
         });
     }
-    async verifyAndFixCode(codeGenerator, basePath, filesToBeFixed, knownFiles) {
+    async verifyAndFixCode(codeGenerator, basePath, filesToBeFixed, allFiles) {
         let attempts = 0;
         let validCode = false;
         for (let i = 0; i <= 3; i++) {
@@ -360,35 +362,10 @@ class StormCodegen {
                 }
                 if (result && !result.valid) {
                     console.debug('Validation error:', result);
-                    const errorStream = await stormClient_1.stormClient.createErrorClassification(result.error, []);
-                    const fixes = new Map();
-                    this.out.on('aborted', () => {
-                        errorStream.abort();
-                    });
-                    errorStream.on('data', (evt) => {
-                        if (evt.type === 'ERROR_CLASSIFIER') {
-                            // find the file that caused the error
-                            // strip base path from event file name, if it exists sometimes the AI sends the full path
-                            const eventFileName = this.removePrefix(basePath + '/', evt.payload.filename);
-                            const file = filesToBeFixed.find((f) => f.filename === eventFileName);
-                            if (!file) {
-                                console.log(`Could not find the file ${eventFileName} in the list of files to be fixed, Henrik might wanna create a new file for this fix`);
-                            }
-                            // read the content of the file
-                            const content = (0, fs_1.readFileSync)((0, path_1.join)(basePath, eventFileName), 'utf8');
-                            const fix = `${evt.payload.potentialFix}\n---\n${knownFiles
-                                .map((e) => e.filename)
-                                .join('\n')}\n---\n${content}`;
-                            //console.log(`trying to fix the code in ${eventFileName}`);
-                            //console.debug(`with the fix:\n${fix}`);
-                            const code = this.codeFix(fix);
-                            fixes.set((0, path_1.join)(basePath, eventFileName), code);
-                        }
-                    });
-                    await errorStream.waitForDone();
-                    for (const [filename, codePromise] of fixes) {
-                        const code = await codePromise;
-                        (0, fs_1.writeFileSync)(filename, code);
+                    const errors = await this.classifyErrors(result.error, basePath);
+                    for (const [filename, fileErrors] of errors.entries()) {
+                        // todo: only try to fix file if it is part of filesToBeFixed
+                        await this.tryToFixFile(basePath, filename, fileErrors, allFiles, codeGenerator);
                     }
                 }
             }
@@ -403,6 +380,127 @@ class StormCodegen {
             console.error(`Validation failed for ${basePath} after ${attempts} attempts`);
         }
     }
+    async tryToFixFile(basePath, filename, fileErrors, allFiles, codeGenerator) {
+        console.log(`Processing ${filename}`);
+        const language = await codeGenerator.language();
+        const relevantFiles = allFiles.filter(file => file.type != codegen_1.AIFileTypes.IGNORE);
+        for (let attempts = 1; attempts <= 5; attempts++) {
+            if (fileErrors.length == 0) {
+                console.log(`No more errors for ${filename}`);
+                return;
+            }
+            console.log(`Errors in ${filename} - requesting error details`);
+            const filesForContext = await this.getErrorDetailsForFile(basePath, filename, fileErrors[0], relevantFiles, language);
+            console.log(`Get error details for ${filename} requesting code fixes`);
+            const fix = this.createFixRequestForFile(basePath, filename, fileErrors[0], filesForContext, relevantFiles, language);
+            const codeFixFiles = await this.codeFix(fix);
+            console.log(`Got fixed code for ${filename}`);
+            for (const codeFixFile of codeFixFiles) {
+                const filePath = codeFixFile.filename.indexOf(basePath) > -1 ? codeFixFile.filename : (0, path_2.join)(basePath, codeFixFile.filename);
+                const existing = (0, fs_1.readFileSync)(filePath);
+                if (existing.toString().replace(/(\r\n|\r|\n)+$/, '') == codeFixFile.content.replace(/(\r\n|\r|\n)+$/, '')) {
+                    console.log(`${filename} not changed by gemini`);
+                    continue;
+                }
+                (0, fs_1.writeFileSync)(filePath, codeFixFile.content);
+            }
+            const result = await codeGenerator.validateForTarget(basePath);
+            if (result && result.valid) {
+                return;
+            }
+            const errors = await this.classifyErrors(result.error, basePath);
+            fileErrors = errors.get(filename) ?? [];
+        }
+    }
+    async classifyErrors(errors, basePath) {
+        const errorStream = await stormClient_1.stormClient.createErrorClassification(errors, []);
+        const fixes = new Map();
+        this.out.on('aborted', () => {
+            errorStream.abort();
+        });
+        errorStream.on('data', (evt) => {
+            if (evt.type === 'ERROR_CLASSIFIER') {
+                const eventFileName = this.removePrefix(basePath + '/', evt.payload.filename);
+                const fix = { error: evt.payload.error, lineNumber: evt.payload.lineNumber, column: evt.payload.column };
+                let existingFixes = fixes.get(eventFileName);
+                if (existingFixes) {
+                    existingFixes.push(fix);
+                }
+                else {
+                    fixes.set(eventFileName, [fix]);
+                }
+            }
+        });
+        await errorStream.waitForDone();
+        return fixes;
+    }
+    async getErrorDetailsForFile(basePath, filename, error, allFiles, language) {
+        const filePath = filename.indexOf(basePath) > -1 ? filename : (0, path_2.join)(basePath, filename); // to compensate when compiler returns absolute path
+        return new Promise(async (resolve, reject) => {
+            const request = {
+                "language": language,
+                "sourceFile": {
+                    "filename": filename,
+                    "content": (0, fs_1.readFileSync)(filePath, 'utf8')
+                },
+                "error": error,
+                "projectFiles": allFiles.map(f => f.filename)
+            };
+            const detailsStream = await stormClient_1.stormClient.createErrorDetails(JSON.stringify(request), []);
+            detailsStream.on('data', (evt) => {
+                if (evt.type === 'ERROR_DETAILS') {
+                    resolve(evt.payload.files);
+                }
+            });
+            this.out.on('aborted', () => {
+                detailsStream.abort();
+            });
+            detailsStream.on('error', (err) => {
+                console.log("error", err);
+                reject(err);
+            });
+            await detailsStream.waitForDone();
+        });
+    }
+    createFixRequestForFile(basePath, filename, error, filesForContext, allFiles, language) {
+        const files = new Set(filesForContext);
+        files.add(filename);
+        const requestedFiles = Array.from(files).flatMap(file => {
+            if (fs.existsSync(file)) {
+                return file;
+            }
+            // file does not exist - look for similar
+            const candidateName = file.split('/').pop();
+            return allFiles
+                .filter(file => file.filename.split('/').pop() === candidateName)
+                .map(f => f.filename);
+        });
+        const filePath = filename.indexOf(basePath) > -1 ? filename : (0, path_2.join)(basePath, filename);
+        const content = (0, fs_1.readFileSync)(filePath, 'utf8');
+        const affectedLine = this.getErrorLine(error, content);
+        const fixRequest = {
+            "language": language,
+            "filename": filename,
+            "errors": error.error,
+            "affectedLine": affectedLine,
+            "projectFiles": requestedFiles.map(filename => {
+                const filePath = filename.indexOf(basePath) > -1 ? filename : (0, path_2.join)(basePath, filename);
+                const content = (0, fs_1.readFileSync)(filePath, 'utf8');
+                return { filename: filename, content: content };
+            }),
+            "conversationId": this.conversationId,
+            "sessionId": (0, uuid_1.v4)()
+        };
+        return JSON.stringify(fixRequest);
+    }
+    getErrorLine(errorDetails, sourceCode) {
+        const lines = sourceCode.split('\n');
+        const errorLine = lines[errorDetails.lineNumber - 1];
+        if (!errorLine) {
+            return "Error: Line number out of range.";
+        }
+        return errorLine;
+    }
     removePrefix(prefix, str) {
         if (str.startsWith(prefix)) {
             return str.slice(prefix.length);
@@ -412,18 +510,19 @@ class StormCodegen {
     /**
      * Sends the code to the AI for a fix
      */
-    async codeFix(fix) {
+    async codeFix(fix, history) {
         return new Promise(async (resolve, reject) => {
-            const fixStream = await stormClient_1.stormClient.createCodeFix(fix, []);
+            const fixStream = await stormClient_1.stormClient.createCodeFix(fix, history);
             fixStream.on('data', (evt) => {
                 if (evt.type === 'CODE_FIX') {
-                    resolve(evt.payload.content);
+                    resolve(evt.payload.content.files);
                 }
             });
             this.out.on('aborted', () => {
                 fixStream.abort();
             });
             fixStream.on('error', (err) => {
+                console.log("error", err);
                 reject(err);
             });
             await fixStream.waitForDone();
@@ -455,7 +554,7 @@ class StormCodegen {
                 created: Date.now(),
                 payload: {
                     filename: file.filename,
-                    path: (0, path_1.join)(basePath, file.filename),
+                    path: (0, path_2.join)(basePath, file.filename),
                     content: file.content,
                     blockName: aiName,
                     blockRef: ref,
@@ -475,7 +574,7 @@ class StormCodegen {
             created: Date.now(),
             payload: {
                 filename: filename,
-                path: (0, path_1.join)(basePath, filename),
+                path: (0, path_2.join)(basePath, filename),
                 content: content,
                 blockName,
                 blockRef: ref,

package/dist/cjs/src/storm/events.d.ts

@@ -111,14 +111,28 @@ export interface StormEventCodeFix {
     reason: string;
     created: number;
     payload: {
-        filename: string;
-        content: string;
+        content: {
+            files: StormEventErrorDetailsFile[];
+        };
     };
 }
 export interface StormEventErrorClassifierInfo {
     error: string;
     filename: string;
-    potentialFix: string;
+    lineNumber: number;
+    column: number;
+}
+export interface StormEventErrorDetailsFile {
+    filename: string;
+    content: string;
+}
+export interface StormEventErrorDetails {
+    type: 'ERROR_DETAILS';
+    reason: string;
+    created: number;
+    payload: {
+        files: string[];
+    };
 }
 export interface ScreenTemplate {
     name: string;
@@ -218,4 +232,4 @@ export interface StormEventPhases {
         phaseType: StormEventPhaseType;
     };
 }
-export type StormEvent = StormEventCreateBlock | StormEventCreateConnection | StormEventCreatePlanProperties | StormEventInvalidResponse | StormEventPlanRetry | StormEventCreateDSL | StormEventCreateDSLResource | StormEventError | StormEventScreen | StormEventScreenCandidate | StormEventFileLogical | StormEventFileState | StormEventFileDone | StormEventFileChunk | StormEventDone | StormEventDefinitionChange | StormEventErrorClassifier | StormEventCodeFix | StormEventBlockReady | StormEventPhases;
+export type StormEvent = StormEventCreateBlock | StormEventCreateConnection | StormEventCreatePlanProperties | StormEventInvalidResponse | StormEventPlanRetry | StormEventCreateDSL | StormEventCreateDSLResource | StormEventError | StormEventScreen | StormEventScreenCandidate | StormEventFileLogical | StormEventFileState | StormEventFileDone | StormEventFileChunk | StormEventDone | StormEventDefinitionChange | StormEventErrorClassifier | StormEventCodeFix | StormEventErrorDetails | StormEventBlockReady | StormEventPhases;

package/dist/cjs/src/storm/stormClient.d.ts

@@ -11,6 +11,7 @@ declare class StormClient {
     createServiceImplementation(prompt: StormFileImplementationPrompt, conversationId?: string): Promise<StormStream>;
     createErrorClassification(prompt: string, history?: ConversationItem[], conversationId?: string): Promise<StormStream>;
     createCodeFix(prompt: string, history?: ConversationItem[], conversationId?: string): Promise<StormStream>;
+    createErrorDetails(prompt: string, history?: ConversationItem[], conversationId?: string): Promise<StormStream>;
 }
 export declare const stormClient: StormClient;
 export {};

package/dist/cjs/src/storm/stormClient.js

@@ -100,5 +100,11 @@ class StormClient {
             prompt,
         });
     }
+    createErrorDetails(prompt, history, conversationId) {
+        return this.send('/v2/code/errordetails', {
+            conversationId: conversationId,
+            prompt,
+        });
+    }
 }
 exports.stormClient = new StormClient();

package/dist/esm/src/storm/codegen.d.ts

@@ -26,6 +26,11 @@ export declare class StormCodegen {
      */
     private processBlockCode;
     private verifyAndFixCode;
+    private tryToFixFile;
+    private classifyErrors;
+    private getErrorDetailsForFile;
+    private createFixRequestForFile;
+    private getErrorLine;
     removePrefix(prefix: string, str: string): string;
     /**
      * Sends the code to the AI for a fix

package/dist/esm/src/storm/codegen.js

@@ -38,11 +38,13 @@ const event_parser_1 = require("./event-parser");
 const stream_1 = require("./stream");
 const nodejs_utils_1 = require("@kapeta/nodejs-utils");
 const promises_1 = require("fs/promises");
-const path_1 = __importStar(require("path"));
+const path_1 = __importDefault(require("path"));
+const path_2 = __importStar(require("path"));
 const node_os_1 = __importDefault(require("node:os"));
 const fs_1 = require("fs");
-const path_2 = __importDefault(require("path"));
 const yaml_1 = __importDefault(require("yaml"));
+const fs = __importStar(require("node:fs"));
+const uuid_1 = require("uuid");
 const SIMULATED_DELAY = 1000;
 class SimulatedFileDelay {
     file;
@@ -197,7 +199,7 @@ class StormCodegen {
                     ...data,
                     payload: {
                         ...data.payload,
-                        path: (0, path_1.join)(basePath, data.payload.filename),
+                        path: (0, path_2.join)(basePath, data.payload.filename),
                         blockName,
                         blockRef,
                         instanceId,
@@ -209,7 +211,7 @@ class StormCodegen {
                     ...data,
                     payload: {
                         ...data.payload,
-                        path: (0, path_1.join)(basePath, data.payload.filename),
+                        path: (0, path_2.join)(basePath, data.payload.filename),
                         blockName,
                         blockRef,
                         instanceId,
@@ -221,7 +223,7 @@ class StormCodegen {
                     ...data,
                     payload: {
                         ...data.payload,
-                        path: (0, path_1.join)(basePath, data.payload.filename),
+                        path: (0, path_2.join)(basePath, data.payload.filename),
                         blockName,
                         blockRef,
                         instanceId,
@@ -241,7 +243,7 @@ class StormCodegen {
                     created: Date.now(),
                     payload: {
                         filename: data.payload.filename,
-                        path: (0, path_1.join)(this.getBasePath(blockUri.fullName), data.payload.filename),
+                        path: (0, path_2.join)(this.getBasePath(blockUri.fullName), data.payload.filename),
                         content: data.payload.content,
                         blockRef: ref,
                         instanceId: event_parser_1.StormEventParser.toInstanceIdFromRef(ref),
@@ -297,7 +299,7 @@ class StormCodegen {
         }
         // Gather the context files for implementation. These will be all be passed to the AI
         const contextFiles = relevantFiles.filter((file) => ![codegen_1.AIFileTypes.SERVICE, codegen_1.AIFileTypes.WEB_SCREEN].includes(file.type));
-        // Send the service and UI templates to the AI. These will be send one-by-one in addition to the context files
+        // Send the service and UI templates to the AI. These will be sent one-by-one in addition to the context files
         const serviceFiles = allFiles.filter((file) => file.type === codegen_1.AIFileTypes.SERVICE);
         if (serviceFiles.length > 0) {
             await this.processTemplates((0, nodejs_utils_1.parseKapetaUri)(block.uri), block.aiName, stormClient_1.stormClient.createServiceImplementation.bind(stormClient_1.stormClient), serviceFiles, contextFiles);
@@ -307,17 +309,17 @@ class StormCodegen {
             return;
         }
         for (const serviceFile of serviceFiles) {
-            const filePath = (0, path_1.join)(basePath, serviceFile.filename);
+            const filePath = (0, path_2.join)(basePath, serviceFile.filename);
             await (0, promises_1.writeFile)(filePath, serviceFile.content);
         }
         for (const serviceFile of contextFiles) {
-            const filePath = (0, path_1.join)(basePath, serviceFile.filename);
+            const filePath = (0, path_2.join)(basePath, serviceFile.filename);
             await (0, promises_1.writeFile)(filePath, serviceFile.content);
         }
-        const kapetaYmlPath = (0, path_1.join)(basePath, 'kapeta.yml');
+        const kapetaYmlPath = (0, path_2.join)(basePath, 'kapeta.yml');
         await (0, promises_1.writeFile)(kapetaYmlPath, yaml_1.default.stringify(block.content));
         for (const screenFile of screenFiles) {
-            const filePath = (0, path_1.join)(basePath, screenFile.payload.filename);
+            const filePath = (0, path_2.join)(basePath, screenFile.payload.filename);
             await (0, promises_1.writeFile)(filePath, screenFile.payload.content);
         }
         const screenFilesConverted = screenFiles.map(screenFile => {
@@ -329,8 +331,8 @@ class StormCodegen {
                 type: codegen_1.AIFileTypes.WEB_SCREEN,
             };
         });
-        const filesToBeFixed = serviceFiles.concat(contextFiles).concat(screenFilesConverted);
         allFiles.push(...screenFilesConverted);
+        const filesToBeFixed = serviceFiles.concat(contextFiles).concat(screenFilesConverted);
         const codeGenerator = new codegen_1.BlockCodeGenerator(block.content);
         await this.verifyAndFixCode(codeGenerator, basePath, filesToBeFixed, allFiles);
         const blockRef = block.uri;
@@ -346,7 +348,7 @@ class StormCodegen {
             },
         });
     }
-    async verifyAndFixCode(codeGenerator, basePath, filesToBeFixed, knownFiles) {
+    async verifyAndFixCode(codeGenerator, basePath, filesToBeFixed, allFiles) {
         let attempts = 0;
         let validCode = false;
         for (let i = 0; i <= 3; i++) {
@@ -360,35 +362,10 @@ class StormCodegen {
                 }
                 if (result && !result.valid) {
                     console.debug('Validation error:', result);
-                    const errorStream = await stormClient_1.stormClient.createErrorClassification(result.error, []);
-                    const fixes = new Map();
-                    this.out.on('aborted', () => {
-                        errorStream.abort();
-                    });
-                    errorStream.on('data', (evt) => {
-                        if (evt.type === 'ERROR_CLASSIFIER') {
-                            // find the file that caused the error
-                            // strip base path from event file name, if it exists sometimes the AI sends the full path
-                            const eventFileName = this.removePrefix(basePath + '/', evt.payload.filename);
-                            const file = filesToBeFixed.find((f) => f.filename === eventFileName);
-                            if (!file) {
-                                console.log(`Could not find the file ${eventFileName} in the list of files to be fixed, Henrik might wanna create a new file for this fix`);
-                            }
-                            // read the content of the file
-                            const content = (0, fs_1.readFileSync)((0, path_1.join)(basePath, eventFileName), 'utf8');
-                            const fix = `${evt.payload.potentialFix}\n---\n${knownFiles
-                                .map((e) => e.filename)
-                                .join('\n')}\n---\n${content}`;
-                            //console.log(`trying to fix the code in ${eventFileName}`);
-                            //console.debug(`with the fix:\n${fix}`);
-                            const code = this.codeFix(fix);
-                            fixes.set((0, path_1.join)(basePath, eventFileName), code);
-                        }
-                    });
-                    await errorStream.waitForDone();
-                    for (const [filename, codePromise] of fixes) {
-                        const code = await codePromise;
-                        (0, fs_1.writeFileSync)(filename, code);
+                    const errors = await this.classifyErrors(result.error, basePath);
+                    for (const [filename, fileErrors] of errors.entries()) {
+                        // todo: only try to fix file if it is part of filesToBeFixed
+                        await this.tryToFixFile(basePath, filename, fileErrors, allFiles, codeGenerator);
                     }
                 }
             }
@@ -403,6 +380,127 @@ class StormCodegen {
             console.error(`Validation failed for ${basePath} after ${attempts} attempts`);
         }
     }
+    async tryToFixFile(basePath, filename, fileErrors, allFiles, codeGenerator) {
+        console.log(`Processing ${filename}`);
+        const language = await codeGenerator.language();
+        const relevantFiles = allFiles.filter(file => file.type != codegen_1.AIFileTypes.IGNORE);
+        for (let attempts = 1; attempts <= 5; attempts++) {
+            if (fileErrors.length == 0) {
+                console.log(`No more errors for ${filename}`);
+                return;
+            }
+            console.log(`Errors in ${filename} - requesting error details`);
+            const filesForContext = await this.getErrorDetailsForFile(basePath, filename, fileErrors[0], relevantFiles, language);
+            console.log(`Get error details for ${filename} requesting code fixes`);
+            const fix = this.createFixRequestForFile(basePath, filename, fileErrors[0], filesForContext, relevantFiles, language);
+            const codeFixFiles = await this.codeFix(fix);
+            console.log(`Got fixed code for ${filename}`);
+            for (const codeFixFile of codeFixFiles) {
+                const filePath = codeFixFile.filename.indexOf(basePath) > -1 ? codeFixFile.filename : (0, path_2.join)(basePath, codeFixFile.filename);
+                const existing = (0, fs_1.readFileSync)(filePath);
+                if (existing.toString().replace(/(\r\n|\r|\n)+$/, '') == codeFixFile.content.replace(/(\r\n|\r|\n)+$/, '')) {
+                    console.log(`${filename} not changed by gemini`);
+                    continue;
+                }
+                (0, fs_1.writeFileSync)(filePath, codeFixFile.content);
+            }
+            const result = await codeGenerator.validateForTarget(basePath);
+            if (result && result.valid) {
+                return;
+            }
+            const errors = await this.classifyErrors(result.error, basePath);
+            fileErrors = errors.get(filename) ?? [];
+        }
+    }
+    async classifyErrors(errors, basePath) {
+        const errorStream = await stormClient_1.stormClient.createErrorClassification(errors, []);
+        const fixes = new Map();
+        this.out.on('aborted', () => {
+            errorStream.abort();
+        });
+        errorStream.on('data', (evt) => {
+            if (evt.type === 'ERROR_CLASSIFIER') {
+                const eventFileName = this.removePrefix(basePath + '/', evt.payload.filename);
+                const fix = { error: evt.payload.error, lineNumber: evt.payload.lineNumber, column: evt.payload.column };
+                let existingFixes = fixes.get(eventFileName);
+                if (existingFixes) {
+                    existingFixes.push(fix);
+                }
+                else {
+                    fixes.set(eventFileName, [fix]);
+                }
+            }
+        });
+        await errorStream.waitForDone();
+        return fixes;
+    }
+    async getErrorDetailsForFile(basePath, filename, error, allFiles, language) {
+        const filePath = filename.indexOf(basePath) > -1 ? filename : (0, path_2.join)(basePath, filename); // to compensate when compiler returns absolute path
+        return new Promise(async (resolve, reject) => {
+            const request = {
+                "language": language,
+                "sourceFile": {
+                    "filename": filename,
+                    "content": (0, fs_1.readFileSync)(filePath, 'utf8')
+                },
+                "error": error,
+                "projectFiles": allFiles.map(f => f.filename)
+            };
+            const detailsStream = await stormClient_1.stormClient.createErrorDetails(JSON.stringify(request), []);
+            detailsStream.on('data', (evt) => {
+                if (evt.type === 'ERROR_DETAILS') {
+                    resolve(evt.payload.files);
+                }
+            });
+            this.out.on('aborted', () => {
+                detailsStream.abort();
+            });
+            detailsStream.on('error', (err) => {
+                console.log("error", err);
+                reject(err);
+            });
+            await detailsStream.waitForDone();
+        });
+    }
+    createFixRequestForFile(basePath, filename, error, filesForContext, allFiles, language) {
+        const files = new Set(filesForContext);
+        files.add(filename);
+        const requestedFiles = Array.from(files).flatMap(file => {
+            if (fs.existsSync(file)) {
+                return file;
+            }
+            // file does not exist - look for similar
+            const candidateName = file.split('/').pop();
+            return allFiles
+                .filter(file => file.filename.split('/').pop() === candidateName)
+                .map(f => f.filename);
+        });
+        const filePath = filename.indexOf(basePath) > -1 ? filename : (0, path_2.join)(basePath, filename);
+        const content = (0, fs_1.readFileSync)(filePath, 'utf8');
+        const affectedLine = this.getErrorLine(error, content);
+        const fixRequest = {
+            "language": language,
+            "filename": filename,
+            "errors": error.error,
+            "affectedLine": affectedLine,
+            "projectFiles": requestedFiles.map(filename => {
+                const filePath = filename.indexOf(basePath) > -1 ? filename : (0, path_2.join)(basePath, filename);
+                const content = (0, fs_1.readFileSync)(filePath, 'utf8');
+                return { filename: filename, content: content };
+            }),
+            "conversationId": this.conversationId,
+            "sessionId": (0, uuid_1.v4)()
+        };
+        return JSON.stringify(fixRequest);
+    }
+    getErrorLine(errorDetails, sourceCode) {
+        const lines = sourceCode.split('\n');
+        const errorLine = lines[errorDetails.lineNumber - 1];
+        if (!errorLine) {
+            return "Error: Line number out of range.";
+        }
+        return errorLine;
+    }
     removePrefix(prefix, str) {
         if (str.startsWith(prefix)) {
             return str.slice(prefix.length);
@@ -412,18 +510,19 @@ class StormCodegen {
     /**
      * Sends the code to the AI for a fix
      */
-    async codeFix(fix) {
+    async codeFix(fix, history) {
         return new Promise(async (resolve, reject) => {
-            const fixStream = await stormClient_1.stormClient.createCodeFix(fix, []);
+            const fixStream = await stormClient_1.stormClient.createCodeFix(fix, history);
             fixStream.on('data', (evt) => {
                 if (evt.type === 'CODE_FIX') {
-                    resolve(evt.payload.content);
+                    resolve(evt.payload.content.files);
                 }
             });
             this.out.on('aborted', () => {
                 fixStream.abort();
             });
             fixStream.on('error', (err) => {
+                console.log("error", err);
                 reject(err);
             });
             await fixStream.waitForDone();
@@ -455,7 +554,7 @@ class StormCodegen {
                 created: Date.now(),
                 payload: {
                     filename: file.filename,
-                    path: (0, path_1.join)(basePath, file.filename),
+                    path: (0, path_2.join)(basePath, file.filename),
                     content: file.content,
                     blockName: aiName,
                     blockRef: ref,
@@ -475,7 +574,7 @@ class StormCodegen {
             created: Date.now(),
             payload: {
                 filename: filename,
-                path: (0, path_1.join)(basePath, filename),
+                path: (0, path_2.join)(basePath, filename),
                 content: content,
                 blockName,
                 blockRef: ref,

package/dist/esm/src/storm/events.d.ts

@@ -111,14 +111,28 @@ export interface StormEventCodeFix {
     reason: string;
     created: number;
     payload: {
-        filename: string;
-        content: string;
+        content: {
+            files: StormEventErrorDetailsFile[];
+        };
     };
 }
 export interface StormEventErrorClassifierInfo {
     error: string;
     filename: string;
-    potentialFix: string;
+    lineNumber: number;
+    column: number;
+}
+export interface StormEventErrorDetailsFile {
+    filename: string;
+    content: string;
+}
+export interface StormEventErrorDetails {
+    type: 'ERROR_DETAILS';
+    reason: string;
+    created: number;
+    payload: {
+        files: string[];
+    };
 }
 export interface ScreenTemplate {
     name: string;
@@ -218,4 +232,4 @@ export interface StormEventPhases {
         phaseType: StormEventPhaseType;
     };
 }
-export type StormEvent = StormEventCreateBlock | StormEventCreateConnection | StormEventCreatePlanProperties | StormEventInvalidResponse | StormEventPlanRetry | StormEventCreateDSL | StormEventCreateDSLResource | StormEventError | StormEventScreen | StormEventScreenCandidate | StormEventFileLogical | StormEventFileState | StormEventFileDone | StormEventFileChunk | StormEventDone | StormEventDefinitionChange | StormEventErrorClassifier | StormEventCodeFix | StormEventBlockReady | StormEventPhases;
+export type StormEvent = StormEventCreateBlock | StormEventCreateConnection | StormEventCreatePlanProperties | StormEventInvalidResponse | StormEventPlanRetry | StormEventCreateDSL | StormEventCreateDSLResource | StormEventError | StormEventScreen | StormEventScreenCandidate | StormEventFileLogical | StormEventFileState | StormEventFileDone | StormEventFileChunk | StormEventDone | StormEventDefinitionChange | StormEventErrorClassifier | StormEventCodeFix | StormEventErrorDetails | StormEventBlockReady | StormEventPhases;

package/dist/esm/src/storm/stormClient.d.ts

@@ -11,6 +11,7 @@ declare class StormClient {
     createServiceImplementation(prompt: StormFileImplementationPrompt, conversationId?: string): Promise<StormStream>;
     createErrorClassification(prompt: string, history?: ConversationItem[], conversationId?: string): Promise<StormStream>;
     createCodeFix(prompt: string, history?: ConversationItem[], conversationId?: string): Promise<StormStream>;
+    createErrorDetails(prompt: string, history?: ConversationItem[], conversationId?: string): Promise<StormStream>;
 }
 export declare const stormClient: StormClient;
 export {};

package/dist/esm/src/storm/stormClient.js

@@ -100,5 +100,11 @@ class StormClient {
             prompt,
         });
     }
+    createErrorDetails(prompt, history, conversationId) {
+        return this.send('/v2/code/errordetails', {
+            conversationId: conversationId,
+            prompt,
+        });
+    }
 }
 exports.stormClient = new StormClient();

package/package.json

@@ -1,6 +1,6 @@
 {
     "name": "@kapeta/local-cluster-service",
-    "version": "0.51.4",
+    "version": "0.52.1",
     "description": "Manages configuration, ports and service discovery for locally running Kapeta systems",
     "type": "commonjs",
     "exports": {
@@ -50,7 +50,7 @@
     },
     "homepage": "https://github.com/kapetacom/local-cluster-service#readme",
     "dependencies": {
-        "@kapeta/codegen": "^1.4.0",
+        "@kapeta/codegen": "^1.5.0",
         "@kapeta/config-mapper": "^1.2.1",
         "@kapeta/kaplang-core": "^1.17.1",
         "@kapeta/local-cluster-config": "^0.4.2",

package/src/storm/codegen.ts

@@ -3,7 +3,7 @@
  * SPDX-License-Identifier: BUSL-1.1
  */
 
-import { Definition } from '@kapeta/local-cluster-config';
+import {Definition} from '@kapeta/local-cluster-config';
 import {
     AIFileTypes,
     BlockCodeGenerator,
@@ -13,22 +13,36 @@ import {
     GeneratedResult,
     MODE_CREATE_ONLY,
 } from '@kapeta/codegen';
-import { BlockDefinition } from '@kapeta/schemas';
-import { codeGeneratorManager } from '../codeGeneratorManager';
-import { STORM_ID, stormClient } from './stormClient';
-import { StormEvent, StormEventFileChunk, StormEventFileDone, StormEventFileLogical } from './events';
-import { BlockDefinitionInfo, StormEventParser } from './event-parser';
-import { StormFileImplementationPrompt, StormFileInfo, StormStream } from './stream';
-import { KapetaURI, parseKapetaUri } from '@kapeta/nodejs-utils';
-import { writeFile } from 'fs/promises';
-import path, { join } from 'path';
+import {BlockDefinition} from '@kapeta/schemas';
+import {codeGeneratorManager} from '../codeGeneratorManager';
+import {STORM_ID, stormClient} from './stormClient';
+import {
+    StormEvent,
+    StormEventErrorDetailsFile,
+    StormEventFileChunk,
+    StormEventFileDone,
+    StormEventFileLogical
+} from './events';
+import {BlockDefinitionInfo, StormEventParser} from './event-parser';
+import {ConversationItem, StormFileImplementationPrompt, StormFileInfo, StormStream} from './stream';
+import {KapetaURI, parseKapetaUri} from '@kapeta/nodejs-utils';
+import {writeFile} from 'fs/promises';
+import path from 'path';
+import Path, {join} from 'path';
 import os from 'node:os';
-import { readFileSync, writeFileSync } from 'fs';
-import Path from 'path';
+import {readFileSync, writeFileSync} from 'fs';
 import YAML from "yaml";
+import * as fs from "node:fs";
+import {v4 as uuidv4} from 'uuid';
 
 type ImplementationGenerator = (prompt: StormFileImplementationPrompt, conversationId?: string) => Promise<StormStream>;
 
+interface ErrorClassification {
+    error: string;
+    lineNumber: number;
+    column: number;
+}
+
 const SIMULATED_DELAY = 1000;
 
 class SimulatedFileDelay {
@@ -321,7 +335,7 @@ export class StormCodegen {
             (file) => ![AIFileTypes.SERVICE, AIFileTypes.WEB_SCREEN].includes(file.type)
         );
 
-        // Send the service and UI templates to the AI. These will be send one-by-one in addition to the context files
+        // Send the service and UI templates to the AI. These will be sent one-by-one in addition to the context files
         const serviceFiles: StormFileInfo[] = allFiles.filter((file) => file.type === AIFileTypes.SERVICE);
         if (serviceFiles.length > 0) {
             await this.processTemplates(
@@ -366,8 +380,9 @@ export class StormCodegen {
                 type: AIFileTypes.WEB_SCREEN,
             };
         });
-        const filesToBeFixed = serviceFiles.concat(contextFiles).concat(screenFilesConverted);
         allFiles.push(...screenFilesConverted);
+
+        const filesToBeFixed = serviceFiles.concat(contextFiles).concat(screenFilesConverted);
         const codeGenerator = new BlockCodeGenerator(block.content as BlockDefinition);
         await this.verifyAndFixCode(codeGenerator, basePath, filesToBeFixed, allFiles);
 
@@ -389,7 +404,7 @@ export class StormCodegen {
         codeGenerator: CodeGenerator,
         basePath: string,
         filesToBeFixed: StormFileInfo[],
-        knownFiles: StormFileInfo[]
+        allFiles: StormFileInfo[]
     ) {
         let attempts = 0;
         let validCode = false;
@@ -405,46 +420,18 @@ export class StormCodegen {
 
                 if (result && !result.valid) {
                     console.debug('Validation error:', result);
-                    const errorStream = await stormClient.createErrorClassification(result.error, []);
-                    const fixes = new Map<string, Promise<string>>();
-
-                    this.out.on('aborted', () => {
-                        errorStream.abort();
-                    });
-
-                    errorStream.on('data', (evt) => {
-                        if (evt.type === 'ERROR_CLASSIFIER') {
-                            // find the file that caused the error
-                            // strip base path from event file name, if it exists sometimes the AI sends the full path
-                            const eventFileName = this.removePrefix(basePath + '/', evt.payload.filename);
-                            const file = filesToBeFixed.find((f) => f.filename === eventFileName);
-                            if (!file) {
-                                console.log(
-                                    `Could not find the file ${eventFileName} in the list of files to be fixed, Henrik might wanna create a new file for this fix`
-                                );
-                            }
-                            // read the content of the file
-                            const content = readFileSync(join(basePath, eventFileName), 'utf8');
-                            const fix = `${evt.payload.potentialFix}\n---\n${knownFiles
-                                .map((e) => e.filename)
-                                .join('\n')}\n---\n${content}`;
-                            //console.log(`trying to fix the code in ${eventFileName}`);
-                            //console.debug(`with the fix:\n${fix}`);
-                            const code = this.codeFix(fix);
-                            fixes.set(join(basePath, eventFileName), code);
-                        }
-                    });
-
-                    await errorStream.waitForDone();
-                    for (const [filename, codePromise] of fixes) {
-                        const code = await codePromise;
-                        writeFileSync(filename, code);
+
+                    const errors = await this.classifyErrors(result.error, basePath);
+                    for (const [filename, fileErrors] of errors.entries()) {
+                        // todo: only try to fix file if it is part of filesToBeFixed
+                        await this.tryToFixFile(basePath, filename, fileErrors, allFiles, codeGenerator);
                     }
                 }
             } catch (e) {
                 console.error('Error:', e);
             }
         }
+
         if (validCode) {
             console.log(`Validation successful after ${attempts} attempts`);
         } else {
@@ -452,6 +439,152 @@ export class StormCodegen {
         }
     }
 
+    private async tryToFixFile(basePath: string, filename: string, fileErrors: ErrorClassification[], allFiles: StormFileInfo[], codeGenerator: CodeGenerator) {
+        console.log(`Processing ${filename}`);
+        const language = await codeGenerator.language();
+        const relevantFiles = allFiles.filter(file => file.type != AIFileTypes.IGNORE);
+
+        for (let attempts = 1;  attempts <= 5; attempts++) {
+            if (fileErrors.length == 0) {
+                console.log(`No more errors for ${filename}`);
+                return;
+            }
+
+            console.log(`Errors in ${filename} - requesting error details`);
+            const filesForContext = await this.getErrorDetailsForFile(basePath, filename, fileErrors[0], relevantFiles, language);
+            console.log(`Get error details for ${filename} requesting code fixes`);
+
+            const fix = this.createFixRequestForFile(basePath, filename, fileErrors[0], filesForContext, relevantFiles, language);
+            const codeFixFiles = await this.codeFix(fix);
+            console.log(`Got fixed code for ${filename}`);
+            for (const codeFixFile of codeFixFiles) {
+                const filePath = codeFixFile.filename.indexOf(basePath) > -1 ? codeFixFile.filename : join(basePath, codeFixFile.filename);
+                const existing = readFileSync(filePath);
+                if (existing.toString().replace(/(\r\n|\r|\n)+$/, '') == codeFixFile.content.replace(/(\r\n|\r|\n)+$/, '')) {
+                    console.log(`${filename} not changed by gemini`);
+                    continue;
+                }
+                writeFileSync(filePath, codeFixFile.content);
+            }
+
+            const result = await codeGenerator.validateForTarget(basePath);
+            if (result && result.valid) {
+                return;
+            }
+
+            const errors = await this.classifyErrors(result.error, basePath);
+            fileErrors = errors.get(filename) ?? [];
+        }
+    }
+
+    private async classifyErrors(errors: string, basePath: string): Promise<Map<string, ErrorClassification[]>> {
+        const errorStream = await stormClient.createErrorClassification(errors, []);
+        const fixes = new Map<string, ErrorClassification[]>();
+
+        this.out.on('aborted', () => {
+            errorStream.abort();
+        });
+
+        errorStream.on('data', (evt) => {
+            if (evt.type === 'ERROR_CLASSIFIER') {
+                const eventFileName = this.removePrefix(basePath + '/', evt.payload.filename);
+                const fix = { error: evt.payload.error, lineNumber: evt.payload.lineNumber, column: evt.payload.column };
+
+                let existingFixes = fixes.get(eventFileName);
+                if (existingFixes) {
+                    existingFixes.push(fix);
+                } else {
+                    fixes.set(eventFileName, [fix]);
+                }
+            }
+        });
+
+        await errorStream.waitForDone();
+
+        return fixes;
+    }
+
+    private async getErrorDetailsForFile(basePath: string, filename: string, error: ErrorClassification, allFiles: StormFileInfo[], language: string): Promise<string[]> {
+        const filePath = filename.indexOf(basePath) > -1 ? filename : join(basePath, filename); // to compensate when compiler returns absolute path
+        return new Promise<string[]>(async (resolve, reject) => {
+            const request = {
+                "language": language,
+                "sourceFile": {
+                    "filename": filename,
+                    "content": readFileSync(filePath, 'utf8')
+                },
+                "error": error,
+                "projectFiles": allFiles.map(f => f.filename)
+            };
+
+            const detailsStream = await stormClient.createErrorDetails(JSON.stringify(request), []);
+            detailsStream.on('data', (evt) => {
+                if (evt.type === 'ERROR_DETAILS') {
+                    resolve(evt.payload.files);
+                }
+            });
+            this.out.on('aborted', () => {
+                detailsStream.abort();
+            });
+            detailsStream.on('error', (err) => {
+                console.log("error", err);
+                reject(err);
+            });
+            await detailsStream.waitForDone();
+        });
+    }
+
+    private createFixRequestForFile(basePath: string, filename: string, error: ErrorClassification, filesForContext: string[], allFiles: StormFileInfo[], language: string): string {
+        const files = new Set(filesForContext);
+        files.add(filename);
+
+        const requestedFiles = Array.from(files).flatMap(file => {
+            if (fs.existsSync(file)) {
+                return file;
+            }
+
+            // file does not exist - look for similar
+            const candidateName = file.split('/').pop();
+            return allFiles
+                .filter(file => file.filename.split('/').pop() === candidateName)
+                .map(f => f.filename);
+        });
+
+        const filePath = filename.indexOf(basePath) > -1 ? filename : join(basePath, filename);
+        const content = readFileSync(filePath, 'utf8');
+        const affectedLine = this.getErrorLine(error, content);
+
+        const fixRequest = {
+            "language": language,
+            "filename": filename,
+            "errors": error.error,
+            "affectedLine": affectedLine,
+            "projectFiles": requestedFiles.map(filename => {
+                const filePath = filename.indexOf(basePath) > -1 ? filename : join(basePath, filename);
+                const content = readFileSync(filePath, 'utf8');
+                return { filename: filename, content: content };
+            }),
+            "conversationId": this.conversationId,
+            "sessionId": uuidv4()
+        };
+
+        return JSON.stringify(fixRequest);
+    }
+
+    private getErrorLine(
+        errorDetails: ErrorClassification,
+        sourceCode: string
+    ): string {
+        const lines = sourceCode.split('\n');
+        const errorLine = lines[errorDetails.lineNumber - 1];
+
+        if (!errorLine) {
+            return "Error: Line number out of range.";
+        }
+
+        return errorLine;
+    }
+
     removePrefix(prefix: string, str: string): string {
         if (str.startsWith(prefix)) {
             return str.slice(prefix.length);
@@ -462,18 +595,19 @@ export class StormCodegen {
     /**
      * Sends the code to the AI for a fix
      */
-    private async codeFix(fix: string): Promise<string> {
-        return new Promise<string>(async (resolve, reject) => {
-            const fixStream = await stormClient.createCodeFix(fix, []);
+    private async codeFix(fix: string, history?: ConversationItem[]): Promise<StormEventErrorDetailsFile[]> {
+        return new Promise<StormEventErrorDetailsFile[]>(async (resolve, reject) => {
+            const fixStream = await stormClient.createCodeFix(fix, history);
             fixStream.on('data', (evt) => {
                 if (evt.type === 'CODE_FIX') {
-                    resolve(evt.payload.content);
+                    resolve(evt.payload.content.files);
                 }
             });
             this.out.on('aborted', () => {
                 fixStream.abort();
             });
             fixStream.on('error', (err) => {
+                console.log("error", err);
                 reject(err);
             });
             await fixStream.waitForDone();

package/src/storm/events.ts

@@ -139,14 +139,31 @@ export interface StormEventCodeFix {
     reason: string;
     created: number;
     payload: {
-        filename: string;
-        content: string;
+        content: {
+            files: StormEventErrorDetailsFile[]
+        }
     };
 }
+
 export interface StormEventErrorClassifierInfo {
     error: string;
     filename: string;
-    potentialFix: string;
+    lineNumber: number;
+    column: number;
+}
+
+export interface StormEventErrorDetailsFile {
+    filename: string;
+    content: string;
+}
+
+export interface StormEventErrorDetails {
+    type: 'ERROR_DETAILS';
+    reason: string;
+    created: number;
+    payload: {
+        files: string[]
+    };
 }
 
 export interface ScreenTemplate {
@@ -280,5 +297,6 @@ export type StormEvent =
     | StormEventDefinitionChange
     | StormEventErrorClassifier
     | StormEventCodeFix
+    | StormEventErrorDetails
     | StormEventBlockReady
     | StormEventPhases;

package/src/storm/stormClient.ts

@@ -125,12 +125,20 @@ class StormClient {
             prompt,
         });
     }
+
     public createCodeFix(prompt: string, history?: ConversationItem[], conversationId?: string) {
         return this.send('/v2/code/fix', {
             conversationId: conversationId,
             prompt,
         });
     }
+
+    public createErrorDetails(prompt: string, history?: ConversationItem[], conversationId?: string) {
+        return this.send('/v2/code/errordetails', {
+            conversationId: conversationId,
+            prompt,
+        });
+    }
 }
 
 export const stormClient = new StormClient();