remember what interaction mode you are in

transformerlab · Sep 16, 2024 · 85336f2 · 85336f2
1 parent 4556a63
commit 85336f2
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 1 deletion.
diff --git a/src/renderer/components/Experiment/Interact/Interact.tsx b/src/renderer/components/Experiment/Interact/Interact.tsx
@@ -43,9 +43,10 @@ export default function Chat({
   experimentInfo,
   experimentInfoMutate,
   setRagEngine,
+  mode,
+  setMode,
 }) {
   const { models, isError, isLoading } = chatAPI.useModelStatus();
-  const [mode, setMode] = React.useState('chat');
   const [conversationId, setConversationId] = React.useState(null);
   const [chats, setChats] = React.useState([]);
   const [isThinking, setIsThinking] = React.useState(false);

diff --git a/src/renderer/components/MainAppPanel.tsx b/src/renderer/components/MainAppPanel.tsx
@@ -27,6 +27,7 @@ import ExperimentNotes from './Experiment/ExperimentNotes';
 import TransformerLabSettings from './TransformerLabSettings';
 import Logs from './Logs';
 import FoundationHome from './Experiment/Foundation';
+import { useState } from 'react';
 
 // This component renders the main content of the app that is shown
 // On the rightmost side, regardless of what menu items are selected
@@ -37,6 +38,8 @@ export default function MainAppPanel({
   experimentInfoMutate,
 }) {
   const navigate = useNavigate();
+  const [selectedInteractSubpage, setSelectedInteractSubpage] =
+    useState('chat');
 
   function setFoundation(model) {
     let model_name = '';
@@ -176,6 +179,8 @@ export default function MainAppPanel({
             experimentInfo={experimentInfo}
             experimentInfoMutate={experimentInfoMutate}
             setRagEngine={setRagEngine}
+            mode={selectedInteractSubpage}
+            setMode={setSelectedInteractSubpage}
           />
         }
       />