@kapeta/local-cluster-service 0.74.0 → 0.74.2

package/src/storm/stormClient.ts

@@ -24,6 +24,7 @@ const fetchWithRetries = createFetch(global.fetch, { retries: 5, retryDelay: 10
 export const STORM_ID = 'storm';
 
 export const ConversationIdHeader = 'Conversation-Id';
+export const SystemIdHeader = 'System-Id';
 
 export interface UIShellsPrompt {
     theme?: string;
@@ -85,11 +86,12 @@ export interface BasePromptRequest {
     skipImprovement?: boolean;
 }
 
-class StormClient {
+export class StormClient {
     private readonly _baseUrl: string;
-
-    constructor() {
+    private readonly _systemId: string;
+    constructor(systemId?: string) {
         this._baseUrl = getRemoteUrl('ai-service', 'https://ai.kapeta.com');
+        this._systemId = systemId || "";
     }
 
     private async createOptions(
@@ -110,7 +112,9 @@ class StormClient {
         if (body.conversationId) {
             headers[ConversationIdHeader] = body.conversationId;
         }
-
+        if (this._systemId) {
+            headers[SystemIdHeader] = this._systemId
+        }
         return {
             url,
             method: method,
@@ -130,7 +134,6 @@ class StormClient {
             prompt: stringPrompt,
             conversationId: body.conversationId,
         });
-
         const abort = new AbortController();
         options.signal = abort.signal;
 
@@ -202,6 +205,7 @@ class StormClient {
     public createUIShells(prompt: UIShellsPrompt, conversationId?: string) {
         return this.send('/v2/ui/shells', {
             prompt: JSON.stringify(prompt),
+            conversationId: conversationId,
         });
     }
 
@@ -245,6 +249,9 @@ class StormClient {
         const response = await fetch(u, {
             method: 'POST',
             body: JSON.stringify(prompt.pages),
+            headers: {
+                'systemId': prompt.systemId,
+            },
         });
         return (await response.json()) as HTMLPage[];
     }
@@ -377,5 +384,3 @@ class StormClient {
         return response;
     }
 }
-
-export const stormClient = new StormClient();

package/src/storm/stream.ts

@@ -99,6 +99,7 @@ export interface ConversationItem {
 
 export interface StormContextRequest<T = string> {
     conversationId?: string;
+    systemId?: string;
     history?: ConversationItem[];
     prompt: T;
 }
@@ -150,6 +151,7 @@ export enum HTMLPageEncoding {
 
 export interface ImplementAPIClients {
     pages: HTMLPage[];
+    systemId: string;
 }
 
 export interface HTMLPage {

package/src/stormService.ts

@@ -5,7 +5,7 @@ import path from 'path';
 import { existsSync } from 'fs';
 import { StormEvent, StormEventModelResponse, StormEventPromptImprove, StormEventUIShell } from './storm/events';
 import * as tar from 'tar';
-import { stormClient } from './storm/stormClient';
+import { StormClient } from './storm/stormClient';
 
 export class StormService {
     private getConversationFile(conversationId: string) {
@@ -118,8 +118,8 @@ export class StormService {
         }
     }
 
-    async uploadConversation(handle: string, conversationId: string) {
-        const tarballFile = this.getConversationTarball(conversationId);
+    async uploadConversation(handle: string, systemId: string) {
+        const tarballFile = this.getConversationTarball(systemId);
         const destDir = path.dirname(tarballFile);
         const tarballName = path.basename(tarballFile);
         await tar.create(
@@ -131,13 +131,15 @@ export class StormService {
             },
             ['.']
         );
-        await stormClient.uploadSystem(handle, conversationId, await fs.readFile(tarballFile));
+        const stormClient = new StormClient(systemId);
+        await stormClient.uploadSystem(handle, systemId, await fs.readFile(tarballFile));
     }
 
-    async installProjectById(handle: string, conversationId: string) {
-        const tarballFile = this.getConversationTarball(conversationId);
+    async installProjectById(handle: string, systemId: string) {
+        const tarballFile = this.getConversationTarball(systemId);
         const destDir = path.dirname(tarballFile);
-        const buffer = await stormClient.downloadSystem(handle, conversationId);
+        const stormClient = new StormClient(systemId);
+        const buffer = await stormClient.downloadSystem(handle, systemId);
         await fs.mkdir(destDir, { recursive: true });
         await fs.writeFile(tarballFile, buffer);
         await tar.extract({