libre-webui 0.21.3 → 0.22.1

package/backend/dist/utils/ttsVoiceCloneUpload.js

@@ -0,0 +1,224 @@
+/*
+ * Libre WebUI
+ * Copyright (C) 2025 Kroonen AI, Inc.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+import multer from 'multer';
+import { acquireSharedCapacity, SharedCapacityExceededError, } from '../platform/coordination/sharedAdmission.js';
+export const TTS_VOICE_CLONE_GLOBAL_MAX_AUDIO_BYTES = 10 * 1024 * 1024;
+export const TTS_VOICE_CLONE_MAX_ACTIVE_PER_USER = 6;
+export const TTS_VOICE_CLONE_MAX_ACTIVE_GLOBAL = 16;
+export class TTSVoiceCloneConcurrencyError extends Error {
+    constructor() {
+        super('Too many concurrent voice-cloning requests');
+        this.name = 'TTSVoiceCloneConcurrencyError';
+    }
+}
+export async function reserveTTSVoiceCloneUpload(userId) {
+    try {
+        return await acquireSharedCapacity({
+            limits: [
+                {
+                    scope: 'voice-clone-upload.global',
+                    capacity: TTS_VOICE_CLONE_MAX_ACTIVE_GLOBAL,
+                },
+                {
+                    scope: 'voice-clone-upload.user',
+                    subject: userId,
+                    capacity: TTS_VOICE_CLONE_MAX_ACTIVE_PER_USER,
+                },
+            ],
+        });
+    }
+    catch (error) {
+        if (error instanceof SharedCapacityExceededError) {
+            throw new TTSVoiceCloneConcurrencyError();
+        }
+        throw error;
+    }
+}
+export const TTS_VOICE_CLONE_DEFAULT_AUDIO_MIME_TYPES = [
+    'audio/wav',
+    'audio/x-wav',
+    'audio/wave',
+    'audio/vnd.wave',
+    'audio/mpeg',
+    'audio/mp3',
+    'audio/flac',
+    'audio/x-flac',
+    'audio/ogg',
+    'application/ogg',
+    'audio/mp4',
+    'audio/m4a',
+    'audio/x-m4a',
+];
+export class TTSVoiceCloneUploadError extends Error {
+    constructor(message, code) {
+        super(message);
+        this.code = code;
+        this.name = 'TTSVoiceCloneUploadError';
+    }
+}
+const normalizeMimeType = (mimeType) => mimeType.split(';', 1)[0].trim().toLowerCase();
+const canonicalFormatForMimeType = (mimeType) => {
+    switch (normalizeMimeType(mimeType)) {
+        case 'audio/wav':
+        case 'audio/x-wav':
+        case 'audio/wave':
+        case 'audio/vnd.wave':
+            return 'wav';
+        case 'audio/mpeg':
+        case 'audio/mp3':
+            return 'mp3';
+        case 'audio/flac':
+        case 'audio/x-flac':
+            return 'flac';
+        case 'audio/ogg':
+        case 'application/ogg':
+            return 'ogg';
+        case 'audio/mp4':
+        case 'audio/m4a':
+        case 'audio/x-m4a':
+            return 'm4a';
+        default:
+            return null;
+    }
+};
+const canonicalMimeTypeForFormat = (format) => ({
+    wav: 'audio/wav',
+    mp3: 'audio/mpeg',
+    flac: 'audio/flac',
+    ogg: 'audio/ogg',
+    m4a: 'audio/mp4',
+})[format];
+const detectAudioFormat = (buffer) => {
+    if (buffer.length >= 12 &&
+        buffer.subarray(0, 4).toString('ascii') === 'RIFF' &&
+        buffer.subarray(8, 12).toString('ascii') === 'WAVE') {
+        return 'wav';
+    }
+    if (buffer.length >= 3 &&
+        (buffer.subarray(0, 3).toString('ascii') === 'ID3' ||
+            (buffer[0] === 0xff && (buffer[1] & 0xe0) === 0xe0))) {
+        return 'mp3';
+    }
+    if (buffer.length >= 4 &&
+        buffer.subarray(0, 4).toString('ascii') === 'fLaC') {
+        return 'flac';
+    }
+    if (buffer.length >= 4 &&
+        buffer.subarray(0, 4).toString('ascii') === 'OggS') {
+        return 'ogg';
+    }
+    if (buffer.length >= 12 &&
+        buffer.subarray(4, 8).toString('ascii') === 'ftyp') {
+        return 'm4a';
+    }
+    return null;
+};
+export function getTTSVoiceCloneMaxAudioBytes(config) {
+    const configuredLimit = config?.clone_max_audio_bytes;
+    if (typeof configuredLimit !== 'number' ||
+        !Number.isFinite(configuredLimit) ||
+        configuredLimit <= 0) {
+        return TTS_VOICE_CLONE_GLOBAL_MAX_AUDIO_BYTES;
+    }
+    return Math.min(Math.floor(configuredLimit), TTS_VOICE_CLONE_GLOBAL_MAX_AUDIO_BYTES);
+}
+export function validateTTSVoiceCloneAudio(file, config) {
+    if (!file) {
+        throw new TTSVoiceCloneUploadError('Reference audio is required', 'missing_file');
+    }
+    const size = file.buffer.length;
+    const maxBytes = getTTSVoiceCloneMaxAudioBytes(config);
+    if (size === 0) {
+        throw new TTSVoiceCloneUploadError('Reference audio cannot be empty', 'signature_mismatch');
+    }
+    if (size > maxBytes) {
+        throw new TTSVoiceCloneUploadError(`Reference audio exceeds the maximum size of ${maxBytes} bytes`, 'file_too_large');
+    }
+    const mimeType = normalizeMimeType(file.mimetype);
+    const configuredMimeTypes = config?.clone_audio_mime_types?.length
+        ? config.clone_audio_mime_types.map(normalizeMimeType)
+        : [...TTS_VOICE_CLONE_DEFAULT_AUDIO_MIME_TYPES];
+    const declaredFormat = canonicalFormatForMimeType(mimeType);
+    const configuredFormats = new Set(configuredMimeTypes
+        .map(canonicalFormatForMimeType)
+        .filter((format) => format !== null));
+    if (!declaredFormat || !configuredFormats.has(declaredFormat)) {
+        throw new TTSVoiceCloneUploadError(`Unsupported reference audio MIME type: ${mimeType || 'unknown'}`, 'unsupported_mime_type');
+    }
+    const detectedFormat = detectAudioFormat(file.buffer);
+    if (!detectedFormat || detectedFormat !== declaredFormat) {
+        throw new TTSVoiceCloneUploadError('Reference audio content does not match its declared MIME type', 'signature_mismatch');
+    }
+    return {
+        buffer: file.buffer,
+        originalname: file.originalname,
+        mimetype: canonicalMimeTypeForFormat(detectedFormat),
+        format: detectedFormat,
+        size,
+    };
+}
+const voiceCloneUpload = multer({
+    storage: multer.memoryStorage(),
+    limits: {
+        fileSize: TTS_VOICE_CLONE_GLOBAL_MAX_AUDIO_BYTES,
+        files: 1,
+        fields: 8,
+        parts: 9,
+        fieldNameSize: 100,
+        fieldSize: 128 * 1024,
+    },
+});
+/** Parse one in-memory `reference_audio` file without persisting it. */
+export function parseTTSVoiceCloneUpload(req, res, signal) {
+    return new Promise((resolve, reject) => {
+        if (signal?.aborted) {
+            reject(signal.reason);
+            return;
+        }
+        let settled = false;
+        const finish = (error) => {
+            if (settled)
+                return;
+            settled = true;
+            signal?.removeEventListener('abort', abortUpload);
+            if (error)
+                reject(error);
+            else
+                resolve();
+        };
+        const abortUpload = () => {
+            const reason = signal?.reason instanceof Error
+                ? signal.reason
+                : new Error('Voice-clone upload admission was lost');
+            // Reject before waiting for a slow sender to finish. Multer receives the
+            // request error in the same turn and destroys its partial Busboy upload;
+            // the settled guard ignores Multer's later callback.
+            finish(reason);
+            if (!req.destroyed)
+                req.emit('error', reason);
+        };
+        voiceCloneUpload.single('reference_audio')(req, res, error => {
+            finish(error);
+        });
+        if (!settled) {
+            signal?.addEventListener('abort', abortUpload, { once: true });
+            if (signal?.aborted)
+                abortUpload();
+        }
+    });
+}
+//# sourceMappingURL=ttsVoiceCloneUpload.js.map

package/backend/dist/utils/ttsVoiceCloneUpload.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"ttsVoiceCloneUpload.js","sourceRoot":"","sources":["../../src/utils/ttsVoiceCloneUpload.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;;GAeG;AAGH,OAAO,MAAM,MAAM,QAAQ,CAAC;AAC5B,OAAO,EACL,qBAAqB,EACrB,2BAA2B,GAE5B,MAAM,6CAA6C,CAAC;AAGrD,MAAM,CAAC,MAAM,sCAAsC,GAAG,EAAE,GAAG,IAAI,GAAG,IAAI,CAAC;AACvE,MAAM,CAAC,MAAM,mCAAmC,GAAG,CAAC,CAAC;AACrD,MAAM,CAAC,MAAM,iCAAiC,GAAG,EAAE,CAAC;AAEpD,MAAM,OAAO,6BAA8B,SAAQ,KAAK;IACtD;QACE,KAAK,CAAC,4CAA4C,CAAC,CAAC;QACpD,IAAI,CAAC,IAAI,GAAG,+BAA+B,CAAC;IAC9C,CAAC;CACF;AAED,MAAM,CAAC,KAAK,UAAU,0BAA0B,CAC9C,MAAc;IAEd,IAAI,CAAC;QACH,OAAO,MAAM,qBAAqB,CAAC;YACjC,MAAM,EAAE;gBACN;oBACE,KAAK,EAAE,2BAA2B;oBAClC,QAAQ,EAAE,iCAAiC;iBAC5C;gBACD;oBACE,KAAK,EAAE,yBAAyB;oBAChC,OAAO,EAAE,MAAM;oBACf,QAAQ,EAAE,mCAAmC;iBAC9C;aACF;SACF,CAAC,CAAC;IACL,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,IAAI,KAAK,YAAY,2BAA2B,EAAE,CAAC;YACjD,MAAM,IAAI,6BAA6B,EAAE,CAAC;QAC5C,CAAC;QACD,MAAM,KAAK,CAAC;IACd,CAAC;AACH,CAAC;AAED,MAAM,CAAC,MAAM,wCAAwC,GAAG;IACtD,WAAW;IACX,aAAa;IACb,YAAY;IACZ,gBAAgB;IAChB,YAAY;IACZ,WAAW;IACX,YAAY;IACZ,cAAc;IACd,WAAW;IACX,iBAAiB;IACjB,WAAW;IACX,WAAW;IACX,aAAa;CACL,CAAC;AAuBX,MAAM,OAAO,wBAAyB,SAAQ,KAAK;IACjD,YACE,OAAe,EACN,IAAkC;QAE3C,KAAK,CAAC,OAAO,CAAC,CAAC;oBAFN,IAAI;QAGb,IAAI,CAAC,IAAI,GAAG,0BAA0B,CAAC;IACzC,CAAC;CACF;AAED,MAAM,iBAAiB,GAAG,CAAC,QAAgB,EAAU,EAAE,CACrD,QAAQ,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC,WAAW,EAAE,CAAC;AAEjD,MAAM,0BAA0B,GAAG,CACjC,QAAgB,EAC8B,EAAE;IAChD,QAAQ,iBAAiB,CAAC,QAAQ,CAAC,EAAE,CAAC;QACpC,KAAK,WAAW,CAAC;QACjB,KAAK,aAAa,CAAC;QACnB,KAAK,YAAY,CAAC;QAClB,KAAK,gBAAgB;YACnB,OAAO,KAAK,CAAC;QACf,KAAK,YAAY,CAAC;QAClB,KAAK,WAAW;YACd,OAAO,KAAK,CAAC;QACf,KAAK,YAAY,CAAC;QAClB,KAAK,cAAc;YACjB,OAAO,MAAM,CAAC;QAChB,KAAK,WAAW,CAAC;QACjB,KAAK,iBAAiB;YACpB,OAAO,KAAK,CAAC;QACf,KAAK,WAAW,CAAC;QACjB,KAAK,WAAW,CAAC;QACjB,KAAK,aAAa;YAChB,OAAO,KAAK,CAAC;QACf;YACE,OAAO,IAAI,CAAC;IAChB,CAAC;AACH,CAAC,CAAC;AAEF,MAAM,0BAA0B,GAAG,CACjC,MAA6C,EACrC,EAAE,CACV,CAAC;IACC,GAAG,EAAE,WAAW;IAChB,GAAG,EAAE,YAAY;IACjB,IAAI,EAAE,YAAY;IAClB,GAAG,EAAE,WAAW;IAChB,GAAG,EAAE,WAAW;CACjB,CAAC,CAAC,MAAM,CAAC,CAAC;AAEb,MAAM,iBAAiB,GAAG,CACxB,MAAc,EACgC,EAAE;IAChD,IACE,MAAM,CAAC,MAAM,IAAI,EAAE;QACnB,MAAM,CAAC,QAAQ,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,QAAQ,CAAC,OAAO,CAAC,KAAK,MAAM;QAClD,MAAM,CAAC,QAAQ,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,QAAQ,CAAC,OAAO,CAAC,KAAK,MAAM,EACnD,CAAC;QACD,OAAO,KAAK,CAAC;IACf,CAAC;IACD,IACE,MAAM,CAAC,MAAM,IAAI,CAAC;QAClB,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,QAAQ,CAAC,OAAO,CAAC,KAAK,KAAK;YAChD,CAAC,MAAM,CAAC,CAAC,CAAC,KAAK,IAAI,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,GAAG,IAAI,CAAC,KAAK,IAAI,CAAC,CAAC,EACtD,CAAC;QACD,OAAO,KAAK,CAAC;IACf,CAAC;IACD,IACE,MAAM,CAAC,MAAM,IAAI,CAAC;QAClB,MAAM,CAAC,QAAQ,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,QAAQ,CAAC,OAAO,CAAC,KAAK,MAAM,EAClD,CAAC;QACD,OAAO,MAAM,CAAC;IAChB,CAAC;IACD,IACE,MAAM,CAAC,MAAM,IAAI,CAAC;QAClB,MAAM,CAAC,QAAQ,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,QAAQ,CAAC,OAAO,CAAC,KAAK,MAAM,EAClD,CAAC;QACD,OAAO,KAAK,CAAC;IACf,CAAC;IACD,IACE,MAAM,CAAC,MAAM,IAAI,EAAE;QACnB,MAAM,CAAC,QAAQ,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,QAAQ,CAAC,OAAO,CAAC,KAAK,MAAM,EAClD,CAAC;QACD,OAAO,KAAK,CAAC;IACf,CAAC;IACD,OAAO,IAAI,CAAC;AACd,CAAC,CAAC;AAEF,MAAM,UAAU,6BAA6B,CAAC,MAAkB;IAC9D,MAAM,eAAe,GAAG,MAAM,EAAE,qBAAqB,CAAC;IACtD,IACE,OAAO,eAAe,KAAK,QAAQ;QACnC,CAAC,MAAM,CAAC,QAAQ,CAAC,eAAe,CAAC;QACjC,eAAe,IAAI,CAAC,EACpB,CAAC;QACD,OAAO,sCAAsC,CAAC;IAChD,CAAC;IACD,OAAO,IAAI,CAAC,GAAG,CACb,IAAI,CAAC,KAAK,CAAC,eAAe,CAAC,EAC3B,sCAAsC,CACvC,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,0BAA0B,CACxC,IAAwC,EACxC,MAAkB;IAElB,IAAI,CAAC,IAAI,EAAE,CAAC;QACV,MAAM,IAAI,wBAAwB,CAChC,6BAA6B,EAC7B,cAAc,CACf,CAAC;IACJ,CAAC;IAED,MAAM,IAAI,GAAG,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC;IAChC,MAAM,QAAQ,GAAG,6BAA6B,CAAC,MAAM,CAAC,CAAC;IACvD,IAAI,IAAI,KAAK,CAAC,EAAE,CAAC;QACf,MAAM,IAAI,wBAAwB,CAChC,iCAAiC,EACjC,oBAAoB,CACrB,CAAC;IACJ,CAAC;IACD,IAAI,IAAI,GAAG,QAAQ,EAAE,CAAC;QACpB,MAAM,IAAI,wBAAwB,CAChC,+CAA+C,QAAQ,QAAQ,EAC/D,gBAAgB,CACjB,CAAC;IACJ,CAAC;IAED,MAAM,QAAQ,GAAG,iBAAiB,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;IAClD,MAAM,mBAAmB,GAAG,MAAM,EAAE,sBAAsB,EAAE,MAAM;QAChE,CAAC,CAAC,MAAM,CAAC,sBAAsB,CAAC,GAAG,CAAC,iBAAiB,CAAC;QACtD,CAAC,CAAC,CAAC,GAAG,wCAAwC,CAAC,CAAC;IAClD,MAAM,cAAc,GAAG,0BAA0B,CAAC,QAAQ,CAAC,CAAC;IAC5D,MAAM,iBAAiB,GAAG,IAAI,GAAG,CAC/B,mBAAmB;SAChB,GAAG,CAAC,0BAA0B,CAAC;SAC/B,MAAM,CACL,CAAC,MAAM,EAAmD,EAAE,CAC1D,MAAM,KAAK,IAAI,CAClB,CACJ,CAAC;IACF,IAAI,CAAC,cAAc,IAAI,CAAC,iBAAiB,CAAC,GAAG,CAAC,cAAc,CAAC,EAAE,CAAC;QAC9D,MAAM,IAAI,wBAAwB,CAChC,0CAA0C,QAAQ,IAAI,SAAS,EAAE,EACjE,uBAAuB,CACxB,CAAC;IACJ,CAAC;IAED,MAAM,cAAc,GAAG,iBAAiB,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;IACtD,IAAI,CAAC,cAAc,IAAI,cAAc,KAAK,cAAc,EAAE,CAAC;QACzD,MAAM,IAAI,wBAAwB,CAChC,+DAA+D,EAC/D,oBAAoB,CACrB,CAAC;IACJ,CAAC;IAED,OAAO;QACL,MAAM,EAAE,IAAI,CAAC,MAAM;QACnB,YAAY,EAAE,IAAI,CAAC,YAAY;QAC/B,QAAQ,EAAE,0BAA0B,CAAC,cAAc,CAAC;QACpD,MAAM,EAAE,cAAc;QACtB,IAAI;KACL,CAAC;AACJ,CAAC;AAED,MAAM,gBAAgB,GAAG,MAAM,CAAC;IAC9B,OAAO,EAAE,MAAM,CAAC,aAAa,EAAE;IAC/B,MAAM,EAAE;QACN,QAAQ,EAAE,sCAAsC;QAChD,KAAK,EAAE,CAAC;QACR,MAAM,EAAE,CAAC;QACT,KAAK,EAAE,CAAC;QACR,aAAa,EAAE,GAAG;QAClB,SAAS,EAAE,GAAG,GAAG,IAAI;KACtB;CACF,CAAC,CAAC;AAEH,wEAAwE;AACxE,MAAM,UAAU,wBAAwB,CACtC,GAAY,EACZ,GAAa,EACb,MAAoB;IAEpB,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;QACrC,IAAI,MAAM,EAAE,OAAO,EAAE,CAAC;YACpB,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;YACtB,OAAO;QACT,CAAC;QACD,IAAI,OAAO,GAAG,KAAK,CAAC;QACpB,MAAM,MAAM,GAAG,CAAC,KAAe,EAAQ,EAAE;YACvC,IAAI,OAAO;gBAAE,OAAO;YACpB,OAAO,GAAG,IAAI,CAAC;YACf,MAAM,EAAE,mBAAmB,CAAC,OAAO,EAAE,WAAW,CAAC,CAAC;YAClD,IAAI,KAAK;gBAAE,MAAM,CAAC,KAAK,CAAC,CAAC;;gBACpB,OAAO,EAAE,CAAC;QACjB,CAAC,CAAC;QACF,MAAM,WAAW,GAAG,GAAS,EAAE;YAC7B,MAAM,MAAM,GACV,MAAM,EAAE,MAAM,YAAY,KAAK;gBAC7B,CAAC,CAAC,MAAM,CAAC,MAAM;gBACf,CAAC,CAAC,IAAI,KAAK,CAAC,uCAAuC,CAAC,CAAC;YACzD,yEAAyE;YACzE,yEAAyE;YACzE,qDAAqD;YACrD,MAAM,CAAC,MAAM,CAAC,CAAC;YACf,IAAI,CAAC,GAAG,CAAC,SAAS;gBAAE,GAAG,CAAC,IAAI,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC;QAChD,CAAC,CAAC;QACF,gBAAgB,CAAC,MAAM,CAAC,iBAAiB,CAAC,CAAC,GAAG,EAAE,GAAG,EAAE,KAAK,CAAC,EAAE;YAC3D,MAAM,CAAC,KAAK,CAAC,CAAC;QAChB,CAAC,CAAC,CAAC;QACH,IAAI,CAAC,OAAO,EAAE,CAAC;YACb,MAAM,EAAE,gBAAgB,CAAC,OAAO,EAAE,WAAW,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC;YAC/D,IAAI,MAAM,EAAE,OAAO;gBAAE,WAAW,EAAE,CAAC;QACrC,CAAC;IACH,CAAC,CAAC,CAAC;AACL,CAAC"}

package/backend/dist/utils/websocketMessages.d.ts

@@ -21,8 +21,9 @@ export declare function sendConnected(ws: WebSocketLike): boolean;
 export declare function sendUserMessage(ws: WebSocketLike, data: unknown): boolean;
 export declare function sendAssistantChunk(ws: WebSocketLike, data: AssistantChunkData, options?: SendOptions): boolean;
 export declare function sendAssistantComplete(ws: WebSocketLike, data: unknown): boolean;
+export declare function sendAssistantCancelled(ws: WebSocketLike, data: unknown, options?: SendOptions): boolean;
 export declare function sendToolStatus(ws: WebSocketLike, data: ToolStatusData, options?: SendOptions): boolean;
 export declare function sendError(ws: WebSocketLike, data: unknown): boolean;
 export declare function buildAssistantFakeStreamChunks(content: string, messageId?: string, batchSize?: number): AssistantChunkData[];
-export declare function streamAssistantFakeChunks(ws: WebSocketLike, content: string, messageId?: string, delayMs?: number): Promise<void>;
+export declare function streamAssistantFakeChunks(ws: WebSocketLike, content: string, messageId?: string, delayMs?: number, signal?: AbortSignal): Promise<void>;
 //# sourceMappingURL=websocketMessages.d.ts.map

package/backend/dist/utils/websocketMessages.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"websocketMessages.d.ts","sourceRoot":"","sources":["../../src/utils/websocketMessages.ts"],"names":[],"mappings":"AAiBA,MAAM,WAAW,aAAa;IAC5B,IAAI,CAAC,IAAI,EAAE,MAAM,GAAG,IAAI,CAAC;CAC1B;AAED,MAAM,WAAW,kBAAkB;IACjC,OAAO,EAAE,MAAM,CAAC;IAChB,KAAK,EAAE,MAAM,CAAC;IACd,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,aAAa,CAAC,EAAE,MAAM,CAAC;IACvB,IAAI,EAAE,OAAO,CAAC;IACd,SAAS,CAAC,EAAE,MAAM,CAAC;CACpB;AAED,MAAM,WAAW,cAAc;IAC7B,UAAU,EAAE,MAAM,CAAC;IACnB,IAAI,EAAE,MAAM,CAAC;IACb,KAAK,EAAE,MAAM,CAAC;CACf;AAED,MAAM,WAAW,WAAW;IAC1B,kBAAkB,CAAC,EAAE,OAAO,CAAC;CAC9B;AA4BD,wBAAgB,aAAa,CAAC,EAAE,EAAE,aAAa,GAAG,OAAO,CAIxD;AAED,wBAAgB,eAAe,CAAC,EAAE,EAAE,aAAa,EAAE,IAAI,EAAE,OAAO,GAAG,OAAO,CAEzE;AAED,wBAAgB,kBAAkB,CAChC,EAAE,EAAE,aAAa,EACjB,IAAI,EAAE,kBAAkB,EACxB,OAAO,CAAC,EAAE,WAAW,GACpB,OAAO,CAET;AAED,wBAAgB,qBAAqB,CACnC,EAAE,EAAE,aAAa,EACjB,IAAI,EAAE,OAAO,GACZ,OAAO,CAET;AAED,wBAAgB,cAAc,CAC5B,EAAE,EAAE,aAAa,EACjB,IAAI,EAAE,cAAc,EACpB,OAAO,CAAC,EAAE,WAAW,GACpB,OAAO,CAET;AAED,wBAAgB,SAAS,CAAC,EAAE,EAAE,aAAa,EAAE,IAAI,EAAE,OAAO,GAAG,OAAO,CAEnE;AAED,wBAAgB,8BAA8B,CAC5C,OAAO,EAAE,MAAM,EACf,SAAS,CAAC,EAAE,MAAM,EAClB,SAAS,GAAE,MAAU,GACpB,kBAAkB,EAAE,CAkBtB;AAED,wBAAsB,yBAAyB,CAC7C,EAAE,EAAE,aAAa,EACjB,OAAO,EAAE,MAAM,EACf,SAAS,CAAC,EAAE,MAAM,EAClB,OAAO,GAAE,MAAY,GACpB,OAAO,CAAC,IAAI,CAAC,CAUf"}
+{"version":3,"file":"websocketMessages.d.ts","sourceRoot":"","sources":["../../src/utils/websocketMessages.ts"],"names":[],"mappings":"AAiBA,MAAM,WAAW,aAAa;IAC5B,IAAI,CAAC,IAAI,EAAE,MAAM,GAAG,IAAI,CAAC;CAC1B;AAED,MAAM,WAAW,kBAAkB;IACjC,OAAO,EAAE,MAAM,CAAC;IAChB,KAAK,EAAE,MAAM,CAAC;IACd,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,aAAa,CAAC,EAAE,MAAM,CAAC;IACvB,IAAI,EAAE,OAAO,CAAC;IACd,SAAS,CAAC,EAAE,MAAM,CAAC;CACpB;AAED,MAAM,WAAW,cAAc;IAC7B,UAAU,EAAE,MAAM,CAAC;IACnB,IAAI,EAAE,MAAM,CAAC;IACb,KAAK,EAAE,MAAM,CAAC;CACf;AAED,MAAM,WAAW,WAAW;IAC1B,kBAAkB,CAAC,EAAE,OAAO,CAAC;CAC9B;AA6BD,wBAAgB,aAAa,CAAC,EAAE,EAAE,aAAa,GAAG,OAAO,CAIxD;AAED,wBAAgB,eAAe,CAAC,EAAE,EAAE,aAAa,EAAE,IAAI,EAAE,OAAO,GAAG,OAAO,CAEzE;AAED,wBAAgB,kBAAkB,CAChC,EAAE,EAAE,aAAa,EACjB,IAAI,EAAE,kBAAkB,EACxB,OAAO,CAAC,EAAE,WAAW,GACpB,OAAO,CAET;AAED,wBAAgB,qBAAqB,CACnC,EAAE,EAAE,aAAa,EACjB,IAAI,EAAE,OAAO,GACZ,OAAO,CAET;AAED,wBAAgB,sBAAsB,CACpC,EAAE,EAAE,aAAa,EACjB,IAAI,EAAE,OAAO,EACb,OAAO,CAAC,EAAE,WAAW,GACpB,OAAO,CAET;AAED,wBAAgB,cAAc,CAC5B,EAAE,EAAE,aAAa,EACjB,IAAI,EAAE,cAAc,EACpB,OAAO,CAAC,EAAE,WAAW,GACpB,OAAO,CAET;AAED,wBAAgB,SAAS,CAAC,EAAE,EAAE,aAAa,EAAE,IAAI,EAAE,OAAO,GAAG,OAAO,CAEnE;AAED,wBAAgB,8BAA8B,CAC5C,OAAO,EAAE,MAAM,EACf,SAAS,CAAC,EAAE,MAAM,EAClB,SAAS,GAAE,MAAU,GACpB,kBAAkB,EAAE,CAkBtB;AAED,wBAAsB,yBAAyB,CAC7C,EAAE,EAAE,aAAa,EACjB,OAAO,EAAE,MAAM,EACf,SAAS,CAAC,EAAE,MAAM,EAClB,OAAO,GAAE,MAAY,EACrB,MAAM,CAAC,EAAE,WAAW,GACnB,OAAO,CAAC,IAAI,CAAC,CAoBf"}

package/backend/dist/utils/websocketMessages.js

@@ -40,6 +40,9 @@ export function sendAssistantChunk(ws, data, options) {
 export function sendAssistantComplete(ws, data) {
     return sendWebSocketMessage(ws, 'assistant_complete', data);
 }
+export function sendAssistantCancelled(ws, data, options) {
+    return sendWebSocketMessage(ws, 'assistant_cancelled', data, options);
+}
 export function sendToolStatus(ws, data, options) {
     return sendWebSocketMessage(ws, 'tool_status', data, options);
 }
@@ -62,12 +65,23 @@ export function buildAssistantFakeStreamChunks(content, messageId, batchSize = 3
     }
     return chunks;
 }
-export async function streamAssistantFakeChunks(ws, content, messageId, delayMs = 100) {
+export async function streamAssistantFakeChunks(ws, content, messageId, delayMs = 100, signal) {
     const chunks = buildAssistantFakeStreamChunks(content, messageId);
     for (const [index, chunk] of chunks.entries()) {
+        if (signal?.aborted)
+            return;
         sendAssistantChunk(ws, chunk);
         if (index < chunks.length - 1) {
-            await new Promise(resolve => setTimeout(resolve, delayMs));
+            await new Promise(resolve => {
+                const timer = setTimeout(finish, delayMs);
+                const abort = () => finish();
+                function finish() {
+                    clearTimeout(timer);
+                    signal?.removeEventListener('abort', abort);
+                    resolve();
+                }
+                signal?.addEventListener('abort', abort, { once: true });
+            });
         }
     }
 }

package/backend/dist/utils/websocketMessages.js.map

@@ -1 +1 @@
-{"version":3,"file":"websocketMessages.js","sourceRoot":"","sources":["../../src/utils/websocketMessages.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;;GAeG;AAiCH,SAAS,oBAAoB,CAC3B,EAAiB,EACjB,IAA0B,EAC1B,IAAa,EACb,OAAO,GAAgB,EAAE;IAEzB,IAAI,CAAC;QACH,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC;QACxC,OAAO,IAAI,CAAC;IACd,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,IAAI,OAAO,CAAC,kBAAkB,EAAE,CAAC;YAC/B,OAAO,KAAK,CAAC;QACf,CAAC;QAED,MAAM,KAAK,CAAC;IACd,CAAC;AACH,CAAC;AAED,MAAM,UAAU,aAAa,CAAC,EAAiB;IAC7C,OAAO,oBAAoB,CAAC,EAAE,EAAE,WAAW,EAAE;QAC3C,OAAO,EAAE,0BAA0B;KACpC,CAAC,CAAC;AACL,CAAC;AAED,MAAM,UAAU,eAAe,CAAC,EAAiB,EAAE,IAAa;IAC9D,OAAO,oBAAoB,CAAC,EAAE,EAAE,cAAc,EAAE,IAAI,CAAC,CAAC;AACxD,CAAC;AAED,MAAM,UAAU,kBAAkB,CAChC,EAAiB,EACjB,IAAwB,EACxB,OAAqB;IAErB,OAAO,oBAAoB,CAAC,EAAE,EAAE,iBAAiB,EAAE,IAAI,EAAE,OAAO,CAAC,CAAC;AACpE,CAAC;AAED,MAAM,UAAU,qBAAqB,CACnC,EAAiB,EACjB,IAAa;IAEb,OAAO,oBAAoB,CAAC,EAAE,EAAE,oBAAoB,EAAE,IAAI,CAAC,CAAC;AAC9D,CAAC;AAED,MAAM,UAAU,cAAc,CAC5B,EAAiB,EACjB,IAAoB,EACpB,OAAqB;IAErB,OAAO,oBAAoB,CAAC,EAAE,EAAE,aAAa,EAAE,IAAI,EAAE,OAAO,CAAC,CAAC;AAChE,CAAC;AAED,MAAM,UAAU,SAAS,CAAC,EAAiB,EAAE,IAAa;IACxD,OAAO,oBAAoB,CAAC,EAAE,EAAE,OAAO,EAAE,IAAI,CAAC,CAAC;AACjD,CAAC;AAED,MAAM,UAAU,8BAA8B,CAC5C,OAAe,EACf,SAAkB,EAClB,SAAS,GAAW,CAAC;IAErB,MAAM,KAAK,GAAG,OAAO,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;IACjC,MAAM,MAAM,GAAyB,EAAE,CAAC;IAExC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,KAAK,CAAC,MAAM,EAAE,CAAC,IAAI,SAAS,EAAE,CAAC;QACjD,MAAM,KAAK,GAAG,KAAK,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,GAAG,SAAS,CAAC,CAAC;QAC5C,MAAM,KAAK,GAAG,KAAK,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,GAAG,KAAK,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QACzD,MAAM,MAAM,GAAG,CAAC,GAAG,SAAS,IAAI,KAAK,CAAC,MAAM,CAAC;QAE7C,MAAM,CAAC,IAAI,CAAC;YACV,OAAO,EAAE,KAAK,CAAC,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,GAAG,CAAC;YAC9C,KAAK;YACL,IAAI,EAAE,MAAM;YACZ,SAAS;SACV,CAAC,CAAC;IACL,CAAC;IAED,OAAO,MAAM,CAAC;AAChB,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,yBAAyB,CAC7C,EAAiB,EACjB,OAAe,EACf,SAAkB,EAClB,OAAO,GAAW,GAAG;IAErB,MAAM,MAAM,GAAG,8BAA8B,CAAC,OAAO,EAAE,SAAS,CAAC,CAAC;IAElE,KAAK,MAAM,CAAC,KAAK,EAAE,KAAK,CAAC,IAAI,MAAM,CAAC,OAAO,EAAE,EAAE,CAAC;QAC9C,kBAAkB,CAAC,EAAE,EAAE,KAAK,CAAC,CAAC;QAE9B,IAAI,KAAK,GAAG,MAAM,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC9B,MAAM,IAAI,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC,CAAC;QAC7D,CAAC;IACH,CAAC;AACH,CAAC"}
+{"version":3,"file":"websocketMessages.js","sourceRoot":"","sources":["../../src/utils/websocketMessages.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;;GAeG;AAkCH,SAAS,oBAAoB,CAC3B,EAAiB,EACjB,IAA0B,EAC1B,IAAa,EACb,OAAO,GAAgB,EAAE;IAEzB,IAAI,CAAC;QACH,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC;QACxC,OAAO,IAAI,CAAC;IACd,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,IAAI,OAAO,CAAC,kBAAkB,EAAE,CAAC;YAC/B,OAAO,KAAK,CAAC;QACf,CAAC;QAED,MAAM,KAAK,CAAC;IACd,CAAC;AACH,CAAC;AAED,MAAM,UAAU,aAAa,CAAC,EAAiB;IAC7C,OAAO,oBAAoB,CAAC,EAAE,EAAE,WAAW,EAAE;QAC3C,OAAO,EAAE,0BAA0B;KACpC,CAAC,CAAC;AACL,CAAC;AAED,MAAM,UAAU,eAAe,CAAC,EAAiB,EAAE,IAAa;IAC9D,OAAO,oBAAoB,CAAC,EAAE,EAAE,cAAc,EAAE,IAAI,CAAC,CAAC;AACxD,CAAC;AAED,MAAM,UAAU,kBAAkB,CAChC,EAAiB,EACjB,IAAwB,EACxB,OAAqB;IAErB,OAAO,oBAAoB,CAAC,EAAE,EAAE,iBAAiB,EAAE,IAAI,EAAE,OAAO,CAAC,CAAC;AACpE,CAAC;AAED,MAAM,UAAU,qBAAqB,CACnC,EAAiB,EACjB,IAAa;IAEb,OAAO,oBAAoB,CAAC,EAAE,EAAE,oBAAoB,EAAE,IAAI,CAAC,CAAC;AAC9D,CAAC;AAED,MAAM,UAAU,sBAAsB,CACpC,EAAiB,EACjB,IAAa,EACb,OAAqB;IAErB,OAAO,oBAAoB,CAAC,EAAE,EAAE,qBAAqB,EAAE,IAAI,EAAE,OAAO,CAAC,CAAC;AACxE,CAAC;AAED,MAAM,UAAU,cAAc,CAC5B,EAAiB,EACjB,IAAoB,EACpB,OAAqB;IAErB,OAAO,oBAAoB,CAAC,EAAE,EAAE,aAAa,EAAE,IAAI,EAAE,OAAO,CAAC,CAAC;AAChE,CAAC;AAED,MAAM,UAAU,SAAS,CAAC,EAAiB,EAAE,IAAa;IACxD,OAAO,oBAAoB,CAAC,EAAE,EAAE,OAAO,EAAE,IAAI,CAAC,CAAC;AACjD,CAAC;AAED,MAAM,UAAU,8BAA8B,CAC5C,OAAe,EACf,SAAkB,EAClB,SAAS,GAAW,CAAC;IAErB,MAAM,KAAK,GAAG,OAAO,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;IACjC,MAAM,MAAM,GAAyB,EAAE,CAAC;IAExC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,KAAK,CAAC,MAAM,EAAE,CAAC,IAAI,SAAS,EAAE,CAAC;QACjD,MAAM,KAAK,GAAG,KAAK,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,GAAG,SAAS,CAAC,CAAC;QAC5C,MAAM,KAAK,GAAG,KAAK,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,GAAG,KAAK,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QACzD,MAAM,MAAM,GAAG,CAAC,GAAG,SAAS,IAAI,KAAK,CAAC,MAAM,CAAC;QAE7C,MAAM,CAAC,IAAI,CAAC;YACV,OAAO,EAAE,KAAK,CAAC,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,GAAG,CAAC;YAC9C,KAAK;YACL,IAAI,EAAE,MAAM;YACZ,SAAS;SACV,CAAC,CAAC;IACL,CAAC;IAED,OAAO,MAAM,CAAC;AAChB,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,yBAAyB,CAC7C,EAAiB,EACjB,OAAe,EACf,SAAkB,EAClB,OAAO,GAAW,GAAG,EACrB,MAAoB;IAEpB,MAAM,MAAM,GAAG,8BAA8B,CAAC,OAAO,EAAE,SAAS,CAAC,CAAC;IAElE,KAAK,MAAM,CAAC,KAAK,EAAE,KAAK,CAAC,IAAI,MAAM,CAAC,OAAO,EAAE,EAAE,CAAC;QAC9C,IAAI,MAAM,EAAE,OAAO;YAAE,OAAO;QAC5B,kBAAkB,CAAC,EAAE,EAAE,KAAK,CAAC,CAAC;QAE9B,IAAI,KAAK,GAAG,MAAM,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC9B,MAAM,IAAI,OAAO,CAAO,OAAO,CAAC,EAAE;gBAChC,MAAM,KAAK,GAAG,UAAU,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;gBAC1C,MAAM,KAAK,GAAG,GAAG,EAAE,CAAC,MAAM,EAAE,CAAC;gBAC7B,SAAS,MAAM;oBACb,YAAY,CAAC,KAAK,CAAC,CAAC;oBACpB,MAAM,EAAE,mBAAmB,CAAC,OAAO,EAAE,KAAK,CAAC,CAAC;oBAC5C,OAAO,EAAE,CAAC;gBACZ,CAAC;gBACD,MAAM,EAAE,gBAAgB,CAAC,OAAO,EAAE,KAAK,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC;YAC3D,CAAC,CAAC,CAAC;QACL,CAAC;IACH,CAAC;AACH,CAAC"}

package/backend/dist/websocketServer.d.ts

@@ -1,4 +1,7 @@
 import type { Server } from 'http';
 export declare const isAllowedWebSocketOrigin: (origin: string | undefined, configuredOrigins?: (string | undefined)[]) => boolean;
-export declare function registerWebSocketServer(server: Server): void;
+export interface RegisteredWebSocketServers {
+    close(): Promise<void>;
+}
+export declare function registerWebSocketServer(server: Server): RegisteredWebSocketServers;
 //# sourceMappingURL=websocketServer.d.ts.map

package/backend/dist/websocketServer.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"websocketServer.d.ts","sourceRoot":"","sources":["../src/websocketServer.ts"],"names":[],"mappings":"AAiBA,OAAO,KAAK,EAAmB,MAAM,EAAE,MAAM,MAAM,CAAC;AAwEpD,eAAO,MAAM,wBAAwB,WAC3B,MAAM,GAAG,SAAS,iDAEzB,OAqBF,CAAC;AAiCF,wBAAgB,uBAAuB,CAAC,MAAM,EAAE,MAAM,GAAG,IAAI,CAmoB5D"}
+{"version":3,"file":"websocketServer.d.ts","sourceRoot":"","sources":["../src/websocketServer.ts"],"names":[],"mappings":"AAiBA,OAAO,KAAK,EAAmB,MAAM,EAAE,MAAM,MAAM,CAAC;AAqGpD,eAAO,MAAM,wBAAwB,WAC3B,MAAM,GAAG,SAAS,iDAEzB,OAqBF,CAAC;AA6CF,MAAM,WAAW,0BAA0B;IACzC,KAAK,IAAI,OAAO,CAAC,IAAI,CAAC,CAAC;CACxB;AASD,wBAAgB,uBAAuB,CACrC,MAAM,EAAE,MAAM,GACb,0BAA0B,CA+4B5B"}