diff --git a/ui/litellm-dashboard/src/components/add_model/add_model_tab.tsx b/ui/litellm-dashboard/src/components/add_model/add_model_tab.tsx index c699c161e3e..59f5ed5cbe4 100644 --- a/ui/litellm-dashboard/src/components/add_model/add_model_tab.tsx +++ b/ui/litellm-dashboard/src/components/add_model/add_model_tab.tsx @@ -9,6 +9,8 @@ import AdvancedSettings from "./advanced_settings"; import { Providers, providerLogoMap, getPlaceholder } from "../provider_info_helpers"; import type { Team } from "../key_team_helpers/key_list"; import { CredentialItem } from "../networking"; +import { testModelConnection } from "./handle_add_model_submit"; + interface AddModelTabProps { form: FormInstance; handleOk: () => void; @@ -22,6 +24,7 @@ interface AddModelTabProps { setShowAdvancedSettings: (show: boolean) => void; teams: Team[] | null; credentials: CredentialItem[]; + accessToken: string; } const { Title, Link } = Typography; @@ -39,7 +42,14 @@ const AddModelTab: React.FC = ({ setShowAdvancedSettings, teams, credentials, + accessToken, }) => { + // Add a function to handle test connection + const handleTestConnection = async () => { + const formValues = form.getFieldsValue(); + await testModelConnection(formValues, accessToken); + }; + return ( <> Add new model @@ -184,7 +194,10 @@ const AddModelTab: React.FC = ({ Need Help? - +
+ + +
diff --git a/ui/litellm-dashboard/src/components/model_dashboard.tsx b/ui/litellm-dashboard/src/components/model_dashboard.tsx index 0a9c26ac4a4..77e422ce433 100644 --- a/ui/litellm-dashboard/src/components/model_dashboard.tsx +++ b/ui/litellm-dashboard/src/components/model_dashboard.tsx @@ -1140,6 +1140,7 @@ const ModelDashboard: React.FC = ({ setShowAdvancedSettings={setShowAdvancedSettings} teams={teams} credentials={credentialsList} + accessToken={accessToken} />