diff --git a/ui/litellm-dashboard/src/components/add_model/provider_specific_fields.test.tsx b/ui/litellm-dashboard/src/components/add_model/provider_specific_fields.test.tsx index 4475cd4ca3f..26df1711ee6 100644 --- a/ui/litellm-dashboard/src/components/add_model/provider_specific_fields.test.tsx +++ b/ui/litellm-dashboard/src/components/add_model/provider_specific_fields.test.tsx @@ -44,4 +44,22 @@ describe("ProviderSpecificFields", () => { expect(apiKeyLabel).toBeInTheDocument(); }); }); + + it("should render the provider specific fields for vLLM", async () => { + const { getByLabelText, getByPlaceholderText } = render( +
, + ); + + await waitFor(() => { + const apiBaseInput = getByPlaceholderText("https://..."); + expect(apiBaseInput).toBeInTheDocument(); + expect(apiBaseInput).toHaveAttribute("type", "text"); + + // Check for API Key field + const apiKeyLabel = getByLabelText("vLLM API Key"); + expect(apiKeyLabel).toBeInTheDocument(); + }); + }); }); diff --git a/ui/litellm-dashboard/src/components/add_model/provider_specific_fields.tsx b/ui/litellm-dashboard/src/components/add_model/provider_specific_fields.tsx index c7686fb0b0c..61500dd8978 100644 --- a/ui/litellm-dashboard/src/components/add_model/provider_specific_fields.tsx +++ b/ui/litellm-dashboard/src/components/add_model/provider_specific_fields.tsx @@ -507,9 +507,9 @@ const PROVIDER_CREDENTIAL_FIELDS: Record