diff --git a/web/helpers/atoms/Assistant.atom.ts b/web/helpers/atoms/Assistant.atom.ts index cb50a0553..29da581c3 100644 --- a/web/helpers/atoms/Assistant.atom.ts +++ b/web/helpers/atoms/Assistant.atom.ts @@ -4,9 +4,15 @@ import { atomWithStorage } from 'jotai/utils' export const assistantsAtom = atom([]) +export const cachedAssistantAtom = atomWithStorage< + ThreadAssistantInfo | undefined +>('activeAssistant', undefined, undefined, { getOnInit: true }) /** * Get the current active assistant */ -export const activeAssistantAtom = atomWithStorage< - ThreadAssistantInfo | undefined ->('activeAssistant', undefined, undefined, { getOnInit: true }) +export const activeAssistantAtom = atom( + (get) => get(cachedAssistantAtom) ?? get(assistantsAtom)[0], + (_get, set, newAssistant: ThreadAssistantInfo) => { + set(cachedAssistantAtom, newAssistant) + } +)