From 7a7ebbba6df79c1e67a52f58b794adcf93390939 Mon Sep 17 00:00:00 2001 From: hacking-racoon Date: Tue, 24 Dec 2024 21:58:41 +0900 Subject: [PATCH] bugfox(app): Change localStorage to sessionStorage --- ui/components/ChatWindow.tsx | 28 +++++++++---------- ui/components/MessageInputActions/Attach.tsx | 4 +-- .../MessageInputActions/AttachSmall.tsx | 4 +-- ui/components/SearchImages.tsx | 8 +++--- ui/components/SearchVideos.tsx | 8 +++--- ui/components/SettingsDialog.tsx | 24 ++++++++-------- ui/lib/actions.ts | 8 +++--- 7 files changed, 42 insertions(+), 42 deletions(-) diff --git a/ui/components/ChatWindow.tsx b/ui/components/ChatWindow.tsx index fd3d0a6..544ad36 100644 --- a/ui/components/ChatWindow.tsx +++ b/ui/components/ChatWindow.tsx @@ -37,10 +37,10 @@ const useSocket = ( useEffect(() => { if (!ws) { const connectWs = async () => { - let chatModel = localStorage.getItem('chatModel'); - let chatModelProvider = localStorage.getItem('chatModelProvider'); - let embeddingModel = localStorage.getItem('embeddingModel'); - let embeddingModelProvider = localStorage.getItem( + let chatModel = sessionStorage.getItem('chatModel'); + let chatModelProvider = sessionStorage.getItem('chatModelProvider'); + let embeddingModel = sessionStorage.getItem('embeddingModel'); + let embeddingModelProvider = sessionStorage.getItem( 'embeddingModelProvider', ); @@ -95,10 +95,10 @@ const useSocket = ( )[0]; } - localStorage.setItem('chatModel', chatModel!); - localStorage.setItem('chatModelProvider', chatModelProvider); - localStorage.setItem('embeddingModel', embeddingModel!); - localStorage.setItem( + sessionStorage.setItem('chatModel', chatModel!); + sessionStorage.setItem('chatModelProvider', chatModelProvider); + sessionStorage.setItem('embeddingModel', embeddingModel!); + sessionStorage.setItem( 'embeddingModelProvider', embeddingModelProvider, ); @@ -111,7 +111,7 @@ const useSocket = ( !chatModelProviders[chatModelProvider] ) { chatModelProvider = Object.keys(chatModelProviders)[0]; - localStorage.setItem('chatModelProvider', chatModelProvider); + sessionStorage.setItem('chatModelProvider', chatModelProvider); } if ( @@ -120,7 +120,7 @@ const useSocket = ( !chatModelProviders[chatModelProvider][chatModel] ) { chatModel = Object.keys(chatModelProviders[chatModelProvider])[0]; - localStorage.setItem('chatModel', chatModel); + sessionStorage.setItem('chatModel', chatModel); } if ( @@ -128,7 +128,7 @@ const useSocket = ( !embeddingModelProviders[embeddingModelProvider] ) { embeddingModelProvider = Object.keys(embeddingModelProviders)[0]; - localStorage.setItem( + sessionStorage.setItem( 'embeddingModelProvider', embeddingModelProvider, ); @@ -141,7 +141,7 @@ const useSocket = ( embeddingModel = Object.keys( embeddingModelProviders[embeddingModelProvider], )[0]; - localStorage.setItem('embeddingModel', embeddingModel); + sessionStorage.setItem('embeddingModel', embeddingModel); } } @@ -154,11 +154,11 @@ const useSocket = ( if (chatModelProvider === 'custom_openai') { searchParams.append( 'openAIApiKey', - localStorage.getItem('openAIApiKey')!, + sessionStorage.getItem('openAIApiKey')!, ); searchParams.append( 'openAIBaseURL', - localStorage.getItem('openAIBaseURL')!, + sessionStorage.getItem('openAIBaseURL')!, ); } diff --git a/ui/components/MessageInputActions/Attach.tsx b/ui/components/MessageInputActions/Attach.tsx index 61cc86a..84b976c 100644 --- a/ui/components/MessageInputActions/Attach.tsx +++ b/ui/components/MessageInputActions/Attach.tsx @@ -33,10 +33,10 @@ const Attach = ({ data.append('files', e.target.files![i]); } - const embeddingModelProvider = localStorage.getItem( + const embeddingModelProvider = sessionStorage.getItem( 'embeddingModelProvider', ); - const embeddingModel = localStorage.getItem('embeddingModel'); + const embeddingModel = sessionStorage.getItem('embeddingModel'); data.append('embedding_model_provider', embeddingModelProvider!); data.append('embedding_model', embeddingModel!); diff --git a/ui/components/MessageInputActions/AttachSmall.tsx b/ui/components/MessageInputActions/AttachSmall.tsx index 3514a58..37fa767 100644 --- a/ui/components/MessageInputActions/AttachSmall.tsx +++ b/ui/components/MessageInputActions/AttachSmall.tsx @@ -31,10 +31,10 @@ const AttachSmall = ({ data.append('files', e.target.files![i]); } - const embeddingModelProvider = localStorage.getItem( + const embeddingModelProvider = sessionStorage.getItem( 'embeddingModelProvider', ); - const embeddingModel = localStorage.getItem('embeddingModel'); + const embeddingModel = sessionStorage.getItem('embeddingModel'); data.append('embedding_model_provider', embeddingModelProvider!); data.append('embedding_model', embeddingModel!); diff --git a/ui/components/SearchImages.tsx b/ui/components/SearchImages.tsx index b083af7..85cefea 100644 --- a/ui/components/SearchImages.tsx +++ b/ui/components/SearchImages.tsx @@ -30,11 +30,11 @@ const SearchImages = ({ onClick={async () => { setLoading(true); - const chatModelProvider = localStorage.getItem('chatModelProvider'); - const chatModel = localStorage.getItem('chatModel'); + const chatModelProvider = sessionStorage.getItem('chatModelProvider'); + const chatModel = sessionStorage.getItem('chatModel'); - const customOpenAIBaseURL = localStorage.getItem('openAIBaseURL'); - const customOpenAIKey = localStorage.getItem('openAIApiKey'); + const customOpenAIBaseURL = sessionStorage.getItem('openAIBaseURL'); + const customOpenAIKey = sessionStorage.getItem('openAIApiKey'); const res = await fetch( `${process.env.NEXT_PUBLIC_API_URL}/images`, diff --git a/ui/components/SearchVideos.tsx b/ui/components/SearchVideos.tsx index 2d820ef..b7de47d 100644 --- a/ui/components/SearchVideos.tsx +++ b/ui/components/SearchVideos.tsx @@ -43,11 +43,11 @@ const Searchvideos = ({ onClick={async () => { setLoading(true); - const chatModelProvider = localStorage.getItem('chatModelProvider'); - const chatModel = localStorage.getItem('chatModel'); + const chatModelProvider = sessionStorage.getItem('chatModelProvider'); + const chatModel = sessionStorage.getItem('chatModel'); - const customOpenAIBaseURL = localStorage.getItem('openAIBaseURL'); - const customOpenAIKey = localStorage.getItem('openAIApiKey'); + const customOpenAIBaseURL = sessionStorage.getItem('openAIBaseURL'); + const customOpenAIKey = sessionStorage.getItem('openAIApiKey'); const res = await fetch( `${process.env.NEXT_PUBLIC_API_URL}/videos`, diff --git a/ui/components/SettingsDialog.tsx b/ui/components/SettingsDialog.tsx index 163857b..d044db4 100644 --- a/ui/components/SettingsDialog.tsx +++ b/ui/components/SettingsDialog.tsx @@ -123,22 +123,22 @@ const SettingsDialog = ({ : ''; const chatModelProvider = - localStorage.getItem('chatModelProvider') || + sessionStorage.getItem('chatModelProvider') || defaultChatModelProvider || ''; const chatModel = - localStorage.getItem('chatModel') || + sessionStorage.getItem('chatModel') || (data.chatModelProviders && data.chatModelProviders[chatModelProvider]?.length > 0 ? data.chatModelProviders[chatModelProvider][0].name : undefined) || ''; const embeddingModelProvider = - localStorage.getItem('embeddingModelProvider') || + sessionStorage.getItem('embeddingModelProvider') || defaultEmbeddingModelProvider || ''; const embeddingModel = - localStorage.getItem('embeddingModel') || + sessionStorage.getItem('embeddingModel') || (data.embeddingModelProviders && data.embeddingModelProviders[embeddingModelProvider]?.[0].name) || ''; @@ -147,8 +147,8 @@ const SettingsDialog = ({ setSelectedChatModel(chatModel); setSelectedEmbeddingModelProvider(embeddingModelProvider); setSelectedEmbeddingModel(embeddingModel); - setCustomOpenAIApiKey(localStorage.getItem('openAIApiKey') || ''); - setCustomOpenAIBaseURL(localStorage.getItem('openAIBaseURL') || ''); + setCustomOpenAIApiKey(sessionStorage.getItem('openAIApiKey') || ''); + setCustomOpenAIBaseURL(sessionStorage.getItem('openAIBaseURL') || ''); setChatModels(data.chatModelProviders || {}); setEmbeddingModels(data.embeddingModelProviders || {}); setIsLoading(false); @@ -171,15 +171,15 @@ const SettingsDialog = ({ body: JSON.stringify(config), }); - localStorage.setItem('chatModelProvider', selectedChatModelProvider!); - localStorage.setItem('chatModel', selectedChatModel!); - localStorage.setItem( + sessionStorage.setItem('chatModelProvider', selectedChatModelProvider!); + sessionStorage.setItem('chatModel', selectedChatModel!); + sessionStorage.setItem( 'embeddingModelProvider', selectedEmbeddingModelProvider!, ); - localStorage.setItem('embeddingModel', selectedEmbeddingModel!); - localStorage.setItem('openAIApiKey', customOpenAIApiKey!); - localStorage.setItem('openAIBaseURL', customOpenAIBaseURL!); + sessionStorage.setItem('embeddingModel', selectedEmbeddingModel!); + sessionStorage.setItem('openAIApiKey', customOpenAIApiKey!); + sessionStorage.setItem('openAIBaseURL', customOpenAIBaseURL!); } catch (err) { console.log(err); } finally { diff --git a/ui/lib/actions.ts b/ui/lib/actions.ts index a4409b0..451fe01 100644 --- a/ui/lib/actions.ts +++ b/ui/lib/actions.ts @@ -1,11 +1,11 @@ import { Message } from '@/components/ChatWindow'; export const getSuggestions = async (chatHisory: Message[]) => { - const chatModel = localStorage.getItem('chatModel'); - const chatModelProvider = localStorage.getItem('chatModelProvider'); + const chatModel = sessionStorage.getItem('chatModel'); + const chatModelProvider = sessionStorage.getItem('chatModelProvider'); - const customOpenAIKey = localStorage.getItem('openAIApiKey'); - const customOpenAIBaseURL = localStorage.getItem('openAIBaseURL'); + const customOpenAIKey = sessionStorage.getItem('openAIApiKey'); + const customOpenAIBaseURL = sessionStorage.getItem('openAIBaseURL'); const res = await fetch(`${process.env.NEXT_PUBLIC_API_URL}/suggestions`, { method: 'POST',