@e-llm-studio/instant-learning 1.3.4 → 1.3.5

package/dist/cjs/features/IL-OTJ/_components/ChatComponents/NormalChat.js

@@ -129,6 +129,10 @@ const NormalChat = ({
   React.useEffect(() => {
     onStreamingStateChange == null ? void 0 : onStreamingStateChange(isStreaming);
   }, [isStreaming, onStreamingStateChange]);
+  const accessTokenRef = React.useRef(user == null ? void 0 : user.accessToken);
+  React.useEffect(() => {
+    accessTokenRef.current = user == null ? void 0 : user.accessToken;
+  }, [user == null ? void 0 : user.accessToken]);
   React.useEffect(() => {
     isMountedRef.current = true;
     return () => {
@@ -559,6 +563,11 @@ const NormalChat = ({
       setIsStreamEnd((_H = (_G = data == null ? void 0 : data.chatStream) == null ? void 0 : _G.continueChatResult) == null ? void 0 : _H.isEnd);
     }
   };
+  const liveHeaders = {
+    get Authorization() {
+      return accessTokenRef.current ? `Bearer ${accessTokenRef.current}` : void 0;
+    }
+  };
   const handleStream = (event, isFirstQuestionSent, userInputParam, isValidationFailure = false, customId, setIsLoading) => {
     var _a, _b;
     const contentSource = userInputParam !== void 0 ? userInputParam : userInput || "";
@@ -654,9 +663,7 @@ const NormalChat = ({
       chatPreviewId: chatPreviewIdRef.current,
       customMetaData,
       aiModel: "",
-      additionalHeaders: (user == null ? void 0 : user.accessToken) ? {
-        Authorization: `Bearer ${user.accessToken}`
-      } : void 0,
+      additionalHeaders: liveHeaders,
       onStreamEvent: (data) => {
         if (!isMountedRef.current || streamGenerationRef.current !== streamGeneration) return;
         handleMessages(
@@ -704,6 +711,7 @@ const NormalChat = ({
       setInitialLoadingMessage(loadingMessage);
       streamStartTimesRef.current.set(params.requestId, Date.now());
       currentStreamRequestIdRef.current = params.requestId;
+      console.log("Chat Stream Params:", params);
       streamingResponse.startChatStream(params);
     }
     setIsLoading && setIsLoading(false);