diff --git a/src/renderer/components/Experiment/Rag/Documents.tsx b/src/renderer/components/Experiment/Rag/Documents.tsx index de27b787de11167c5610d7754eab406c1c483947..142c7402bd16fdc43a6e246ab749e798a67b96d9 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 e008b8d7c426fbe64493ac1e319a0b69c0b4ee44..7a754ee38daa1ab9d2e71f6ceca70e9453e5dec9 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} />