chore: Minor code cleanup

tjtanjin · Sep 15, 2024 · d142265 · d142265
1 parent 77aa8d2
commit d142265
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 6 deletions.
diff --git a/src/hooks/internal/useMessagesInternal.ts b/src/hooks/internal/useMessagesInternal.ts
@@ -211,7 +211,7 @@ export const useMessagesInternal = () => {
 			return updatedMessages;
 		});
 		return streamMessageMap.current.get(sender) || null;
-	}, []);
+	},[]);
 
 	/**
 	 * Sets the streaming mode of the chatbot.
@@ -233,9 +233,12 @@ export const useMessagesInternal = () => {
 			return true;
 		}
 
+		const messageId = streamMessageMap.current.get(sender);
+		const messageToEndStreamFor = messages.find((message) => message.id === messageId);
+
 		// handles stop stream message event
 		if (settings.event?.rcbStopStreamMessage) {
-			const event = callRcbEvent(RcbEvent.STOP_STREAM_MESSAGE, {});
+			const event = callRcbEvent(RcbEvent.STOP_STREAM_MESSAGE, {messageToEndStreamFor});
 			if (event.defaultPrevented) {
 				return false;
 			}
@@ -245,7 +248,7 @@ export const useMessagesInternal = () => {
 		streamMessageMap.current.delete(sender);
 		saveChatHistory(messages);
 		return true;
-	}, [messages]);
+	}, [messages])
 
 	return {
 		endStreamMessage,

diff --git a/src/hooks/internal/useRcbEventInternal.ts b/src/hooks/internal/useRcbEventInternal.ts
@@ -3,9 +3,6 @@ import { useCallback } from "react";
 import { emitRcbEvent } from "../../services/RcbEventService";
 import { usePaths } from "../usePaths";
 import { useBotRefsContext } from "../../context/BotRefsContext";
-import { useSettingsContext } from "../../context/SettingsContext";
-import { useStylesContext } from "../../context/StylesContext";
-import { useMessagesContext } from "../../context/MessagesContext";
 import { RcbEvent } from "../../constants/RcbEvent";
 
 /**