From faf55b47733804eb7dc97e73698b67a6476920b1 Mon Sep 17 00:00:00 2001
From: ali asaria <aliasaria@users.noreply.github.com>
Date: Fri, 21 Feb 2025 15:50:40 -0500
Subject: [PATCH] make it so you can set a fixed folder from outside the
 documents component

---
 .../components/Experiment/Rag/Documents.tsx        | 14 ++++----------
 src/renderer/components/Experiment/Rag/index.tsx   |  2 +-
 2 files changed, 5 insertions(+), 11 deletions(-)

diff --git a/src/renderer/components/Experiment/Rag/Documents.tsx b/src/renderer/components/Experiment/Rag/Documents.tsx
index de27b787..142c7402 100644
--- a/src/renderer/components/Experiment/Rag/Documents.tsx
+++ b/src/renderer/components/Experiment/Rag/Documents.tsx
@@ -140,6 +140,7 @@ export default function Documents({
   experimentInfo,
   fullPage = false,
   additionalMessage = false,
+  fixedFolder = '',
 }) {
   const [doc, setDoc] = React.useState<Doc>('desc');
   const [selected, setSelected] = React.useState<readonly string[]>([]);
@@ -154,13 +155,7 @@ export default function Documents({
 
   const [loading, setLoading] = React.useState(false);
 
-  const [currentFolder, setCurrentFolder] = React.useState('');
-
-  React.useEffect(() => {
-    if (!additionalMessage) {
-      setCurrentFolder('rag');
-    }
-  }, [additionalMessage]);
+  const [currentFolder, setCurrentFolder] = React.useState(fixedFolder);
 
   const {
     data: rows,
@@ -474,10 +469,9 @@ export default function Documents({
           ) : (
             <>
               <Link
+                disabled={fixedFolder !== ''}
                 onClick={() => {
-                  additionalMessage
-                    ? setCurrentFolder('')
-                    : setCurrentFolder('rag');
+                  setCurrentFolder('');
                 }}
               >
                 Documents /
diff --git a/src/renderer/components/Experiment/Rag/index.tsx b/src/renderer/components/Experiment/Rag/index.tsx
index e008b8d7..7a754ee3 100644
--- a/src/renderer/components/Experiment/Rag/index.tsx
+++ b/src/renderer/components/Experiment/Rag/index.tsx
@@ -117,7 +117,7 @@ export default function DocumentSearch({ experimentInfo, setRagEngine }) {
             flexDirection: 'column',
           }}
         >
-          <Documents experimentInfo={experimentInfo} />
+          <Documents experimentInfo={experimentInfo} setFolder="rag" />
         </Box>
         <Box sx={{ flex: 1, display: 'flex', flexDirection: 'column' }}>
           <Query experimentInfo={experimentInfo} />
-- 
GitLab