@mulingai-npm/redis 3.29.2 → 3.29.4

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

@@ -18,6 +18,7 @@ export type StreamingChunkData = {
 export type MulingstreamChunkData = {
     chunkId: string;
     roomId: string;
+    userId: string;
     chunkNumber: number;
     language: string;
     targetLanguages: string[];
@@ -39,6 +40,8 @@ export type MulingstreamChunkData = {
             duration?: number;
         };
     };
+    isComplete: boolean;
+    isSaved: boolean;
 };
 export declare class MulingstreamChunkManager {
     private redisClient;
@@ -60,6 +63,7 @@ export declare class MulingstreamChunkManager {
      */
     addStreamingChunk(params: {
         roomId: string;
+        userId: string;
         chunkNumber: number;
         language: string;
         transcription: string;
@@ -96,6 +100,26 @@ export declare class MulingstreamChunkManager {
         duration?: number;
     }): Promise<MulingstreamChunkData | null>;
     areTranslationsProcessed(roomId: string, n: number): Promise<boolean>;
+    /**
+     * Check if all TTS for all target languages are done (READY, USED, or DISCARDED - not INIT)
+     */
+    isChunkComplete(roomId: string, n: number): Promise<boolean>;
+    /**
+     * Mark a chunk as complete (all TTS finished for all target languages)
+     */
+    markChunkComplete(roomId: string, n: number): Promise<MulingstreamChunkData | null>;
+    /**
+     * Mark a chunk as saved to database
+     */
+    markChunkSaved(roomId: string, n: number): Promise<MulingstreamChunkData | null>;
+    /**
+     * Get all complete but unsaved chunks for a room
+     */
+    getCompleteUnsavedChunks(roomId: string): Promise<MulingstreamChunkData[]>;
+    /**
+     * Mark multiple chunks as saved in batch
+     */
+    markChunksSaved(roomId: string, chunkNumbers: number[]): Promise<void>;
     /**
      * @deprecated This function is replaced by ChunkSequencer (5-sequencer-handler.ts)
      *

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

@@ -27,6 +27,7 @@ class MulingstreamChunkManager {
         return {
             chunkId: h.chunkId,
             roomId: h.roomId,
+            userId: h.userId,
             chunkNumber: parseInt(h.chunkNumber, 10),
             language: h.language,
             targetLanguages: this.deserialize(h.targetLanguages),
@@ -35,7 +36,9 @@ class MulingstreamChunkManager {
             createdAt: parseInt(h.createdAt, 10),
             streamingChunk: this.deserialize(h.streamingChunk),
             translation: this.deserialize(h.translation),
-            tts: this.deserialize(h.tts)
+            tts: this.deserialize(h.tts),
+            isComplete: h.isComplete === 'true',
+            isSaved: h.isSaved === 'true'
         };
     }
     getTimeout() {
@@ -72,7 +75,7 @@ class MulingstreamChunkManager {
      */
     async addStreamingChunk(params) {
         var _a, _b;
-        const { roomId, chunkNumber, language, transcription, sttProvider, targetLanguages, shortCodeTargetLanguages, streamingData } = params;
+        const { roomId, userId, chunkNumber, language, transcription, sttProvider, targetLanguages, shortCodeTargetLanguages, streamingData } = params;
         // Clear room if this is first chunk (new session)
         if (chunkNumber === 1) {
             const old = await this.redisClient.zrange(this.roomZsetKey(roomId), 0, -1);
@@ -116,6 +119,7 @@ class MulingstreamChunkManager {
         const chunk = {
             chunkId,
             roomId,
+            userId,
             chunkNumber,
             language,
             targetLanguages,
@@ -124,11 +128,14 @@ class MulingstreamChunkManager {
             createdAt,
             streamingChunk,
             translation,
-            tts
+            tts,
+            isComplete: false,
+            isSaved: false
         };
         const hash = {
             chunkId,
             roomId,
+            userId,
             chunkNumber: String(chunkNumber),
             language,
             targetLanguages: this.serialize(targetLanguages),
@@ -137,7 +144,9 @@ class MulingstreamChunkManager {
             createdAt: String(createdAt),
             streamingChunk: this.serialize(streamingChunk),
             translation: this.serialize(translation),
-            tts: this.serialize(tts)
+            tts: this.serialize(tts),
+            isComplete: 'false',
+            isSaved: 'false'
         };
         const pipe = this.redisClient.pipeline();
         pipe.hset(this.chunkHashKey(chunkId), hash);
@@ -185,7 +194,9 @@ class MulingstreamChunkManager {
             finalTranscription: chunk.finalTranscription,
             translation: this.serialize(chunk.translation),
             tts: this.serialize(chunk.tts),
-            streamingChunk: this.serialize(chunk.streamingChunk)
+            streamingChunk: this.serialize(chunk.streamingChunk),
+            isComplete: String(chunk.isComplete),
+            isSaved: String(chunk.isSaved)
         });
         p.expire(key, EXPIRATION);
         await p.exec();
@@ -259,6 +270,47 @@ class MulingstreamChunkManager {
         const c = await this.getMulingstreamChunkById(roomId, n);
         return !!c && Object.values(c.translation).every((t) => t.status !== 'INIT');
     }
+    /**
+     * Check if all TTS for all target languages are done (READY, USED, or DISCARDED - not INIT)
+     */
+    async isChunkComplete(roomId, n) {
+        const c = await this.getMulingstreamChunkById(roomId, n);
+        if (!c)
+            return false;
+        return Object.values(c.tts).every((t) => t.status !== 'INIT');
+    }
+    /**
+     * Mark a chunk as complete (all TTS finished for all target languages)
+     */
+    async markChunkComplete(roomId, n) {
+        return this.withChunk(roomId, n, (c) => {
+            c.isComplete = true;
+        });
+    }
+    /**
+     * Mark a chunk as saved to database
+     */
+    async markChunkSaved(roomId, n) {
+        return this.withChunk(roomId, n, (c) => {
+            c.isSaved = true;
+        });
+    }
+    /**
+     * Get all complete but unsaved chunks for a room
+     */
+    async getCompleteUnsavedChunks(roomId) {
+        var _a;
+        const chunks = (_a = (await this.getMulingstreamChunksByRoom(roomId))) !== null && _a !== void 0 ? _a : [];
+        return chunks.filter((c) => c.isComplete && !c.isSaved);
+    }
+    /**
+     * Mark multiple chunks as saved in batch
+     */
+    async markChunksSaved(roomId, chunkNumbers) {
+        for (const n of chunkNumbers) {
+            await this.markChunkSaved(roomId, n);
+        }
+    }
     /**
      * @deprecated This function is replaced by ChunkSequencer (5-sequencer-handler.ts)
      *

package/package.json

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