diff --git a/ui/litellm-dashboard/src/components/add_model/ConnectionErrorDisplay.tsx b/ui/litellm-dashboard/src/components/add_model/ConnectionErrorDisplay.tsx index f52c3b7df0..a46a1b5cfa 100644 --- a/ui/litellm-dashboard/src/components/add_model/ConnectionErrorDisplay.tsx +++ b/ui/litellm-dashboard/src/components/add_model/ConnectionErrorDisplay.tsx @@ -45,10 +45,9 @@ const ConnectionErrorDisplay: React.FC = ({ console.log("result from prepareModelAddRequest:", result); const { litellmParamsObj, modelInfoObj, modelName: returnedModelName } = result; - const requestBody = { ...litellmParamsObj, mode: testMode }; - const response = await testConnectionRequest(accessToken, requestBody); + const response = await testConnectionRequest(accessToken, litellmParamsObj, modelInfoObj?.mode); if (response.status === "success") { message.success("Connection test successful!"); setError(null); @@ -56,7 +55,7 @@ const ConnectionErrorDisplay: React.FC = ({ } else { const errorMessage = response.result?.error || response.message || "Unknown error"; setError(errorMessage); - setRawRequest(requestBody); + setRawRequest(litellmParamsObj); setRawResponse(response.result?.raw_request_typed_dict); setIsSuccess(false); } diff --git a/ui/litellm-dashboard/src/components/networking.tsx b/ui/litellm-dashboard/src/components/networking.tsx index 81b9bb93e0..bb06608b11 100644 --- a/ui/litellm-dashboard/src/components/networking.tsx +++ b/ui/litellm-dashboard/src/components/networking.tsx @@ -2276,10 +2276,11 @@ export const keyInfoCall = async (accessToken: String, keys: String[]) => { export const testConnectionRequest = async ( accessToken: string, - requestBody: Record + litellm_params: Record, + mode: string, ) => { try { - console.log("Sending model connection test request:", JSON.stringify(requestBody)); + console.log("Sending model connection test request:", JSON.stringify(litellm_params)); // Construct the URL based on environment const url = proxyBaseUrl ? `${proxyBaseUrl}/health/test_connection` : `/health/test_connection`; @@ -2292,7 +2293,8 @@ export const testConnectionRequest = async ( }, body: JSON.stringify( { - litellm_params: requestBody + litellm_params: litellm_params, + mode: mode, } ) });