ai-workspace-services/litellm
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

ui - add nvidia triton models (#10456)

Browse Source
This commit is contained in:
Ishaan Jaff 2025-04-30 21:42:15 -07:00 committed by GitHub
parent 711601e22a
commit 7c679abe85
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
4 changed files with 20 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

BIN
ui/litellm-dashboard/out/assets/logos/nvidia_triton.png Normal file
View File

Binary file not shown.
Side by Side

After

Width:  |  Height:  |  Size: 5.6 KiB

BIN
ui/litellm-dashboard/public/assets/logos/nvidia_triton.png Normal file
View File

Binary file not shown.
Side by Side

After

Width:  |  Height:  |  Size: 5.6 KiB

15
ui/litellm-dashboard/src/components/add_model/provider_specific_fields.tsx
View File

@ -327,7 +327,20 @@ const PROVIDER_CREDENTIAL_FIELDS: Record<Providers, ProviderCredentialField[]> =
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<ProviderSpecificFieldsProps> = ({

9
ui/litellm-dashboard/src/components/provider_info_helpers.tsx
View File

@ -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<string, string> = {
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<string, string> = {
[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 } => {