Fix mediaChunks errors

Author: dlarocque

packages/ai/integration/live.test.ts

@@ -23,7 +23,7 @@ import {
   LiveServerContent,
   LiveServerToolCall,
   LiveServerToolCallCancellation,
-  ResponseModality,
+  ResponseModality
 } from '../src';
 import { liveTestConfigs } from './constants';
 import { HELLO_AUDIO_PCM_BASE64 } from './sample-data/hello-audio';
@@ -177,12 +177,6 @@ describe('Live', function () {
         });
 
         it('should send multiple audio chunks in a single batch call', async () => {
-          // Sending more than one mediaChunk in a message to Google AI results in the server
-          // closing the WebSocket connection with 'Request Contains an Invalid Argument.'.
-          // Skip this test for Google AI.
-          if (testConfig.ai.backend.backendType === BackendType.GOOGLE_AI) {
-            return;
-          }
           const model = getLiveGenerativeModel(testConfig.ai, {
             model: testConfig.model,
             generationConfig: textLiveGenerationConfig
@@ -241,7 +235,7 @@ describe('Live', function () {
        * These tests are currently very unreliable. Their behavior seems to change frequently.
        * Skipping them for now.
        */
-      /**
+      /*
       describe('function calling', () => {
         // When this tests runs against the Google AI backend, the first message we get back 
         // has an `executableCode` part, and then 

packages/ai/package.json

@@ -41,7 +41,7 @@
     "test:browser": "yarn testsetup && karma start",
     "test:node": "TS_NODE_COMPILER_OPTIONS='{\"module\":\"commonjs\"}' mocha --require ts-node/register --require src/index.node.ts 'src/**/*.test.ts' --config ../../config/mocharc.node.js",
     "test:integration": "karma start --integration",
-    "test:integration:node": "TS_NODE_COMPILER_OPTIONS='{\"module\":\"commonjs\"}' mocha integration/live.test.ts --config ../../config/mocharc.node.js",
+    "test:integration:node": "TS_NODE_COMPILER_OPTIONS='{\"module\":\"commonjs\"}' mocha integration/**/*.test.ts --config ../../config/mocharc.node.js",
     "api-report": "api-extractor run --local --verbose",
     "typings:public": "node ../../scripts/build/use_typings.js ./dist/ai-public.d.ts",
     "trusted-type-check": "tsec -p tsconfig.json --noEmit"

packages/ai/src/methods/live-session.ts

@@ -101,10 +101,14 @@ export class LiveSession {
       );
     }
 
-    const message: _LiveClientRealtimeInput = {
-      realtimeInput: { mediaChunks }
-    };
-    this.webSocketHandler.send(JSON.stringify(message));
+    // The backend does not support sending more than one mediaChunk in one message.
+    // Work around this limitation by sending mediaChunks in separate messages.
+    mediaChunks.forEach(mediaChunk => {
+      const message: _LiveClientRealtimeInput = {
+        realtimeInput: { mediaChunks: [mediaChunk] }
+      };
+      this.webSocketHandler.send(JSON.stringify(message));
+    });
   }
 
   /**