diff --git a/ui/litellm-dashboard/src/components/policies/attachment_table.test.tsx b/ui/litellm-dashboard/src/components/policies/attachment_table.test.tsx index 4372983b71a..cb9319dfe33 100644 --- a/ui/litellm-dashboard/src/components/policies/attachment_table.test.tsx +++ b/ui/litellm-dashboard/src/components/policies/attachment_table.test.tsx @@ -15,6 +15,10 @@ vi.mock("@heroicons/react/outline", () => ({ SwitchVerticalIcon: function SwitchVerticalIcon() { return null; }, ChevronUpIcon: function ChevronUpIcon() { return null; }, ChevronDownIcon: function ChevronDownIcon() { return null; }, + PencilAltIcon: function PencilAltIcon() { return null; }, + PlayIcon: function PlayIcon() { return null; }, + RefreshIcon: function RefreshIcon() { return null; }, + ExternalLinkIcon: function ExternalLinkIcon() { return null; }, })); vi.mock("@tremor/react", async (importOriginal) => { diff --git a/ui/litellm-dashboard/src/components/policies/attachment_table.tsx b/ui/litellm-dashboard/src/components/policies/attachment_table.tsx index d9de8378a8a..11ff32b34ff 100644 --- a/ui/litellm-dashboard/src/components/policies/attachment_table.tsx +++ b/ui/litellm-dashboard/src/components/policies/attachment_table.tsx @@ -1,6 +1,6 @@ import React, { useState } from "react"; -import { Table, TableBody, TableCell, TableHead, TableHeaderCell, TableRow, Icon, Badge } from "@tremor/react"; -import { TrashIcon, SwitchVerticalIcon, ChevronUpIcon, ChevronDownIcon } from "@heroicons/react/outline"; +import { Table, TableBody, TableCell, TableHead, TableHeaderCell, TableRow, Badge } from "@tremor/react"; +import { SwitchVerticalIcon, ChevronUpIcon, ChevronDownIcon } from "@heroicons/react/outline"; import { Tooltip, Tag } from "antd"; import { ColumnDef, @@ -12,6 +12,7 @@ import { } from "@tanstack/react-table"; import { PolicyAttachment } from "./types"; import ImpactPopover from "./impact_popover"; +import TableIconActionButton from "../common_components/IconActionButton/TableIconActionButtons/TableIconActionButton"; interface AttachmentTableProps { attachments: PolicyAttachment[]; @@ -201,14 +202,11 @@ const AttachmentTable: React.FC = ({
{isAdmin && ( - - onDeleteClick(attachment.attachment_id)} - className="cursor-pointer hover:text-red-500" - /> - + onDeleteClick(attachment.attachment_id)} + /> )}
);