fix: fix stream send message content

PiperOrigin-RevId: 590343492

Author: vertex-sdk-bot

.release-please-manifest.json

@@ -1,3 +1,3 @@
 {
-  ".": "0.1.0"
+  ".": "0.0.0"
 }

CHANGELOG.md

@@ -1,11 +0,0 @@
-# Changelog
-
-## 0.1.0 (2023-12-12)
-
-
-### Features
-
-* add streamGenerateContent method ([6263800](https://github.com/googleapis/nodejs-vertexai/commit/626380039d7bb2fb9af9219f70ad549950b5f490))
-* add streamSendMessage method ([598b1dd](https://github.com/googleapis/nodejs-vertexai/commit/598b1dd7ca8d84c9b32e633a65634abea232f7de))
-* add generateContent method ([d7f1f0f](https://github.com/googleapis/nodejs-vertexai/commit/d7f1f0f66b7bf22c2cb59a8ef698b426cf7e3b8b))
-* add countTokens method ([e0265a3](https://github.com/googleapis/nodejs-vertexai/commit/e0265a36d73b460c66062a0b520b5556d0aa894b))

src/index.ts

@@ -145,6 +145,7 @@ export declare interface StartChatSessionRequest extends StartChatParams {
 export class ChatSession {
   private project: string;
   private location: string;
+  private _send_stream_promise: Promise<void> = Promise.resolve();
 
   private historyInternal: Content[];
   private _vertex_instance: VertexAI_Internal;
@@ -195,21 +196,12 @@ export class ChatSession {
 
     return Promise.resolve({response: generateContentResponse});
   }
-
-  async sendMessageStream(request: string|Array<string|Part>):
-      Promise<StreamGenerateContentResult> {
-    const newContent: Content = formulateNewContent(request);
-    const generateContentrequest: GenerateContentRequest = {
-      contents: this.historyInternal.concat([newContent]),
-      safety_settings: this.safety_settings,
-      generation_config: this.generation_config,
-    };
-
-    const streamGenerateContentResultPromise =
-        this._model_instance.generateContentStream(
-            generateContentrequest);
-    const streamGenerateContentResult =
-      await streamGenerateContentResultPromise;
+  
+  async appendHistory(
+      streamGenerateContentResultPromise: Promise<StreamGenerateContentResult>,
+      newContent: Content,
+      ): Promise<void> {
+    const streamGenerateContentResult = await streamGenerateContentResultPromise;
     const streamGenerateContentResponse =
         await streamGenerateContentResult.response;
     // Only push the latest message to history if the response returned a result
@@ -225,7 +217,22 @@ export class ChatSession {
       // TODO: handle promptFeedback in the response
       throw new Error('Did not get a candidate from the model');
     }
+  }
+
+  async sendMessageStream(request: string|Array<string|Part>):
+      Promise<StreamGenerateContentResult> {
+    const newContent: Content = formulateNewContent(request);
+    const generateContentrequest: GenerateContentRequest = {
+      contents: this.historyInternal.concat([newContent]),
+      safety_settings: this.safety_settings,
+      generation_config: this.generation_config,
+    };
+
+    const streamGenerateContentResultPromise =
+        this._model_instance.generateContentStream(
+            generateContentrequest);
 
+    this._send_stream_promise = this.appendHistory(streamGenerateContentResultPromise, newContent);
     return streamGenerateContentResultPromise;
   }
 }