@kapeta/local-cluster-service 0.74.0 → 0.74.1

package/CHANGELOG.md

@@ -1,3 +1,10 @@
+## [0.74.1](https://github.com/kapetacom/local-cluster-service/compare/v0.74.0...v0.74.1) (2024-09-27)
+
+
+### Bug Fixes
+
+* adding system id to all calls ([b3e978f](https://github.com/kapetacom/local-cluster-service/commit/b3e978ffb74fe0ab1a46006412e52c63fe49eff8))
+
 # [0.74.0](https://github.com/kapetacom/local-cluster-service/compare/v0.73.0...v0.74.0) (2024-09-26)
 
 

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

@@ -240,8 +240,9 @@ class PageQueue extends node_events_1.EventEmitter {
             console.warn('Skipping image reference of type %s for url %s', mimeType, prompt.url);
             return;
         }
+        const client = new stormClient_1.StormClient(this.systemId);
         this.images.set(prompt.url, prompt.description);
-        const result = await stormClient_1.stormClient.createImage(`Create an image for the url "${prompt.url}" with this description: ${prompt.description}`.trim());
+        const result = await client.createImage(`Create an image for the url "${prompt.url}" with this description: ${prompt.description}`.trim());
         let imageEvent = null;
         result.on('data', (event) => {
             if (event.type === 'IMAGE') {
@@ -256,7 +257,8 @@ class PageQueue extends node_events_1.EventEmitter {
         this.emit('image', imageEvent, prompt);
     }
     async generate(prompt, conversationId) {
-        const screenStream = await stormClient_1.stormClient.createUIPage(prompt, conversationId);
+        const client = new stormClient_1.StormClient(this.systemId);
+        const screenStream = await client.createUIPage(prompt, conversationId);
         let pageEvent = null;
         screenStream.on('data', (event) => {
             if (event.type === 'PAGE') {
@@ -273,7 +275,8 @@ class PageQueue extends node_events_1.EventEmitter {
         await this.processPageEventWithReferences(pageEvent);
     }
     async resolveReferences(content) {
-        const referenceStream = await stormClient_1.stormClient.classifyUIReferences(content);
+        const client = new stormClient_1.StormClient(this.systemId);
+        const referenceStream = await client.classifyUIReferences(content);
         const references = [];
         referenceStream.on('data', (referenceData) => {
             if (referenceData.type !== 'REF_CLASSIFICATION') {

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

@@ -311,9 +311,10 @@ class StormCodegen {
             permissions: '0644',
         });
         const uiEvents = [];
+        const stormClient = new stormClient_1.StormClient(this.uiSystemId);
         // generate screens
         if (uiTemplates.length) {
-            const screenStream = await stormClient_1.stormClient.listScreens({
+            const screenStream = await stormClient.listScreens({
                 events: filteredEvents,
                 templates: uiTemplates,
                 context: relevantFiles,
@@ -346,7 +347,7 @@ class StormCodegen {
                     context: relevantFiles.concat([getScreenEventsFile()]),
                     prompt: this.userPrompt,
                 };
-                const uiStream = await stormClient_1.stormClient.createUIImplementation(payload);
+                const uiStream = await stormClient.createUIImplementation(payload);
                 uiStream.on('data', (evt) => {
                     const uiFile = this.handleUiOutput(blockUri, block.aiName, evt);
                     if (uiFile != undefined) {
@@ -374,13 +375,13 @@ class StormCodegen {
         });
         allFiles.push(...screenFilesConverted);
         let webRouters = allFiles.filter((file) => file.type === codegen_1.AIFileTypes.WEB_ROUTER);
-        webRouters = await this.processTemplates(blockUri, block.aiName, stormClient_1.stormClient.generateCode.bind(stormClient_1.stormClient), webRouters, screenFilesConverted.concat([getScreenEventsFile()]));
+        webRouters = await this.processTemplates(blockUri, block.aiName, stormClient.generateCode.bind(stormClient), webRouters, screenFilesConverted.concat([getScreenEventsFile()]));
         // Gather the context files for implementation. These will be all be passed to the AI
         const contextFiles = relevantFiles.filter((file) => ![codegen_1.AIFileTypes.SERVICE, codegen_1.AIFileTypes.WEB_SCREEN, codegen_1.AIFileTypes.WEB_ROUTER].includes(file.type));
         // Send the service and UI templates to the AI. These will be sent one-by-one in addition to the context files
         let serviceFiles = allFiles.filter((file) => file.type === codegen_1.AIFileTypes.SERVICE);
         if (serviceFiles.length > 0) {
-            serviceFiles = await this.processTemplates(blockUri, block.aiName, stormClient_1.stormClient.createServiceImplementation.bind(stormClient_1.stormClient), serviceFiles, contextFiles);
+            serviceFiles = await this.processTemplates(blockUri, block.aiName, stormClient.createServiceImplementation.bind(stormClient), serviceFiles, contextFiles);
         }
         if (this.isAborted()) {
             return;
@@ -510,7 +511,8 @@ class StormCodegen {
         }
     }
     async classifyErrors(errors, basePath) {
-        const errorStream = await stormClient_1.stormClient.createErrorClassification(errors, []);
+        const stormClient = new stormClient_1.StormClient(this.uiSystemId);
+        const errorStream = await stormClient.createErrorClassification(errors, []);
         const fixes = new Map();
         this.out.on('aborted', () => {
             errorStream.abort();
@@ -547,7 +549,8 @@ class StormCodegen {
                 error: error,
                 projectFiles: allFiles.map((f) => f.filename),
             };
-            const detailsStream = await stormClient_1.stormClient.createErrorDetails(JSON.stringify(request), []);
+            const stormClient = new stormClient_1.StormClient(this.uiSystemId);
+            const detailsStream = await stormClient.createErrorDetails(JSON.stringify(request), []);
             detailsStream.on('data', (evt) => {
                 if (evt.type === 'ERROR_DETAILS') {
                     resolve(evt.payload.files);
@@ -611,7 +614,8 @@ class StormCodegen {
     async codeFix(blockUri, blockName, fix, history) {
         return new Promise(async (resolve, reject) => {
             try {
-                const fixStream = await stormClient_1.stormClient.createCodeFix(fix, history, this.conversationId);
+                const stormClient = new stormClient_1.StormClient(this.uiSystemId);
+                const fixStream = await stormClient.createCodeFix(fix, history, this.conversationId);
                 let resolved = false;
                 fixStream.on('data', (evt) => {
                     if (this.handleFileEvents(blockUri, blockName, evt)) {

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

@@ -117,8 +117,10 @@ router.post('/ui/create-system/:handle/:systemId', async (req, res) => {
     res.set(stormClient_1.ConversationIdHeader, systemId);
     sendEvent(res, (0, event_parser_1.createPhaseStartEvent)(events_1.StormEventPhaseType.IMPLEMENT_APIS));
     const pagesFromDisk = (0, utils_1.readFilesAndContent)(srcDir);
-    const pagesWithImplementation = await stormClient_1.stormClient.replaceMockWithAPICall({
+    const client = new stormClient_1.StormClient(systemId);
+    const pagesWithImplementation = await client.replaceMockWithAPICall({
         pages: pagesFromDisk,
+        systemId: systemId,
     });
     await (0, utils_1.copyDirectory)(srcDir, destDir, (fileName, content) => {
         // find the page from result1 and write the content to the file
@@ -131,7 +133,7 @@ router.post('/ui/create-system/:handle/:systemId', async (req, res) => {
     const pageContents = pagesWithImplementation.map((page) => {
         return page.content;
     });
-    const prompt = await stormClient_1.stormClient.generatePrompt(pageContents);
+    const prompt = await client.generatePrompt(pageContents);
     sendEvent(res, (0, event_parser_1.createPhaseEndEvent)(events_1.StormEventPhaseType.COMPOSE_SYSTEM_PROMPT));
     req.query.systemId = systemId;
     const promptRequest = {
@@ -149,10 +151,12 @@ router.post('/ui/create-system-simple/:handle/:systemId', async (req, res) => {
     //res.set('Access-Control-Expose-Headers', ConversationIdHeader);
     //res.set(ConversationIdHeader, systemId);
     //sendEvent(res, createPhaseStartEvent(StormEventPhaseType.IMPLEMENT_APIS));
+    const client = new stormClient_1.StormClient(systemId);
     try {
         const pagesFromDisk = (0, utils_1.readFilesAndContent)(srcDir);
-        const pagesWithImplementation = await stormClient_1.stormClient.replaceMockWithAPICall({
+        const pagesWithImplementation = await client.replaceMockWithAPICall({
             pages: pagesFromDisk,
+            systemId: systemId,
         });
         //sendEvent(res, createPhaseEndEvent(StormEventPhaseType.IMPLEMENT_APIS));
         //sendEvent(res, createPhaseStartEvent(StormEventPhaseType.COMPOSE_SYSTEM));
@@ -165,7 +169,7 @@ router.post('/ui/create-system-simple/:handle/:systemId', async (req, res) => {
             }
             return page;
         });
-        const systemUrl = await stormClient_1.stormClient.createSimpleBackend(handle, systemId, { pages: allFiles });
+        const systemUrl = await client.createSimpleBackend(handle, systemId, { pages: allFiles });
         //sendEvent(res, {type: 'SYSTEM_READY', created: Math.floor(Date.now() / 1000), reason: 'System Ready', payload: { systemUrl: systemUrl }});
         //sendEvent(res, createPhaseEndEvent(StormEventPhaseType.COMPOSE_SYSTEM));
         //sendDone(res);
@@ -250,7 +254,8 @@ router.post('/:handle/ui/iterative', async (req, res) => {
     try {
         const conversationId = req.headers[stormClient_1.ConversationIdHeader.toLowerCase()];
         const aiRequest = JSON.parse(req.stringBody ?? '{}');
-        const landingPagesStream = await stormClient_1.stormClient.createUILandingPages(aiRequest, conversationId);
+        const client = new stormClient_1.StormClient(conversationId); //todo is this correct we are using the landing page getConversationId down below as well
+        const landingPagesStream = await client.createUILandingPages(aiRequest, conversationId);
         onRequestAborted(req, res, () => {
             landingPagesStream.abort();
         });
@@ -339,8 +344,9 @@ router.post('/:handle/ui', async (req, res) => {
     try {
         const outerConversationId = req.headers[stormClient_1.ConversationIdHeader.toLowerCase()] || (0, crypto_1.randomUUID)();
         const aiRequest = JSON.parse(req.stringBody ?? '{}');
+        const stormClient = new stormClient_1.StormClient(outerConversationId);
         // Get user journeys
-        const userJourneysStream = await stormClient_1.stormClient.createUIUserJourneys(aiRequest, outerConversationId);
+        const userJourneysStream = await stormClient.createUIUserJourneys(aiRequest, outerConversationId);
         onRequestAborted(req, res, () => {
             userJourneysStream.abort();
         });
@@ -378,7 +384,7 @@ router.post('/:handle/ui', async (req, res) => {
         });
         let theme = '';
         try {
-            const themeStream = await stormClient_1.stormClient.createTheme(aiRequest, outerConversationId);
+            const themeStream = await stormClient.createTheme(aiRequest, outerConversationId);
             onRequestAborted(req, res, () => {
                 themeStream.abort();
             });
@@ -417,7 +423,7 @@ router.post('/:handle/ui', async (req, res) => {
         }
         await waitForStormStream(userJourneysStream);
         // Get the UI shells
-        const shellsStream = await stormClient_1.stormClient.createUIShells({
+        const shellsStream = await stormClient.createUIShells({
             theme: theme || undefined,
             pages: Object.values(uniqueUserJourneyScreens).map((screen) => ({
                 name: screen.name,
@@ -574,7 +580,8 @@ router.post('/ui/vote', async (req, res) => {
     const aiRequest = JSON.parse(req.stringBody ?? '{}');
     const { topic, vote, mainConversationId } = aiRequest;
     try {
-        await stormClient_1.stormClient.voteUIPage(topic, conversationId, vote, mainConversationId);
+        const stormClient = new stormClient_1.StormClient(mainConversationId);
+        await stormClient.voteUIPage(topic, conversationId, vote, mainConversationId);
     }
     catch (e) {
         res.status(500).send({ error: e.message });
@@ -585,7 +592,8 @@ router.post('/ui/get-vote', async (req, res) => {
     const aiRequest = JSON.parse(req.stringBody ?? '{}');
     const { topic, mainConversationId } = aiRequest;
     try {
-        const vote = await stormClient_1.stormClient.getVoteUIPage(topic, conversationId, mainConversationId);
+        const stormClient = new stormClient_1.StormClient(mainConversationId);
+        const vote = await stormClient.getVoteUIPage(topic, conversationId, mainConversationId);
         res.send({ vote });
     }
     catch (e) {
@@ -609,7 +617,8 @@ async function handleAll(req, res) {
         const eventParser = new event_parser_1.StormEventParser(stormOptions);
         const conversationId = req.headers[stormClient_1.ConversationIdHeader.toLowerCase()];
         const aiRequest = JSON.parse(req.stringBody ?? '{}');
-        const metaStream = await stormClient_1.stormClient.createMetadata(aiRequest, conversationId);
+        const stormClient = new stormClient_1.StormClient(systemId);
+        const metaStream = await stormClient.createMetadata(aiRequest, conversationId);
         onRequestAborted(req, res, () => {
             metaStream.abort();
         });

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

@@ -4,6 +4,7 @@ import { ConversationItem, CreateSimpleBackendRequest, HTMLPage, ImplementAPICli
 import { Page, StormEventPageUrl } from './events';
 export declare const STORM_ID = "storm";
 export declare const ConversationIdHeader = "Conversation-Id";
+export declare const SystemIdHeader = "System-Id";
 export interface UIShellsPrompt {
     theme?: string;
     pages: {
@@ -55,9 +56,10 @@ export interface BasePromptRequest {
     prompt: string;
     skipImprovement?: boolean;
 }
-declare class StormClient {
+export declare class StormClient {
     private readonly _baseUrl;
-    constructor();
+    private readonly _systemId;
+    constructor(systemId?: string);
     private createOptions;
     private send;
     createMetadata(prompt: BasePromptRequest, conversationId?: string): Promise<StormStream>;
@@ -88,5 +90,3 @@ declare class StormClient {
     downloadSystem(handle: string, conversationId: string): Promise<Buffer>;
     uploadSystem(handle: string, conversationId: string, buffer: Buffer): Promise<Response>;
 }
-export declare const stormClient: StormClient;
-export {};

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

@@ -3,7 +3,7 @@ var __importDefault = (this && this.__importDefault) || function (mod) {
     return (mod && mod.__esModule) ? mod : { "default": mod };
 };
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.stormClient = exports.ConversationIdHeader = exports.STORM_ID = void 0;
+exports.StormClient = exports.SystemIdHeader = exports.ConversationIdHeader = exports.STORM_ID = void 0;
 /**
  * Copyright 2023 Kapeta Inc.
  * SPDX-License-Identifier: BUSL-1.1
@@ -17,10 +17,13 @@ const fetch_retry_1 = __importDefault(require("fetch-retry"));
 const fetchWithRetries = (0, fetch_retry_1.default)(global.fetch, { retries: 5, retryDelay: 10 });
 exports.STORM_ID = 'storm';
 exports.ConversationIdHeader = 'Conversation-Id';
+exports.SystemIdHeader = 'System-Id';
 class StormClient {
     _baseUrl;
-    constructor() {
+    _systemId;
+    constructor(systemId) {
         this._baseUrl = (0, utils_1.getRemoteUrl)('ai-service', 'https://ai.kapeta.com');
+        this._systemId = systemId || "";
     }
     async createOptions(path, method, body) {
         const url = `${this._baseUrl}${path}`;
@@ -35,6 +38,9 @@ class StormClient {
         if (body.conversationId) {
             headers[exports.ConversationIdHeader] = body.conversationId;
         }
+        if (this._systemId) {
+            headers[exports.SystemIdHeader] = this._systemId;
+        }
         return {
             url,
             method: method,
@@ -103,6 +109,7 @@ class StormClient {
     createUIShells(prompt, conversationId) {
         return this.send('/v2/ui/shells', {
             prompt: JSON.stringify(prompt),
+            conversationId: conversationId,
         });
     }
     createUILandingPages(prompt, conversationId) {
@@ -138,6 +145,9 @@ class StormClient {
         const response = await fetch(u, {
             method: 'POST',
             body: JSON.stringify(prompt.pages),
+            headers: {
+                'systemId': prompt.systemId,
+            },
         });
         return (await response.json());
     }
@@ -252,4 +262,4 @@ class StormClient {
         return response;
     }
 }
-exports.stormClient = new StormClient();
+exports.StormClient = StormClient;

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

@@ -35,6 +35,7 @@ export interface ConversationItem {
 }
 export interface StormContextRequest<T = string> {
     conversationId?: string;
+    systemId?: string;
     history?: ConversationItem[];
     prompt: T;
 }
@@ -78,6 +79,7 @@ export declare enum HTMLPageEncoding {
 }
 export interface ImplementAPIClients {
     pages: HTMLPage[];
+    systemId: string;
 }
 export interface HTMLPage {
     fileName: string;

package/dist/cjs/src/stormService.d.ts

@@ -13,8 +13,8 @@ export declare class StormService {
     saveConversation(conversationId: string, events: StormEvent[]): Promise<void>;
     appendConversation(conversationId: string, events: StormEvent[]): Promise<void>;
     deleteConversation(conversationId: string): Promise<void>;
-    uploadConversation(handle: string, conversationId: string): Promise<void>;
-    installProjectById(handle: string, conversationId: string): Promise<void>;
+    uploadConversation(handle: string, systemId: string): Promise<void>;
+    installProjectById(handle: string, systemId: string): Promise<void>;
 }
 declare const _default: StormService;
 export default _default;

package/dist/cjs/src/stormService.js

@@ -119,8 +119,8 @@ class StormService {
             await promises_1.default.rm(conversationDir, { recursive: true, force: true });
         }
     }
-    async uploadConversation(handle, conversationId) {
-        const tarballFile = this.getConversationTarball(conversationId);
+    async uploadConversation(handle, systemId) {
+        const tarballFile = this.getConversationTarball(systemId);
         const destDir = path_1.default.dirname(tarballFile);
         const tarballName = path_1.default.basename(tarballFile);
         await tar.create({
@@ -129,12 +129,14 @@ class StormService {
             gzip: true,
             filter: (entry) => !entry.includes(tarballName),
         }, ['.']);
-        await stormClient_1.stormClient.uploadSystem(handle, conversationId, await promises_1.default.readFile(tarballFile));
+        const stormClient = new stormClient_1.StormClient(systemId);
+        await stormClient.uploadSystem(handle, systemId, await promises_1.default.readFile(tarballFile));
     }
-    async installProjectById(handle, conversationId) {
-        const tarballFile = this.getConversationTarball(conversationId);
+    async installProjectById(handle, systemId) {
+        const tarballFile = this.getConversationTarball(systemId);
         const destDir = path_1.default.dirname(tarballFile);
-        const buffer = await stormClient_1.stormClient.downloadSystem(handle, conversationId);
+        const stormClient = new stormClient_1.StormClient(systemId);
+        const buffer = await stormClient.downloadSystem(handle, systemId);
         await promises_1.default.mkdir(destDir, { recursive: true });
         await promises_1.default.writeFile(tarballFile, buffer);
         await tar.extract({

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

@@ -240,8 +240,9 @@ class PageQueue extends node_events_1.EventEmitter {
             console.warn('Skipping image reference of type %s for url %s', mimeType, prompt.url);
             return;
         }
+        const client = new stormClient_1.StormClient(this.systemId);
         this.images.set(prompt.url, prompt.description);
-        const result = await stormClient_1.stormClient.createImage(`Create an image for the url "${prompt.url}" with this description: ${prompt.description}`.trim());
+        const result = await client.createImage(`Create an image for the url "${prompt.url}" with this description: ${prompt.description}`.trim());
         let imageEvent = null;
         result.on('data', (event) => {
             if (event.type === 'IMAGE') {
@@ -256,7 +257,8 @@ class PageQueue extends node_events_1.EventEmitter {
         this.emit('image', imageEvent, prompt);
     }
     async generate(prompt, conversationId) {
-        const screenStream = await stormClient_1.stormClient.createUIPage(prompt, conversationId);
+        const client = new stormClient_1.StormClient(this.systemId);
+        const screenStream = await client.createUIPage(prompt, conversationId);
         let pageEvent = null;
         screenStream.on('data', (event) => {
             if (event.type === 'PAGE') {
@@ -273,7 +275,8 @@ class PageQueue extends node_events_1.EventEmitter {
         await this.processPageEventWithReferences(pageEvent);
     }
     async resolveReferences(content) {
-        const referenceStream = await stormClient_1.stormClient.classifyUIReferences(content);
+        const client = new stormClient_1.StormClient(this.systemId);
+        const referenceStream = await client.classifyUIReferences(content);
         const references = [];
         referenceStream.on('data', (referenceData) => {
             if (referenceData.type !== 'REF_CLASSIFICATION') {

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

@@ -311,9 +311,10 @@ class StormCodegen {
             permissions: '0644',
         });
         const uiEvents = [];
+        const stormClient = new stormClient_1.StormClient(this.uiSystemId);
         // generate screens
         if (uiTemplates.length) {
-            const screenStream = await stormClient_1.stormClient.listScreens({
+            const screenStream = await stormClient.listScreens({
                 events: filteredEvents,
                 templates: uiTemplates,
                 context: relevantFiles,
@@ -346,7 +347,7 @@ class StormCodegen {
                     context: relevantFiles.concat([getScreenEventsFile()]),
                     prompt: this.userPrompt,
                 };
-                const uiStream = await stormClient_1.stormClient.createUIImplementation(payload);
+                const uiStream = await stormClient.createUIImplementation(payload);
                 uiStream.on('data', (evt) => {
                     const uiFile = this.handleUiOutput(blockUri, block.aiName, evt);
                     if (uiFile != undefined) {
@@ -374,13 +375,13 @@ class StormCodegen {
         });
         allFiles.push(...screenFilesConverted);
         let webRouters = allFiles.filter((file) => file.type === codegen_1.AIFileTypes.WEB_ROUTER);
-        webRouters = await this.processTemplates(blockUri, block.aiName, stormClient_1.stormClient.generateCode.bind(stormClient_1.stormClient), webRouters, screenFilesConverted.concat([getScreenEventsFile()]));
+        webRouters = await this.processTemplates(blockUri, block.aiName, stormClient.generateCode.bind(stormClient), webRouters, screenFilesConverted.concat([getScreenEventsFile()]));
         // Gather the context files for implementation. These will be all be passed to the AI
         const contextFiles = relevantFiles.filter((file) => ![codegen_1.AIFileTypes.SERVICE, codegen_1.AIFileTypes.WEB_SCREEN, codegen_1.AIFileTypes.WEB_ROUTER].includes(file.type));
         // Send the service and UI templates to the AI. These will be sent one-by-one in addition to the context files
         let serviceFiles = allFiles.filter((file) => file.type === codegen_1.AIFileTypes.SERVICE);
         if (serviceFiles.length > 0) {
-            serviceFiles = await this.processTemplates(blockUri, block.aiName, stormClient_1.stormClient.createServiceImplementation.bind(stormClient_1.stormClient), serviceFiles, contextFiles);
+            serviceFiles = await this.processTemplates(blockUri, block.aiName, stormClient.createServiceImplementation.bind(stormClient), serviceFiles, contextFiles);
         }
         if (this.isAborted()) {
             return;
@@ -510,7 +511,8 @@ class StormCodegen {
         }
     }
     async classifyErrors(errors, basePath) {
-        const errorStream = await stormClient_1.stormClient.createErrorClassification(errors, []);
+        const stormClient = new stormClient_1.StormClient(this.uiSystemId);
+        const errorStream = await stormClient.createErrorClassification(errors, []);
         const fixes = new Map();
         this.out.on('aborted', () => {
             errorStream.abort();
@@ -547,7 +549,8 @@ class StormCodegen {
                 error: error,
                 projectFiles: allFiles.map((f) => f.filename),
             };
-            const detailsStream = await stormClient_1.stormClient.createErrorDetails(JSON.stringify(request), []);
+            const stormClient = new stormClient_1.StormClient(this.uiSystemId);
+            const detailsStream = await stormClient.createErrorDetails(JSON.stringify(request), []);
             detailsStream.on('data', (evt) => {
                 if (evt.type === 'ERROR_DETAILS') {
                     resolve(evt.payload.files);
@@ -611,7 +614,8 @@ class StormCodegen {
     async codeFix(blockUri, blockName, fix, history) {
         return new Promise(async (resolve, reject) => {
             try {
-                const fixStream = await stormClient_1.stormClient.createCodeFix(fix, history, this.conversationId);
+                const stormClient = new stormClient_1.StormClient(this.uiSystemId);
+                const fixStream = await stormClient.createCodeFix(fix, history, this.conversationId);
                 let resolved = false;
                 fixStream.on('data', (evt) => {
                     if (this.handleFileEvents(blockUri, blockName, evt)) {

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

@@ -117,8 +117,10 @@ router.post('/ui/create-system/:handle/:systemId', async (req, res) => {
     res.set(stormClient_1.ConversationIdHeader, systemId);
     sendEvent(res, (0, event_parser_1.createPhaseStartEvent)(events_1.StormEventPhaseType.IMPLEMENT_APIS));
     const pagesFromDisk = (0, utils_1.readFilesAndContent)(srcDir);
-    const pagesWithImplementation = await stormClient_1.stormClient.replaceMockWithAPICall({
+    const client = new stormClient_1.StormClient(systemId);
+    const pagesWithImplementation = await client.replaceMockWithAPICall({
         pages: pagesFromDisk,
+        systemId: systemId,
     });
     await (0, utils_1.copyDirectory)(srcDir, destDir, (fileName, content) => {
         // find the page from result1 and write the content to the file
@@ -131,7 +133,7 @@ router.post('/ui/create-system/:handle/:systemId', async (req, res) => {
     const pageContents = pagesWithImplementation.map((page) => {
         return page.content;
     });
-    const prompt = await stormClient_1.stormClient.generatePrompt(pageContents);
+    const prompt = await client.generatePrompt(pageContents);
     sendEvent(res, (0, event_parser_1.createPhaseEndEvent)(events_1.StormEventPhaseType.COMPOSE_SYSTEM_PROMPT));
     req.query.systemId = systemId;
     const promptRequest = {
@@ -149,10 +151,12 @@ router.post('/ui/create-system-simple/:handle/:systemId', async (req, res) => {
     //res.set('Access-Control-Expose-Headers', ConversationIdHeader);
     //res.set(ConversationIdHeader, systemId);
     //sendEvent(res, createPhaseStartEvent(StormEventPhaseType.IMPLEMENT_APIS));
+    const client = new stormClient_1.StormClient(systemId);
     try {
         const pagesFromDisk = (0, utils_1.readFilesAndContent)(srcDir);
-        const pagesWithImplementation = await stormClient_1.stormClient.replaceMockWithAPICall({
+        const pagesWithImplementation = await client.replaceMockWithAPICall({
             pages: pagesFromDisk,
+            systemId: systemId,
         });
         //sendEvent(res, createPhaseEndEvent(StormEventPhaseType.IMPLEMENT_APIS));
         //sendEvent(res, createPhaseStartEvent(StormEventPhaseType.COMPOSE_SYSTEM));
@@ -165,7 +169,7 @@ router.post('/ui/create-system-simple/:handle/:systemId', async (req, res) => {
             }
             return page;
         });
-        const systemUrl = await stormClient_1.stormClient.createSimpleBackend(handle, systemId, { pages: allFiles });
+        const systemUrl = await client.createSimpleBackend(handle, systemId, { pages: allFiles });
         //sendEvent(res, {type: 'SYSTEM_READY', created: Math.floor(Date.now() / 1000), reason: 'System Ready', payload: { systemUrl: systemUrl }});
         //sendEvent(res, createPhaseEndEvent(StormEventPhaseType.COMPOSE_SYSTEM));
         //sendDone(res);
@@ -250,7 +254,8 @@ router.post('/:handle/ui/iterative', async (req, res) => {
     try {
         const conversationId = req.headers[stormClient_1.ConversationIdHeader.toLowerCase()];
         const aiRequest = JSON.parse(req.stringBody ?? '{}');
-        const landingPagesStream = await stormClient_1.stormClient.createUILandingPages(aiRequest, conversationId);
+        const client = new stormClient_1.StormClient(conversationId); //todo is this correct we are using the landing page getConversationId down below as well
+        const landingPagesStream = await client.createUILandingPages(aiRequest, conversationId);
         onRequestAborted(req, res, () => {
             landingPagesStream.abort();
         });
@@ -339,8 +344,9 @@ router.post('/:handle/ui', async (req, res) => {
     try {
         const outerConversationId = req.headers[stormClient_1.ConversationIdHeader.toLowerCase()] || (0, crypto_1.randomUUID)();
         const aiRequest = JSON.parse(req.stringBody ?? '{}');
+        const stormClient = new stormClient_1.StormClient(outerConversationId);
         // Get user journeys
-        const userJourneysStream = await stormClient_1.stormClient.createUIUserJourneys(aiRequest, outerConversationId);
+        const userJourneysStream = await stormClient.createUIUserJourneys(aiRequest, outerConversationId);
         onRequestAborted(req, res, () => {
             userJourneysStream.abort();
         });
@@ -378,7 +384,7 @@ router.post('/:handle/ui', async (req, res) => {
         });
         let theme = '';
         try {
-            const themeStream = await stormClient_1.stormClient.createTheme(aiRequest, outerConversationId);
+            const themeStream = await stormClient.createTheme(aiRequest, outerConversationId);
             onRequestAborted(req, res, () => {
                 themeStream.abort();
             });
@@ -417,7 +423,7 @@ router.post('/:handle/ui', async (req, res) => {
         }
         await waitForStormStream(userJourneysStream);
         // Get the UI shells
-        const shellsStream = await stormClient_1.stormClient.createUIShells({
+        const shellsStream = await stormClient.createUIShells({
             theme: theme || undefined,
             pages: Object.values(uniqueUserJourneyScreens).map((screen) => ({
                 name: screen.name,
@@ -574,7 +580,8 @@ router.post('/ui/vote', async (req, res) => {
     const aiRequest = JSON.parse(req.stringBody ?? '{}');
     const { topic, vote, mainConversationId } = aiRequest;
     try {
-        await stormClient_1.stormClient.voteUIPage(topic, conversationId, vote, mainConversationId);
+        const stormClient = new stormClient_1.StormClient(mainConversationId);
+        await stormClient.voteUIPage(topic, conversationId, vote, mainConversationId);
     }
     catch (e) {
         res.status(500).send({ error: e.message });
@@ -585,7 +592,8 @@ router.post('/ui/get-vote', async (req, res) => {
     const aiRequest = JSON.parse(req.stringBody ?? '{}');
     const { topic, mainConversationId } = aiRequest;
     try {
-        const vote = await stormClient_1.stormClient.getVoteUIPage(topic, conversationId, mainConversationId);
+        const stormClient = new stormClient_1.StormClient(mainConversationId);
+        const vote = await stormClient.getVoteUIPage(topic, conversationId, mainConversationId);
         res.send({ vote });
     }
     catch (e) {
@@ -609,7 +617,8 @@ async function handleAll(req, res) {
         const eventParser = new event_parser_1.StormEventParser(stormOptions);
         const conversationId = req.headers[stormClient_1.ConversationIdHeader.toLowerCase()];
         const aiRequest = JSON.parse(req.stringBody ?? '{}');
-        const metaStream = await stormClient_1.stormClient.createMetadata(aiRequest, conversationId);
+        const stormClient = new stormClient_1.StormClient(systemId);
+        const metaStream = await stormClient.createMetadata(aiRequest, conversationId);
         onRequestAborted(req, res, () => {
             metaStream.abort();
         });

package/dist/esm/src/storm/stormClient.d.ts

@@ -4,6 +4,7 @@ import { ConversationItem, CreateSimpleBackendRequest, HTMLPage, ImplementAPICli
 import { Page, StormEventPageUrl } from './events';
 export declare const STORM_ID = "storm";
 export declare const ConversationIdHeader = "Conversation-Id";
+export declare const SystemIdHeader = "System-Id";
 export interface UIShellsPrompt {
     theme?: string;
     pages: {
@@ -55,9 +56,10 @@ export interface BasePromptRequest {
     prompt: string;
     skipImprovement?: boolean;
 }
-declare class StormClient {
+export declare class StormClient {
     private readonly _baseUrl;
-    constructor();
+    private readonly _systemId;
+    constructor(systemId?: string);
     private createOptions;
     private send;
     createMetadata(prompt: BasePromptRequest, conversationId?: string): Promise<StormStream>;
@@ -88,5 +90,3 @@ declare class StormClient {
     downloadSystem(handle: string, conversationId: string): Promise<Buffer>;
     uploadSystem(handle: string, conversationId: string, buffer: Buffer): Promise<Response>;
 }
-export declare const stormClient: StormClient;
-export {};

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

@@ -3,7 +3,7 @@ var __importDefault = (this && this.__importDefault) || function (mod) {
     return (mod && mod.__esModule) ? mod : { "default": mod };
 };
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.stormClient = exports.ConversationIdHeader = exports.STORM_ID = void 0;
+exports.StormClient = exports.SystemIdHeader = exports.ConversationIdHeader = exports.STORM_ID = void 0;
 /**
  * Copyright 2023 Kapeta Inc.
  * SPDX-License-Identifier: BUSL-1.1
@@ -17,10 +17,13 @@ const fetch_retry_1 = __importDefault(require("fetch-retry"));
 const fetchWithRetries = (0, fetch_retry_1.default)(global.fetch, { retries: 5, retryDelay: 10 });
 exports.STORM_ID = 'storm';
 exports.ConversationIdHeader = 'Conversation-Id';
+exports.SystemIdHeader = 'System-Id';
 class StormClient {
     _baseUrl;
-    constructor() {
+    _systemId;
+    constructor(systemId) {
         this._baseUrl = (0, utils_1.getRemoteUrl)('ai-service', 'https://ai.kapeta.com');
+        this._systemId = systemId || "";
     }
     async createOptions(path, method, body) {
         const url = `${this._baseUrl}${path}`;
@@ -35,6 +38,9 @@ class StormClient {
         if (body.conversationId) {
             headers[exports.ConversationIdHeader] = body.conversationId;
         }
+        if (this._systemId) {
+            headers[exports.SystemIdHeader] = this._systemId;
+        }
         return {
             url,
             method: method,
@@ -103,6 +109,7 @@ class StormClient {
     createUIShells(prompt, conversationId) {
         return this.send('/v2/ui/shells', {
             prompt: JSON.stringify(prompt),
+            conversationId: conversationId,
         });
     }
     createUILandingPages(prompt, conversationId) {
@@ -138,6 +145,9 @@ class StormClient {
         const response = await fetch(u, {
             method: 'POST',
             body: JSON.stringify(prompt.pages),
+            headers: {
+                'systemId': prompt.systemId,
+            },
         });
         return (await response.json());
     }
@@ -252,4 +262,4 @@ class StormClient {
         return response;
     }
 }
-exports.stormClient = new StormClient();
+exports.StormClient = StormClient;

package/dist/esm/src/storm/stream.d.ts

@@ -35,6 +35,7 @@ export interface ConversationItem {
 }
 export interface StormContextRequest<T = string> {
     conversationId?: string;
+    systemId?: string;
     history?: ConversationItem[];
     prompt: T;
 }
@@ -78,6 +79,7 @@ export declare enum HTMLPageEncoding {
 }
 export interface ImplementAPIClients {
     pages: HTMLPage[];
+    systemId: string;
 }
 export interface HTMLPage {
     fileName: string;

package/dist/esm/src/stormService.d.ts

@@ -13,8 +13,8 @@ export declare class StormService {
     saveConversation(conversationId: string, events: StormEvent[]): Promise<void>;
     appendConversation(conversationId: string, events: StormEvent[]): Promise<void>;
     deleteConversation(conversationId: string): Promise<void>;
-    uploadConversation(handle: string, conversationId: string): Promise<void>;
-    installProjectById(handle: string, conversationId: string): Promise<void>;
+    uploadConversation(handle: string, systemId: string): Promise<void>;
+    installProjectById(handle: string, systemId: string): Promise<void>;
 }
 declare const _default: StormService;
 export default _default;

package/dist/esm/src/stormService.js

@@ -119,8 +119,8 @@ class StormService {
             await promises_1.default.rm(conversationDir, { recursive: true, force: true });
         }
     }
-    async uploadConversation(handle, conversationId) {
-        const tarballFile = this.getConversationTarball(conversationId);
+    async uploadConversation(handle, systemId) {
+        const tarballFile = this.getConversationTarball(systemId);
         const destDir = path_1.default.dirname(tarballFile);
         const tarballName = path_1.default.basename(tarballFile);
         await tar.create({
@@ -129,12 +129,14 @@ class StormService {
             gzip: true,
             filter: (entry) => !entry.includes(tarballName),
         }, ['.']);
-        await stormClient_1.stormClient.uploadSystem(handle, conversationId, await promises_1.default.readFile(tarballFile));
+        const stormClient = new stormClient_1.StormClient(systemId);
+        await stormClient.uploadSystem(handle, systemId, await promises_1.default.readFile(tarballFile));
     }
-    async installProjectById(handle, conversationId) {
-        const tarballFile = this.getConversationTarball(conversationId);
+    async installProjectById(handle, systemId) {
+        const tarballFile = this.getConversationTarball(systemId);
         const destDir = path_1.default.dirname(tarballFile);
-        const buffer = await stormClient_1.stormClient.downloadSystem(handle, conversationId);
+        const stormClient = new stormClient_1.StormClient(systemId);
+        const buffer = await stormClient.downloadSystem(handle, systemId);
         await promises_1.default.mkdir(destDir, { recursive: true });
         await promises_1.default.writeFile(tarballFile, buffer);
         await tar.extract({

package/package.json

@@ -1,6 +1,6 @@
 {
     "name": "@kapeta/local-cluster-service",
-    "version": "0.74.0",
+    "version": "0.74.1",
     "description": "Manages configuration, ports and service discovery for locally running Kapeta systems",
     "type": "commonjs",
     "exports": {

package/src/storm/PageGenerator.ts

@@ -4,7 +4,7 @@
  */
 
 import uuid from 'node-uuid';
-import { stormClient, UIPagePrompt } from './stormClient';
+import { StormClient, UIPagePrompt } from './stormClient';
 import { ReferenceClassification, StormEvent, StormEventPage, StormImage, UIShell } from './events';
 import { EventEmitter } from 'node:events';
 import PQueue from 'p-queue';
@@ -265,8 +265,9 @@ export class PageQueue extends EventEmitter {
             return;
         }
 
+        const client = new StormClient(this.systemId);
         this.images.set(prompt.url, prompt.description);
-        const result = await stormClient.createImage(
+        const result = await client.createImage(
             `Create an image for the url "${prompt.url}" with this description: ${prompt.description}`.trim()
         );
 
@@ -287,7 +288,8 @@ export class PageQueue extends EventEmitter {
     }
 
     public async generate(prompt: UIPagePrompt, conversationId: string) {
-        const screenStream = await stormClient.createUIPage(prompt, conversationId);
+        const client = new StormClient(this.systemId);
+        const screenStream = await client.createUIPage(prompt, conversationId);
         let pageEvent: StormEventPage | null = null;
         screenStream.on('data', (event: StormEvent) => {
             if (event.type === 'PAGE') {
@@ -306,7 +308,8 @@ export class PageQueue extends EventEmitter {
     }
 
     private async resolveReferences(content: string): Promise<ReferenceClassification[]> {
-        const referenceStream = await stormClient.classifyUIReferences(content);
+        const client = new StormClient(this.systemId);
+        const referenceStream = await client.classifyUIReferences(content);
 
         const references: ReferenceClassification[] = [];
 

package/src/storm/codegen.ts

@@ -17,7 +17,7 @@ import {
 
 import { BlockDefinition } from '@kapeta/schemas';
 import { codeGeneratorManager } from '../codeGeneratorManager';
-import { STORM_ID, stormClient } from './stormClient';
+import { STORM_ID, StormClient } from './stormClient';
 import {
     StormEvent,
     StormEventBlockStatusType,
@@ -371,6 +371,7 @@ export class StormCodegen {
         });
         const uiEvents = [];
 
+        const stormClient = new StormClient(this.uiSystemId);
         // generate screens
         if (uiTemplates.length) {
             const screenStream = await stormClient.listScreens({
@@ -663,6 +664,7 @@ export class StormCodegen {
     }
 
     private async classifyErrors(errors: string, basePath: string): Promise<Map<string, ErrorClassification[]>> {
+        const stormClient = new StormClient(this.uiSystemId);
         const errorStream = await stormClient.createErrorClassification(errors, []);
         const fixes = new Map<string, ErrorClassification[]>();
 
@@ -711,7 +713,7 @@ export class StormCodegen {
                 error: error,
                 projectFiles: allFiles.map((f) => f.filename),
             };
-
+            const stormClient = new StormClient(this.uiSystemId);
             const detailsStream = await stormClient.createErrorDetails(JSON.stringify(request), []);
             detailsStream.on('data', (evt) => {
                 if (evt.type === 'ERROR_DETAILS') {
@@ -799,6 +801,7 @@ export class StormCodegen {
     ): Promise<StormEventErrorDetailsFile> {
         return new Promise<StormEventErrorDetailsFile>(async (resolve, reject) => {
             try {
+                const stormClient = new StormClient(this.uiSystemId);
                 const fixStream = await stormClient.createCodeFix(fix, history, this.conversationId);
                 let resolved = false;
                 fixStream.on('data', (evt) => {

package/src/storm/routes.ts

@@ -21,12 +21,12 @@ import {
 
 import {
     ConversationIdHeader,
-    stormClient,
     UIPagePrompt,
     UIPageEditRequest,
     BasePromptRequest,
     UIPageVoteRequest,
     UIPageGetVoteRequest,
+    StormClient,
 } from './stormClient';
 import { StormEvent, StormEventPage, StormEventPhaseType, UserJourneyScreen } from './events';
 
@@ -162,8 +162,10 @@ router.post('/ui/create-system/:handle/:systemId', async (req: KapetaBodyRequest
     sendEvent(res, createPhaseStartEvent(StormEventPhaseType.IMPLEMENT_APIS));
 
     const pagesFromDisk = readFilesAndContent(srcDir);
-    const pagesWithImplementation = await stormClient.replaceMockWithAPICall({
+    const client = new StormClient(systemId)
+    const pagesWithImplementation = await client.replaceMockWithAPICall({
         pages: pagesFromDisk,
+        systemId: systemId,
     });
 
     await copyDirectory(srcDir, destDir, (fileName, content) => {
@@ -181,7 +183,7 @@ router.post('/ui/create-system/:handle/:systemId', async (req: KapetaBodyRequest
         return page.content;
     });
 
-    const prompt = await stormClient.generatePrompt(pageContents);
+    const prompt = await client.generatePrompt(pageContents);
 
     sendEvent(res, createPhaseEndEvent(StormEventPhaseType.COMPOSE_SYSTEM_PROMPT));
 
@@ -204,11 +206,12 @@ router.post('/ui/create-system-simple/:handle/:systemId', async (req: KapetaBody
     //res.set(ConversationIdHeader, systemId);
 
     //sendEvent(res, createPhaseStartEvent(StormEventPhaseType.IMPLEMENT_APIS));
-
+    const client = new StormClient(systemId);
     try {
         const pagesFromDisk = readFilesAndContent(srcDir);
-        const pagesWithImplementation = await stormClient.replaceMockWithAPICall({
+        const pagesWithImplementation = await client.replaceMockWithAPICall({
             pages: pagesFromDisk,
+            systemId: systemId,
         });
 
         //sendEvent(res, createPhaseEndEvent(StormEventPhaseType.IMPLEMENT_APIS));
@@ -227,7 +230,7 @@ router.post('/ui/create-system-simple/:handle/:systemId', async (req: KapetaBody
             return page;
         });
 
-        const systemUrl = await stormClient.createSimpleBackend(handle, systemId, { pages: allFiles });
+        const systemUrl = await client.createSimpleBackend(handle, systemId, { pages: allFiles });
         //sendEvent(res, {type: 'SYSTEM_READY', created: Math.floor(Date.now() / 1000), reason: 'System Ready', payload: { systemUrl: systemUrl }});
 
         //sendEvent(res, createPhaseEndEvent(StormEventPhaseType.COMPOSE_SYSTEM));
@@ -328,8 +331,8 @@ router.post('/:handle/ui/iterative', async (req: KapetaBodyRequest, res: Respons
         const conversationId = req.headers[ConversationIdHeader.toLowerCase()] as string | undefined;
 
         const aiRequest: BasePromptRequest = JSON.parse(req.stringBody ?? '{}');
-
-        const landingPagesStream = await stormClient.createUILandingPages(aiRequest, conversationId);
+        const client = new StormClient(conversationId); //todo is this correct we are using the landing page getConversationId down below as well
+        const landingPagesStream = await client.createUILandingPages(aiRequest, conversationId);
 
         onRequestAborted(req, res, () => {
             landingPagesStream.abort();
@@ -437,7 +440,7 @@ router.post('/:handle/ui', async (req: KapetaBodyRequest, res: Response) => {
             (req.headers[ConversationIdHeader.toLowerCase()] as string | undefined) || randomUUID();
 
         const aiRequest: BasePromptRequest = JSON.parse(req.stringBody ?? '{}');
-
+        const stormClient = new StormClient(outerConversationId);
         // Get user journeys
         const userJourneysStream = await stormClient.createUIUserJourneys(aiRequest, outerConversationId);
 
@@ -719,6 +722,7 @@ router.post('/ui/vote', async (req: KapetaBodyRequest, res: Response) => {
     const aiRequest: UIPageVoteRequest = JSON.parse(req.stringBody ?? '{}');
     const { topic, vote, mainConversationId } = aiRequest;
     try {
+        const stormClient = new StormClient(mainConversationId);
         await stormClient.voteUIPage(topic, conversationId, vote, mainConversationId);
     } catch (e: any) {
         res.status(500).send({ error: e.message });
@@ -730,6 +734,7 @@ router.post('/ui/get-vote', async (req: KapetaBodyRequest, res: Response) => {
     const aiRequest: UIPageGetVoteRequest = JSON.parse(req.stringBody ?? '{}');
     const { topic, mainConversationId } = aiRequest;
     try {
+        const stormClient = new StormClient(mainConversationId);
         const vote = await stormClient.getVoteUIPage(topic, conversationId, mainConversationId);
         res.send({ vote });
     } catch (e: any) {
@@ -761,6 +766,7 @@ async function handleAll(req: KapetaBodyRequest, res: Response) {
         const conversationId = req.headers[ConversationIdHeader.toLowerCase()] as string | undefined;
 
         const aiRequest: BasePromptRequest = JSON.parse(req.stringBody ?? '{}');
+        const stormClient = new StormClient(systemId);
         const metaStream = await stormClient.createMetadata(aiRequest, conversationId);
 
         onRequestAborted(req, res, () => {

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({