@kapeta/local-cluster-service 0.48.5 → 0.50.0

package/src/storm/stream.ts

@@ -5,10 +5,12 @@
 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 = '';
     private lines: string[] = [];
+    private aborted: boolean = false;
 
     constructor(prompt: string = '', conversationId?: string | null) {
         super();
@@ -19,6 +21,10 @@ export class StormStream extends EventEmitter {
         return this.conversationId;
     }
 
+    isAborted() {
+        return this.aborted;
+    }
+
     addJSONLine(line: string) {
         try {
             this.lines.push(line);
@@ -37,6 +43,7 @@ export class StormStream extends EventEmitter {
     }
 
     on(event: 'end', listener: () => void): this;
+    on(event: 'aborted', listener: () => void): this;
     on(event: 'error', listener: (e: Error) => void): this;
     on(event: 'data', listener: (data: StormEvent) => void): this;
     on(event: string, listener: (...args: any[]) => void): this {
@@ -44,6 +51,7 @@ export class StormStream extends EventEmitter {
     }
 
     emit(event: 'end'): boolean;
+    emit(event: 'aborted'): void;
     emit(event: 'error', e: Error): boolean;
     emit(event: 'data', data: StormEvent): boolean;
     emit(eventName: string | symbol, ...args: any[]): boolean {
@@ -61,6 +69,14 @@ export class StormStream extends EventEmitter {
             });
         });
     }
+
+    abort() {
+        if (this.aborted) {
+            return;
+        }
+        this.aborted = true;
+        this.emit('aborted');
+    }
 }
 
 export interface ConversationItem {
@@ -73,6 +89,12 @@ export interface StormContextRequest<T = string> {
     prompt: T;
 }
 
+export interface StormCreateBlockRequest {
+    definition: BlockDefinition;
+    tmpPath: string;
+    newPath: string;
+}
+
 export interface StormFileInfo extends GeneratedFile {
     type: AIFileTypes;
 }