@kapeta/local-cluster-service 0.48.4 → 0.49.0

package/src/storm/events.ts

@@ -127,6 +127,28 @@ export interface StormEventError {
     };
 }
 
+export interface StormEventErrorClassifier {
+    type: 'ERROR_CLASSIFIER';
+    reason: string;
+    created: number;
+    payload: StormEventErrorClassifierInfo;
+}
+
+export interface StormEventCodeFix {
+    type: 'CODE_FIX';
+    reason: string;
+    created: number;
+    payload: {
+        filename: string;
+        content: string;
+    };
+}
+export interface StormEventErrorClassifierInfo {
+    error: string;
+    filename: string;
+    potentialFix: string;
+}
+
 export interface ScreenTemplate {
     name: string;
     template: string;
@@ -174,6 +196,18 @@ export interface StormEventFile {
     };
 }
 
+export interface StormEventBlockReady {
+    type: 'BLOCK_READY';
+    reason: string;
+    created: number;
+    payload: {
+        path: string;
+        blockName: string;
+        blockRef: string;
+        instanceId: string;
+    };
+}
+
 export interface StormEventDone {
     type: 'DONE';
     created: number;
@@ -199,4 +233,7 @@ export type StormEvent =
     | StormEventScreenCandidate
     | StormEventFile
     | StormEventDone
-    | StormEventDefinitionChange;
+    | StormEventDefinitionChange
+    | StormEventErrorClassifier
+    | StormEventCodeFix
+    | StormEventBlockReady;

package/src/storm/routes.ts

@@ -4,15 +4,19 @@
  */
 
 import Router from 'express-promise-router';
+import FS from 'fs-extra';
 import { Response } from 'express';
 import { corsHandler } from '../middleware/cors';
 import { stringBody } from '../middleware/stringBody';
 import { KapetaBodyRequest } from '../types';
-import { StormContextRequest, StormFileImplementationPrompt, StormFileInfo, StormStream } from './stream';
+import { StormContextRequest, StormCreateBlockRequest, StormStream } from './stream';
 import { ConversationIdHeader, stormClient } from './stormClient';
 import { StormEvent } from './events';
 import { resolveOptions, StormDefinitions, StormEventParser } from './event-parser';
 import { StormCodegen } from './codegen';
+import { assetManager } from '../assetManager';
+import Path from 'path';
+import { normalizeKapetaUri } from '@kapeta/nodejs-utils';
 
 const router = Router();
 
@@ -62,7 +66,12 @@ router.post('/:handle/all', async (req: KapetaBodyRequest, res: Response) => {
         sendDefinitions(res, result);
 
         if (!req.query.skipCodegen) {
-            const stormCodegen = new StormCodegen(aiRequest.prompt, result.blocks, eventParser.getEvents());
+            const stormCodegen = new StormCodegen(
+                metaStream.getConversationId(),
+                aiRequest.prompt,
+                result.blocks,
+                eventParser.getEvents()
+            );
 
             const codegenPromise = streamStormPartialResponse(stormCodegen.getStream(), res);
 
@@ -78,6 +87,34 @@ router.post('/:handle/all', async (req: KapetaBodyRequest, res: Response) => {
     }
 });
 
+router.post('/block/create', async (req: KapetaBodyRequest, res: Response) => {
+    const createRequest: StormCreateBlockRequest = JSON.parse(req.stringBody ?? '{}');
+
+    try {
+        const ymlPath = Path.join(createRequest.newPath, 'kapeta.yml');
+
+        console.log('Creating block at', ymlPath);
+
+        const [asset] = await assetManager.createAsset(ymlPath, createRequest.definition);
+
+        if (await FS.pathExists(createRequest.tmpPath)) {
+            console.log('Moving block from', createRequest.tmpPath, 'to', createRequest.newPath);
+
+            await FS.move(createRequest.tmpPath, createRequest.newPath, {
+                overwrite: true,
+            });
+
+            console.log('Updating asset', asset.ref);
+
+            res.send(await assetManager.updateAsset(asset.ref, createRequest.definition));
+        } else {
+            res.send(asset);
+        }
+    } catch (err: any) {
+        res.status(500).send({ error: err.message });
+    }
+});
+
 function sendDefinitions(res: Response, result: StormDefinitions) {
     sendEvent(res, {
         type: 'DEFINITION_CHANGE',

package/src/storm/stormClient.ts

@@ -10,7 +10,6 @@ import {
     ConversationItem,
     StormContextRequest,
     StormFileImplementationPrompt,
-    StormFileInfo,
     StormStream,
     StormUIImplementationPrompt,
 } from './stream';
@@ -111,6 +110,19 @@ class StormClient {
             conversationId,
         });
     }
+
+    public createErrorClassification(prompt: string, history?: ConversationItem[], conversationId?: string) {
+        return this.send('/v2/code/errorclassifier', {
+            conversationId: conversationId,
+            prompt,
+        });
+    }
+    public createCodeFix(prompt: string, history?: ConversationItem[], conversationId?: string) {
+        return this.send('/v2/code/fix', {
+            conversationId: conversationId,
+            prompt,
+        });
+    }
 }
 
 export const stormClient = new StormClient();

package/src/storm/stream.ts

@@ -5,6 +5,7 @@
 import { EventEmitter } from 'node:events';
 import { StormEvent } from './events';
 import { AIFileTypes, GeneratedFile } from '@kapeta/codegen';
+import { BlockDefinition } from '@kapeta/schemas';
 
 export class StormStream extends EventEmitter {
     private conversationId: string = '';
@@ -73,6 +74,12 @@ export interface StormContextRequest<T = string> {
     prompt: T;
 }
 
+export interface StormCreateBlockRequest {
+    definition: BlockDefinition;
+    tmpPath: string;
+    newPath: string;
+}
+
 export interface StormFileInfo extends GeneratedFile {
     type: AIFileTypes;
 }