move more to context

Author: aliasaria

src/renderer/components/Experiment/Documents/index.tsx

@@ -1,7 +1,7 @@
 import { Sheet } from '@mui/joy';
 import Documents from '../../Shared/Documents';
 
-export default function BigDocumentsPage({ experimentInfo }) {
+export default function BigDocumentsPage({}) {
   return (
     <Sheet
       sx={{
@@ -12,7 +12,7 @@ export default function BigDocumentsPage({ experimentInfo }) {
         pb: 2,
       }}
     >
-      <Documents experimentInfo={experimentInfo} fullPage additionalMessage />
+      <Documents fullPage additionalMessage />
     </Sheet>
   );
 }

src/renderer/components/Experiment/Eval/Eval.tsx

@@ -2,10 +2,12 @@
 
 import Sheet from '@mui/joy/Sheet';
 
+import { useExperimentInfo } from 'renderer/lib/ExperimentInfoContext.js';
 import EvalJobsTable from './EvalJobsTable.tsx';
 import EvalTasksTable from './EvalTasksTable';
 
-export default function Eval({ experimentInfo, experimentInfoMutate }) {
+export default function Eval({}) {
+  const { experimentInfo } = useExperimentInfo();
   if (!experimentInfo) {
     return 'No experiment selected';
   }

src/renderer/components/Experiment/Generate/Generate.tsx

@@ -9,6 +9,7 @@ import { Typography, Option, Stack } from '@mui/joy';
 
 import GenerateJobsTable from './GenerateJobsTable';
 import GenerateTasksTable from './GenerateTasksTable';
+import { useExperimentInfo } from 'renderer/lib/ExperimentInfoContext';
 
 function getTemplateParametersForPlugin(pluginName, plugins) {
   if (!pluginName || !plugins) {
@@ -28,11 +29,8 @@ function getTemplateParametersForPlugin(pluginName, plugins) {
 
 const fetcher = (url) => fetch(url).then((res) => res.json());
 
-export default function Generate({
-  experimentInfo,
-  addGeneration,
-  experimentInfoMutate,
-}) {
+export default function Generate({ addGeneration }) {
+  const { experimentInfo, experimentInfoMutate } = useExperimentInfo();
   const [currentPlugin, setCurrentPlugin] = useState('');
   const [currentGenerationId, setCurrentGenerationId] = useState('');
 

src/renderer/components/Experiment/Train/TrainLoRA.tsx

@@ -51,6 +51,7 @@ import TensorboardModal from './TensorboardModal';
 import ViewOutputModal from './ViewOutputModal';
 import ImportRecipeModal from './ImportRecipeModal';
 import ViewEvalImagesModal from './ViewEvalImagesModal';
+import { useExperimentInfo } from 'renderer/lib/ExperimentInfoContext';
 dayjs.extend(relativeTime);
 var duration = require('dayjs/plugin/duration');
 dayjs.extend(duration);
@@ -96,7 +97,8 @@ function formatJobConfig(c): ReactElement {
 
 const fetcher = (url) => fetch(url).then((res) => res.json());
 
-export default function TrainLoRA({ experimentInfo }) {
+export default function TrainLoRA({}) {
+  const { experimentInfo } = useExperimentInfo();
   const [open, setOpen] = useState(false);
   const [currentTensorboardForModal, setCurrentTensorboardForModal] =
     useState(-1);

src/renderer/components/MainAppPanel.tsx

@@ -363,34 +363,16 @@ export default function MainAppPanel({ setLogsDrawerOpen = null }) {
         />
         <Route path="/experiment/embeddings" element={<Embeddings />} />
         <Route path="/experiment/tokenize" element={<Tokenize />} />
-        <Route
-          path="/experiment/training"
-          element={<TrainLoRA experimentInfo={experimentInfo} />}
-        />
+        <Route path="/experiment/training" element={<TrainLoRA />} />
         <Route
           path="/experiment/eval"
-          element={
-            <Eval
-              experimentInfo={experimentInfo}
-              addEvaluation={experimentAddEvaluation}
-              experimentInfoMutate={experimentInfoMutate}
-            />
-          }
+          element={<Eval addEvaluation={experimentAddEvaluation} />}
         />
         <Route
           path="/experiment/generate"
-          element={
-            <Generate
-              experimentInfo={experimentInfo}
-              addGeneration={experimentAddGeneration}
-              experimentInfoMutate={experimentInfoMutate}
-            />
-          }
-        />
-        <Route
-          path="/experiment/documents"
-          element={<Documents experimentInfo={experimentInfo} />}
+          element={<Generate addGeneration={experimentAddGeneration} />}
         />
+        <Route path="/experiment/documents" element={<Documents />} />
         <Route
           path="/experiment/rag"
           element={

src/renderer/components/Shared/Documents/index.tsx

@@ -53,6 +53,7 @@ import { FaRegFilePdf } from 'react-icons/fa6';
 import { LuFileJson } from 'react-icons/lu';
 import TinyButton from 'renderer/components/Shared/TinyButton';
 import * as chatAPI from '../../../lib/transformerlab-api-sdk';
+import { useExperimentInfo } from 'renderer/lib/ExperimentInfoContext';
 
 function RowMenu({ experimentInfo, filename, foldername, mutate, row }) {
   return (
@@ -314,11 +315,11 @@ const fetcher = (url) => fetch(url).then((res) => res.json());
 type Order = 'asc' | 'desc';
 
 export default function Documents({
-  experimentInfo,
   fullPage = false,
   additionalMessage = false,
   fixedFolder = '',
 }) {
+  const { experimentInfo } = useExperimentInfo();
   const [doc, setDoc] = React.useState<Doc>('desc');
   const [open, setOpen] = React.useState(false);
   const [dropzoneActive, setDropzoneActive] = React.useState(false);