diff --git a/ui/litellm-dashboard/src/components/templates/key_edit_view.tsx b/ui/litellm-dashboard/src/components/templates/key_edit_view.tsx index 8a8a018f267..7e1b83fc977 100644 --- a/ui/litellm-dashboard/src/components/templates/key_edit_view.tsx +++ b/ui/litellm-dashboard/src/components/templates/key_edit_view.tsx @@ -6,6 +6,7 @@ import { InfoCircleOutlined } from "@ant-design/icons"; import { TextInput, Button as TremorButton } from "@tremor/react"; import { Form, Input, Select, Switch, Tooltip } from "antd"; import { useEffect, useState } from "react"; +import { rolesWithWriteAccess } from "../../utils/roles"; import AgentSelector from "../agent_management/AgentSelector"; import AccessGroupSelector from "../common_components/AccessGroupSelector"; import { mapInternalToDisplayNames } from "../callback_info_helpers"; @@ -84,6 +85,7 @@ export function KeyEditView({ userRole, premiumUser = false, }: KeyEditViewProps) { + const canEditGuardrails = premiumUser || (userRole != null && rolesWithWriteAccess.includes(userRole)); const [form] = Form.useForm(); const [promptsList, setPromptsList] = useState([]); const [tagsList, setTagsList] = useState>({}); @@ -443,7 +445,7 @@ export function KeyEditView({ form.setFieldValue("guardrails", v); }} accessToken={accessToken} - disabled={!premiumUser} + disabled={!canEditGuardrails} /> )} @@ -460,7 +462,7 @@ export function KeyEditView({ name="disable_global_guardrails" valuePropName="checked" > - +