Fix: The files in the knowledge base folder on the file management page should not be deleted #9975 (#9976)

### What problem does this PR solve?

Fix: The files in the knowledge base folder on the file management page
should not be deleted #9975

### Type of change

- [x] Bug Fix (non-breaking change which fixes an issue)

web/src/pages/files/action-cell.tsx

@@ -21,8 +21,9 @@ import {
   UseHandleConnectToKnowledgeReturnType,
   UseRenameCurrentFileReturnType,
 } from './hooks';
+import { useHandleDeleteFile } from './use-delete-file';
 import { UseMoveDocumentShowType } from './use-move-file';
-import { isFolderType } from './util';
+import { isFolderType, isKnowledgeBaseType } from './util';
 
 type IProps = Pick<CellContext<IFile, unknown>, 'row'> &
   Pick<UseHandleConnectToKnowledgeReturnType, 'showConnectToKnowledgeModal'> &
@@ -40,6 +41,7 @@ export function ActionCell({
   const { downloadFile } = useDownloadFile();
   const isFolder = isFolderType(record.type);
   const extension = getExtension(record.name);
+  const isKnowledgeBase = isKnowledgeBaseType(record.source_type);
 
   const handleShowConnectToKnowledgeModal = useCallback(() => {
     showConnectToKnowledgeModal(record);
@@ -60,34 +62,44 @@ export function ActionCell({
     showMoveFileModal([record.id]);
   }, [record, showMoveFileModal]);
 
+  const { handleRemoveFile } = useHandleDeleteFile();
+
+  const onRemoveFile = useCallback(() => {
+    handleRemoveFile([documentId]);
+  }, [handleRemoveFile, documentId]);
+
   return (
     <section className="flex gap-4 items-center text-text-sub-title-invert opacity-0 group-hover:opacity-100 transition-opacity">
-      <Button
-        variant="transparent"
-        className="border-none hover:bg-bg-card text-text-primary"
-        size={'sm'}
-        onClick={handleShowConnectToKnowledgeModal}
-      >
-        <Link2 />
-      </Button>
-      <Button
-        variant="transparent"
-        className="border-none hover:bg-bg-card text-text-primary"
-        size={'sm'}
-        onClick={handleShowMoveFileModal}
-      >
-        <FolderInput />
-      </Button>
-
-      <Button
-        variant="transparent"
-        className="border-none hover:bg-bg-card text-text-primary"
-        size={'sm'}
-        onClick={handleShowFileRenameModal}
-      >
-        <FolderPen />
-      </Button>
-
+      {isKnowledgeBase || (
+        <Button
+          variant="transparent"
+          className="border-none hover:bg-bg-card text-text-primary"
+          size={'sm'}
+          onClick={handleShowConnectToKnowledgeModal}
+        >
+          <Link2 />
+        </Button>
+      )}
+      {isKnowledgeBase || (
+        <Button
+          variant="transparent"
+          className="border-none hover:bg-bg-card text-text-primary"
+          size={'sm'}
+          onClick={handleShowMoveFileModal}
+        >
+          <FolderInput />
+        </Button>
+      )}
+      {isKnowledgeBase || (
+        <Button
+          variant="transparent"
+          className="border-none hover:bg-bg-card text-text-primary"
+          size={'sm'}
+          onClick={handleShowFileRenameModal}
+        >
+          <FolderPen />
+        </Button>
+      )}
       {isFolder || (
         <Button
           variant="transparent"
@@ -138,15 +150,17 @@ export function ActionCell({
           )}
         </DropdownMenuContent>
       </DropdownMenu> */}
-      <ConfirmDeleteDialog>
-        <Button
-          variant="transparent"
-          className="border-none hover:bg-bg-card text-text-primary"
-          size={'sm'}
-        >
-          <Trash2 />
-        </Button>
-      </ConfirmDeleteDialog>
+      {isKnowledgeBase || (
+        <ConfirmDeleteDialog onOk={onRemoveFile}>
+          <Button
+            variant="transparent"
+            className="border-none hover:bg-bg-card text-text-primary"
+            size={'sm'}
+          >
+            <Trash2 />
+          </Button>
+        </ConfirmDeleteDialog>
+      )}
     </section>
   );
 }

web/src/pages/files/files-table.tsx

@@ -48,7 +48,7 @@ import { KnowledgeCell } from './knowledge-cell';
 import { LinkToDatasetDialog } from './link-to-dataset-dialog';
 import { UseMoveDocumentShowType } from './use-move-file';
 import { useNavigateToOtherFolder } from './use-navigate-to-folder';
-import { isFolderType } from './util';
+import { isFolderType, isKnowledgeBaseType } from './util';
 
 type FilesTableProps = Pick<
   ReturnType<typeof useFetchFileList>,
@@ -112,6 +112,7 @@ export function FilesTable({
           checked={row.getIsSelected()}
           onCheckedChange={(value) => row.toggleSelected(!!value)}
           aria-label="Select row"
+          disabled={!row.getCanSelect()}
         />
       ),
       enableSorting: false,
@@ -247,7 +248,9 @@ export function FilesTable({
     onRowSelectionChange: setRowSelection,
 
     manualPagination: true, //we're doing manual "server-side" pagination
-
+    enableRowSelection(row) {
+      return !isKnowledgeBaseType(row.original.source_type);
+    },
     state: {
       sorting,
       columnFilters,

web/src/pages/files/util.ts

@@ -1,3 +1,7 @@
 export function isFolderType(type: string) {
   return type === 'folder';
 }
+
+export function isKnowledgeBaseType(sourceType: string) {
+  return sourceType === 'knowledgebase';
+}