feat: preload previous response

Author: Henrrypg

backend/openedx_ai_extensions/workflows/configs/default_cms.json

@@ -17,7 +17,8 @@
         "config": {
           "titleText": "AI Assistant",
           "buttonText": "Start",
-          "customMessage": "Use an AI workflow to create multiple answer questions from this unit in a content library"
+          "customMessage": "Use an AI workflow to create multiple answer questions from this unit in a content library",
+          "preloadPreviousSession": true
         }
       },
       "response": {

backend/openedx_ai_extensions/workflows/orchestrators.py

@@ -113,6 +113,13 @@ def run(self, input_data):
 class EducatorAssistantOrchestrator(SessionBasedOrchestrator):
     """Orchestrator for educator assistant workflows."""
 
+    def get_current_session_response(self, _):
+        """Retrieve the current session's LLM response."""
+        submission = self.get_submission()
+        if submission and submission.get("answer"):
+            return {"response": submission["answer"]["collection_url"]}
+        return {"response": None}
+
     def run(self, input_data):
         # Prepare context
         context = {

backend/tests/test_submission_processor.py

@@ -181,3 +181,53 @@ def test_process_retrieves_existing_submissions(
 
     # Should return response with messages
     assert "response" in result or "error" in result
+
+
+@pytest.mark.django_db
+@patch("openedx_ai_extensions.processors.openedx.submission_processor.submissions_api")
+def test_process_respects_max_context_messages_limit(
+    mock_submissions_api, user_session  # pylint: disable=redefined-outer-name
+):
+    """
+    Test that process() respects the max_context_messages configuration limit.
+    """
+    # Create processor with max_context_messages=2
+    config = {
+        "SubmissionProcessor": {
+            "max_context_messages": 2,
+        }
+    }
+    processor = SubmissionProcessor(config=config, user_session=user_session)
+
+    # Mock multiple submissions exceeding the limit
+    mock_submissions = [
+        {
+            "uuid": "submission-1",
+            "answer": {"messages": [{"role": "user", "content": "Message 1"}]},
+            "created_at": "2025-01-01T00:00:00Z",
+        },
+        {
+            "uuid": "submission-2",
+            "answer": {"messages": [{"role": "assistant", "content": "Response 1"}]},
+            "created_at": "2025-01-01T00:01:00Z",
+        },
+        {
+            "uuid": "submission-3",
+            "answer": {"messages": [{"role": "user", "content": "Message 2"}]},
+            "created_at": "2025-01-01T00:02:00Z",
+        },
+        {
+            "uuid": "submission-4",
+            "answer": {"messages": [{"role": "assistant", "content": "Response 2"}]},
+            "created_at": "2025-01-01T00:03:00Z",
+        },
+    ]
+    mock_submissions_api.get_submissions.return_value = mock_submissions
+
+    result = processor.process(context={}, input_data=None)
+
+    # Verify get_submissions was called
+    mock_submissions_api.get_submissions.assert_called_once()
+
+    # Should return response
+    assert "response" in result or "error" in result

frontend/src/components/AIEducatorLibraryAssistComponent.jsx

@@ -1,4 +1,4 @@
-import React, { useState, useEffect } from 'react';
+import React, { useState, useEffect, useRef } from 'react';
 import PropTypes from 'prop-types';
 import {
   Button,
@@ -26,6 +26,7 @@ const AIEducatorLibraryAssistComponent = ({
   libraries: librariesProp,
   titleText,
   buttonText,
+  preloadPreviousSession,
   customMessage,
   onSuccess,
   onError,
@@ -45,6 +46,9 @@ const AIEducatorLibraryAssistComponent = ({
   const [numberOfQuestions, setNumberOfQuestions] = useState(5);
   const [additionalInstructions, setAdditionalInstructions] = useState('');
 
+  // Track if we've already attempted to load previous session
+  const hasLoadedSession = useRef(false);
+
   /**
    * Fetch libraries from API
    * Only called when user opens the form
@@ -95,6 +99,53 @@ const AIEducatorLibraryAssistComponent = ({
     }
   }, [librariesProp]);
 
+  // Preload previous session if enabled
+  useEffect(() => {
+    const loadPreviousSession = async () => {
+      if (!preloadPreviousSession || hasAsked || hasLoadedSession.current) {
+        return;
+      }
+
+      hasLoadedSession.current = true;
+      setIsLoading(true);
+      try {
+        const contextData = prepareContextData({
+          courseId,
+          unitId,
+        });
+
+        const data = await callWorkflowService({
+          context: contextData,
+          action: 'get_current_session_response',
+          payload: {
+            requestId: `ai-request-${Date.now()}`,
+            courseId,
+          },
+        });
+
+        // Handle response - only set if there's actual data
+        if (data.response && data.response !== null) {
+          setResponse(data.response);
+          setHasAsked(true);
+        } else if (debug) {
+          // No previous session or empty response - do nothing, show normal component
+          // eslint-disable-next-line no-console
+          console.log('No previous session found or empty response');
+        }
+      } catch (err) {
+        // Silent fail - no previous session is not an error
+        if (debug) {
+          // eslint-disable-next-line no-console
+          console.log('Error loading previous session:', err);
+        }
+      } finally {
+        setIsLoading(false);
+      }
+    };
+
+    loadPreviousSession();
+  }, [preloadPreviousSession, hasAsked, courseId, unitId, setResponse, setHasAsked, debug]);
+
   // Early return after all hooks have been called
   if (hasAsked && !isLoading) {
     return null;
@@ -381,6 +432,7 @@ AIEducatorLibraryAssistComponent.propTypes = {
   ),
   titleText: PropTypes.string,
   buttonText: PropTypes.string,
+  preloadPreviousSession: PropTypes.bool,
   customMessage: PropTypes.string,
   onSuccess: PropTypes.func,
   onError: PropTypes.func,
@@ -392,6 +444,7 @@ AIEducatorLibraryAssistComponent.defaultProps = {
   titleText: 'AI Assistant',
   buttonText: 'Start',
   customMessage: 'Use an AI workflow to create multiple answer questions from this unit in a content library',
+  preloadPreviousSession: false,
   onSuccess: null,
   onError: null,
   debug: false,

frontend/src/components/AIEducatorLibraryResponseComponent.jsx

@@ -4,6 +4,7 @@ import { Button, Alert, Card } from '@openedx/paragon';
 import {
   Warning,
 } from '@openedx/paragon/icons';
+import { prepareContextData, callWorkflowService } from '../services';
 
 /**
  * AI Response Component
@@ -18,6 +19,7 @@ const AIEducatorLibraryResponseComponent = ({
   customMessage,
   titleText,
   hyperlinkText,
+  contextData,
 }) => {
   // Don't render if no response or error
   if (!response && !error) {
@@ -28,6 +30,35 @@ const AIEducatorLibraryResponseComponent = ({
   const baseUrl = window.location.origin;
   const hyperlinkUrl = `${baseUrl}/${response}`;
 
+  const handleClearSession = async () => {
+    try {
+      // Prepare context data
+      const preparedContext = prepareContextData({
+        ...contextData,
+      });
+
+      // Make API call
+      await callWorkflowService({
+        context: preparedContext,
+        action: 'clear_session',
+        payload: {
+          requestId: `ai-request-${Date.now()}`,
+          courseId: preparedContext.courseId || null,
+        },
+      });
+    } catch (err) {
+      // eslint-disable-next-line no-console
+      console.error('[AISidebarResponse] Clear session error:', err);
+    }
+  };
+
+  const handleClearAndClose = async () => {
+    await handleClearSession();
+    if (onClear) {
+      onClear();
+    }
+  };
+
   return (
     <Card className="ai-educator-library-response mt-3 mb-3">
       <Card.Section>
@@ -74,12 +105,12 @@ const AIEducatorLibraryResponseComponent = ({
                   </a>
                 )}
               </div>
-              {onClear && (
+              {handleClearAndClose && (
                 <div className="d-flex justify-content-end mt-3">
                   <Button
                     variant="outline-secondary"
                     size="sm"
-                    onClick={onClear}
+                    onClick={handleClearAndClose}
                     className="py-1 px-2"
                   >
                     Clear
@@ -103,6 +134,7 @@ AIEducatorLibraryResponseComponent.propTypes = {
   customMessage: PropTypes.string,
   titleText: PropTypes.string,
   hyperlinkText: PropTypes.string,
+  contextData: PropTypes.shape({}),
 };
 
 AIEducatorLibraryResponseComponent.defaultProps = {
@@ -114,6 +146,7 @@ AIEducatorLibraryResponseComponent.defaultProps = {
   customMessage: 'Question generation success.',
   titleText: 'AI Assistant',
   hyperlinkText: 'View content ›',
+  contextData: {},
 };
 
 export default AIEducatorLibraryResponseComponent;