jamesf/Perplexica
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

bugfox(app): Change localStorage to sessionStorage

Browse source
This commit is contained in:
hacking-racoon 2024-12-24 21:58:41 +09:00
parent ee68095157
commit 7a7ebbba6d
7 changed files with 42 additions and 42 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

28
ui/components/ChatWindow.tsx
View file

@ -37,10 +37,10 @@ const useSocket = (
useEffect(() => { useEffect(() => {
if (!ws) { if (!ws) {
const connectWs = async () => { const connectWs = async () => {
let chatModel = localStorage.getItem('chatModel'); let chatModel = sessionStorage.getItem('chatModel');
let chatModelProvider = localStorage.getItem('chatModelProvider'); let chatModelProvider = sessionStorage.getItem('chatModelProvider');
let embeddingModel = localStorage.getItem('embeddingModel'); let embeddingModel = sessionStorage.getItem('embeddingModel');
let embeddingModelProvider = localStorage.getItem( let embeddingModelProvider = sessionStorage.getItem(
'embeddingModelProvider', 'embeddingModelProvider',
); );
@ -95,10 +95,10 @@ const useSocket = (
)[0]; )[0];
} }
localStorage.setItem('chatModel', chatModel!); sessionStorage.setItem('chatModel', chatModel!);
localStorage.setItem('chatModelProvider', chatModelProvider); sessionStorage.setItem('chatModelProvider', chatModelProvider);
localStorage.setItem('embeddingModel', embeddingModel!); sessionStorage.setItem('embeddingModel', embeddingModel!);
localStorage.setItem( sessionStorage.setItem(
'embeddingModelProvider', 'embeddingModelProvider',
embeddingModelProvider, embeddingModelProvider,
); );
@ -111,7 +111,7 @@ const useSocket = (
!chatModelProviders[chatModelProvider] !chatModelProviders[chatModelProvider]
) { ) {
chatModelProvider = Object.keys(chatModelProviders)[0]; chatModelProvider = Object.keys(chatModelProviders)[0];
localStorage.setItem('chatModelProvider', chatModelProvider); sessionStorage.setItem('chatModelProvider', chatModelProvider);
} }
if ( if (
@ -120,7 +120,7 @@ const useSocket = (
!chatModelProviders[chatModelProvider][chatModel] !chatModelProviders[chatModelProvider][chatModel]
) { ) {
chatModel = Object.keys(chatModelProviders[chatModelProvider])[0]; chatModel = Object.keys(chatModelProviders[chatModelProvider])[0];
localStorage.setItem('chatModel', chatModel); sessionStorage.setItem('chatModel', chatModel);
} }
if ( if (
@ -128,7 +128,7 @@ const useSocket = (
!embeddingModelProviders[embeddingModelProvider] !embeddingModelProviders[embeddingModelProvider]
) { ) {
embeddingModelProvider = Object.keys(embeddingModelProviders)[0]; embeddingModelProvider = Object.keys(embeddingModelProviders)[0];
localStorage.setItem( sessionStorage.setItem(
'embeddingModelProvider', 'embeddingModelProvider',
embeddingModelProvider, embeddingModelProvider,
); );
@ -141,7 +141,7 @@ const useSocket = (
embeddingModel = Object.keys( embeddingModel = Object.keys(
embeddingModelProviders[embeddingModelProvider], embeddingModelProviders[embeddingModelProvider],
)[0]; )[0];
localStorage.setItem('embeddingModel', embeddingModel); sessionStorage.setItem('embeddingModel', embeddingModel);
} }
} }
@ -154,11 +154,11 @@ const useSocket = (
if (chatModelProvider === 'custom_openai') { if (chatModelProvider === 'custom_openai') {
searchParams.append( searchParams.append(
'openAIApiKey', 'openAIApiKey',
localStorage.getItem('openAIApiKey')!, sessionStorage.getItem('openAIApiKey')!,
); );
searchParams.append( searchParams.append(
'openAIBaseURL', 'openAIBaseURL',
localStorage.getItem('openAIBaseURL')!, sessionStorage.getItem('openAIBaseURL')!,
); );
} }

4
ui/components/MessageInputActions/Attach.tsx
View file

@ -33,10 +33,10 @@ const Attach = ({
data.append('files', e.target.files![i]); data.append('files', e.target.files![i]);
} }
const embeddingModelProvider = localStorage.getItem( const embeddingModelProvider = sessionStorage.getItem(
'embeddingModelProvider', 'embeddingModelProvider',
); );
const embeddingModel = localStorage.getItem('embeddingModel'); const embeddingModel = sessionStorage.getItem('embeddingModel');
data.append('embedding_model_provider', embeddingModelProvider!); data.append('embedding_model_provider', embeddingModelProvider!);
data.append('embedding_model', embeddingModel!); data.append('embedding_model', embeddingModel!);

4
ui/components/MessageInputActions/AttachSmall.tsx
View file

@ -31,10 +31,10 @@ const AttachSmall = ({
data.append('files', e.target.files![i]); data.append('files', e.target.files![i]);
} }
const embeddingModelProvider = localStorage.getItem( const embeddingModelProvider = sessionStorage.getItem(
'embeddingModelProvider', 'embeddingModelProvider',
); );
const embeddingModel = localStorage.getItem('embeddingModel'); const embeddingModel = sessionStorage.getItem('embeddingModel');
data.append('embedding_model_provider', embeddingModelProvider!); data.append('embedding_model_provider', embeddingModelProvider!);
data.append('embedding_model', embeddingModel!); data.append('embedding_model', embeddingModel!);

8
ui/components/SearchImages.tsx
View file

@ -30,11 +30,11 @@ const SearchImages = ({
onClick={async () => { onClick={async () => {
setLoading(true); setLoading(true);
const chatModelProvider = localStorage.getItem('chatModelProvider'); const chatModelProvider = sessionStorage.getItem('chatModelProvider');
const chatModel = localStorage.getItem('chatModel'); const chatModel = sessionStorage.getItem('chatModel');
const customOpenAIBaseURL = localStorage.getItem('openAIBaseURL'); const customOpenAIBaseURL = sessionStorage.getItem('openAIBaseURL');
const customOpenAIKey = localStorage.getItem('openAIApiKey'); const customOpenAIKey = sessionStorage.getItem('openAIApiKey');
const res = await fetch( const res = await fetch(
`${process.env.NEXT_PUBLIC_API_URL}/images`, `${process.env.NEXT_PUBLIC_API_URL}/images`,

8
ui/components/SearchVideos.tsx
View file

@ -43,11 +43,11 @@ const Searchvideos = ({
onClick={async () => { onClick={async () => {
setLoading(true); setLoading(true);
const chatModelProvider = localStorage.getItem('chatModelProvider'); const chatModelProvider = sessionStorage.getItem('chatModelProvider');
const chatModel = localStorage.getItem('chatModel'); const chatModel = sessionStorage.getItem('chatModel');
const customOpenAIBaseURL = localStorage.getItem('openAIBaseURL'); const customOpenAIBaseURL = sessionStorage.getItem('openAIBaseURL');
const customOpenAIKey = localStorage.getItem('openAIApiKey'); const customOpenAIKey = sessionStorage.getItem('openAIApiKey');
const res = await fetch( const res = await fetch(
`${process.env.NEXT_PUBLIC_API_URL}/videos`, `${process.env.NEXT_PUBLIC_API_URL}/videos`,

24
ui/components/SettingsDialog.tsx
View file

@ -123,22 +123,22 @@ const SettingsDialog = ({
: ''; : '';
const chatModelProvider = const chatModelProvider =
localStorage.getItem('chatModelProvider') || sessionStorage.getItem('chatModelProvider') ||
defaultChatModelProvider || defaultChatModelProvider ||
''; '';
const chatModel = const chatModel =
localStorage.getItem('chatModel') || sessionStorage.getItem('chatModel') ||
(data.chatModelProviders && (data.chatModelProviders &&
data.chatModelProviders[chatModelProvider]?.length > 0 data.chatModelProviders[chatModelProvider]?.length > 0
? data.chatModelProviders[chatModelProvider][0].name ? data.chatModelProviders[chatModelProvider][0].name
: undefined) || : undefined) ||
''; '';
const embeddingModelProvider = const embeddingModelProvider =
localStorage.getItem('embeddingModelProvider') || sessionStorage.getItem('embeddingModelProvider') ||
defaultEmbeddingModelProvider || defaultEmbeddingModelProvider ||
''; '';
const embeddingModel = const embeddingModel =
localStorage.getItem('embeddingModel') || sessionStorage.getItem('embeddingModel') ||
(data.embeddingModelProviders && (data.embeddingModelProviders &&
data.embeddingModelProviders[embeddingModelProvider]?.[0].name) || data.embeddingModelProviders[embeddingModelProvider]?.[0].name) ||
''; '';
@ -147,8 +147,8 @@ const SettingsDialog = ({
setSelectedChatModel(chatModel); setSelectedChatModel(chatModel);
setSelectedEmbeddingModelProvider(embeddingModelProvider); setSelectedEmbeddingModelProvider(embeddingModelProvider);
setSelectedEmbeddingModel(embeddingModel); setSelectedEmbeddingModel(embeddingModel);
setCustomOpenAIApiKey(localStorage.getItem('openAIApiKey') || ''); setCustomOpenAIApiKey(sessionStorage.getItem('openAIApiKey') || '');
setCustomOpenAIBaseURL(localStorage.getItem('openAIBaseURL') || ''); setCustomOpenAIBaseURL(sessionStorage.getItem('openAIBaseURL') || '');
setChatModels(data.chatModelProviders || {}); setChatModels(data.chatModelProviders || {});
setEmbeddingModels(data.embeddingModelProviders || {}); setEmbeddingModels(data.embeddingModelProviders || {});
setIsLoading(false); setIsLoading(false);
@ -171,15 +171,15 @@ const SettingsDialog = ({
body: JSON.stringify(config), body: JSON.stringify(config),
}); });
localStorage.setItem('chatModelProvider', selectedChatModelProvider!); sessionStorage.setItem('chatModelProvider', selectedChatModelProvider!);
localStorage.setItem('chatModel', selectedChatModel!); sessionStorage.setItem('chatModel', selectedChatModel!);
localStorage.setItem( sessionStorage.setItem(
'embeddingModelProvider', 'embeddingModelProvider',
selectedEmbeddingModelProvider!, selectedEmbeddingModelProvider!,
); );
localStorage.setItem('embeddingModel', selectedEmbeddingModel!); sessionStorage.setItem('embeddingModel', selectedEmbeddingModel!);
localStorage.setItem('openAIApiKey', customOpenAIApiKey!); sessionStorage.setItem('openAIApiKey', customOpenAIApiKey!);
localStorage.setItem('openAIBaseURL', customOpenAIBaseURL!); sessionStorage.setItem('openAIBaseURL', customOpenAIBaseURL!);
} catch (err) { } catch (err) {
console.log(err); console.log(err);
} finally { } finally {

8
ui/lib/actions.ts
View file

@ -1,11 +1,11 @@
import { Message } from '@/components/ChatWindow'; import { Message } from '@/components/ChatWindow';
export const getSuggestions = async (chatHisory: Message[]) => { export const getSuggestions = async (chatHisory: Message[]) => {
const chatModel = localStorage.getItem('chatModel'); const chatModel = sessionStorage.getItem('chatModel');
const chatModelProvider = localStorage.getItem('chatModelProvider'); const chatModelProvider = sessionStorage.getItem('chatModelProvider');
const customOpenAIKey = localStorage.getItem('openAIApiKey'); const customOpenAIKey = sessionStorage.getItem('openAIApiKey');
const customOpenAIBaseURL = localStorage.getItem('openAIBaseURL'); const customOpenAIBaseURL = sessionStorage.getItem('openAIBaseURL');
const res = await fetch(`${process.env.NEXT_PUBLIC_API_URL}/suggestions`, { const res = await fetch(`${process.env.NEXT_PUBLIC_API_URL}/suggestions`, {
method: 'POST', method: 'POST',