mirror of
https://github.com/BerriAI/litellm.git
synced 2025-04-26 11:14:04 +00:00
Co-authored-by: Jaswanth Karani <karani.jaswanth@gmail.com>
This commit is contained in:
parent
b76b380bc8
commit
e41bc5f32b
4 changed files with 9 additions and 4 deletions
|
@ -6,7 +6,8 @@ import { modelCreateCall, Model } from "../networking";
|
|||
export const handleAddModelSubmit = async (
|
||||
formValues: Record<string, any>,
|
||||
accessToken: string,
|
||||
form: any
|
||||
form: any,
|
||||
callback?: ()=>void
|
||||
) => {
|
||||
try {
|
||||
console.log("handling submit for formValues:", formValues);
|
||||
|
@ -137,6 +138,7 @@ export const handleAddModelSubmit = async (
|
|||
};
|
||||
|
||||
const response: any = await modelCreateCall(accessToken, new_model);
|
||||
callback && callback()
|
||||
|
||||
console.log(`response for model create call: ${response["data"]}`);
|
||||
});
|
||||
|
|
|
@ -13,11 +13,13 @@ import { TrashIcon } from "@heroicons/react/outline";
|
|||
interface DeleteModelProps {
|
||||
modelID: string;
|
||||
accessToken: string;
|
||||
callback?: ()=>void;
|
||||
}
|
||||
|
||||
const DeleteModelButton: React.FC<DeleteModelProps> = ({
|
||||
modelID,
|
||||
accessToken,
|
||||
callback
|
||||
}) => {
|
||||
const [isModalVisible, setIsModalVisible] = useState(false);
|
||||
|
||||
|
@ -30,6 +32,7 @@ const DeleteModelButton: React.FC<DeleteModelProps> = ({
|
|||
console.log("model delete Response:", response);
|
||||
message.success(`Model ${modelID} deleted successfully`);
|
||||
setIsModalVisible(false);
|
||||
callback && setTimeout(callback, 4000) //added timeout of 4 seconds as deleted model is taking time to reflect in get models
|
||||
} catch (error) {
|
||||
console.error("Error deleting the model:", error);
|
||||
}
|
||||
|
|
|
@ -1031,7 +1031,7 @@ const ModelDashboard: React.FC<ModelDashboardProps> = ({
|
|||
form
|
||||
.validateFields()
|
||||
.then((values) => {
|
||||
handleAddModelSubmit(values, accessToken, form);
|
||||
handleAddModelSubmit(values, accessToken, form, handleRefreshClick);
|
||||
// form.resetFields();
|
||||
})
|
||||
.catch((error) => {
|
||||
|
@ -1450,6 +1450,7 @@ const ModelDashboard: React.FC<ModelDashboardProps> = ({
|
|||
<DeleteModelButton
|
||||
modelID={model.model_info.id}
|
||||
accessToken={accessToken}
|
||||
callback={handleRefreshClick}
|
||||
/>
|
||||
</Col>
|
||||
</Grid>
|
||||
|
|
|
@ -98,7 +98,7 @@ export const modelCreateCall = async (
|
|||
const data = await response.json();
|
||||
console.log("API Response:", data);
|
||||
message.success(
|
||||
"Model created successfully. Wait 60s and refresh on 'All Models' page"
|
||||
"Model created successfully"
|
||||
);
|
||||
return data;
|
||||
} catch (error) {
|
||||
|
@ -168,7 +168,6 @@ export const modelDeleteCall = async (
|
|||
|
||||
const data = await response.json();
|
||||
console.log("API Response:", data);
|
||||
message.success("Model deleted successfully. Restart server to see this.");
|
||||
return data;
|
||||
} catch (error) {
|
||||
console.error("Failed to create key:", error);
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue