fix: add upgrade eligibility gate for refresh

src/components/MessageForm/index.jsx

@@ -10,11 +10,14 @@ import {
   getChatResponse,
   updateCurrentMessage,
 } from '../../data/thunks';
+import { useCourseUpgrade } from '../../hooks';
 import { usePromptExperimentDecision } from '../../experiments';
 
 import './MessageForm.scss';
 
 const MessageForm = ({ courseId, shouldAutofocus, unitId }) => {
+  const { upgradeable } = useCourseUpgrade();
+
   const { apiIsLoading, currentMessage, apiError } = useSelector(state => state.learningAssistant);
   const dispatch = useDispatch();
   const inputRef = useRef();
@@ -35,7 +38,7 @@ const MessageForm = ({ courseId, shouldAutofocus, unitId }) => {
     if (currentMessage) {
       dispatch(acknowledgeDisclosure(true));
       dispatch(addChatMessage('user', currentMessage, courseId, promptExperimentVariationKey));
-      dispatch(getChatResponse(courseId, unitId, promptExperimentVariationKey));
+      dispatch(getChatResponse(courseId, unitId, upgradeable, promptExperimentVariationKey));
     }
   };
 

src/components/MessageForm/index.test.jsx

@@ -10,11 +10,17 @@ import {
   acknowledgeDisclosure,
   addChatMessage,
   getChatResponse,
+  getLearningAssistantChatSummary,
   updateCurrentMessage,
 } from '../../data/thunks';
+import { useCourseUpgrade } from '../../hooks';
 
 import MessageForm from '.';
 
+jest.mock('../../hooks', () => ({
+  useCourseUpgrade: jest.fn(),
+}));
+
 jest.mock('../../utils/surveyMonkey', () => ({
   showControlSurvey: jest.fn(),
   showVariationSurvey: jest.fn(),
@@ -81,6 +87,9 @@ describe('<MessageForm />', () => {
   beforeEach(() => {
     jest.resetAllMocks();
     usePromptExperimentDecision.mockReturnValue([]);
+    useCourseUpgrade.mockReturnValue({
+      upgradeable: true,
+    });
   });
 
   describe('when rendered', () => {
@@ -137,7 +146,7 @@ describe('<MessageForm />', () => {
 
       expect(acknowledgeDisclosure).toHaveBeenCalledWith(true);
       expect(addChatMessage).toHaveBeenCalledWith('user', currentMessage, defaultProps.courseId, undefined);
-      expect(getChatResponse).toHaveBeenCalledWith(defaultProps.courseId, defaultProps.unitId, undefined);
+      expect(getChatResponse).toHaveBeenCalledWith(defaultProps.courseId, defaultProps.unitId, true, undefined);
       expect(mockDispatch).toHaveBeenCalledTimes(3);
     });
 
@@ -187,6 +196,7 @@ describe('<MessageForm />', () => {
       expect(getChatResponse).toHaveBeenCalledWith(
         defaultProps.courseId,
         defaultProps.unitId,
+        true,
         OPTIMIZELY_PROMPT_EXPERIMENT_VARIATION_KEYS.UPDATED_PROMPT,
       );
       expect(mockDispatch).toHaveBeenCalledTimes(3);

src/data/thunks.js

@@ -56,7 +56,7 @@ export function addChatMessage(role, content, courseId, promptExperimentVariatio
   };
 }
 
-export function getChatResponse(courseId, unitId, promptExperimentVariationKey = undefined) {
+export function getChatResponse(courseId, unitId, upgradeable, promptExperimentVariationKey = undefined) {
   return async (dispatch, getState) => {
     const { userId } = getAuthenticatedUser();
     const { messageList } = getState().learningAssistant;
@@ -69,9 +69,9 @@ export function getChatResponse(courseId, unitId, promptExperimentVariationKey =
       const customQueryParams = promptExperimentVariationKey ? { responseVariation: promptExperimentVariationKey } : {};
       const message = await fetchChatResponse(courseId, messageList, unitId, customQueryParams);
 
-      // Refresh chat summary only on the first message so we can tell if the user has initiated an audit trial
-      // NOTE: This is a bit of a hacky solution that may be refined later
-      if (messageList.length === 1) {
+      // Refresh chat summary only on the first message for an upgrade eligible user
+      // so we can tell if the user has just initiated an audit trial
+      if (messageList.length === 1 && upgradeable) {
         // eslint-disable-next-line no-use-before-define
         dispatch(getLearningAssistantChatSummary(courseId));
       }

src/hooks/use-course-upgrade.js

@@ -26,7 +26,6 @@ export default function useCourseUpgrade() {
     auditTrialDaysRemaining = Math.ceil((auditTrialExpirationDate - Date.now()) / millisecondsInOneDay);
 
     auditTrialExpired = auditTrialDaysRemaining < 0;
-    // console.log({ upgradeUrl }, { auditTrialDaysRemaining });
   }
 
   return {

src/widgets/Xpert.jsx

@@ -30,7 +30,7 @@ const Xpert = ({
   };
 
   useEffect(() => {
-    dispatch(getLearningAssistantChatSummary(courseId));
+    dispatch(getLearningAssistantChatSummary(courseId, isUpgradeEligible));
   }, [dispatch, courseId]);
 
   return isEnabled ? (