@kapeta/local-cluster-service 0.45.0 → 0.46.0

package/CHANGELOG.md

@@ -1,3 +1,10 @@
+# [0.46.0](https://github.com/kapetacom/local-cluster-service/compare/v0.45.0...v0.46.0) (2024-05-28)
+
+
+### Features
+
+* Change how the UI AI work to get templates from codegen ([#148](https://github.com/kapetacom/local-cluster-service/issues/148)) ([f757ec8](https://github.com/kapetacom/local-cluster-service/commit/f757ec8e4fab0dbb58b96a1b9e25f3e734ddb10f))
+
 # [0.45.0](https://github.com/kapetacom/local-cluster-service/compare/v0.44.0...v0.45.0) (2024-05-24)
 
 

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

@@ -2,15 +2,19 @@
  * Copyright 2023 Kapeta Inc.
  * SPDX-License-Identifier: BUSL-1.1
  */
+import { StormEvent } from './events';
 import { BlockDefinitionInfo } from './event-parser';
 import { StormStream } from './stream';
 export declare class StormCodegen {
     private readonly userPrompt;
     private readonly blocks;
     private readonly out;
-    constructor(userPrompt: string, blocks: BlockDefinitionInfo[]);
+    private readonly events;
+    constructor(userPrompt: string, blocks: BlockDefinitionInfo[], events: StormEvent[]);
     process(): Promise<void>;
     getStream(): StormStream;
+    private handleTemplateFileOutput;
+    private handleUiOutput;
     private handleFileOutput;
     /**
      * Generates the code for a block and sends it to the AI

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

@@ -13,12 +13,13 @@ class StormCodegen {
     userPrompt;
     blocks;
     out = new stream_1.StormStream();
-    constructor(userPrompt, blocks) {
+    events;
+    constructor(userPrompt, blocks, events) {
         this.userPrompt = userPrompt;
         this.blocks = blocks;
+        this.events = events;
     }
     async process() {
-        console.log('Processing blocks', this.blocks.length);
         for (const block of this.blocks) {
             await this.processBlockCode(block);
         }
@@ -27,13 +28,42 @@ class StormCodegen {
     getStream() {
         return this.out;
     }
-    handleFileOutput(blockUri, template, data) {
+    handleTemplateFileOutput(blockUri, template, data) {
         switch (data.type) {
             case 'FILE':
                 template.filename = data.payload.filename;
                 template.content = data.payload.content;
+                return this.handleFileOutput(blockUri, data);
+        }
+    }
+    handleUiOutput(blockUri, data) {
+        switch (data.type) {
+            case 'SCREEN':
+                this.out.emit('data', {
+                    type: 'SCREEN',
+                    reason: data.reason,
+                    created: Date.now(),
+                    payload: {
+                        ...data.payload,
+                        blockName: blockUri.toNormalizedString(),
+                    },
+                });
+            case 'FILE':
+                return this.handleFileOutput(blockUri, data);
+        }
+    }
+    handleFileOutput(blockUri, data) {
+        switch (data.type) {
+            case 'FILE':
                 this.emitFile(blockUri, data.payload.filename, data.payload.content, data.reason);
-                break;
+                return {
+                    type: 'FILE',
+                    created: Date.now(),
+                    payload: {
+                        filename: data.payload.filename,
+                        content: data.payload.content,
+                    },
+                };
         }
     }
     /**
@@ -41,22 +71,35 @@ class StormCodegen {
      */
     async processBlockCode(block) {
         // Generate the code for the block using the standard codegen templates
-        const generatedResult = await this.generateBlock(block.content, block.screens);
+        const generatedResult = await this.generateBlock(block.content);
         if (!generatedResult) {
-            console.warn('No generated result for block', block.uri);
             return;
         }
         const allFiles = this.toStormFiles(generatedResult);
         // Send all the non-ai files to the stream
         this.emitFiles(block.uri, allFiles);
-        const implementFiles = [codegen_1.AIFileTypes.SERVICE, codegen_1.AIFileTypes.WEB_SCREEN];
-        // Gather the context files. These will be all be passed to the AI
-        const contextFiles = allFiles.filter((file) => file.type !== codegen_1.AIFileTypes.IGNORE && !implementFiles.includes(file.type));
+        const relevantFiles = allFiles.filter((file) => file.type !== codegen_1.AIFileTypes.IGNORE && file.type !== codegen_1.AIFileTypes.WEB_SCREEN);
+        const uiTemplates = allFiles.filter((file) => file.type === codegen_1.AIFileTypes.WEB_SCREEN);
+        if (uiTemplates.length > 0) {
+            const uiStream = await stormClient_1.stormClient.createUIImplementation({
+                events: this.events,
+                templates: uiTemplates,
+                context: relevantFiles,
+                blockName: block.aiName,
+                prompt: this.userPrompt,
+            });
+            uiStream.on('data', (evt) => {
+                this.handleUiOutput(block.uri, evt);
+            });
+            await uiStream.waitForDone();
+        }
+        // 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
         const serviceFiles = allFiles.filter((file) => file.type === codegen_1.AIFileTypes.SERVICE);
-        await this.processTemplates(block.uri, stormClient_1.stormClient.createServiceImplementation.bind(stormClient_1.stormClient), serviceFiles, contextFiles);
-        //const uiTemplates: StormFileInfo[] = allFiles.filter((file) => file.type === 'ui');
-        //await this.processTemplates(stormClient.createUIImplementation, uiTemplates, contextFiles);
+        if (serviceFiles.length > 0) {
+            await this.processTemplates(block.uri, stormClient_1.stormClient.createServiceImplementation.bind(stormClient_1.stormClient), serviceFiles, contextFiles);
+        }
     }
     /**
      * Emits the text-based files to the stream
@@ -94,17 +137,25 @@ class StormCodegen {
     /**
      * Sends the template to the AI and processes the response
      */
-    processTemplates(blockUri, generator, templates, contextFiles) {
+    async processTemplates(blockUri, generator, templates, contextFiles) {
         const promises = templates.map(async (templateFile) => {
             const stream = await generator({
                 context: contextFiles,
                 template: templateFile,
                 prompt: this.userPrompt,
             });
-            stream.on('data', (evt) => this.handleFileOutput(blockUri, templateFile, evt));
-            return stream.waitForDone();
+            const files = [];
+            stream.on('data', (evt) => {
+                const file = this.handleTemplateFileOutput(blockUri, templateFile, evt);
+                if (file) {
+                    files.push(file);
+                }
+            });
+            await stream.waitForDone();
+            return files;
         });
-        return Promise.all(promises);
+        const fileChunks = await Promise.all(promises);
+        return fileChunks.flat();
     }
     /**
      * Converts the generated files to a format that can be sent to the AI
@@ -143,7 +194,7 @@ class StormCodegen {
     /**
      * Generates the code using codegen for a given block.
      */
-    async generateBlock(yamlContent, screens) {
+    async generateBlock(yamlContent) {
         if (!yamlContent.spec.target?.kind) {
             //Not all block types have targets
             return;
@@ -153,7 +204,6 @@ class StormCodegen {
         }
         const codeGenerator = new codegen_1.BlockCodeGenerator(yamlContent);
         codeGenerator.withOption('AIContext', stormClient_1.STORM_ID);
-        codeGenerator.withOption('AIScreens', screens ?? []);
         return codeGenerator.generate();
     }
 }

package/dist/cjs/src/storm/event-parser.d.ts

@@ -2,13 +2,13 @@
  * Copyright 2023 Kapeta Inc.
  * SPDX-License-Identifier: BUSL-1.1
  */
-import { ScreenTemplate, StormEvent } from './events';
+import { StormEvent } from './events';
 import { BlockDefinition, Plan } from '@kapeta/schemas';
 import { KapetaURI } from '@kapeta/nodejs-utils';
 export interface BlockDefinitionInfo {
     uri: KapetaURI;
     content: BlockDefinition;
-    screens: ScreenTemplate[];
+    aiName: string;
 }
 export interface ParsedResult {
     plan: Plan;

package/dist/cjs/src/storm/event-parser.js

@@ -106,7 +106,6 @@ class StormEventParser {
         this.connections = [];
     }
     addEvent(evt) {
-        console.log('Processing storm event', evt);
         this.events.push(evt);
         switch (evt.type) {
             case 'CREATE_PLAN_PROPERTIES':
@@ -119,7 +118,6 @@ class StormEventParser {
                     apis: [],
                     models: [],
                     types: [],
-                    screens: [],
                 };
                 break;
             case 'PLAN_RETRY':
@@ -141,18 +139,9 @@ class StormEventParser {
             case 'CREATE_CONNECTION':
                 this.connections.push(evt.payload);
                 break;
-            case 'SCREEN':
-                this.blocks[evt.payload.blockName].screens.push({
-                    name: evt.payload.name,
-                    description: evt.payload.description,
-                    url: evt.payload.url,
-                    template: evt.payload.template,
-                });
-                break;
             default:
             case 'SCREEN_CANDIDATE':
             case 'FILE':
-                console.warn('Unhandled event: %s', evt.type, evt);
                 break;
         }
     }
@@ -190,15 +179,6 @@ class StormEventParser {
                 },
             };
         });
-        Object.values(this.blocks).forEach((blockInfo) => {
-            const blockRef = this.toRef(handle, blockInfo.name);
-            const block = blockDefinitions[blockRef.toNormalizedString()];
-            if (!block) {
-                console.warn('Block not found: %s', blockInfo.name);
-                return;
-            }
-            screens[blockRef.fullName] = blockInfo.screens;
-        });
         // Copy API methods from API provider to CLIENT consumer
         this.connections
             .filter((connection) => connection.fromResourceType === 'API' && connection.toResourceType === 'CLIENT')
@@ -207,29 +187,28 @@ class StormEventParser {
             const clientConsumerRef = this.toRef(handle, apiConnection.toComponent);
             const apiProviderBlock = blockDefinitions[apiProviderRef.toNormalizedString()];
             if (!apiProviderBlock) {
-                console.warn('API provider not found: %s', apiConnection.fromComponent);
+                console.warn('API provider not found: %s', apiConnection.fromComponent, apiConnection);
                 return;
             }
             const clientConsumerBlock = blockDefinitions[clientConsumerRef.toNormalizedString()];
             if (!clientConsumerBlock) {
-                console.warn('Client consumer not found: %s', apiConnection.toComponent);
+                console.warn('Client consumer not found: %s', apiConnection.toComponent, apiConnection);
                 return;
             }
             const apiResource = apiProviderBlock.content.spec.providers?.find((p) => p.kind === this.options.apiKind && p.metadata.name === apiConnection.fromResource);
             if (!apiResource) {
-                console.warn('API resource not found: %s on %s', apiConnection.fromResource, apiProviderRef.toNormalizedString());
+                console.warn('API resource not found: %s on %s', apiConnection.fromResource, apiProviderRef.toNormalizedString(), apiConnection);
                 return;
             }
             const clientResource = clientConsumerBlock.content.spec.consumers?.find((clientResource) => {
                 if (clientResource.kind !== this.options.clientKind) {
-                    return;
-                }
-                if (clientResource.metadata.name !== apiConnection.toResource) {
-                    return;
+                    console.warn('Client resource kind mismatch: %s', clientResource.kind, this.options.clientKind);
+                    return false;
                 }
+                return clientResource.metadata.name === apiConnection.toResource;
             });
             if (!clientResource) {
-                console.warn('Client resource not found: %s on %s', apiConnection.toResource, clientConsumerRef.toNormalizedString());
+                console.warn('Client resource not found: %s on %s', apiConnection.toResource, clientConsumerRef.toNormalizedString(), apiConnection);
                 return;
             }
             clientResource.spec.methods = apiResource.spec.methods;
@@ -284,6 +263,7 @@ class StormEventParser {
             const blockRef = this.toRef(handle, blockInfo.name);
             const blockDefinitionInfo = {
                 uri: blockRef,
+                aiName: blockInfo.name,
                 content: {
                     kind: this.toBlockKind(blockInfo.type),
                     metadata: {
@@ -305,7 +285,6 @@ class StormEventParser {
                         consumers: [],
                     },
                 },
-                screens: blockInfo.screens,
             };
             const blockSpec = blockDefinitionInfo.content.spec;
             let apiResource = undefined;

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

@@ -18,7 +18,6 @@ export interface StormBlockInfoFilled extends StormBlockInfo {
     apis: string[];
     types: string[];
     models: string[];
-    screens: ScreenTemplate[];
 }
 export interface StormEventCreateBlock {
     type: 'CREATE_BLOCK';

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

@@ -21,7 +21,6 @@ router.post('/:handle/all', async (req, res) => {
     try {
         const stormOptions = await (0, event_parser_1.resolveOptions)();
         const eventParser = new event_parser_1.StormEventParser(stormOptions);
-        console.log('Got prompt', req.stringBody);
         const aiRequest = JSON.parse(req.stringBody ?? '{}');
         const metaStream = await stormClient_1.stormClient.createMetadata(aiRequest.prompt, aiRequest.history);
         res.set('Content-Type', 'application/x-ndjson');
@@ -41,8 +40,7 @@ router.post('/:handle/all', async (req, res) => {
             return;
         }
         const result = eventParser.toResult(handle);
-        console.log('RESULT\n', JSON.stringify(result, null, 2));
-        const stormCodegen = new codegen_1.StormCodegen(aiRequest.prompt, result.blocks);
+        const stormCodegen = new codegen_1.StormCodegen(aiRequest.prompt, result.blocks, eventParser.getEvents());
         const codegenStream = streamStormPartialResponse(stormCodegen.getStream(), res);
         await stormCodegen.process();
         await codegenStream;
@@ -52,26 +50,6 @@ router.post('/:handle/all', async (req, res) => {
         sendError(err, res);
     }
 });
-router.post('/metadata', async (req, res) => {
-    const aiRequest = JSON.parse(req.stringBody ?? '{}');
-    const result = await stormClient_1.stormClient.createMetadata(aiRequest.prompt, aiRequest.history);
-    await streamStormResponse(result, res);
-});
-router.post('/services/implement', async (req, res) => {
-    const aiRequest = JSON.parse(req.stringBody ?? '{}');
-    const result = await stormClient_1.stormClient.createServiceImplementation(aiRequest.prompt, aiRequest.history);
-    await streamStormResponse(result, res);
-});
-router.post('/ui/implement', async (req, res) => {
-    const aiRequest = JSON.parse(req.stringBody ?? '{}');
-    const result = await stormClient_1.stormClient.createUIImplementation(aiRequest.prompt, aiRequest.history);
-    await streamStormResponse(result, res);
-});
-async function streamStormResponse(result, res) {
-    res.set('Content-Type', 'application/x-ndjson');
-    await streamStormPartialResponse(result, res);
-    sendDone(res);
-}
 function sendDone(res) {
     res.write(JSON.stringify({
         type: 'DONE',

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

@@ -1,4 +1,4 @@
-import { ConversationItem, StormFileImplementationPrompt, StormStream } from './stream';
+import { ConversationItem, StormFileImplementationPrompt, StormStream, StormUIImplementationPrompt } from './stream';
 export declare const STORM_ID = "storm";
 declare class StormClient {
     private readonly _baseUrl;
@@ -6,7 +6,7 @@ declare class StormClient {
     private createOptions;
     private send;
     createMetadata(prompt: string, history?: ConversationItem[]): Promise<StormStream>;
-    createUIImplementation(prompt: StormFileImplementationPrompt, history?: ConversationItem[]): Promise<StormStream>;
+    createUIImplementation(prompt: StormUIImplementationPrompt, history?: ConversationItem[]): Promise<StormStream>;
     createServiceImplementation(prompt: StormFileImplementationPrompt, history?: ConversationItem[]): Promise<StormStream>;
 }
 export declare const stormClient: StormClient;

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

@@ -53,12 +53,6 @@ class StormClient {
         jsonLStream.on('error', (error) => {
             out.emit('error', error);
         });
-        jsonLStream.on('pause', () => {
-            console.log('paused');
-        });
-        jsonLStream.on('resume', () => {
-            console.log('resumed');
-        });
         jsonLStream.on('close', () => {
             out.end();
         });
@@ -77,7 +71,6 @@ class StormClient {
         });
     }
     createServiceImplementation(prompt, history) {
-        console.log('SENDING SERVICE PROMPT', JSON.stringify(prompt, null, 2));
         return this.send('/v2/services/merge', {
             history: history ?? [],
             prompt,

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

@@ -36,3 +36,10 @@ export interface StormFileImplementationPrompt {
     template: StormFileInfo;
     prompt: string;
 }
+export interface StormUIImplementationPrompt {
+    events: StormEvent[];
+    templates: StormFileInfo[];
+    context: StormFileInfo[];
+    blockName: string;
+    prompt: string;
+}

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

@@ -2,15 +2,19 @@
  * Copyright 2023 Kapeta Inc.
  * SPDX-License-Identifier: BUSL-1.1
  */
+import { StormEvent } from './events';
 import { BlockDefinitionInfo } from './event-parser';
 import { StormStream } from './stream';
 export declare class StormCodegen {
     private readonly userPrompt;
     private readonly blocks;
     private readonly out;
-    constructor(userPrompt: string, blocks: BlockDefinitionInfo[]);
+    private readonly events;
+    constructor(userPrompt: string, blocks: BlockDefinitionInfo[], events: StormEvent[]);
     process(): Promise<void>;
     getStream(): StormStream;
+    private handleTemplateFileOutput;
+    private handleUiOutput;
     private handleFileOutput;
     /**
      * Generates the code for a block and sends it to the AI

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

@@ -13,12 +13,13 @@ class StormCodegen {
     userPrompt;
     blocks;
     out = new stream_1.StormStream();
-    constructor(userPrompt, blocks) {
+    events;
+    constructor(userPrompt, blocks, events) {
         this.userPrompt = userPrompt;
         this.blocks = blocks;
+        this.events = events;
     }
     async process() {
-        console.log('Processing blocks', this.blocks.length);
         for (const block of this.blocks) {
             await this.processBlockCode(block);
         }
@@ -27,13 +28,42 @@ class StormCodegen {
     getStream() {
         return this.out;
     }
-    handleFileOutput(blockUri, template, data) {
+    handleTemplateFileOutput(blockUri, template, data) {
         switch (data.type) {
             case 'FILE':
                 template.filename = data.payload.filename;
                 template.content = data.payload.content;
+                return this.handleFileOutput(blockUri, data);
+        }
+    }
+    handleUiOutput(blockUri, data) {
+        switch (data.type) {
+            case 'SCREEN':
+                this.out.emit('data', {
+                    type: 'SCREEN',
+                    reason: data.reason,
+                    created: Date.now(),
+                    payload: {
+                        ...data.payload,
+                        blockName: blockUri.toNormalizedString(),
+                    },
+                });
+            case 'FILE':
+                return this.handleFileOutput(blockUri, data);
+        }
+    }
+    handleFileOutput(blockUri, data) {
+        switch (data.type) {
+            case 'FILE':
                 this.emitFile(blockUri, data.payload.filename, data.payload.content, data.reason);
-                break;
+                return {
+                    type: 'FILE',
+                    created: Date.now(),
+                    payload: {
+                        filename: data.payload.filename,
+                        content: data.payload.content,
+                    },
+                };
         }
     }
     /**
@@ -41,22 +71,35 @@ class StormCodegen {
      */
     async processBlockCode(block) {
         // Generate the code for the block using the standard codegen templates
-        const generatedResult = await this.generateBlock(block.content, block.screens);
+        const generatedResult = await this.generateBlock(block.content);
         if (!generatedResult) {
-            console.warn('No generated result for block', block.uri);
             return;
         }
         const allFiles = this.toStormFiles(generatedResult);
         // Send all the non-ai files to the stream
         this.emitFiles(block.uri, allFiles);
-        const implementFiles = [codegen_1.AIFileTypes.SERVICE, codegen_1.AIFileTypes.WEB_SCREEN];
-        // Gather the context files. These will be all be passed to the AI
-        const contextFiles = allFiles.filter((file) => file.type !== codegen_1.AIFileTypes.IGNORE && !implementFiles.includes(file.type));
+        const relevantFiles = allFiles.filter((file) => file.type !== codegen_1.AIFileTypes.IGNORE && file.type !== codegen_1.AIFileTypes.WEB_SCREEN);
+        const uiTemplates = allFiles.filter((file) => file.type === codegen_1.AIFileTypes.WEB_SCREEN);
+        if (uiTemplates.length > 0) {
+            const uiStream = await stormClient_1.stormClient.createUIImplementation({
+                events: this.events,
+                templates: uiTemplates,
+                context: relevantFiles,
+                blockName: block.aiName,
+                prompt: this.userPrompt,
+            });
+            uiStream.on('data', (evt) => {
+                this.handleUiOutput(block.uri, evt);
+            });
+            await uiStream.waitForDone();
+        }
+        // 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
         const serviceFiles = allFiles.filter((file) => file.type === codegen_1.AIFileTypes.SERVICE);
-        await this.processTemplates(block.uri, stormClient_1.stormClient.createServiceImplementation.bind(stormClient_1.stormClient), serviceFiles, contextFiles);
-        //const uiTemplates: StormFileInfo[] = allFiles.filter((file) => file.type === 'ui');
-        //await this.processTemplates(stormClient.createUIImplementation, uiTemplates, contextFiles);
+        if (serviceFiles.length > 0) {
+            await this.processTemplates(block.uri, stormClient_1.stormClient.createServiceImplementation.bind(stormClient_1.stormClient), serviceFiles, contextFiles);
+        }
     }
     /**
      * Emits the text-based files to the stream
@@ -94,17 +137,25 @@ class StormCodegen {
     /**
      * Sends the template to the AI and processes the response
      */
-    processTemplates(blockUri, generator, templates, contextFiles) {
+    async processTemplates(blockUri, generator, templates, contextFiles) {
         const promises = templates.map(async (templateFile) => {
             const stream = await generator({
                 context: contextFiles,
                 template: templateFile,
                 prompt: this.userPrompt,
             });
-            stream.on('data', (evt) => this.handleFileOutput(blockUri, templateFile, evt));
-            return stream.waitForDone();
+            const files = [];
+            stream.on('data', (evt) => {
+                const file = this.handleTemplateFileOutput(blockUri, templateFile, evt);
+                if (file) {
+                    files.push(file);
+                }
+            });
+            await stream.waitForDone();
+            return files;
         });
-        return Promise.all(promises);
+        const fileChunks = await Promise.all(promises);
+        return fileChunks.flat();
     }
     /**
      * Converts the generated files to a format that can be sent to the AI
@@ -143,7 +194,7 @@ class StormCodegen {
     /**
      * Generates the code using codegen for a given block.
      */
-    async generateBlock(yamlContent, screens) {
+    async generateBlock(yamlContent) {
         if (!yamlContent.spec.target?.kind) {
             //Not all block types have targets
             return;
@@ -153,7 +204,6 @@ class StormCodegen {
         }
         const codeGenerator = new codegen_1.BlockCodeGenerator(yamlContent);
         codeGenerator.withOption('AIContext', stormClient_1.STORM_ID);
-        codeGenerator.withOption('AIScreens', screens ?? []);
         return codeGenerator.generate();
     }
 }

package/dist/esm/src/storm/event-parser.d.ts

@@ -2,13 +2,13 @@
  * Copyright 2023 Kapeta Inc.
  * SPDX-License-Identifier: BUSL-1.1
  */
-import { ScreenTemplate, StormEvent } from './events';
+import { StormEvent } from './events';
 import { BlockDefinition, Plan } from '@kapeta/schemas';
 import { KapetaURI } from '@kapeta/nodejs-utils';
 export interface BlockDefinitionInfo {
     uri: KapetaURI;
     content: BlockDefinition;
-    screens: ScreenTemplate[];
+    aiName: string;
 }
 export interface ParsedResult {
     plan: Plan;

package/dist/esm/src/storm/event-parser.js

@@ -106,7 +106,6 @@ class StormEventParser {
         this.connections = [];
     }
     addEvent(evt) {
-        console.log('Processing storm event', evt);
         this.events.push(evt);
         switch (evt.type) {
             case 'CREATE_PLAN_PROPERTIES':
@@ -119,7 +118,6 @@ class StormEventParser {
                     apis: [],
                     models: [],
                     types: [],
-                    screens: [],
                 };
                 break;
             case 'PLAN_RETRY':
@@ -141,18 +139,9 @@ class StormEventParser {
             case 'CREATE_CONNECTION':
                 this.connections.push(evt.payload);
                 break;
-            case 'SCREEN':
-                this.blocks[evt.payload.blockName].screens.push({
-                    name: evt.payload.name,
-                    description: evt.payload.description,
-                    url: evt.payload.url,
-                    template: evt.payload.template,
-                });
-                break;
             default:
             case 'SCREEN_CANDIDATE':
             case 'FILE':
-                console.warn('Unhandled event: %s', evt.type, evt);
                 break;
         }
     }
@@ -190,15 +179,6 @@ class StormEventParser {
                 },
             };
         });
-        Object.values(this.blocks).forEach((blockInfo) => {
-            const blockRef = this.toRef(handle, blockInfo.name);
-            const block = blockDefinitions[blockRef.toNormalizedString()];
-            if (!block) {
-                console.warn('Block not found: %s', blockInfo.name);
-                return;
-            }
-            screens[blockRef.fullName] = blockInfo.screens;
-        });
         // Copy API methods from API provider to CLIENT consumer
         this.connections
             .filter((connection) => connection.fromResourceType === 'API' && connection.toResourceType === 'CLIENT')
@@ -207,29 +187,28 @@ class StormEventParser {
             const clientConsumerRef = this.toRef(handle, apiConnection.toComponent);
             const apiProviderBlock = blockDefinitions[apiProviderRef.toNormalizedString()];
             if (!apiProviderBlock) {
-                console.warn('API provider not found: %s', apiConnection.fromComponent);
+                console.warn('API provider not found: %s', apiConnection.fromComponent, apiConnection);
                 return;
             }
             const clientConsumerBlock = blockDefinitions[clientConsumerRef.toNormalizedString()];
             if (!clientConsumerBlock) {
-                console.warn('Client consumer not found: %s', apiConnection.toComponent);
+                console.warn('Client consumer not found: %s', apiConnection.toComponent, apiConnection);
                 return;
             }
             const apiResource = apiProviderBlock.content.spec.providers?.find((p) => p.kind === this.options.apiKind && p.metadata.name === apiConnection.fromResource);
             if (!apiResource) {
-                console.warn('API resource not found: %s on %s', apiConnection.fromResource, apiProviderRef.toNormalizedString());
+                console.warn('API resource not found: %s on %s', apiConnection.fromResource, apiProviderRef.toNormalizedString(), apiConnection);
                 return;
             }
             const clientResource = clientConsumerBlock.content.spec.consumers?.find((clientResource) => {
                 if (clientResource.kind !== this.options.clientKind) {
-                    return;
-                }
-                if (clientResource.metadata.name !== apiConnection.toResource) {
-                    return;
+                    console.warn('Client resource kind mismatch: %s', clientResource.kind, this.options.clientKind);
+                    return false;
                 }
+                return clientResource.metadata.name === apiConnection.toResource;
             });
             if (!clientResource) {
-                console.warn('Client resource not found: %s on %s', apiConnection.toResource, clientConsumerRef.toNormalizedString());
+                console.warn('Client resource not found: %s on %s', apiConnection.toResource, clientConsumerRef.toNormalizedString(), apiConnection);
                 return;
             }
             clientResource.spec.methods = apiResource.spec.methods;
@@ -284,6 +263,7 @@ class StormEventParser {
             const blockRef = this.toRef(handle, blockInfo.name);
             const blockDefinitionInfo = {
                 uri: blockRef,
+                aiName: blockInfo.name,
                 content: {
                     kind: this.toBlockKind(blockInfo.type),
                     metadata: {
@@ -305,7 +285,6 @@ class StormEventParser {
                         consumers: [],
                     },
                 },
-                screens: blockInfo.screens,
             };
             const blockSpec = blockDefinitionInfo.content.spec;
             let apiResource = undefined;

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

@@ -18,7 +18,6 @@ export interface StormBlockInfoFilled extends StormBlockInfo {
     apis: string[];
     types: string[];
     models: string[];
-    screens: ScreenTemplate[];
 }
 export interface StormEventCreateBlock {
     type: 'CREATE_BLOCK';

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

@@ -21,7 +21,6 @@ router.post('/:handle/all', async (req, res) => {
     try {
         const stormOptions = await (0, event_parser_1.resolveOptions)();
         const eventParser = new event_parser_1.StormEventParser(stormOptions);
-        console.log('Got prompt', req.stringBody);
         const aiRequest = JSON.parse(req.stringBody ?? '{}');
         const metaStream = await stormClient_1.stormClient.createMetadata(aiRequest.prompt, aiRequest.history);
         res.set('Content-Type', 'application/x-ndjson');
@@ -41,8 +40,7 @@ router.post('/:handle/all', async (req, res) => {
             return;
         }
         const result = eventParser.toResult(handle);
-        console.log('RESULT\n', JSON.stringify(result, null, 2));
-        const stormCodegen = new codegen_1.StormCodegen(aiRequest.prompt, result.blocks);
+        const stormCodegen = new codegen_1.StormCodegen(aiRequest.prompt, result.blocks, eventParser.getEvents());
         const codegenStream = streamStormPartialResponse(stormCodegen.getStream(), res);
         await stormCodegen.process();
         await codegenStream;
@@ -52,26 +50,6 @@ router.post('/:handle/all', async (req, res) => {
         sendError(err, res);
     }
 });
-router.post('/metadata', async (req, res) => {
-    const aiRequest = JSON.parse(req.stringBody ?? '{}');
-    const result = await stormClient_1.stormClient.createMetadata(aiRequest.prompt, aiRequest.history);
-    await streamStormResponse(result, res);
-});
-router.post('/services/implement', async (req, res) => {
-    const aiRequest = JSON.parse(req.stringBody ?? '{}');
-    const result = await stormClient_1.stormClient.createServiceImplementation(aiRequest.prompt, aiRequest.history);
-    await streamStormResponse(result, res);
-});
-router.post('/ui/implement', async (req, res) => {
-    const aiRequest = JSON.parse(req.stringBody ?? '{}');
-    const result = await stormClient_1.stormClient.createUIImplementation(aiRequest.prompt, aiRequest.history);
-    await streamStormResponse(result, res);
-});
-async function streamStormResponse(result, res) {
-    res.set('Content-Type', 'application/x-ndjson');
-    await streamStormPartialResponse(result, res);
-    sendDone(res);
-}
 function sendDone(res) {
     res.write(JSON.stringify({
         type: 'DONE',

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

@@ -1,4 +1,4 @@
-import { ConversationItem, StormFileImplementationPrompt, StormStream } from './stream';
+import { ConversationItem, StormFileImplementationPrompt, StormStream, StormUIImplementationPrompt } from './stream';
 export declare const STORM_ID = "storm";
 declare class StormClient {
     private readonly _baseUrl;
@@ -6,7 +6,7 @@ declare class StormClient {
     private createOptions;
     private send;
     createMetadata(prompt: string, history?: ConversationItem[]): Promise<StormStream>;
-    createUIImplementation(prompt: StormFileImplementationPrompt, history?: ConversationItem[]): Promise<StormStream>;
+    createUIImplementation(prompt: StormUIImplementationPrompt, history?: ConversationItem[]): Promise<StormStream>;
     createServiceImplementation(prompt: StormFileImplementationPrompt, history?: ConversationItem[]): Promise<StormStream>;
 }
 export declare const stormClient: StormClient;

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

@@ -53,12 +53,6 @@ class StormClient {
         jsonLStream.on('error', (error) => {
             out.emit('error', error);
         });
-        jsonLStream.on('pause', () => {
-            console.log('paused');
-        });
-        jsonLStream.on('resume', () => {
-            console.log('resumed');
-        });
         jsonLStream.on('close', () => {
             out.end();
         });
@@ -77,7 +71,6 @@ class StormClient {
         });
     }
     createServiceImplementation(prompt, history) {
-        console.log('SENDING SERVICE PROMPT', JSON.stringify(prompt, null, 2));
         return this.send('/v2/services/merge', {
             history: history ?? [],
             prompt,

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

@@ -36,3 +36,10 @@ export interface StormFileImplementationPrompt {
     template: StormFileInfo;
     prompt: string;
 }
+export interface StormUIImplementationPrompt {
+    events: StormEvent[];
+    templates: StormFileInfo[];
+    context: StormFileInfo[];
+    blockName: string;
+    prompt: string;
+}

package/package.json

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

package/src/storm/codegen.ts

@@ -8,7 +8,7 @@ import { AIFileTypes, BlockCodeGenerator, GeneratedFile, GeneratedResult } from
 import { BlockDefinition } from '@kapeta/schemas';
 import { codeGeneratorManager } from '../codeGeneratorManager';
 import { STORM_ID, stormClient } from './stormClient';
-import { ScreenTemplate, StormEvent, StormEventFile } from './events';
+import { ScreenTemplate, StormEvent, StormEventFile, StormEventScreen } from './events';
 import { BlockDefinitionInfo } from './event-parser';
 import { ConversationItem, StormFileImplementationPrompt, StormFileInfo, StormStream } from './stream';
 import { KapetaURI } from '@kapeta/nodejs-utils';
@@ -22,15 +22,15 @@ export class StormCodegen {
     private readonly userPrompt: string;
     private readonly blocks: BlockDefinitionInfo[];
     private readonly out = new StormStream();
+    private readonly events: StormEvent[];
 
-    constructor(userPrompt: string, blocks: BlockDefinitionInfo[]) {
+    constructor(userPrompt: string, blocks: BlockDefinitionInfo[], events: StormEvent[]) {
         this.userPrompt = userPrompt;
         this.blocks = blocks;
+        this.events = events;
     }
 
     public async process() {
-        console.log('Processing blocks', this.blocks.length);
-
         for (const block of this.blocks) {
             await this.processBlockCode(block);
         }
@@ -42,13 +42,44 @@ export class StormCodegen {
         return this.out;
     }
 
-    private handleFileOutput(blockUri: KapetaURI, template: StormFileInfo, data: StormEvent) {
+    private handleTemplateFileOutput(blockUri: KapetaURI, template: StormFileInfo, data: StormEvent) {
         switch (data.type) {
             case 'FILE':
                 template.filename = data.payload.filename;
                 template.content = data.payload.content;
+                return this.handleFileOutput(blockUri, data);
+        }
+    }
+
+    private handleUiOutput(blockUri: KapetaURI, data: StormEvent) {
+        switch (data.type) {
+            case 'SCREEN':
+                this.out.emit('data', {
+                    type: 'SCREEN',
+                    reason: data.reason,
+                    created: Date.now(),
+                    payload: {
+                        ...data.payload,
+                        blockName: blockUri.toNormalizedString(),
+                    },
+                });
+            case 'FILE':
+                return this.handleFileOutput(blockUri, data);
+        }
+    }
+
+    private handleFileOutput(blockUri: KapetaURI, data: StormEvent) {
+        switch (data.type) {
+            case 'FILE':
                 this.emitFile(blockUri, data.payload.filename, data.payload.content, data.reason);
-                break;
+                return {
+                    type: 'FILE',
+                    created: Date.now(),
+                    payload: {
+                        filename: data.payload.filename,
+                        content: data.payload.content,
+                    },
+                } as StormEventFile;
         }
     }
 
@@ -57,9 +88,8 @@ export class StormCodegen {
      */
     private async processBlockCode(block: BlockDefinitionInfo) {
         // Generate the code for the block using the standard codegen templates
-        const generatedResult = await this.generateBlock(block.content, block.screens);
+        const generatedResult = await this.generateBlock(block.content);
         if (!generatedResult) {
-            console.warn('No generated result for block', block.uri);
             return;
         }
 
@@ -68,24 +98,41 @@ export class StormCodegen {
         // Send all the non-ai files to the stream
         this.emitFiles(block.uri, allFiles);
 
-        const implementFiles = [AIFileTypes.SERVICE, AIFileTypes.WEB_SCREEN];
+        const relevantFiles: StormFileInfo[] = allFiles.filter(
+            (file) => file.type !== AIFileTypes.IGNORE && file.type !== AIFileTypes.WEB_SCREEN
+        );
+        const uiTemplates: StormFileInfo[] = allFiles.filter((file) => file.type === AIFileTypes.WEB_SCREEN);
+        if (uiTemplates.length > 0) {
+            const uiStream = await stormClient.createUIImplementation({
+                events: this.events,
+                templates: uiTemplates,
+                context: relevantFiles,
+                blockName: block.aiName,
+                prompt: this.userPrompt,
+            });
+
+            uiStream.on('data', (evt) => {
+                this.handleUiOutput(block.uri, evt);
+            });
+
+            await uiStream.waitForDone();
+        }
 
-        // Gather the context files. These will be all be passed to the AI
-        const contextFiles: StormFileInfo[] = allFiles.filter(
-            (file) => file.type !== AIFileTypes.IGNORE && !implementFiles.includes(file.type)
+        // Gather the context files for implementation. These will be all be passed to the AI
+        const contextFiles: StormFileInfo[] = relevantFiles.filter(
+            (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
         const serviceFiles: StormFileInfo[] = allFiles.filter((file) => file.type === AIFileTypes.SERVICE);
-        await this.processTemplates(
-            block.uri,
-            stormClient.createServiceImplementation.bind(stormClient),
-            serviceFiles,
-            contextFiles
-        );
-
-        //const uiTemplates: StormFileInfo[] = allFiles.filter((file) => file.type === 'ui');
-        //await this.processTemplates(stormClient.createUIImplementation, uiTemplates, contextFiles);
+        if (serviceFiles.length > 0) {
+            await this.processTemplates(
+                block.uri,
+                stormClient.createServiceImplementation.bind(stormClient),
+                serviceFiles,
+                contextFiles
+            );
+        }
     }
 
     /**
@@ -129,7 +176,7 @@ export class StormCodegen {
     /**
      * Sends the template to the AI and processes the response
      */
-    private processTemplates(
+    private async processTemplates(
         blockUri: KapetaURI,
         generator: ImplementationGenerator,
         templates: StormFileInfo[],
@@ -142,12 +189,22 @@ export class StormCodegen {
                 prompt: this.userPrompt,
             });
 
-            stream.on('data', (evt) => this.handleFileOutput(blockUri, templateFile, evt));
+            const files: StormEventFile[] = [];
 
-            return stream.waitForDone();
+            stream.on('data', (evt) => {
+                const file = this.handleTemplateFileOutput(blockUri, templateFile, evt);
+                if (file) {
+                    files.push(file);
+                }
+            });
+
+            await stream.waitForDone();
+            return files;
         });
 
-        return Promise.all(promises);
+        const fileChunks = await Promise.all(promises);
+
+        return fileChunks.flat();
     }
 
     /**
@@ -191,7 +248,7 @@ export class StormCodegen {
     /**
      * Generates the code using codegen for a given block.
      */
-    private async generateBlock(yamlContent: Definition, screens: ScreenTemplate[] | undefined) {
+    private async generateBlock(yamlContent: Definition) {
         if (!yamlContent.spec.target?.kind) {
             //Not all block types have targets
             return;
@@ -203,7 +260,6 @@ export class StormCodegen {
 
         const codeGenerator = new BlockCodeGenerator(yamlContent as BlockDefinition);
         codeGenerator.withOption('AIContext', STORM_ID);
-        codeGenerator.withOption('AIScreens', screens ?? []);
 
         return codeGenerator.generate();
     }

package/src/storm/event-parser.ts

@@ -28,7 +28,7 @@ import { definitionsManager } from '../definitionsManager';
 export interface BlockDefinitionInfo {
     uri: KapetaURI;
     content: BlockDefinition;
-    screens: ScreenTemplate[];
+    aiName: string;
 }
 
 export interface ParsedResult {
@@ -197,7 +197,6 @@ export class StormEventParser {
     }
 
     public addEvent(evt: StormEvent): void {
-        console.log('Processing storm event', evt);
         this.events.push(evt);
         switch (evt.type) {
             case 'CREATE_PLAN_PROPERTIES':
@@ -210,7 +209,6 @@ export class StormEventParser {
                     apis: [],
                     models: [],
                     types: [],
-                    screens: [],
                 };
                 break;
             case 'PLAN_RETRY':
@@ -232,19 +230,10 @@ export class StormEventParser {
             case 'CREATE_CONNECTION':
                 this.connections.push(evt.payload);
                 break;
-            case 'SCREEN':
-                this.blocks[evt.payload.blockName].screens.push({
-                    name: evt.payload.name,
-                    description: evt.payload.description,
-                    url: evt.payload.url,
-                    template: evt.payload.template,
-                });
-                break;
 
             default:
             case 'SCREEN_CANDIDATE':
             case 'FILE':
-                console.warn('Unhandled event: %s', evt.type, evt);
                 break;
         }
     }
@@ -288,17 +277,6 @@ export class StormEventParser {
             } satisfies BlockInstance;
         });
 
-        Object.values(this.blocks).forEach((blockInfo) => {
-            const blockRef = this.toRef(handle, blockInfo.name);
-            const block = blockDefinitions[blockRef.toNormalizedString()];
-            if (!block) {
-                console.warn('Block not found: %s', blockInfo.name);
-                return;
-            }
-
-            screens[blockRef.fullName] = blockInfo.screens;
-        });
-
         // Copy API methods from API provider to CLIENT consumer
         this.connections
             .filter((connection) => connection.fromResourceType === 'API' && connection.toResourceType === 'CLIENT')
@@ -307,12 +285,12 @@ export class StormEventParser {
                 const clientConsumerRef = this.toRef(handle, apiConnection.toComponent);
                 const apiProviderBlock = blockDefinitions[apiProviderRef.toNormalizedString()];
                 if (!apiProviderBlock) {
-                    console.warn('API provider not found: %s', apiConnection.fromComponent);
+                    console.warn('API provider not found: %s', apiConnection.fromComponent, apiConnection);
                     return;
                 }
                 const clientConsumerBlock = blockDefinitions[clientConsumerRef.toNormalizedString()];
                 if (!clientConsumerBlock) {
-                    console.warn('Client consumer not found: %s', apiConnection.toComponent);
+                    console.warn('Client consumer not found: %s', apiConnection.toComponent, apiConnection);
                     return;
                 }
 
@@ -324,25 +302,27 @@ export class StormEventParser {
                     console.warn(
                         'API resource not found: %s on %s',
                         apiConnection.fromResource,
-                        apiProviderRef.toNormalizedString()
+                        apiProviderRef.toNormalizedString(),
+                        apiConnection
                     );
                     return;
                 }
 
                 const clientResource = clientConsumerBlock.content.spec.consumers?.find((clientResource) => {
                     if (clientResource.kind !== this.options.clientKind) {
-                        return;
-                    }
-                    if (clientResource.metadata.name !== apiConnection.toResource) {
-                        return;
+                        console.warn('Client resource kind mismatch: %s', clientResource.kind, this.options.clientKind);
+                        return false;
                     }
+
+                    return clientResource.metadata.name === apiConnection.toResource;
                 });
 
                 if (!clientResource) {
                     console.warn(
                         'Client resource not found: %s on %s',
                         apiConnection.toResource,
-                        clientConsumerRef.toNormalizedString()
+                        clientConsumerRef.toNormalizedString(),
+                        apiConnection
                     );
                     return;
                 }
@@ -407,6 +387,7 @@ export class StormEventParser {
 
             const blockDefinitionInfo: BlockDefinitionInfo = {
                 uri: blockRef,
+                aiName: blockInfo.name,
                 content: {
                     kind: this.toBlockKind(blockInfo.type),
                     metadata: {
@@ -428,7 +409,6 @@ export class StormEventParser {
                         consumers: [],
                     },
                 },
-                screens: blockInfo.screens,
             };
 
             const blockSpec = blockDefinitionInfo.content.spec;

package/src/storm/events.ts

@@ -35,7 +35,6 @@ export interface StormBlockInfoFilled extends StormBlockInfo {
     apis: string[];
     types: string[];
     models: string[];
-    screens: ScreenTemplate[];
 }
 
 export interface StormEventCreateBlock {

package/src/storm/routes.ts

@@ -27,7 +27,6 @@ router.post('/:handle/all', async (req: KapetaBodyRequest, res: Response) => {
 
         const eventParser = new StormEventParser(stormOptions);
 
-        console.log('Got prompt', req.stringBody);
         const aiRequest: StormContextRequest = JSON.parse(req.stringBody ?? '{}');
         const metaStream = await stormClient.createMetadata(aiRequest.prompt, aiRequest.history);
 
@@ -52,9 +51,7 @@ router.post('/:handle/all', async (req: KapetaBodyRequest, res: Response) => {
         }
         const result = eventParser.toResult(handle);
 
-        console.log('RESULT\n', JSON.stringify(result, null, 2));
-
-        const stormCodegen = new StormCodegen(aiRequest.prompt, result.blocks);
+        const stormCodegen = new StormCodegen(aiRequest.prompt, result.blocks, eventParser.getEvents());
 
         const codegenStream = streamStormPartialResponse(stormCodegen.getStream(), res);
 
@@ -68,35 +65,6 @@ router.post('/:handle/all', async (req: KapetaBodyRequest, res: Response) => {
     }
 });
 
-router.post('/metadata', async (req: KapetaBodyRequest, res: Response) => {
-    const aiRequest: StormContextRequest = JSON.parse(req.stringBody ?? '{}');
-    const result = await stormClient.createMetadata(aiRequest.prompt, aiRequest.history);
-
-    await streamStormResponse(result, res);
-});
-
-router.post('/services/implement', async (req: KapetaBodyRequest, res: Response) => {
-    const aiRequest: StormContextRequest<StormFileImplementationPrompt> = JSON.parse(req.stringBody ?? '{}');
-    const result = await stormClient.createServiceImplementation(aiRequest.prompt, aiRequest.history);
-
-    await streamStormResponse(result, res);
-});
-
-router.post('/ui/implement', async (req: KapetaBodyRequest, res: Response) => {
-    const aiRequest: StormContextRequest<StormFileImplementationPrompt> = JSON.parse(req.stringBody ?? '{}');
-    const result = await stormClient.createUIImplementation(aiRequest.prompt, aiRequest.history);
-
-    await streamStormResponse(result, res);
-});
-
-async function streamStormResponse(result: StormStream, res: Response) {
-    res.set('Content-Type', 'application/x-ndjson');
-
-    await streamStormPartialResponse(result, res);
-
-    sendDone(res);
-}
-
 function sendDone(res: Response) {
     res.write(
         JSON.stringify({

package/src/storm/stormClient.ts

@@ -12,6 +12,7 @@ import {
     StormFileImplementationPrompt,
     StormFileInfo,
     StormStream,
+    StormUIImplementationPrompt,
 } from './stream';
 
 export const STORM_ID = 'storm';
@@ -73,14 +74,6 @@ class StormClient {
             out.emit('error', error);
         });
 
-        jsonLStream.on('pause', () => {
-            console.log('paused');
-        });
-
-        jsonLStream.on('resume', () => {
-            console.log('resumed');
-        });
-
         jsonLStream.on('close', () => {
             out.end();
         });
@@ -95,15 +88,14 @@ class StormClient {
         });
     }
 
-    public createUIImplementation(prompt: StormFileImplementationPrompt, history?: ConversationItem[]) {
-        return this.send<StormFileImplementationPrompt>('/v2/ui/merge', {
+    public createUIImplementation(prompt: StormUIImplementationPrompt, history?: ConversationItem[]) {
+        return this.send<StormUIImplementationPrompt>('/v2/ui/merge', {
             history: history ?? [],
             prompt,
         });
     }
 
     public createServiceImplementation(prompt: StormFileImplementationPrompt, history?: ConversationItem[]) {
-        console.log('SENDING SERVICE PROMPT', JSON.stringify(prompt, null, 2));
         return this.send<StormFileImplementationPrompt>('/v2/services/merge', {
             history: history ?? [],
             prompt,

package/src/storm/stream.ts

@@ -86,3 +86,11 @@ export interface StormFileImplementationPrompt {
     template: StormFileInfo;
     prompt: string;
 }
+
+export interface StormUIImplementationPrompt {
+    events: StormEvent[];
+    templates: StormFileInfo[];
+    context: StormFileInfo[];
+    blockName: string;
+    prompt: string;
+}