Nicholai/jan
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: performance issue with atom storage persistence

Browse Source
This commit is contained in:
Louis 2024-12-17 15:44:13 +07:00
parent f2db31781e
commit 2b0a4c433c
No known key found for this signature in database
GPG Key ID: 44FA9F4D33C37DE2
1 changed files with 5 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
web/helpers/atoms/ChatMessage.atom.ts
View File

@ -6,7 +6,7 @@ import {
} from '@janhq/core' } from '@janhq/core'
import { atom } from 'jotai' import { atom } from 'jotai'
import { atomWithStorage } from 'jotai/utils' import { atomWithStorage, createJSONStorage } from 'jotai/utils'
import { import {
getActiveThreadIdAtom, getActiveThreadIdAtom,
@ -16,13 +16,16 @@ import {
import { TokenSpeed } from '@/types/token' import { TokenSpeed } from '@/types/token'
const CHAT_MESSAGE_NAME = 'chatMessages' const CHAT_MESSAGE_NAME = 'chatMessages'
const storage = createJSONStorage<Record<string, ThreadMessage[]>>(
() => sessionStorage
)
/** /**
* Stores all chat messages for all threads * Stores all chat messages for all threads
*/ */
export const chatMessages = atomWithStorage<Record<string, ThreadMessage[]>>( export const chatMessages = atomWithStorage<Record<string, ThreadMessage[]>>(
CHAT_MESSAGE_NAME, CHAT_MESSAGE_NAME,
{}, {},
undefined, storage,
{ getOnInit: true } { getOnInit: true }
) )