@kapeta/local-cluster-service 0.73.0 → 0.74.1

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

@@ -7,13 +7,14 @@ export declare class StormService {
         id: string;
         description: string;
         title: string;
+        url?: string | undefined;
     }[]>;
     getConversation(conversationId: string): Promise<string>;
     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

@@ -54,22 +54,43 @@ class StormService {
         // Returns list of UUIDs - probably want to make it more useful than that
         const conversations = [];
         for (const file of eventFiles) {
-            const nldContents = await promises_1.default.readFile(file, 'utf8');
-            const events = nldContents.split('\n').map((e) => JSON.parse(e));
-            // find the shell and get the title tag
-            const shellEvent = events.find((e) => e.type === 'AI' && e.event.type === 'UI_SHELL')?.event;
-            const html = shellEvent?.payload.content;
-            const title = html?.match(/<title>(.*?)<\/title>/)?.[1];
-            const id = events.find((e) => e.type === 'AI')?.systemId;
-            const initialPrompt = events.find((e) => e.type === 'AI' && e.event.type === 'PROMPT_IMPROVE')?.event?.payload?.prompt || events[0].text;
-            if (!id) {
-                continue;
+            try {
+                const nldContents = await promises_1.default.readFile(file, 'utf8');
+                const events = nldContents.split('\n').map((e) => JSON.parse(e));
+                // find the shell and get the title tag
+                const shellEvent = events.find((e) => e.type === 'AI' && e.event.type === 'UI_SHELL')?.event;
+                const html = shellEvent?.payload.content;
+                const title = html?.match(/<title>(.*?)<\/title>/)?.[1];
+                const id = events.find((e) => e.type === 'AI')?.systemId;
+                const initialPrompt = events.find((e) => e.type === 'AI' && e.event.type === 'PROMPT_IMPROVE')?.event?.payload?.prompt || events[0].text;
+                if (!id) {
+                    continue;
+                }
+                let url = undefined;
+                // Find the last model response event that has a URL in the payload (in case it changed over time)
+                for (const evt of [...events].reverse()) {
+                    const event = evt.event;
+                    if (evt.type === 'AI' && event.type === 'MODEL_RESPONSE') {
+                        // Look for a URL in the model response markdown
+                        const regex = /\[(.*?)\]\((.*?)\)/g;
+                        const match = regex.exec(event.payload.text);
+                        const [, _linkText, linkUrl] = match || [];
+                        if (linkUrl?.startsWith('http')) {
+                            url = linkUrl;
+                            break;
+                        }
+                    }
+                }
+                conversations.push({
+                    id,
+                    description: initialPrompt,
+                    title: title || 'New system',
+                    url,
+                });
+            }
+            catch (e) {
+                console.error('Failed to load conversation at %s', file, e);
             }
-            conversations.push({
-                id,
-                description: initialPrompt,
-                title: title || 'New system',
-            });
         }
         return conversations;
     }
@@ -98,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({
@@ -108,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.73.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/events.ts

@@ -480,7 +480,16 @@ export interface StormEventSystemReady {
     created: number;
     payload: {
         systemUrl: string;
-    }
+    };
+}
+
+export interface StormEventModelResponse {
+    type: 'MODEL_RESPONSE';
+    reason: string;
+    created: number;
+    payload: {
+        text: string;
+    };
 }
 
 export type StormEvent =
@@ -518,4 +527,5 @@ export type StormEvent =
     | StormEventApiBase
     | StormEventUIStarted
     | StormImage
-    | StormEventSystemReady;
+    | StormEventSystemReady
+    | StormEventModelResponse;

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,9 +162,12 @@ 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) => {
         // find the page from result1 and write the content to the file
         const page = pagesWithImplementation.find((p) => p.fileName === fileName);
@@ -180,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));
 
@@ -203,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));
@@ -226,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));
@@ -327,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();
@@ -436,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);
 
@@ -718,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 });
@@ -729,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) {
@@ -760,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 {