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