diff --git a/web/src/components/message-input/next.tsx b/web/src/components/message-input/next.tsx index 52b4029130..3c37e6b72f 100644 --- a/web/src/components/message-input/next.tsx +++ b/web/src/components/message-input/next.tsx @@ -19,6 +19,7 @@ import * as React from 'react'; import { useCallback, useEffect, useMemo, useState } from 'react'; import { useTranslation } from 'react-i18next'; import { toast } from 'sonner'; +import storage from '@/utils/authorization-util'; import { SelectWithSearch } from '../originui/select-with-search'; import { AudioButton } from '../ui/audio-button'; @@ -73,7 +74,9 @@ export function NextMessageInput({ null, ); - const [enableThinking, setEnableThinking] = useState('1'); + const [enableThinking, setEnableThinking] = useState( + () => storage.getThinkingLevel(), + ); const [enableInternet, setEnableInternet] = useState(false); const thinkingOptions = useMemo( @@ -105,6 +108,7 @@ export function NextMessageInput({ const handleThinkingChange = useCallback((value: string) => { setEnableThinking(value); + storage.setThinkingLevel(value); }, []); const handleInternetToggle = useCallback(() => { diff --git a/web/src/constants/authorization.ts b/web/src/constants/authorization.ts index 85b0930bf0..5516be221a 100644 --- a/web/src/constants/authorization.ts +++ b/web/src/constants/authorization.ts @@ -1,3 +1,4 @@ export const Authorization = 'Authorization'; export const Token = 'token'; export const UserInfo = 'userInfo'; +export const ThinkingLevel = 'thinkingLevel'; diff --git a/web/src/utils/authorization-util.ts b/web/src/utils/authorization-util.ts index e25e4915f1..97bfd66c3b 100644 --- a/web/src/utils/authorization-util.ts +++ b/web/src/utils/authorization-util.ts @@ -1,4 +1,9 @@ -import { Authorization, Token, UserInfo } from '@/constants/authorization'; +import { + Authorization, + ThinkingLevel, + Token, + UserInfo, +} from '@/constants/authorization'; import { getSearchValue } from './common-util'; const KeySet = [Authorization, Token, UserInfo]; @@ -45,6 +50,12 @@ const storage = { getLanguage: (): string => { return localStorage.getItem('lng') as string; }, + setThinkingLevel: (level: string) => { + localStorage.setItem(ThinkingLevel, level); + }, + getThinkingLevel: (): string => { + return localStorage.getItem(ThinkingLevel) || '1'; + }, }; export const getAuthorization = () => {