@mulingai-npm/redis 2.2.0 → 2.3.0

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

@@ -6,8 +6,8 @@ export type MulingstreamChunkData = {
     chunkNumber: number;
     language: string;
     sttProviders: SttProvider[];
-    translationTargetLanguages: string[];
-    ttsTargetLanguages: string[];
+    targetLanguages: string[];
+    shortCodeTargetLanguages: string[];
     finalTranscription: string;
     sttStatus: StepStatus;
     audioChunk: {
@@ -71,8 +71,8 @@ export declare class MulingstreamChunkManager {
         isLast: boolean;
         theme: string;
         sttProviders: SttProvider[];
-        translationTargetLanguages: string[];
-        ttsTargetLanguages: string[];
+        targetLanguages: string[];
+        shortCodeTargetLanguages: string[];
     }): Promise<void>;
     /**
      * Given roomId and chunkNumber, return the single chunk from the array
@@ -100,4 +100,8 @@ export declare class MulingstreamChunkManager {
      * Discards a specific language in both translation and tts for a chunk.
      */
     discardLanguage(roomId: string, chunkNumber: number, language: string): Promise<boolean>;
+    updateTranslation(roomId: string, chunkNumber: number, language: string, options: {
+        translation?: string;
+        status?: StepStatus;
+    }): Promise<boolean>;
 }

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

@@ -52,7 +52,7 @@ class MulingstreamChunkManager {
     }
     async addMulingstreamChunk(params) {
         var _a;
-        const { roomId, chunkNumber, language, start, end, duration, isFirst, isLast, theme, sttProviders, translationTargetLanguages, ttsTargetLanguages } = params;
+        const { roomId, chunkNumber, language, start, end, duration, isFirst, isLast, theme, sttProviders, targetLanguages, shortCodeTargetLanguages } = params;
         const audioChunk = {
             start,
             end,
@@ -71,13 +71,11 @@ class MulingstreamChunkManager {
         }
         const translation = {};
         const tts = {};
-        for (const lang of translationTargetLanguages) {
+        for (const lang of targetLanguages) {
             translation[lang] = {
                 translation: '',
                 status: 'INIT'
             };
-        }
-        for (const lang of ttsTargetLanguages) {
             tts[lang] = {
                 ttsAudioPath: '',
                 status: 'INIT',
@@ -89,8 +87,8 @@ class MulingstreamChunkManager {
             chunkNumber,
             language,
             sttProviders,
-            translationTargetLanguages,
-            ttsTargetLanguages,
+            targetLanguages,
+            shortCodeTargetLanguages,
             finalTranscription: '',
             sttStatus: 'INIT',
             audioChunk,
@@ -215,5 +213,30 @@ class MulingstreamChunkManager {
         await this.redisClient.jsonSet(`[${roomId}]`, '.', chunks);
         return true;
     }
+    async updateTranslation(roomId, chunkNumber, language, options) {
+        // Fetch all chunks for this room
+        const chunks = await this.getMulingstreamChunksByRoom(roomId);
+        if (!chunks)
+            return false;
+        // Locate the target chunk
+        const chunkIndex = chunks.findIndex((c) => c.chunkNumber === chunkNumber);
+        if (chunkIndex === -1)
+            return false;
+        const chunk = chunks[chunkIndex];
+        // Make sure the requested language exists in this chunk
+        if (!chunk.translation[language]) {
+            return false;
+        }
+        if (options.translation !== undefined) {
+            chunk.translation[language].translation = options.translation;
+        }
+        if (options.status !== undefined) {
+            chunk.translation[language].status = options.status;
+        }
+        // Persist the whole array back to Redis
+        chunks[chunkIndex] = chunk;
+        await this.redisClient.jsonSet(`[${roomId}]`, '.', chunks);
+        return true;
+    }
 }
 exports.MulingstreamChunkManager = MulingstreamChunkManager;

package/package.json

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