diff --git a/ui/litellm-dashboard/out/assets/logos/nvidia_triton.png b/ui/litellm-dashboard/out/assets/logos/nvidia_triton.png new file mode 100644 index 00000000000..b6d7a5b2bb6 Binary files /dev/null and b/ui/litellm-dashboard/out/assets/logos/nvidia_triton.png differ diff --git a/ui/litellm-dashboard/public/assets/logos/nvidia_triton.png b/ui/litellm-dashboard/public/assets/logos/nvidia_triton.png new file mode 100644 index 00000000000..b6d7a5b2bb6 Binary files /dev/null and b/ui/litellm-dashboard/public/assets/logos/nvidia_triton.png differ 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 f44a374fd8a..6d6fd4288fe 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 @@ -327,7 +327,20 @@ const PROVIDER_CREDENTIAL_FIELDS: Record = label: "API Key", type: "password", required: true - }] + }], + [Providers.Triton]: [{ + key: "api_key", + label: "API Key", + type: "password", + required: false + }, + { + key: "api_base", + label: "API Base", + placeholder: "http://localhost:8000/generate", + required: false + } +] }; const ProviderSpecificFields: React.FC = ({ diff --git a/ui/litellm-dashboard/src/components/provider_info_helpers.tsx b/ui/litellm-dashboard/src/components/provider_info_helpers.tsx index 6b675aa6176..54bc17e4847 100644 --- a/ui/litellm-dashboard/src/components/provider_info_helpers.tsx +++ b/ui/litellm-dashboard/src/components/provider_info_helpers.tsx @@ -25,7 +25,8 @@ export enum Providers { Perplexity = "Perplexity", TogetherAI = "TogetherAI", Openrouter = "Openrouter", - FireworksAI = "Fireworks AI" + FireworksAI = "Fireworks AI", + Triton = "Triton" } @@ -53,7 +54,8 @@ export const provider_map: Record = { Perplexity: "perplexity", TogetherAI: "together_ai", Openrouter: "openrouter", - FireworksAI: "fireworks_ai" + FireworksAI: "fireworks_ai", + Triton: "triton" }; const asset_logos_folder = '/ui/assets/logos/'; @@ -82,7 +84,8 @@ export const providerLogoMap: Record = { [Providers.Sambanova]: `${asset_logos_folder}sambanova.svg`, [Providers.TogetherAI]: `${asset_logos_folder}togetherai.svg`, [Providers.Vertex_AI]: `${asset_logos_folder}google.svg`, - [Providers.xAI]: `${asset_logos_folder}xai.svg` + [Providers.xAI]: `${asset_logos_folder}xai.svg`, + [Providers.Triton]: `${asset_logos_folder}nvidia_triton.png` }; export const getProviderLogoAndName = (providerValue: string): { logo: string, displayName: string } => {