From 1566e3c74d2a313f6ce174d1533911389422ecbb Mon Sep 17 00:00:00 2001
From: Debanjum <debanjum@gmail.com>
Date: Fri, 27 Jun 2025 14:43:11 -0700
Subject: [PATCH] Ease bulk (de-)selecting of files to add/remove to agent
 knowledge base

Add select all, deselect all buttons to select all (filtered) files to
add, remove from an agent's knowledge base.
---
 .../app/components/agentCard/agentCard.tsx    | 43 ++++++++++++++++++-
 1 file changed, 42 insertions(+), 1 deletion(-)

diff --git a/src/interface/web/app/components/agentCard/agentCard.tsx b/src/interface/web/app/components/agentCard/agentCard.tsx
index 04bc26a3..da7e8889 100644
--- a/src/interface/web/app/components/agentCard/agentCard.tsx
+++ b/src/interface/web/app/components/agentCard/agentCard.tsx
@@ -533,6 +533,7 @@ export function AgentModificationForm(props: AgentModificationFormProps) {
     const [uploadedFiles, setUploadedFiles] = useState<string[]>([]);
     const [allFileOptions, setAllFileOptions] = useState<string[]>([]);
     const [currentStep, setCurrentStep] = useState(0);
+    const [fileSearchValue, setFileSearchValue] = useState("");
 
     const [showSubscribeDialog, setShowSubscribeDialog] = useState(false);
 
@@ -1044,10 +1045,50 @@ export function AgentModificationForm(props: AgentModificationFormProps) {
                                                     </div>
                                                 </div>
                                             </div>
-                                            <CommandInput placeholder="Select files..." />
+                                            <CommandInput
+                                                placeholder="Select files..."
+                                                value={fileSearchValue}
+                                                onValueChange={setFileSearchValue}
+                                            />
                                             <CommandList>
                                                 <CommandEmpty>No files found.</CommandEmpty>
                                                 <CommandGroup>
+                                                    <div className="flex gap-2 px-2 py-1 border-b">
+                                                        <Button
+                                                            type="button"
+                                                            variant="ghost"
+                                                            size="sm"
+                                                            className="h-6 px-2 text-xs"
+                                                            onClick={(e) => {
+                                                                e.stopPropagation();
+                                                                const filteredFiles = allFileOptions.filter(file =>
+                                                                    file.toLowerCase().includes(fileSearchValue.toLowerCase())
+                                                                );
+                                                                const currentFiles = props.form.getValues("files") || [];
+                                                                const newFiles = [...new Set([...currentFiles, ...filteredFiles])];
+                                                                props.form.setValue("files", newFiles);
+                                                            }}
+                                                        >
+                                                            Select All
+                                                        </Button>
+                                                        <Button
+                                                            type="button"
+                                                            variant="ghost"
+                                                            size="sm"
+                                                            className="h-6 px-2 text-xs"
+                                                            onClick={(e) => {
+                                                                e.stopPropagation();
+                                                                const filteredFiles = allFileOptions.filter(file =>
+                                                                    file.toLowerCase().includes(fileSearchValue.toLowerCase())
+                                                                );
+                                                                const currentFiles = props.form.getValues("files") || [];
+                                                                const newFiles = currentFiles.filter(file => !filteredFiles.includes(file));
+                                                                props.form.setValue("files", newFiles);
+                                                            }}
+                                                        >
+                                                            Deselect All
+                                                        </Button>
+                                                    </div>
                                                     {allFileOptions.map((file) => (
                                                         <CommandItem
                                                             value={file}