helper.ts 696 B

123456789101112131415161718192021222324252627282930
  1. import { ss } from '@/utils/storage'
  2. const LOCAL_NAME = 'settingsStorage'
  3. export interface SettingsState {
  4. systemMessage: string
  5. temperature: number
  6. top_p: number
  7. }
  8. export function defaultSetting(): SettingsState {
  9. return {
  10. systemMessage: 'Follow the user\'s instructions carefully. Respond using markdown.',
  11. temperature: 0.8,
  12. top_p: 1,
  13. }
  14. }
  15. export function getLocalState(): SettingsState {
  16. const localSetting: SettingsState | undefined = ss.get(LOCAL_NAME)
  17. return { ...defaultSetting(), ...localSetting }
  18. }
  19. export function setLocalState(setting: SettingsState): void {
  20. ss.set(LOCAL_NAME, setting)
  21. }
  22. export function removeLocalState() {
  23. ss.remove(LOCAL_NAME)
  24. }