@mulingai-npm/redis 1.13.0 → 1.13.2

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,6 +3,8 @@ export type SttProvider = 'azure' | 'whisper' | 'google' | 'aws';
 export type MulingstreamRoomData = {
     mulingstreamRoomTrackerId: string;
     roomId: string;
+    sttProviders: SttProvider[];
+    targetLanguages: string[];
     lastReadyChunk: number;
     lastUsedChunk: number;
     lastDiscardedChunk: number;

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

@@ -7,12 +7,17 @@ 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),
@@ -71,6 +76,8 @@ class MulingstreamRoomTracker {
         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(),

package/package.json

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