phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-26 11:14:04 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix params to test connection

Browse source
This commit is contained in:
Ishaan Jaff 2025-03-14 18:57:39 -07:00
parent 5b1867feca
commit e23b51d7d4
2 changed files with 7 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

5
ui/litellm-dashboard/src/components/add_model/ConnectionErrorDisplay.tsx
View file

@ -45,10 +45,9 @@ const ConnectionErrorDisplay: React.FC<ConnectionErrorDisplayProps> = ({
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<ConnectionErrorDisplayProps> = ({
} 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);
}

8
ui/litellm-dashboard/src/components/networking.tsx
View file

@ -2276,10 +2276,11 @@ export const keyInfoCall = async (accessToken: String, keys: String[]) => {
export const testConnectionRequest = async (
accessToken: string,
requestBody: Record<string, any>
litellm_params: Record<string, any>,
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,
}
)
});