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(() => {
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')!,
);
}

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

@ -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!);

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

@ -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!);

8
ui/components/SearchImages.tsx
View file

@ -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`,

8
ui/components/SearchVideos.tsx
View file

@ -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`,

24
ui/components/SettingsDialog.tsx
View file

@ -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 {

8
ui/lib/actions.ts
View file

@ -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',