From 69bb78c8befc75f87968e7d86c4541a5ac6e276e Mon Sep 17 00:00:00 2001
From: Enrico Ros <enrico.ros@gmail.com>
Date: Tue, 12 Mar 2024 18:54:45 -0700
Subject: [PATCH] Beam: reduce direct open calls

---
 src/apps/beam/AppBeam.tsx | 5 ++---
 1 file changed, 2 insertions(+), 3 deletions(-)

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' */}
     <Button size='sm' variant='plain' color='neutral' onClick={() => {
       conversation.current = initTestConversation();
-      beamStoreApi.getState().open(conversation.current.messages, useModelsStore.getState().chatLLMId);
+      initTestBeamStore(conversation.current.messages, beamStoreApi);
     }}>
       .open
     </Button>