@kapeta/local-cluster-service 0.54.3 → 0.54.5

package/CHANGELOG.md

@@ -1,3 +1,20 @@
+## [0.54.5](https://github.com/kapetacom/local-cluster-service/compare/v0.54.4...v0.54.5) (2024-06-21)
+
+
+### Bug Fixes
+
+* enable filesToBeFixed inclusion list in codefix ([4eaf272](https://github.com/kapetacom/local-cluster-service/commit/4eaf2720bc49b5709c170f88a85614b1a1a8bb8b))
+* forward FILE_FAILED events from streams ([fc22b68](https://github.com/kapetacom/local-cluster-service/commit/fc22b68e129546166c117ea5d13efeebf4bd9d72))
+
+## [0.54.4](https://github.com/kapetacom/local-cluster-service/compare/v0.54.3...v0.54.4) (2024-06-20)
+
+
+### Bug Fixes
+
+* merge error ([4c29cab](https://github.com/kapetacom/local-cluster-service/commit/4c29cab33c311a3ad21913b974d92207968a15db))
+* minors ([c36e825](https://github.com/kapetacom/local-cluster-service/commit/c36e8258cf0d7a3fb0915080f64d524ffd925929))
+* processTemplate will possibly return a list of files for each file when enhancing services ([3f186a3](https://github.com/kapetacom/local-cluster-service/commit/3f186a3afbb6b527c933ff051d2659e07ab39bbd))
+
 ## [0.54.3](https://github.com/kapetacom/local-cluster-service/compare/v0.54.2...v0.54.3) (2024-06-19)
 
 

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

@@ -117,16 +117,13 @@ class StormCodegen {
     getStream() {
         return this.out;
     }
-    handleTemplateFileOutput(blockUri, aiName, template, data) {
+    handleTemplateFileOutput(blockUri, aiName, data) {
         if (this.handleFileEvents(blockUri, aiName, data)) {
             return;
         }
         switch (data.type) {
             case 'FILE_DONE':
-                template.filename = data.payload.filename;
-                template.content = data.payload.content;
-                this.handleFileDoneOutput(blockUri, aiName, data);
-                break;
+                return this.handleFileDoneOutput(blockUri, aiName, data);
         }
     }
     handleUiOutput(blockUri, blockName, data) {
@@ -198,6 +195,17 @@ class StormCodegen {
                     },
                 });
                 return true;
+            case 'FILE_FAILED':
+                this.out.emit('data', {
+                    ...data,
+                    payload: {
+                        ...data.payload,
+                        path: (0, path_2.join)(basePath, data.payload.filename),
+                        blockName,
+                        blockRef,
+                        instanceId,
+                    },
+                });
         }
         return false;
     }
@@ -229,8 +237,10 @@ class StormCodegen {
         if (this.isAborted()) {
             return;
         }
+        const kapetaYaml = yaml_1.default.stringify(block.content);
+        const blockDefinition = block.content;
         // Generate the code for the block using the standard codegen templates
-        const generatedResult = await this.generateBlock(block.content);
+        const generatedResult = await this.generateBlock(blockDefinition);
         if (!generatedResult) {
             return;
         }
@@ -321,7 +331,7 @@ class StormCodegen {
         if (this.isAborted()) {
             return;
         }
-        const basePath = this.getBasePath(block.content.metadata.name);
+        const basePath = this.getBasePath(blockDefinition.metadata.name);
         const screenFilesConverted = screenFiles.map((screenFile) => {
             return {
                 filename: screenFile.payload.filename,
@@ -342,7 +352,7 @@ class StormCodegen {
             };
             const stream = await stormClient_1.stormClient.generateCode(payload);
             stream.on('data', (evt) => {
-                this.handleTemplateFileOutput(blockUri, block.aiName, webRouter, evt);
+                this.handleTemplateFileOutput(blockUri, block.aiName, evt);
             });
             this.out.on('aborted', () => {
                 stream.abort();
@@ -352,36 +362,37 @@ 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, codegen_1.AIFileTypes.WEB_ROUTER].includes(file.type));
         // 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);
+        let serviceFiles = allFiles.filter((file) => file.type === codegen_1.AIFileTypes.SERVICE);
         if (serviceFiles.length > 0) {
-            await this.processTemplates(blockUri, block.aiName, stormClient_1.stormClient.createServiceImplementation.bind(stormClient_1.stormClient), serviceFiles, contextFiles);
+            serviceFiles = await this.processTemplates(blockUri, block.aiName, stormClient_1.stormClient.createServiceImplementation.bind(stormClient_1.stormClient), serviceFiles, contextFiles);
         }
         if (this.isAborted()) {
             return;
         }
+        for (const contextFile of contextFiles) {
+            const filePath = (0, path_2.join)(basePath, contextFile.filename);
+            await (0, promises_1.writeFile)(filePath, contextFile.content);
+        }
         for (const screenFile of screenFilesConverted) {
             const filePath = (0, path_2.join)(basePath, screenFile.filename);
             await (0, promises_1.writeFile)(filePath, screenFile.content);
         }
+        // this might decide to overwrite a file that is also a context file
         for (const serviceFile of serviceFiles) {
             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_2.join)(basePath, serviceFile.filename);
-            await (0, promises_1.writeFile)(filePath, serviceFile.content);
-        }
         // Write again after modifications
         for (const webRouterFile of webRouters) {
             const filePath = (0, path_2.join)(basePath, webRouterFile.filename);
             await (0, promises_1.writeFile)(filePath, webRouterFile.content);
         }
         const kapetaYmlPath = (0, path_2.join)(basePath, 'kapeta.yml');
-        await (0, promises_1.writeFile)(kapetaYmlPath, yaml_1.default.stringify(block.content));
+        await (0, promises_1.writeFile)(kapetaYmlPath, kapetaYaml);
         const blockRef = block.uri;
         this.emitBlockStatus(blockUri, block.aiName, events_1.StormEventBlockStatusType.QA);
         const filesToBeFixed = serviceFiles.concat(contextFiles).concat(screenFilesConverted);
-        const codeGenerator = new codegen_1.BlockCodeGenerator(block.content);
+        const codeGenerator = new codegen_1.BlockCodeGenerator(blockDefinition);
         this.emitBlockStatus(blockUri, block.aiName, events_1.StormEventBlockStatusType.BUILDING);
         await this.verifyAndFixCode(blockUri, block.aiName, codeGenerator, basePath, filesToBeFixed, allFiles);
         this.out.emit('data', {
@@ -428,8 +439,9 @@ class StormCodegen {
                     if (errors.size > 0) {
                         this.emitBlockStatus(blockUri, blockName, events_1.StormEventBlockStatusType.FIXING);
                         const promises = Array.from(errors.entries()).map(([filename, fileErrors]) => {
-                            // todo: only try to fix file if it is part of filesToBeFixed
-                            return this.tryToFixFile(blockUri, blockName, basePath, filename, fileErrors, allFiles, codeGenerator);
+                            if (filesToBeFixed.some((file) => file.filename === filename)) {
+                                return this.tryToFixFile(blockUri, blockName, basePath, filename, fileErrors, allFiles, codeGenerator);
+                            }
                         });
                         await Promise.all(promises);
                     }
@@ -676,6 +688,7 @@ class StormCodegen {
      */
     async processTemplates(blockUri, aiName, generator, templates, contextFiles) {
         const promises = templates.map(async (templateFile) => {
+            let changedFiles = [];
             const stream = await generator({
                 context: contextFiles,
                 template: templateFile,
@@ -685,11 +698,28 @@ class StormCodegen {
                 stream.abort();
             });
             stream.on('data', (evt) => {
-                this.handleTemplateFileOutput(blockUri, aiName, templateFile, evt);
+                let changedFile = this.handleTemplateFileOutput(blockUri, aiName, evt);
+                if (changedFile) {
+                    changedFiles.push(...changedFiles, changedFile);
+                }
             });
             await stream.waitForDone();
+            return changedFiles;
         });
-        await Promise.all(promises);
+        const changedFiles = await Promise.all(promises);
+        let allFiles = templates.concat(contextFiles);
+        let result = [];
+        for (const changedFile of changedFiles.flat()) {
+            const find = allFiles.find((file) => file.filename === changedFile.payload.filename);
+            if (find) {
+                result.push({ type: find.type, filename: find.filename, mode: find.mode, permissions: find.permissions,
+                    content: changedFile.payload.content });
+            }
+            else {
+                console.warn("processTemplates: AI changed a file that wasn't in the input [" + changedFile.payload.filename + "]");
+            }
+        }
+        return result;
     }
     /**
      * Converts the generated files to a format that can be sent to the AI

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

@@ -200,6 +200,12 @@ export interface StormEventFileDone extends StormEventFileBase {
         content: string;
     };
 }
+export interface StormEventFileFailed extends StormEventFileBase {
+    type: 'FILE_FAILED';
+    payload: StormEventFileBasePayload & {
+        error: string;
+    };
+}
 export interface StormEventFileChunk extends StormEventFileBase {
     type: 'FILE_CHUNK';
     payload: StormEventFileBasePayload & {
@@ -259,4 +265,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 | StormEventErrorDetails | StormEventBlockReady | StormEventPhases | StormEventBlockStatus | StormEventCreateDSLRetry;
+export type StormEvent = StormEventCreateBlock | StormEventCreateConnection | StormEventCreatePlanProperties | StormEventInvalidResponse | StormEventPlanRetry | StormEventCreateDSL | StormEventCreateDSLResource | StormEventError | StormEventScreen | StormEventScreenCandidate | StormEventFileLogical | StormEventFileState | StormEventFileDone | StormEventFileFailed | StormEventFileChunk | StormEventDone | StormEventDefinitionChange | StormEventErrorClassifier | StormEventCodeFix | StormEventErrorDetails | StormEventBlockReady | StormEventPhases | StormEventBlockStatus | StormEventCreateDSLRetry;

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

@@ -117,16 +117,13 @@ class StormCodegen {
     getStream() {
         return this.out;
     }
-    handleTemplateFileOutput(blockUri, aiName, template, data) {
+    handleTemplateFileOutput(blockUri, aiName, data) {
         if (this.handleFileEvents(blockUri, aiName, data)) {
             return;
         }
         switch (data.type) {
             case 'FILE_DONE':
-                template.filename = data.payload.filename;
-                template.content = data.payload.content;
-                this.handleFileDoneOutput(blockUri, aiName, data);
-                break;
+                return this.handleFileDoneOutput(blockUri, aiName, data);
         }
     }
     handleUiOutput(blockUri, blockName, data) {
@@ -198,6 +195,17 @@ class StormCodegen {
                     },
                 });
                 return true;
+            case 'FILE_FAILED':
+                this.out.emit('data', {
+                    ...data,
+                    payload: {
+                        ...data.payload,
+                        path: (0, path_2.join)(basePath, data.payload.filename),
+                        blockName,
+                        blockRef,
+                        instanceId,
+                    },
+                });
         }
         return false;
     }
@@ -229,8 +237,10 @@ class StormCodegen {
         if (this.isAborted()) {
             return;
         }
+        const kapetaYaml = yaml_1.default.stringify(block.content);
+        const blockDefinition = block.content;
         // Generate the code for the block using the standard codegen templates
-        const generatedResult = await this.generateBlock(block.content);
+        const generatedResult = await this.generateBlock(blockDefinition);
         if (!generatedResult) {
             return;
         }
@@ -321,7 +331,7 @@ class StormCodegen {
         if (this.isAborted()) {
             return;
         }
-        const basePath = this.getBasePath(block.content.metadata.name);
+        const basePath = this.getBasePath(blockDefinition.metadata.name);
         const screenFilesConverted = screenFiles.map((screenFile) => {
             return {
                 filename: screenFile.payload.filename,
@@ -342,7 +352,7 @@ class StormCodegen {
             };
             const stream = await stormClient_1.stormClient.generateCode(payload);
             stream.on('data', (evt) => {
-                this.handleTemplateFileOutput(blockUri, block.aiName, webRouter, evt);
+                this.handleTemplateFileOutput(blockUri, block.aiName, evt);
             });
             this.out.on('aborted', () => {
                 stream.abort();
@@ -352,36 +362,37 @@ 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, codegen_1.AIFileTypes.WEB_ROUTER].includes(file.type));
         // 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);
+        let serviceFiles = allFiles.filter((file) => file.type === codegen_1.AIFileTypes.SERVICE);
         if (serviceFiles.length > 0) {
-            await this.processTemplates(blockUri, block.aiName, stormClient_1.stormClient.createServiceImplementation.bind(stormClient_1.stormClient), serviceFiles, contextFiles);
+            serviceFiles = await this.processTemplates(blockUri, block.aiName, stormClient_1.stormClient.createServiceImplementation.bind(stormClient_1.stormClient), serviceFiles, contextFiles);
         }
         if (this.isAborted()) {
             return;
         }
+        for (const contextFile of contextFiles) {
+            const filePath = (0, path_2.join)(basePath, contextFile.filename);
+            await (0, promises_1.writeFile)(filePath, contextFile.content);
+        }
         for (const screenFile of screenFilesConverted) {
             const filePath = (0, path_2.join)(basePath, screenFile.filename);
             await (0, promises_1.writeFile)(filePath, screenFile.content);
         }
+        // this might decide to overwrite a file that is also a context file
         for (const serviceFile of serviceFiles) {
             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_2.join)(basePath, serviceFile.filename);
-            await (0, promises_1.writeFile)(filePath, serviceFile.content);
-        }
         // Write again after modifications
         for (const webRouterFile of webRouters) {
             const filePath = (0, path_2.join)(basePath, webRouterFile.filename);
             await (0, promises_1.writeFile)(filePath, webRouterFile.content);
         }
         const kapetaYmlPath = (0, path_2.join)(basePath, 'kapeta.yml');
-        await (0, promises_1.writeFile)(kapetaYmlPath, yaml_1.default.stringify(block.content));
+        await (0, promises_1.writeFile)(kapetaYmlPath, kapetaYaml);
         const blockRef = block.uri;
         this.emitBlockStatus(blockUri, block.aiName, events_1.StormEventBlockStatusType.QA);
         const filesToBeFixed = serviceFiles.concat(contextFiles).concat(screenFilesConverted);
-        const codeGenerator = new codegen_1.BlockCodeGenerator(block.content);
+        const codeGenerator = new codegen_1.BlockCodeGenerator(blockDefinition);
         this.emitBlockStatus(blockUri, block.aiName, events_1.StormEventBlockStatusType.BUILDING);
         await this.verifyAndFixCode(blockUri, block.aiName, codeGenerator, basePath, filesToBeFixed, allFiles);
         this.out.emit('data', {
@@ -428,8 +439,9 @@ class StormCodegen {
                     if (errors.size > 0) {
                         this.emitBlockStatus(blockUri, blockName, events_1.StormEventBlockStatusType.FIXING);
                         const promises = Array.from(errors.entries()).map(([filename, fileErrors]) => {
-                            // todo: only try to fix file if it is part of filesToBeFixed
-                            return this.tryToFixFile(blockUri, blockName, basePath, filename, fileErrors, allFiles, codeGenerator);
+                            if (filesToBeFixed.some((file) => file.filename === filename)) {
+                                return this.tryToFixFile(blockUri, blockName, basePath, filename, fileErrors, allFiles, codeGenerator);
+                            }
                         });
                         await Promise.all(promises);
                     }
@@ -676,6 +688,7 @@ class StormCodegen {
      */
     async processTemplates(blockUri, aiName, generator, templates, contextFiles) {
         const promises = templates.map(async (templateFile) => {
+            let changedFiles = [];
             const stream = await generator({
                 context: contextFiles,
                 template: templateFile,
@@ -685,11 +698,28 @@ class StormCodegen {
                 stream.abort();
             });
             stream.on('data', (evt) => {
-                this.handleTemplateFileOutput(blockUri, aiName, templateFile, evt);
+                let changedFile = this.handleTemplateFileOutput(blockUri, aiName, evt);
+                if (changedFile) {
+                    changedFiles.push(...changedFiles, changedFile);
+                }
             });
             await stream.waitForDone();
+            return changedFiles;
         });
-        await Promise.all(promises);
+        const changedFiles = await Promise.all(promises);
+        let allFiles = templates.concat(contextFiles);
+        let result = [];
+        for (const changedFile of changedFiles.flat()) {
+            const find = allFiles.find((file) => file.filename === changedFile.payload.filename);
+            if (find) {
+                result.push({ type: find.type, filename: find.filename, mode: find.mode, permissions: find.permissions,
+                    content: changedFile.payload.content });
+            }
+            else {
+                console.warn("processTemplates: AI changed a file that wasn't in the input [" + changedFile.payload.filename + "]");
+            }
+        }
+        return result;
     }
     /**
      * Converts the generated files to a format that can be sent to the AI

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

@@ -200,6 +200,12 @@ export interface StormEventFileDone extends StormEventFileBase {
         content: string;
     };
 }
+export interface StormEventFileFailed extends StormEventFileBase {
+    type: 'FILE_FAILED';
+    payload: StormEventFileBasePayload & {
+        error: string;
+    };
+}
 export interface StormEventFileChunk extends StormEventFileBase {
     type: 'FILE_CHUNK';
     payload: StormEventFileBasePayload & {
@@ -259,4 +265,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 | StormEventErrorDetails | StormEventBlockReady | StormEventPhases | StormEventBlockStatus | StormEventCreateDSLRetry;
+export type StormEvent = StormEventCreateBlock | StormEventCreateConnection | StormEventCreatePlanProperties | StormEventInvalidResponse | StormEventPlanRetry | StormEventCreateDSL | StormEventCreateDSLResource | StormEventError | StormEventScreen | StormEventScreenCandidate | StormEventFileLogical | StormEventFileState | StormEventFileDone | StormEventFileFailed | StormEventFileChunk | StormEventDone | StormEventDefinitionChange | StormEventErrorClassifier | StormEventCodeFix | StormEventErrorDetails | StormEventBlockReady | StormEventPhases | StormEventBlockStatus | StormEventCreateDSLRetry;

package/package.json

@@ -1,6 +1,6 @@
 {
     "name": "@kapeta/local-cluster-service",
-    "version": "0.54.3",
+    "version": "0.54.5",
     "description": "Manages configuration, ports and service discovery for locally running Kapeta systems",
     "type": "commonjs",
     "exports": {

package/src/storm/codegen.ts

@@ -36,7 +36,6 @@ import Path, { join } from 'path';
 import os from 'node:os';
 import { readFileSync, writeFileSync, existsSync } from 'fs';
 import YAML from 'yaml';
-import assert from 'assert';
 
 type ImplementationGenerator<T = StormFileImplementationPrompt> = (
     prompt: T,
@@ -134,19 +133,15 @@ export class StormCodegen {
     private handleTemplateFileOutput(
         blockUri: KapetaURI,
         aiName: string,
-        template: StormFileInfo,
         data: StormEvent
-    ): void {
+    ): StormEventFileDone | undefined {
         if (this.handleFileEvents(blockUri, aiName, data)) {
             return;
         }
 
         switch (data.type) {
             case 'FILE_DONE':
-                template.filename = data.payload.filename;
-                template.content = data.payload.content;
-                this.handleFileDoneOutput(blockUri, aiName, data);
-                break;
+                return this.handleFileDoneOutput(blockUri, aiName, data);
         }
     }
 
@@ -222,12 +217,23 @@ export class StormCodegen {
                     },
                 });
                 return true;
+            case 'FILE_FAILED':
+                this.out.emit('data', {
+                    ...data,
+                    payload: {
+                        ...data.payload,
+                        path: join(basePath, data.payload.filename),
+                        blockName,
+                        blockRef,
+                        instanceId,
+                    },
+                });
         }
 
         return false;
     }
 
-    private handleFileDoneOutput(blockUri: KapetaURI, aiName: string, data: StormEvent) {
+    private handleFileDoneOutput(blockUri: KapetaURI, aiName: string, data: StormEvent): StormEventFileDone | undefined {
         switch (data.type) {
             case 'FILE_DONE':
                 const ref = blockUri.toNormalizedString();
@@ -257,8 +263,12 @@ export class StormCodegen {
         if (this.isAborted()) {
             return;
         }
+
+        const kapetaYaml = YAML.stringify(block.content);
+        const blockDefinition = block.content;
+
         // Generate the code for the block using the standard codegen templates
-        const generatedResult = await this.generateBlock(block.content);
+        const generatedResult = await this.generateBlock(blockDefinition);
         if (!generatedResult) {
             return;
         }
@@ -374,7 +384,7 @@ export class StormCodegen {
         if (this.isAborted()) {
             return;
         }
-        const basePath = this.getBasePath(block.content.metadata.name);
+        const basePath = this.getBasePath(blockDefinition.metadata.name);
 
         const screenFilesConverted = screenFiles.map((screenFile) => {
             return {
@@ -400,7 +410,7 @@ export class StormCodegen {
                 const stream = await stormClient.generateCode(payload);
 
                 stream.on('data', (evt) => {
-                    this.handleTemplateFileOutput(blockUri, block.aiName, webRouter, evt);
+                    this.handleTemplateFileOutput(blockUri, block.aiName, evt);
                 });
 
                 this.out.on('aborted', () => {
@@ -417,9 +427,9 @@ export class StormCodegen {
         );
 
         // 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);
+        let serviceFiles: StormFileInfo[] = allFiles.filter((file) => file.type === AIFileTypes.SERVICE);
         if (serviceFiles.length > 0) {
-            await this.processTemplates(
+            serviceFiles = await this.processTemplates(
                 blockUri,
                 block.aiName,
                 stormClient.createServiceImplementation.bind(stormClient),
@@ -432,21 +442,22 @@ export class StormCodegen {
             return;
         }
 
+        for (const contextFile of contextFiles) {
+            const filePath = join(basePath, contextFile.filename);
+            await writeFile(filePath, contextFile.content);
+        }
+
         for (const screenFile of screenFilesConverted) {
             const filePath = join(basePath, screenFile.filename);
             await writeFile(filePath, screenFile.content);
         }
 
+        // this might decide to overwrite a file that is also a context file
         for (const serviceFile of serviceFiles) {
             const filePath = join(basePath, serviceFile.filename);
             await writeFile(filePath, serviceFile.content);
         }
 
-        for (const serviceFile of contextFiles) {
-            const filePath = join(basePath, serviceFile.filename);
-            await writeFile(filePath, serviceFile.content);
-        }
-
         // Write again after modifications
         for (const webRouterFile of webRouters) {
             const filePath = join(basePath, webRouterFile.filename);
@@ -454,14 +465,14 @@ export class StormCodegen {
         }
 
         const kapetaYmlPath = join(basePath, 'kapeta.yml');
-        await writeFile(kapetaYmlPath, YAML.stringify(block.content));
+        await writeFile(kapetaYmlPath, kapetaYaml);
 
         const blockRef = block.uri;
 
         this.emitBlockStatus(blockUri, block.aiName, StormEventBlockStatusType.QA);
 
         const filesToBeFixed = serviceFiles.concat(contextFiles).concat(screenFilesConverted);
-        const codeGenerator = new BlockCodeGenerator(block.content);
+        const codeGenerator = new BlockCodeGenerator(blockDefinition);
 
         this.emitBlockStatus(blockUri, block.aiName, StormEventBlockStatusType.BUILDING);
         await this.verifyAndFixCode(blockUri, block.aiName, codeGenerator, basePath, filesToBeFixed, allFiles);
@@ -524,16 +535,17 @@ export class StormCodegen {
                         this.emitBlockStatus(blockUri, blockName, StormEventBlockStatusType.FIXING);
 
                         const promises = Array.from(errors.entries()).map(([filename, fileErrors]) => {
-                            // todo: only try to fix file if it is part of filesToBeFixed
-                            return this.tryToFixFile(
-                                blockUri,
-                                blockName,
-                                basePath,
-                                filename,
-                                fileErrors,
-                                allFiles,
-                                codeGenerator
-                            );
+                            if (filesToBeFixed.some((file) => file.filename === filename)) {
+                                return this.tryToFixFile(
+                                    blockUri,
+                                    blockName,
+                                    basePath,
+                                    filename,
+                                    fileErrors,
+                                    allFiles,
+                                    codeGenerator
+                                );
+                            }
                         });
 
                         await Promise.all(promises);
@@ -864,8 +876,9 @@ export class StormCodegen {
         generator: ImplementationGenerator,
         templates: StormFileInfo[],
         contextFiles: StormFileInfo[]
-    ): Promise<void> {
+    ): Promise<StormFileInfo[]> {
         const promises = templates.map(async (templateFile) => {
+            let changedFiles: StormEventFileDone[] = [];
             const stream = await generator({
                 context: contextFiles,
                 template: templateFile,
@@ -877,13 +890,29 @@ export class StormCodegen {
             });
 
             stream.on('data', (evt) => {
-                this.handleTemplateFileOutput(blockUri, aiName, templateFile, evt);
+                let changedFile = this.handleTemplateFileOutput(blockUri, aiName, evt);
+                if (changedFile) {
+                    changedFiles.push(...changedFiles, changedFile);
+                }
             });
 
             await stream.waitForDone();
+            return changedFiles;
         });
 
-        await Promise.all(promises);
+        const changedFiles: StormEventFileDone[][] = await Promise.all(promises);
+        let allFiles = templates.concat(contextFiles);
+        let result: StormFileInfo[] = [];
+        for (const changedFile of changedFiles.flat()) {
+            const find = allFiles.find((file) => file.filename === changedFile.payload.filename);
+            if (find) {
+                result.push({ type: find.type, filename: find.filename, mode: find.mode, permissions: find.permissions,
+                    content: changedFile.payload.content });
+            } else {
+                console.warn("processTemplates: AI changed a file that wasn't in the input [" + changedFile.payload.filename + "]");
+            }
+        }
+        return result;
     }
 
     /**

package/src/storm/events.ts

@@ -241,6 +241,13 @@ export interface StormEventFileDone extends StormEventFileBase {
     };
 }
 
+export interface StormEventFileFailed extends StormEventFileBase {
+    type: 'FILE_FAILED';
+    payload: StormEventFileBasePayload & {
+        error: string;
+    };
+}
+
 export interface StormEventFileChunk extends StormEventFileBase {
     type: 'FILE_CHUNK';
     payload: StormEventFileBasePayload & {
@@ -322,6 +329,7 @@ export type StormEvent =
     | StormEventFileLogical
     | StormEventFileState
     | StormEventFileDone
+    | StormEventFileFailed
     | StormEventFileChunk
     | StormEventDone
     | StormEventDefinitionChange