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

fixed issues #8126 and #8127 (#8275) (#8299)

Browse source 
Co-authored-by: Jaswanth Karani <karani.jaswanth@gmail.com>
This commit is contained in:
Ishaan Jaff 2025-02-05 18:52:58 -08:00 • committed by GitHub
parent b76b380bc8
commit e41bc5f32b
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
4 changed files with 9 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -6,7 +6,8 @@ import { modelCreateCall, Model } from "../networking";
export const handleAddModelSubmit = async ( export const handleAddModelSubmit = async (
formValues: Record<string, any>, formValues: Record<string, any>,
accessToken: string, accessToken: string,
form: any form: any,
callback?: ()=>void
) => { ) => {
try { try {
console.log("handling submit for formValues:", formValues); console.log("handling submit for formValues:", formValues);
@ -137,6 +138,7 @@ export const handleAddModelSubmit = async (
}; };
const response: any = await modelCreateCall(accessToken, new_model); const response: any = await modelCreateCall(accessToken, new_model);
callback && callback()
console.log(`response for model create call: ${response["data"]}`); console.log(`response for model create call: ${response["data"]}`);
}); });

3
ui/litellm-dashboard/src/components/delete_model_button.tsx
View file

@ -13,11 +13,13 @@ import { TrashIcon } from "@heroicons/react/outline";
interface DeleteModelProps { interface DeleteModelProps {
modelID: string; modelID: string;
accessToken: string; accessToken: string;
callback?: ()=>void;
} }
const DeleteModelButton: React.FC<DeleteModelProps> = ({ const DeleteModelButton: React.FC<DeleteModelProps> = ({
modelID, modelID,
accessToken, accessToken,
callback
}) => { }) => {
const [isModalVisible, setIsModalVisible] = useState(false); const [isModalVisible, setIsModalVisible] = useState(false);
@ -30,6 +32,7 @@ const DeleteModelButton: React.FC<DeleteModelProps> = ({
console.log("model delete Response:", response); console.log("model delete Response:", response);
message.success(`Model ${modelID} deleted successfully`); message.success(`Model ${modelID} deleted successfully`);
setIsModalVisible(false); setIsModalVisible(false);
callback && setTimeout(callback, 4000) //added timeout of 4 seconds as deleted model is taking time to reflect in get models
} catch (error) { } catch (error) {
console.error("Error deleting the model:", error); console.error("Error deleting the model:", error);
} }

3
ui/litellm-dashboard/src/components/model_dashboard.tsx
View file

@ -1031,7 +1031,7 @@ const ModelDashboard: React.FC<ModelDashboardProps> = ({
form form
.validateFields() .validateFields()
.then((values) => { .then((values) => {
handleAddModelSubmit(values, accessToken, form); handleAddModelSubmit(values, accessToken, form, handleRefreshClick);
// form.resetFields(); // form.resetFields();
}) })
.catch((error) => { .catch((error) => {
@ -1450,6 +1450,7 @@ const ModelDashboard: React.FC<ModelDashboardProps> = ({
<DeleteModelButton <DeleteModelButton
modelID={model.model_info.id} modelID={model.model_info.id}
accessToken={accessToken} accessToken={accessToken}
callback={handleRefreshClick}
/> />
</Col> </Col>
</Grid> </Grid>

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

@ -98,7 +98,7 @@ export const modelCreateCall = async (
const data = await response.json(); const data = await response.json();
console.log("API Response:", data); console.log("API Response:", data);
message.success( message.success(
"Model created successfully. Wait 60s and refresh on 'All Models' page" "Model created successfully"
); );
return data; return data;
} catch (error) { } catch (error) {
@ -168,7 +168,6 @@ export const modelDeleteCall = async (
const data = await response.json(); const data = await response.json();
console.log("API Response:", data); console.log("API Response:", data);
message.success("Model deleted successfully. Restart server to see this.");
return data; return data;
} catch (error) { } catch (error) {
console.error("Failed to create key:", error); console.error("Failed to create key:", error);