@mulingai-npm/redis 1.13.0 → 1.14.0

package/dist/enums/redis-database.d.ts

@@ -1,5 +1,5 @@
 export declare enum RedisDatabase {
     MULINGSTREAM_LISTENER = 0,
-    MULINGSTREAM_ROOM = 1,
+    MULINGSTREAM_ROOM_TRACKER = 1,
     MULINGSTREAM_CHUNK = 2
 }

package/dist/enums/redis-database.js

@@ -4,6 +4,6 @@ exports.RedisDatabase = void 0;
 var RedisDatabase;
 (function (RedisDatabase) {
     RedisDatabase[RedisDatabase["MULINGSTREAM_LISTENER"] = 0] = "MULINGSTREAM_LISTENER";
-    RedisDatabase[RedisDatabase["MULINGSTREAM_ROOM"] = 1] = "MULINGSTREAM_ROOM";
+    RedisDatabase[RedisDatabase["MULINGSTREAM_ROOM_TRACKER"] = 1] = "MULINGSTREAM_ROOM_TRACKER";
     RedisDatabase[RedisDatabase["MULINGSTREAM_CHUNK"] = 2] = "MULINGSTREAM_CHUNK";
 })(RedisDatabase = exports.RedisDatabase || (exports.RedisDatabase = {}));

package/dist/managers/mulingstream-chunk-manager.d.ts

@@ -67,4 +67,19 @@ export declare class MulingstreamChunkManager {
     updateFinalTranscription(mulingstreamChunkId: string, finalTranscription: string): Promise<boolean>;
     updateSttStatus(mulingstreamChunkId: string, service: SttProvider, status: StepStatus): Promise<boolean>;
     updateTranscription(mulingstreamChunkId: string, service: SttProvider, transcription: string, sttStatus: StepStatus): Promise<boolean>;
+    /****
+     * Discards all post-STT steps for a given chunk, i.e. sets
+     * all translation[].status and all tts[].status to "DISCARDED".
+     ****/
+    discardAllPostStt(mulingstreamChunkId: string): Promise<boolean>;
+    /****
+     * Discards all post-translation steps for a given chunk, i.e. sets
+     * all tts[].status to "DISCARDED" (leaves translation alone).
+     ****/
+    discardAllPostTranslation(mulingstreamChunkId: string): Promise<boolean>;
+    /****
+     * Discards a specific language in both the translation and tts objects,
+     * setting status to DISCARDED for just that one language.
+     ****/
+    discardLanguage(mulingstreamChunkId: string, language: string): Promise<boolean>;
 }

package/dist/managers/mulingstream-chunk-manager.js

@@ -206,5 +206,83 @@ class MulingstreamChunkManager {
         });
         return true;
     }
+    /****
+     * Discards all post-STT steps for a given chunk, i.e. sets
+     * all translation[].status and all tts[].status to "DISCARDED".
+     ****/
+    async discardAllPostStt(mulingstreamChunkId) {
+        // Fetch the chunk
+        const data = await this.redisClient.hgetall(`mulingstreamChunk:${mulingstreamChunkId}`);
+        if (!data || !data.mulingstreamChunkId) {
+            return false;
+        }
+        // Parse the translation and tts objects
+        const translation = JSON.parse(data.translation);
+        const tts = JSON.parse(data.tts);
+        // Set every language status to DISCARDED in translation
+        for (const lang of Object.keys(translation)) {
+            translation[lang].status = 'DISCARDED';
+        }
+        // Set every language status to DISCARDED in tts
+        for (const lang of Object.keys(tts)) {
+            tts[lang].status = 'DISCARDED';
+        }
+        // Write back
+        await this.redisClient.hset(`mulingstreamChunk:${mulingstreamChunkId}`, {
+            translation: JSON.stringify(translation),
+            tts: JSON.stringify(tts)
+        });
+        return true;
+    }
+    /****
+     * Discards all post-translation steps for a given chunk, i.e. sets
+     * all tts[].status to "DISCARDED" (leaves translation alone).
+     ****/
+    async discardAllPostTranslation(mulingstreamChunkId) {
+        // Fetch the chunk
+        const data = await this.redisClient.hgetall(`mulingstreamChunk:${mulingstreamChunkId}`);
+        if (!data || !data.mulingstreamChunkId) {
+            return false;
+        }
+        // Parse the tts object
+        const tts = JSON.parse(data.tts);
+        // Set every language status to DISCARDED in tts
+        for (const lang of Object.keys(tts)) {
+            tts[lang].status = 'DISCARDED';
+        }
+        // Write back
+        await this.redisClient.hset(`mulingstreamChunk:${mulingstreamChunkId}`, {
+            tts: JSON.stringify(tts)
+        });
+        return true;
+    }
+    /****
+     * Discards a specific language in both the translation and tts objects,
+     * setting status to DISCARDED for just that one language.
+     ****/
+    async discardLanguage(mulingstreamChunkId, language) {
+        // Fetch the chunk
+        const data = await this.redisClient.hgetall(`mulingstreamChunk:${mulingstreamChunkId}`);
+        if (!data || !data.mulingstreamChunkId) {
+            return false;
+        }
+        // Parse the translation and tts objects
+        const translation = JSON.parse(data.translation);
+        const tts = JSON.parse(data.tts);
+        // If the language exists in translation, discard
+        if (translation[language]) {
+            translation[language].status = 'DISCARDED';
+        }
+        // If the language exists in tts, discard
+        if (tts[language]) {
+            tts[language].status = 'DISCARDED';
+        }
+        // Write back
+        await this.redisClient.hset(`mulingstreamChunk:${mulingstreamChunkId}`, {
+            translation: JSON.stringify(translation),
+            tts: JSON.stringify(tts)
+        });
+        return true;
+    }
 }
 exports.MulingstreamChunkManager = MulingstreamChunkManager;

package/dist/managers/mulingstream-room-tracker.d.ts

@@ -3,9 +3,14 @@ export type SttProvider = 'azure' | 'whisper' | 'google' | 'aws';
 export type MulingstreamRoomData = {
     mulingstreamRoomTrackerId: string;
     roomId: string;
+    sttProviders: SttProvider[];
+    targetLanguages: string[];
     lastReadyChunk: number;
     lastUsedChunk: number;
     lastDiscardedChunk: number;
+    lastReadyTranscription: number;
+    lastUsedTranscription: number;
+    lastDiscardedTranscription: number;
     stt: {
         [sttProvider: string]: {
             lastReadyStt: number;
@@ -58,6 +63,11 @@ export declare class MulingstreamRoomTracker {
      * Any argument passed as null means "do not update that field."
      */
     updateChunkTracker(roomId: string, lastReadyChunk: number | null, lastUsedChunk: number | null, lastDiscardedChunk: number | null): Promise<boolean>;
+    /**
+     * Updates the transcription tracker fields (lastReadyTranscription, lastUsedTranscription , lastDiscardedTranscription).
+     * Any argument passed as null means "do not update that field."
+     */
+    updateTranscriptionTracker(roomId: string, lastReadyTranscription: number | null, lastUsedTranscription: number | null, lastDiscardedTranscription: number | null): Promise<boolean>;
     /**
      * Updates the stt tracker for a specific STT provider in the room.
      * If any argument is null, we do not update that field.

package/dist/managers/mulingstream-room-tracker.js

@@ -7,15 +7,23 @@ class MulingstreamRoomTracker {
         this.redisClient = redisClient;
     }
     parseHashData(data) {
+        // Parse any JSON fields
+        const sttProviders = JSON.parse(data.sttProviders);
+        const targetLanguages = JSON.parse(data.targetLanguages);
         const stt = JSON.parse(data.stt);
         const translation = JSON.parse(data.translation);
         const tts = JSON.parse(data.tts);
         return {
             mulingstreamRoomTrackerId: data.mulingstreamRoomTrackerId,
             roomId: data.roomId,
+            sttProviders,
+            targetLanguages,
             lastReadyChunk: parseInt(data.lastReadyChunk, 10),
             lastUsedChunk: parseInt(data.lastUsedChunk, 10),
             lastDiscardedChunk: parseInt(data.lastDiscardedChunk, 10),
+            lastReadyTranscription: parseInt(data.lastReadyTranscription, 10),
+            lastUsedTranscription: parseInt(data.lastUsedTranscription, 10),
+            lastDiscardedTranscription: parseInt(data.lastDiscardedTranscription, 10),
             stt,
             translation,
             tts
@@ -67,13 +75,21 @@ class MulingstreamRoomTracker {
         const lastReadyChunk = -1;
         const lastUsedChunk = -1;
         const lastDiscardedChunk = -1;
+        const lastReadyTranscription = -1;
+        const lastUsedTranscription = -1;
+        const lastDiscardedTranscription = -1;
         // Store everything in Redis
         await this.redisClient.hset(`mulingstreamRoomTracker:${mulingstreamRoomTrackerId}`, {
             mulingstreamRoomTrackerId,
             roomId,
+            sttProviders: JSON.stringify(sttProviders),
+            targetLanguages: JSON.stringify(targetLanguages),
             lastReadyChunk: lastReadyChunk.toString(),
             lastUsedChunk: lastUsedChunk.toString(),
             lastDiscardedChunk: lastDiscardedChunk.toString(),
+            lastReadyTranscription: lastReadyTranscription.toString(),
+            lastUsedTranscription: lastUsedTranscription.toString(),
+            lastDiscardedTranscription: lastDiscardedTranscription.toString(),
             stt: JSON.stringify(stt),
             translation: JSON.stringify(translation),
             tts: JSON.stringify(tts)
@@ -136,6 +152,31 @@ class MulingstreamRoomTracker {
         }
         return true;
     }
+    /**
+     * Updates the transcription tracker fields (lastReadyTranscription, lastUsedTranscription , lastDiscardedTranscription).
+     * Any argument passed as null means "do not update that field."
+     */
+    async updateTranscriptionTracker(roomId, lastReadyTranscription, lastUsedTranscription, lastDiscardedTranscription) {
+        const mulingstreamRoomTrackerId = `[${roomId}]`;
+        const data = await this.redisClient.hgetall(`mulingstreamRoomTracker:${mulingstreamRoomTrackerId}`);
+        if (!data || !data.mulingstreamRoomTrackerId) {
+            return false;
+        }
+        const updates = {};
+        if (lastReadyTranscription !== null) {
+            updates.lastReadyTranscription = lastReadyTranscription.toString();
+        }
+        if (lastUsedTranscription !== null) {
+            updates.lastUsedTranscription = lastUsedTranscription.toString();
+        }
+        if (lastDiscardedTranscription !== null) {
+            updates.lastDiscardedTranscription = lastDiscardedTranscription.toString();
+        }
+        if (Object.keys(updates).length > 0) {
+            await this.redisClient.hset(`mulingstreamRoomTracker:${mulingstreamRoomTrackerId}`, updates);
+        }
+        return true;
+    }
     /**
      * Updates the stt tracker for a specific STT provider in the room.
      * If any argument is null, we do not update that field.

package/package.json

@@ -1,6 +1,6 @@
 {
     "name": "@mulingai-npm/redis",
-    "version": "1.13.0",
+    "version": "1.14.0",
     "main": "dist/index.js",
     "types": "dist/index.d.ts",
     "repository": {