@kapeta/local-cluster-service 0.53.3 → 0.53.5

package/CHANGELOG.md

@@ -1,3 +1,20 @@
+## [0.53.5](https://github.com/kapetacom/local-cluster-service/compare/v0.53.4...v0.53.5) (2024-06-17)
+
+
+### Bug Fixes
+
+* actually map to correct api ([dc88af0](https://github.com/kapetacom/local-cluster-service/commit/dc88af0f7a76867ed9f96e5653a1ceb7f9002f6d))
+* pretty ([9efc6ed](https://github.com/kapetacom/local-cluster-service/commit/9efc6edd9bb7717889d37a2d1c0c227b10e57d47))
+* split apis into multiple named resources ([da3b59f](https://github.com/kapetacom/local-cluster-service/commit/da3b59fb238f2035235b452491f3b170768e49d9))
+* use type to determine controller or not ([bb88e9f](https://github.com/kapetacom/local-cluster-service/commit/bb88e9f0d349bc23fc7973e8525578e6dcb1d663))
+
+## [0.53.4](https://github.com/kapetacom/local-cluster-service/compare/v0.53.3...v0.53.4) (2024-06-13)
+
+
+### Bug Fixes
+
+* handle API_RETRY and MODEL_RETRY events ([e784047](https://github.com/kapetacom/local-cluster-service/commit/e784047fbc7d851e03c5301506ab5d663db9e989))
+
 ## [0.53.3](https://github.com/kapetacom/local-cluster-service/compare/v0.53.2...v0.53.3) (2024-06-13)
 
 

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

@@ -242,9 +242,16 @@ class StormCodegen {
         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);
         const screenFiles = [];
+        let filteredEvents = [];
+        for (const event of this.events) {
+            filteredEvents.push(event);
+            if (event.type === 'PLAN_RETRY') {
+                filteredEvents = [];
+            }
+        }
         if (uiTemplates.length > 0) {
             const uiStream = await stormClient_1.stormClient.createUIImplementation({
-                events: this.events,
+                events: filteredEvents,
                 templates: uiTemplates,
                 context: relevantFiles,
                 blockName: block.aiName,

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

@@ -210,6 +210,17 @@ class StormEventParser {
                 evt.payload.toBlockId = StormEventParser.toInstanceId(handle, evt.payload.toComponent);
                 this.connections.push(evt.payload);
                 break;
+            case 'API_RETRY':
+                Object.values(this.blocks).forEach((block) => {
+                    block.types = [];
+                    block.apis = [];
+                });
+                break;
+            case 'MODEL_RETRY':
+                Object.values(this.blocks).forEach((block) => {
+                    block.models = [];
+                });
+                break;
         }
         return this.toResult(handle);
     }
@@ -373,7 +384,7 @@ class StormEventParser {
                 },
             };
             const blockSpec = blockDefinitionInfo.content.spec;
-            let apiResource = undefined;
+            const apiResources = {};
             let dbResource = undefined;
             blockInfo.resources.forEach((resource) => {
                 const port = {
@@ -381,10 +392,7 @@ class StormEventParser {
                 };
                 switch (resource.type) {
                     case 'API':
-                        if (apiResource) {
-                            break;
-                        }
-                        apiResource = {
+                        const apiResource = {
                             kind: this.toResourceKind(resource.type),
                             metadata: {
                                 name: resource.name,
@@ -400,6 +408,7 @@ class StormEventParser {
                                 },
                             },
                         };
+                        apiResources[resource.name] = apiResource;
                         blockSpec.providers.push(apiResource);
                         break;
                     case 'CLIENT':
@@ -480,11 +489,28 @@ class StormEventParser {
                         });
                 }
             });
-            if (apiResource) {
-                blockInfo.apis.forEach((api) => {
-                    apiResource.spec.source.value += api + '\n\n';
+            blockInfo.apis.forEach((api) => {
+                const dslApi = kaplang_core_1.DSLAPIParser.parse(api, {
+                    ignoreSemantics: true,
                 });
-            }
+                let exactMatch = false;
+                if (dslApi[0] && dslApi[0].type == kaplang_core_1.DSLEntityType.CONTROLLER) {
+                    const name = dslApi[0].name.toLowerCase();
+                    const apiResourceName = Object.keys(apiResources).find((key) => key.indexOf(name) > -1);
+                    if (apiResourceName) {
+                        const exactResource = apiResources[apiResourceName];
+                        exactResource.spec.source.value += api + '\n\n';
+                        exactMatch = true;
+                    }
+                }
+                if (!exactMatch) {
+                    // if we couldn't place the given api on the exact resource we just park it on the first
+                    // available rest resource
+                    const firstKey = Object.keys(apiResources)[0];
+                    const firstEntry = apiResources[firstKey];
+                    firstEntry.spec.source.value += api + '\n\n';
+                }
+            });
             blockInfo.types.forEach((type) => {
                 blockSpec.entities.source.value += type + '\n';
             });

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

@@ -92,6 +92,14 @@ export interface StormEventCreateDSLResource extends Omit<StormEventCreateDSL, '
         resourceName?: string;
     };
 }
+export interface StormEventCreateDSLRetry {
+    type: 'API_RETRY' | 'MODEL_RETRY';
+    reason: string;
+    created: number;
+    payload: {
+        error: string[];
+    };
+}
 export interface StormEventError {
     type: 'INVALID_RESPONSE' | 'ERROR_INTERNAL';
     reason: string;
@@ -250,4 +258,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;
+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;

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

@@ -3,8 +3,12 @@
  * Copyright 2023 Kapeta Inc.
  * SPDX-License-Identifier: BUSL-1.1
  */
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
 Object.defineProperty(exports, "__esModule", { value: true });
 const event_parser_1 = require("../../src/storm/event-parser");
+const simple_blog_events_json_1 = __importDefault(require("./simple-blog-events.json"));
 const parserOptions = {
     serviceKind: 'kapeta/block-service:local',
     serviceLanguage: 'kapeta/language-target-nodejs-ts:local',
@@ -158,4 +162,18 @@ describe('event-parser', () => {
         expect(result.plan.spec.connections[0].provider.blockId).toBe(serviceBlockInstance.id);
         expect(result.plan.spec.connections[0].provider.resourceName).toBe(apiResource?.metadata.name);
     });
+    it('it will split api into correct provider', () => {
+        const events = simple_blog_events_json_1.default;
+        const parser = new event_parser_1.StormEventParser(parserOptions);
+        events.forEach((event) => parser.processEvent('kapeta', event));
+        const result = parser.toResult('kapeta');
+        const blogService = result.blocks.find((block) => block.aiName === 'blog-service');
+        expect(blogService).toBeDefined();
+        expect(blogService?.content).toBeDefined();
+        const apiProviders = blogService?.content?.spec?.providers?.filter((provider) => provider.kind === 'kapeta/block-type-api:local');
+        expect(apiProviders).toBeDefined();
+        expect(apiProviders.length).toBe(2);
+        expect(apiProviders["0"].spec.source.value).not.toBe('');
+        expect(apiProviders["1"].spec.source.value).not.toBe('');
+    });
 });

package/dist/cjs/test/storm/simple-blog-events.json

@@ -0,0 +1,470 @@
+[
+    {
+        "type": "MODEL_RESPONSE",
+        "reason": "",
+        "payload": {
+            "text": "Certainly! Here's how you can design a simple blog system. It will consist of a backend block for managing posts and users, a frontend block for displaying the blog, and a gateway block for making the blog publicly accessible."
+        },
+        "created": 1718365003668
+    },
+    {
+        "type": "CREATE_PLAN_PROPERTIES",
+        "reason": "Define the properties for the plan itself",
+        "payload": {
+            "description": "A simple blog system with a backend for managing posts and users, a frontend for displaying the blog, and a gateway for making the blog publicly accessible.",
+            "name": "Simple Blog"
+        },
+        "created": 1718365004184
+    },
+    {
+        "type": "CREATE_BLOCK",
+        "reason": "The system needs a backend to handle posts and users",
+        "payload": {
+            "description": "Handles posts and users",
+            "name": "blog-service",
+            "resources": [
+                {
+                    "description": "Stores blog posts",
+                    "name": "posts",
+                    "type": "DATABASE"
+                },
+                {
+                    "description": "Stores user information",
+                    "name": "users",
+                    "type": "DATABASE"
+                },
+                {
+                    "description": "Handles blog posts",
+                    "name": "posts",
+                    "type": "API"
+                },
+                {
+                    "description": "Handles user information",
+                    "name": "users",
+                    "type": "API"
+                },
+                {
+                    "description": "Handles JWT token generation and validation",
+                    "name": "jwt-provider",
+                    "type": "JWTPROVIDER"
+                },
+                {
+                    "description": "Sends emails for user registration and password reset",
+                    "name": "smtp-client",
+                    "type": "SMTPCLIENT"
+                }
+            ],
+            "type": "BACKEND",
+            "blockRef": "kapeta://kapeta/blog-service:local",
+            "instanceId": "d8d25816-bb83-5797-9881-a6282fe7a20b"
+        },
+        "created": 1718365005843
+    },
+    {
+        "type": "CREATE_BLOCK",
+        "reason": "The system needs a frontend to display the blog",
+        "payload": {
+            "description": "Displays the blog",
+            "name": "blog-ui",
+            "resources": [
+                {
+                    "description": "Fetches blog posts from the backend",
+                    "name": "posts",
+                    "type": "CLIENT"
+                },
+                {
+                    "description": "Handles user authentication and registration",
+                    "name": "users",
+                    "type": "CLIENT"
+                },
+                {
+                    "description": "Displays the main page of the blog",
+                    "name": "main",
+                    "type": "WEBPAGE"
+                }
+            ],
+            "type": "FRONTEND",
+            "blockRef": "kapeta://kapeta/blog-ui:local",
+            "instanceId": "79761687-1528-5f8f-8036-b8d49e23b49d"
+        },
+        "created": 1718365006823
+    },
+    {
+        "type": "CREATE_BLOCK",
+        "reason": "The system needs a gateway to make the blog publicly accessible",
+        "payload": {
+            "description": "Makes the blog publicly accessible",
+            "name": "gateway",
+            "resources": [
+                {
+                    "description": "Makes the posts API public under '/posts' path",
+                    "name": "posts",
+                    "path": "/posts",
+                    "type": "CLIENT"
+                },
+                {
+                    "description": "Makes the main page public under '/' path",
+                    "name": "main",
+                    "path": "/",
+                    "type": "WEBFRAGMENT"
+                }
+            ],
+            "type": "GATEWAY",
+            "blockRef": "kapeta://kapeta/gateway:local",
+            "instanceId": "f7327a25-7fc2-57b6-99e6-1b7e07378790"
+        },
+        "created": 1718365007689
+    },
+    {
+        "type": "CREATE_CONNECTION",
+        "reason": "The frontend needs to fetch blog posts from the backend",
+        "payload": {
+            "fromComponent": "blog-service",
+            "fromResource": "posts",
+            "fromResourceType": "API",
+            "toComponent": "blog-ui",
+            "toResource": "posts",
+            "toResourceType": "CLIENT",
+            "fromBlockId": "d8d25816-bb83-5797-9881-a6282fe7a20b",
+            "toBlockId": "79761687-1528-5f8f-8036-b8d49e23b49d"
+        },
+        "created": 1718365008108
+    },
+    {
+        "type": "CREATE_CONNECTION",
+        "reason": "The frontend needs to handle user authentication and registration",
+        "payload": {
+            "fromComponent": "blog-service",
+            "fromResource": "users",
+            "fromResourceType": "API",
+            "toComponent": "blog-ui",
+            "toResource": "users",
+            "toResourceType": "CLIENT",
+            "fromBlockId": "d8d25816-bb83-5797-9881-a6282fe7a20b",
+            "toBlockId": "79761687-1528-5f8f-8036-b8d49e23b49d"
+        },
+        "created": 1718365008526
+    },
+    {
+        "type": "CREATE_CONNECTION",
+        "reason": "The gateway needs to make the posts API publicly accessible",
+        "payload": {
+            "fromComponent": "blog-service",
+            "fromResource": "posts",
+            "fromResourceType": "API",
+            "toComponent": "gateway",
+            "toResource": "posts",
+            "toResourceType": "CLIENT",
+            "fromBlockId": "d8d25816-bb83-5797-9881-a6282fe7a20b",
+            "toBlockId": "f7327a25-7fc2-57b6-99e6-1b7e07378790"
+        },
+        "created": 1718365008990
+    },
+    {
+        "type": "CREATE_CONNECTION",
+        "reason": "The gateway needs to make the main page publicly accessible",
+        "payload": {
+            "fromComponent": "blog-ui",
+            "fromResource": "main",
+            "fromResourceType": "WEBPAGE",
+            "toComponent": "gateway",
+            "toResource": "main",
+            "toResourceType": "WEBFRAGMENT",
+            "fromBlockId": "79761687-1528-5f8f-8036-b8d49e23b49d",
+            "toBlockId": "f7327a25-7fc2-57b6-99e6-1b7e07378790"
+        },
+        "created": 1718365009823
+    },
+    {
+        "type": "CREATE_CONNECTION",
+        "reason": "The blog service needs to be able to authenticate requests",
+        "payload": {
+            "fromComponent": "blog-service",
+            "fromResource": "jwt-provider",
+            "fromResourceType": "JWTPROVIDER",
+            "toComponent": "blog-service",
+            "toResource": "jwt-consumer",
+            "toResourceType": "JWTCONSUMER",
+            "fromBlockId": "d8d25816-bb83-5797-9881-a6282fe7a20b",
+            "toBlockId": "d8d25816-bb83-5797-9881-a6282fe7a20b"
+        },
+        "created": 1718365009974
+    },
+    {
+        "type": "PLAN_RETRY",
+        "reason": "Found issues in the response but will try again",
+        "payload": {
+            "agent": "plan",
+            "errors": [
+                "Resource jwt-consumer is not defined for component blog-service in connection: 'blog-service.jwt-provider.JWTPROVIDER' to 'blog-service.jwt-consumer.JWTCONSUMER'"
+            ]
+        },
+        "created": 1718365010012
+    },
+    {
+        "type": "MODEL_RESPONSE",
+        "reason": "",
+        "payload": {
+            "text": "Certainly! Here's how you can design a simple blog system. It will consist of a backend block for managing posts and users, a frontend block for displaying the blog, and a gateway block for making the blog publicly accessible."
+        },
+        "created": 1718365011059
+    },
+    {
+        "type": "CREATE_PLAN_PROPERTIES",
+        "reason": "Define the properties for the plan itself",
+        "payload": {
+            "description": "A simple blog system with a backend for managing posts and users, a frontend for displaying the blog, and a gateway for making the blog publicly accessible.",
+            "name": "Simple Blog"
+        },
+        "created": 1718365011625
+    },
+    {
+        "type": "CREATE_BLOCK",
+        "reason": "The system needs a backend to handle posts and users",
+        "payload": {
+            "description": "Handles posts and users",
+            "name": "blog-service",
+            "resources": [
+                {
+                    "description": "Stores blog posts",
+                    "name": "posts",
+                    "type": "DATABASE"
+                },
+                {
+                    "description": "Stores user information",
+                    "name": "users",
+                    "type": "DATABASE"
+                },
+                {
+                    "description": "Handles blog posts",
+                    "name": "posts",
+                    "type": "API"
+                },
+                {
+                    "description": "Handles user information",
+                    "name": "users",
+                    "type": "API"
+                },
+                {
+                    "description": "Handles JWT token generation and validation",
+                    "name": "jwt-provider",
+                    "type": "JWTPROVIDER"
+                },
+                {
+                    "description": "Handles JWT token validation",
+                    "name": "jwt-consumer",
+                    "type": "JWTCONSUMER"
+                },
+                {
+                    "description": "Sends emails for user registration and password reset",
+                    "name": "smtp-client",
+                    "type": "SMTPCLIENT"
+                }
+            ],
+            "type": "BACKEND",
+            "blockRef": "kapeta://kapeta/blog-service:local",
+            "instanceId": "d8d25816-bb83-5797-9881-a6282fe7a20b"
+        },
+        "created": 1718365013238
+    },
+    {
+        "type": "CREATE_BLOCK",
+        "reason": "The system needs a frontend to display the blog",
+        "payload": {
+            "description": "Displays the blog",
+            "name": "blog-ui",
+            "resources": [
+                {
+                    "description": "Fetches blog posts from the backend",
+                    "name": "posts",
+                    "type": "CLIENT"
+                },
+                {
+                    "description": "Handles user authentication and registration",
+                    "name": "users",
+                    "type": "CLIENT"
+                },
+                {
+                    "description": "Displays the main page of the blog",
+                    "name": "main",
+                    "type": "WEBPAGE"
+                }
+            ],
+            "type": "FRONTEND",
+            "blockRef": "kapeta://kapeta/blog-ui:local",
+            "instanceId": "79761687-1528-5f8f-8036-b8d49e23b49d"
+        },
+        "created": 1718365014077
+    },
+    {
+        "type": "CREATE_BLOCK",
+        "reason": "The system needs a gateway to make the blog publicly accessible",
+        "payload": {
+            "description": "Makes the blog publicly accessible",
+            "name": "gateway",
+            "resources": [
+                {
+                    "description": "Makes the posts API public under '/posts' path",
+                    "name": "posts",
+                    "path": "/posts",
+                    "type": "CLIENT"
+                },
+                {
+                    "description": "Makes the main page public under '/' path",
+                    "name": "main",
+                    "path": "/",
+                    "type": "WEBFRAGMENT"
+                }
+            ],
+            "type": "GATEWAY",
+            "blockRef": "kapeta://kapeta/gateway:local",
+            "instanceId": "f7327a25-7fc2-57b6-99e6-1b7e07378790"
+        },
+        "created": 1718365015029
+    },
+    {
+        "type": "CREATE_CONNECTION",
+        "reason": "The frontend needs to fetch blog posts from the backend",
+        "payload": {
+            "fromComponent": "blog-service",
+            "fromResource": "posts",
+            "fromResourceType": "API",
+            "toComponent": "blog-ui",
+            "toResource": "posts",
+            "toResourceType": "CLIENT",
+            "fromBlockId": "d8d25816-bb83-5797-9881-a6282fe7a20b",
+            "toBlockId": "79761687-1528-5f8f-8036-b8d49e23b49d"
+        },
+        "created": 1718365015463
+    },
+    {
+        "type": "CREATE_CONNECTION",
+        "reason": "The frontend needs to handle user authentication and registration",
+        "payload": {
+            "fromComponent": "blog-service",
+            "fromResource": "users",
+            "fromResourceType": "API",
+            "toComponent": "blog-ui",
+            "toResource": "users",
+            "toResourceType": "CLIENT",
+            "fromBlockId": "d8d25816-bb83-5797-9881-a6282fe7a20b",
+            "toBlockId": "79761687-1528-5f8f-8036-b8d49e23b49d"
+        },
+        "created": 1718365016463
+    },
+    {
+        "type": "CREATE_CONNECTION",
+        "reason": "The gateway needs to make the posts API publicly accessible",
+        "payload": {
+            "fromComponent": "blog-service",
+            "fromResource": "posts",
+            "fromResourceType": "API",
+            "toComponent": "gateway",
+            "toResource": "posts",
+            "toResourceType": "CLIENT",
+            "fromBlockId": "d8d25816-bb83-5797-9881-a6282fe7a20b",
+            "toBlockId": "f7327a25-7fc2-57b6-99e6-1b7e07378790"
+        },
+        "created": 1718365016884
+    },
+    {
+        "type": "CREATE_CONNECTION",
+        "reason": "The gateway needs to make the main page publicly accessible",
+        "payload": {
+            "fromComponent": "blog-ui",
+            "fromResource": "main",
+            "fromResourceType": "WEBPAGE",
+            "toComponent": "gateway",
+            "toResource": "main",
+            "toResourceType": "WEBFRAGMENT",
+            "fromBlockId": "79761687-1528-5f8f-8036-b8d49e23b49d",
+            "toBlockId": "f7327a25-7fc2-57b6-99e6-1b7e07378790"
+        },
+        "created": 1718365017299
+    },
+    {
+        "type": "CREATE_CONNECTION",
+        "reason": "The blog service needs to be able to authenticate requests",
+        "payload": {
+            "fromComponent": "blog-service",
+            "fromResource": "jwt-provider",
+            "fromResourceType": "JWTPROVIDER",
+            "toComponent": "blog-service",
+            "toResource": "jwt-consumer",
+            "toResourceType": "JWTCONSUMER",
+            "fromBlockId": "d8d25816-bb83-5797-9881-a6282fe7a20b",
+            "toBlockId": "d8d25816-bb83-5797-9881-a6282fe7a20b"
+        },
+        "created": 1718365017692
+    },
+    {
+        "type": "CREATE_TYPE",
+        "reason": "",
+        "payload": {
+            "blockName": "blog-service",
+            "content": "type Post {\n\tid: string\n\ttitle: string\n\tcontent: string\n\tauthorId: string\n\tcreatedAt: date\n\tupdatedAt: date\n}",
+            "blockRef": "kapeta://kapeta/blog-service:local",
+            "instanceId": "d8d25816-bb83-5797-9881-a6282fe7a20b"
+        },
+        "created": 1718365018773
+    },
+    {
+        "type": "CREATE_TYPE",
+        "reason": "",
+        "payload": {
+            "blockName": "blog-service",
+            "content": "type User {\n\tid: string\n\tusername: string\n\temail: string\n\tpassword: string\n\tcreatedAt: date\n\tupdatedAt: date\n}",
+            "blockRef": "kapeta://kapeta/blog-service:local",
+            "instanceId": "d8d25816-bb83-5797-9881-a6282fe7a20b"
+        },
+        "created": 1718365019329
+    },
+    {
+        "type": "CREATE_API",
+        "reason": "",
+        "payload": {
+            "blockName": "blog-service",
+            "content": "controller Post(\"/posts\") {\n\n\t@GET(\"/\")\n\tgetPosts():Post[]\n\t\n\t@GET(\"/{id}\")\n\tgetPost(@Path(\"id\") id:string):Post\n\t\n\t@POST(\"/\")\n\tcreatePost(@Body post:Post):Post\n\t\n\t@PUT(\"/{id}\")\n\tupdatePost(@Path(\"id\") id:string, @Body post:Post):Post\n\t\n\t@DELETE(\"/{id}\")\n\tdeletePost(@Path(\"id\") id:string):void\n}",
+            "blockRef": "kapeta://kapeta/blog-service:local",
+            "instanceId": "d8d25816-bb83-5797-9881-a6282fe7a20b",
+            "resourceName": "posts"
+        },
+        "created": 1718365020235
+    },
+    {
+        "type": "CREATE_API",
+        "reason": "",
+        "payload": {
+            "blockName": "blog-service",
+            "content": "controller User(\"/users\") {\n\n\t@GET(\"/\")\n\tgetUsers():User[]\n\t\n\t@GET(\"/{id}\")\n\tgetUser(@Path(\"id\") id:string):User\n\t\n\t@POST(\"/\")\n\tcreateUser(@Body user:User):User\n\t\n\t@PUT(\"/{id}\")\n\tupdateUser(@Path(\"id\") id:string, @Body user:User):User\n\t\n\t@DELETE(\"/{id}\")\n\tdeleteUser(@Path(\"id\") id:string):void\n}",
+            "blockRef": "kapeta://kapeta/blog-service:local",
+            "instanceId": "d8d25816-bb83-5797-9881-a6282fe7a20b",
+            "resourceName": "posts"
+        },
+        "created": 1718365020558
+    },
+    {
+        "type": "CREATE_MODEL",
+        "reason": "",
+        "payload": {
+            "blockName": "blog-service",
+            "content": "type Post {\n\t@Id\n\tid: string\n\ttitle: string\n\tcontent: string\n\tauthorId: string\n\tcreatedAt: date\n\tupdatedAt: date\n}",
+            "blockRef": "kapeta://kapeta/blog-service:local",
+            "instanceId": "d8d25816-bb83-5797-9881-a6282fe7a20b",
+            "resourceName": "posts"
+        },
+        "created": 1718365022396
+    },
+    {
+        "type": "CREATE_MODEL",
+        "reason": "",
+        "payload": {
+            "blockName": "blog-service",
+            "content": "type User {\n\t@Id\n\tid: string\n\tusername: string\n\temail: string\n\tpassword: string\n\tcreatedAt: date\n\tupdatedAt: date\n}",
+            "blockRef": "kapeta://kapeta/blog-service:local",
+            "instanceId": "d8d25816-bb83-5797-9881-a6282fe7a20b",
+            "resourceName": "posts"
+        },
+        "created": 1718365022826
+    }
+]

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

@@ -242,9 +242,16 @@ class StormCodegen {
         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);
         const screenFiles = [];
+        let filteredEvents = [];
+        for (const event of this.events) {
+            filteredEvents.push(event);
+            if (event.type === 'PLAN_RETRY') {
+                filteredEvents = [];
+            }
+        }
         if (uiTemplates.length > 0) {
             const uiStream = await stormClient_1.stormClient.createUIImplementation({
-                events: this.events,
+                events: filteredEvents,
                 templates: uiTemplates,
                 context: relevantFiles,
                 blockName: block.aiName,