@e-llm-studio/instant-learning 1.3.3 → 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 () => {
@@ -290,7 +294,7 @@ const NormalChat = ({
       method: "get",
       maxBodyLength: Infinity,
       url: `${url}/api/chatbot/getChatSettings?name=${chatAssistant}&organizationName=${organizationName}`,
-      headers: {}
+      headers: (user == null ? void 0 : user.accessToken) ? { Authorization: `Bearer ${user.accessToken}` } : {}
     };
     transport.http.request(config).then((response) => {
       var _a, _b, _c;
@@ -309,6 +313,7 @@ const NormalChat = ({
       const requestOptions = {
         method: "GET",
         params: { guestSessionId: guestSessionIdRef.current },
+        headers: (user == null ? void 0 : user.accessToken) ? { Authorization: `Bearer ${user.accessToken}` } : {},
         withCredentials: true
       };
       transport.http.get(getHistoryURL, requestOptions).then((response) => {
@@ -558,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 || "";
@@ -653,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(
@@ -703,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);