Merge pull request #791 from arc53/bug/sticky-prompts

Add getLocalPrompt and setLocalPrompt functions to preferenceApi.ts

frontend/src/preferences/preferenceApi.ts

@@ -66,10 +66,19 @@ export function getLocalRecentDocs(): string | null {
   return doc;
 }
 
+export function getLocalPrompt(): string | null {
+  const prompt = localStorage.getItem('DocsGPTPrompt');
+  return prompt;
+}
+
 export function setLocalApiKey(key: string): void {
   localStorage.setItem('DocsGPTApiKey', key);
 }
 
+export function setLocalPrompt(prompt: string): void {
+  localStorage.setItem('DocsGPTPrompt', prompt);
+}
+
 export function setLocalRecentDocs(doc: Doc): void {
   localStorage.setItem('DocsGPTRecentDocs', JSON.stringify(doc));
   let namePath = doc.name;

frontend/src/preferences/preferenceSlice.ts

@@ -81,6 +81,16 @@ prefListenerMiddleware.startListening({
   },
 });
 
+prefListenerMiddleware.startListening({
+  matcher: isAnyOf(setPrompt),
+  effect: (action, listenerApi) => {
+    localStorage.setItem(
+      'DocsGPTPrompt',
+      JSON.stringify((listenerApi.getState() as RootState).preference.prompt),
+    );
+  },
+});
+
 export const selectApiKey = (state: RootState) => state.preference.apiKey;
 export const selectApiKeyStatus = (state: RootState) =>
   !!state.preference.apiKey;

frontend/src/store.ts

@@ -6,14 +6,18 @@ import {
 } from './preferences/preferenceSlice';
 
 const key = localStorage.getItem('DocsGPTApiKey');
+const prompt = localStorage.getItem('DocsGPTPrompt');
 const doc = localStorage.getItem('DocsGPTRecentDocs');
 
 const store = configureStore({
   preloadedState: {
     preference: {
       apiKey: key ?? '',
       selectedDocs: doc !== null ? JSON.parse(doc) : null,
-      prompt: { name: 'default', id: 'default', type: 'private' },
+      prompt:
+        prompt !== null
+          ? JSON.parse(prompt)
+          : { name: 'default', id: 'default', type: 'private' },
       conversations: null,
       sourceDocs: [
         {