diff --git a/src/renderer/components/Experiment/Generate/GenerateModal.tsx b/src/renderer/components/Experiment/Generate/GenerateModal.tsx index 292ed8c62e1cf27be60a08faa46fdfa737288031..d7469eda5ddc7353c6b90b6c523500f41555f3a7 100644 --- a/src/renderer/components/Experiment/Generate/GenerateModal.tsx +++ b/src/renderer/components/Experiment/Generate/GenerateModal.tsx @@ -305,6 +305,7 @@ export default function GenerateModal({ value={selectedDocs} onChange={setSelectedDocs} name="docs" + required /> <FormHelperText>Select documents to upload</FormHelperText> </FormControl> diff --git a/src/renderer/components/Experiment/Rag/PickADocumentMenu.tsx b/src/renderer/components/Experiment/Rag/PickADocumentMenu.tsx index 120bc50c313a60cd24f955941274bb8554b1f415..f67856a89cb84015d1b0294fe91f5ae4af3f833f 100644 --- a/src/renderer/components/Experiment/Rag/PickADocumentMenu.tsx +++ b/src/renderer/components/Experiment/Rag/PickADocumentMenu.tsx @@ -10,6 +10,7 @@ export default function PickADocumentMenu({ value, onChange, defaultValue = [], + required = false, showFoldersOnly = false, }) { const { @@ -24,7 +25,13 @@ export default function PickADocumentMenu({ } return ( - <Select multiple onChange={handleChange} value={value} name={name}> + <Select + multiple + onChange={handleChange} + value={value} + name={name} + required={required} + > {rows?.map((row) => showFoldersOnly ? ( row?.type === 'folder' && (