@kapeta/local-cluster-service 0.49.0 → 0.50.0

package/src/storm/stormClient.ts

@@ -13,6 +13,7 @@ import {
     StormStream,
     StormUIImplementationPrompt,
 } from './stream';
+import { getRawAsset } from 'node:sea';
 
 export const STORM_ID = 'storm';
 
@@ -60,6 +61,9 @@ class StormClient {
             conversationId: body.conversationId,
         });
 
+        const abort = new AbortController();
+        options.signal = abort.signal;
+
         const response = await fetch(options.url, options);
 
         if (response.status !== 200) {
@@ -69,7 +73,6 @@ class StormClient {
         }
 
         const conversationId = response.headers.get(ConversationIdHeader);
-        console.log('Received conversationId', conversationId);
 
         const out = new StormStream(stringPrompt, conversationId);
 
@@ -87,6 +90,10 @@ class StormClient {
             out.end();
         });
 
+        out.on('aborted', () => {
+            abort.abort();
+        });
+
         return out;
     }
 

package/src/storm/stream.ts

@@ -10,6 +10,7 @@ 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();
@@ -20,6 +21,10 @@ export class StormStream extends EventEmitter {
         return this.conversationId;
     }
 
+    isAborted() {
+        return this.aborted;
+    }
+
     addJSONLine(line: string) {
         try {
             this.lines.push(line);
@@ -38,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 {
@@ -45,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 {
@@ -62,6 +69,14 @@ export class StormStream extends EventEmitter {
             });
         });
     }
+
+    abort() {
+        if (this.aborted) {
+            return;
+        }
+        this.aborted = true;
+        this.emit('aborted');
+    }
 }
 
 export interface ConversationItem {