add rag to types of allowed plugins

transformerlab · Mar 30, 2024 · 35b6ba5 · 35b6ba5
1 parent 0a17929
commit 35b6ba5
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 5 deletions.
diff --git a/src/renderer/components/Plugins/PluginCard.tsx b/src/renderer/components/Plugins/PluginCard.tsx
@@ -5,6 +5,7 @@ import Typography from '@mui/joy/Typography';
 import {
   ArrowRightFromLineIcon,
   DownloadIcon,
+  FolderSearch2Icon,
   GraduationCapIcon,
   HelpCircleIcon,
   RocketIcon,
@@ -28,6 +29,8 @@ function getIcon(type: string) {
       return <RocketIcon color="#FFA732" />;
     case 'exporter':
       return <ArrowRightFromLineIcon color="#711DB0" />;
+    case 'rag':
+      return <FolderSearch2Icon color="skyblue" />;
     default:
       return null;
   }

diff --git a/src/renderer/components/Plugins/PluginGallery.tsx b/src/renderer/components/Plugins/PluginGallery.tsx
@@ -41,11 +41,13 @@ export default function PluginGallery({ experimentInfo }) {
             setFilters({ ...filters, type: newValue });
           }}
         >
-          {['All', 'trainer', 'evaluator', 'loader', 'exporter'].map((type) => (
-            <Option value={type}>
-              <Chip>{type}</Chip>
-            </Option>
-          ))}
+          {['All', 'trainer', 'evaluator', 'loader', 'exporter', 'rag'].map(
+            (type) => (
+              <Option value={type}>
+                <Chip>{type}</Chip>
+              </Option>
+            )
+          )}
         </Select>
       </FormControl>
     </>