diff --git a/src/apps/beam/AppBeam.tsx b/src/apps/beam/AppBeam.tsx index e0fad9c00..5091f82aa 100644 --- a/src/apps/beam/AppBeam.tsx +++ b/src/apps/beam/AppBeam.tsx @@ -19,8 +19,7 @@ function initTestConversation(): DConversation { return conversation; } -function initTestBeamStore(messages: DMessage[]): BeamStoreApi { - const beamStore = createBeamStore(); +function initTestBeamStore(messages: DMessage[], beamStore: BeamStoreApi = createBeamStore()): BeamStoreApi { beamStore.getState().open(messages, useModelsStore.getState().chatLLMId); return beamStore; } @@ -47,7 +46,7 @@ export function AppBeam() { {/* 'open' */}