scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-09 03:18:44 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 108

fix mode

Browse source
This commit is contained in:
Ishaan Jaff 2025-03-14 18:53:14 -07:00
parent fa25c4ba7a
commit 66dc49ab92
2 changed files with 10 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

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

@ -42,9 +42,12 @@ const ConnectionErrorDisplay: React.FC<ConnectionErrorDisplayProps> = ({
const result = await prepareModelAddRequest(formValues, accessToken, null);
if (!result) throw new Error("Failed to prepare model data");
const { litellmParamsObj } = result;
console.log("result from prepareModelAddRequest:", result);
const { litellmParamsObj, modelInfoObj, modelName: returnedModelName } = result;
const requestBody = { ...litellmParamsObj, mode: testMode };
const response = await testConnectionRequest(accessToken, requestBody);
if (response.status === "success") {
message.success("Connection test successful!");

11
ui/litellm-dashboard/src/components/add_model/handle_add_model_submit.tsx
View file

@ -72,7 +72,6 @@ export const prepareModelAddRequest = async (
} else if (key == "model") {
continue;
}
// Check if key is "base_model"
else if (key === "base_model") {
// Add key-value pair to model_info dictionary
@ -82,7 +81,11 @@ export const prepareModelAddRequest = async (
modelInfoObj["team_id"] = value;
}
else if (key == "mode") {
console.log("placing mode in modelInfo")
modelInfoObj["mode"] = value;
// remove "mode" from litellmParams
delete litellmParamsObj["mode"];
}
else if (key === "custom_model_name") {
litellmParamsObj["model"] = value;
@ -147,13 +150,13 @@ export const prepareModelAddRequest = async (
};
export const handleAddModelSubmit = async (
values: any,
accessToken: string,
form: any,
callback?: () => void,
) => {
try {
const formValues = form.getFieldsValue();
const result = await prepareModelAddRequest(formValues, accessToken, form);
const result = await prepareModelAddRequest(values, accessToken, form);
if (!result) {
return; // Exit if preparation failed
@ -172,8 +175,6 @@ export const handleAddModelSubmit = async (
callback && callback();
form.resetFields();
message.success("Model added successfully");
} catch (error) {
message.error("Failed to add model: " + error, 10);
}