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 26df1711ee6..454caf9e601 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 @@ -62,4 +62,39 @@ describe("ProviderSpecificFields", () => { expect(apiKeyLabel).toBeInTheDocument(); }); }); + + it("should render the provider specific fields for Azure", async () => { + const { getByLabelText, getByPlaceholderText } = render( +
, + ); + + await waitFor(() => { + // Check for API Base field + const apiBaseInput = getByPlaceholderText("https://..."); + expect(apiBaseInput).toBeInTheDocument(); + expect(apiBaseInput).toHaveAttribute("type", "text"); + + // Check for API Version field + const apiVersionInput = getByPlaceholderText("2023-07-01-preview"); + expect(apiVersionInput).toBeInTheDocument(); + + // Check for Base Model field + const baseModelInput = getByPlaceholderText("azure/gpt-3.5-turbo"); + expect(baseModelInput).toBeInTheDocument(); + + // Check for API Key field + const apiKeyInput = getByLabelText("Azure API Key"); + expect(apiKeyInput).toBeInTheDocument(); + expect(apiKeyInput).toHaveAttribute("type", "password"); + expect(apiKeyInput).toHaveAttribute("placeholder", "Enter your Azure API Key"); + + // Check for Azure AD Token field + const azureAdTokenInput = getByLabelText("Azure AD Token"); + expect(azureAdTokenInput).toBeInTheDocument(); + expect(azureAdTokenInput).toHaveAttribute("type", "password"); + expect(azureAdTokenInput).toHaveAttribute("placeholder", "Enter your Azure AD Token"); + }); + }); }); 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 61500dd8978..2f294611408 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 @@ -164,7 +164,15 @@ const PROVIDER_CREDENTIAL_FIELDS: Record