Store last conversation in in localstorage #6993

Author: lgestc

x-pack/packages/kbn-elastic-assistant/impl/assistant/assistant_overlay/index.tsx

@@ -32,7 +32,7 @@ export const AssistantOverlay: React.FC = React.memo(() => {
     WELCOME_CONVERSATION_TITLE
   );
   const [promptContextId, setPromptContextId] = useState<string | undefined>();
-  const { setShowAssistantOverlay } = useAssistantContext();
+  const { setShowAssistantOverlay, localStorageLastConversationId } = useAssistantContext();
 
   // Bind `showAssistantOverlay` in SecurityAssistantContext to this modal instance
   const showOverlay = useCallback(
@@ -52,15 +52,25 @@ export const AssistantOverlay: React.FC = React.memo(() => {
     setShowAssistantOverlay(showOverlay);
   }, [setShowAssistantOverlay, showOverlay]);
 
+  // Called whenever platform specific shortcut for assistant is pressed
+  const handleShortcutPress = useCallback(() => {
+    // Try to restore the last conversation on shortcut pressed
+    if (!isModalVisible) {
+      setConversationId(localStorageLastConversationId || WELCOME_CONVERSATION_TITLE);
+    }
+
+    setIsModalVisible(!isModalVisible);
+  }, [isModalVisible, localStorageLastConversationId]);
+
   // Register keyboard listener to show the modal when cmd + ; is pressed
   const onKeyDown = useCallback(
     (event: KeyboardEvent) => {
       if (event.key === ';' && (isMac ? event.metaKey : event.ctrlKey)) {
         event.preventDefault();
-        setIsModalVisible(!isModalVisible);
+        handleShortcutPress();
       }
     },
-    [isModalVisible]
+    [handleShortcutPress]
   );
   useEvent('keydown', onKeyDown);
 

x-pack/packages/kbn-elastic-assistant/impl/assistant/index.tsx

@@ -78,6 +78,7 @@ const AssistantComponent: React.FC<Props> = ({
     getComments,
     http,
     promptContexts,
+    setLastConversationId,
     title,
   } = useAssistantContext();
   const [selectedPromptContexts, setSelectedPromptContexts] = useState<
@@ -98,6 +99,11 @@ const AssistantComponent: React.FC<Props> = ({
     [conversationId, conversations, createConversation, selectedConversationId]
   );
 
+  // Remember last selection for reuse after keyboard shortcut is pressed
+  useEffect(() => {
+    setLastConversationId(selectedConversationId);
+  }, [selectedConversationId, setLastConversationId]);
+
   // Welcome conversation is a special 'setup' case when no connector exists, mostly extracted to `ConnectorSetup` component,
   // but currently a bit of state is littered throughout the assistant component. TODO: clean up/isolate this state
   const welcomeConversation = useMemo(

x-pack/packages/kbn-elastic-assistant/impl/assistant_context/constants.tsx

@@ -8,3 +8,4 @@
 export const DEFAULT_ASSISTANT_NAMESPACE = 'elasticAssistantDefault';
 export const QUICK_PROMPT_LOCAL_STORAGE_KEY = 'quickPrompts';
 export const SYSTEM_PROMPT_LOCAL_STORAGE_KEY = 'systemPrompts';
+export const LAST_CONVERSATION_ID_LOCAL_STORAGE_KEY = 'lastConversationId';

x-pack/packages/kbn-elastic-assistant/impl/assistant_context/index.tsx

@@ -27,6 +27,7 @@ import { Prompt } from '../assistant/types';
 import { BASE_SYSTEM_PROMPTS } from '../content/prompts/system';
 import {
   DEFAULT_ASSISTANT_NAMESPACE,
+  LAST_CONVERSATION_ID_LOCAL_STORAGE_KEY,
   QUICK_PROMPT_LOCAL_STORAGE_KEY,
   SYSTEM_PROMPT_LOCAL_STORAGE_KEY,
 } from './constants';
@@ -96,6 +97,7 @@ interface UseAssistantContext {
     showAnonymizedValues: boolean;
   }) => EuiCommentProps[];
   http: HttpSetup;
+  localStorageLastConversationId: string | undefined;
   promptContexts: Record<string, PromptContext>;
   nameSpace: string;
   registerPromptContext: RegisterPromptContext;
@@ -104,6 +106,7 @@ interface UseAssistantContext {
   setConversations: React.Dispatch<React.SetStateAction<Record<string, Conversation>>>;
   setDefaultAllow: React.Dispatch<React.SetStateAction<string[]>>;
   setDefaultAllowReplacement: React.Dispatch<React.SetStateAction<string[]>>;
+  setLastConversationId: React.Dispatch<React.SetStateAction<string | undefined>>;
   setShowAssistantOverlay: (showAssistantOverlay: ShowAssistantOverlay) => void;
   showAssistantOverlay: ShowAssistantOverlay;
   title: string;
@@ -148,6 +151,9 @@ export const AssistantProvider: React.FC<AssistantProviderProps> = ({
     baseSystemPrompts
   );
 
+  const [localStorageLastConversationId, setLocalStorageLastConversationId] =
+    useLocalStorage<string>(`${nameSpace}.${LAST_CONVERSATION_ID_LOCAL_STORAGE_KEY}`);
+
   /**
    * Prompt contexts are used to provide components a way to register and make their data available to the assistant.
    */
@@ -248,6 +254,8 @@ export const AssistantProvider: React.FC<AssistantProviderProps> = ({
       showAssistantOverlay,
       title,
       unRegisterPromptContext,
+      localStorageLastConversationId,
+      setLastConversationId: setLocalStorageLastConversationId,
     }),
     [
       actionTypeRegistry,
@@ -263,6 +271,7 @@ export const AssistantProvider: React.FC<AssistantProviderProps> = ({
       defaultAllowReplacement,
       getComments,
       http,
+      localStorageLastConversationId,
       localStorageQuickPrompts,
       localStorageSystemPrompts,
       nameSpace,
@@ -271,6 +280,7 @@ export const AssistantProvider: React.FC<AssistantProviderProps> = ({
       registerPromptContext,
       setDefaultAllow,
       setDefaultAllowReplacement,
+      setLocalStorageLastConversationId,
       setLocalStorageQuickPrompts,
       setLocalStorageSystemPrompts,
       showAssistantOverlay,