@carbon/ai-chat 1.2.0-rc.2 → 1.2.1

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
@@ -3059,31 +3059,31 @@ class MessageService {
3059
3059
  }
3060
3060
  }
3061
3061
  markCurrentMessageAsStreaming(messageID) {
3062
- if (this.queue.current) {
3063
- this.queue.current.isStreaming = true;
3064
- this.streamingMessageID = this.queue.current.message.id;
3065
- if (messageID && messageID !== this.queue.current.message.id) {
3066
- const controller = this.queue.current.sendMessageController;
3067
- if (controller) {
3068
- this.messageAbortControllers.set(messageID, controller);
3069
- }
3062
+ const current = this.queue.current;
3063
+ this.streamingMessageID = messageID ?? current?.message.id;
3064
+ if (messageID && this.lastProcessedMessageID) {
3065
+ const controller = this.messageAbortControllers.get(this.lastProcessedMessageID);
3066
+ if (controller) {
3067
+ this.messageAbortControllers.set(messageID, controller);
3070
3068
  }
3071
- } else if (messageID) {
3072
- this.streamingMessageID = messageID;
3073
- if (this.lastProcessedMessageID) {
3074
- const controller = this.messageAbortControllers.get(this.lastProcessedMessageID);
3075
- if (controller) {
3076
- this.messageAbortControllers.set(messageID, controller);
3077
- }
3069
+ }
3070
+ if (!current) {
3071
+ return;
3072
+ }
3073
+ current.isStreaming = true;
3074
+ if (messageID && messageID !== current.message.id) {
3075
+ const controller = current.sendMessageController;
3076
+ if (controller) {
3077
+ this.messageAbortControllers.set(messageID, controller);
3078
3078
  }
3079
3079
  }
3080
3080
  }
3081
3081
  finalizeStreamingMessage(messageID) {
3082
- this.streamingMessageID = null;
3083
- this.messageAbortControllers.delete(messageID);
3084
- if (this.queue.current && this.queue.current.message.id === messageID) {
3082
+ if (this.streamingMessageID === messageID) {
3085
3083
  this.moveToNextQueueItem();
3086
3084
  }
3085
+ this.messageAbortControllers.delete(messageID);
3086
+ this.streamingMessageID = null;
3087
3087
  }
3088
3088
  async cancelCurrentMessageRequest(reason = CancellationReason.STOP_STREAMING) {
3089
3089
  if (this.streamingMessageID) {