fix(compass-assistant): also send the system context after the chat was cleared COMPASS-10182 (#7659)

also send the system context after the chat was cleared;

Author: lerouxb

packages/compass-assistant/src/compass-assistant-provider.spec.tsx

@@ -431,7 +431,7 @@ describe('CompassAssistantProvider', function () {
         });
       }
 
-      const contextMessages = mockChat.messages.filter(
+      let contextMessages = mockChat.messages.filter(
         (message) => message.metadata?.isSystemContext
       );
 
@@ -456,6 +456,30 @@ describe('CompassAssistantProvider', function () {
           ],
         },
       ]);
+
+      // if we clear the chat it will send the context again next time
+      sendMessageSpy.resetHistory();
+      mockChat.messages = [];
+
+      userEvent.type(
+        screen.getByPlaceholderText('Ask a question'),
+        'How about now?'
+      );
+      userEvent.click(screen.getByLabelText('Send message'));
+
+      await waitFor(() => {
+        expect(sendMessageSpy.callCount).to.equal(1);
+        expect(sendMessageSpy.getCall(0).args[0]).to.deep.include({
+          text: 'How about now?',
+        });
+
+        expect(screen.getByText('How about now?')).to.exist;
+      });
+
+      contextMessages = mockChat.messages.filter(
+        (message) => message.metadata?.isSystemContext
+      );
+      expect(contextMessages).to.have.lengthOf(1);
     });
 
     it('will not send new messages if the user does not opt in', async function () {

packages/compass-assistant/src/compass-assistant-provider.tsx

@@ -258,10 +258,15 @@ export const AssistantProvider: React.FunctionComponent<
           ? contextPrompt.parts[0].text
           : '';
 
+      const hasSystemContextMessage = chat.messages.some((message) => {
+        return message.metadata?.isSystemContext;
+      });
+
       const shouldSendContextPrompt =
         message?.metadata?.sendContext &&
         (!lastContextPromptRef.current ||
-          lastContextPromptRef.current !== contextPromptText);
+          lastContextPromptRef.current !== contextPromptText ||
+          !hasSystemContextMessage);
       if (shouldSendContextPrompt) {
         lastContextPromptRef.current = contextPromptText;
         chat.messages = [...chat.messages, contextPrompt];