diff --git a/src/renderer/components/Experiment/Rag/Documents.tsx b/src/renderer/components/Experiment/Rag/Documents.tsx index 28cf6a116f7f696f616be9ab029c0e67d3a681c5..6a84c3251dee594eeff307a704657e282574dd33 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, @@ -405,6 +400,10 @@ export default function Documents({ </FormControl> </React.Fragment> ); + + /**** + * Main Documents Component is Here + */ return ( <> <Modal @@ -470,9 +469,9 @@ export default function Documents({ ) : ( <> <Link + disabled={fixedFolder !== ''} onClick={() => { - additionalMessage? setCurrentFolder('') : - setCurrentFolder('rag'); + setCurrentFolder(''); }} > Documents / @@ -712,6 +711,11 @@ export default function Documents({ </tr> </thead> <tbody> + {rows?.status == 'error' && ( + <tr> + <td colSpan={2}>{/*rows?.message*/}</td> + </tr> + )} {rows?.length == 0 && ( <tr> <td colSpan={2} style={{ padding: '2rem' }}> @@ -745,7 +749,8 @@ export default function Documents({ </Stack> {additionalMessage && ( <Typography level="body-xs" mt={1}> - Documents for RAG should be uploaded in a folder called "rag" and only those will be indexed for RAG. + Documents for RAG should be uploaded in a folder called "rag" and only + those will be indexed for RAG. </Typography> )} </> diff --git a/src/renderer/components/Experiment/Rag/index.tsx b/src/renderer/components/Experiment/Rag/index.tsx index e008b8d7c426fbe64493ac1e319a0b69c0b4ee44..06665de771693feb9184ba0f171c0f573a5d4d08 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} fixedFolder="rag" /> </Box> <Box sx={{ flex: 1, display: 'flex', flexDirection: 'column' }}> <Query experimentInfo={experimentInfo} />