models - instant update ui on editing values

This commit is contained in:
Ishaan Jaff 2025-03-14 09:43:44 -07:00
parent 3875df666b
commit aed8cd7ac6
2 changed files with 23 additions and 4 deletions

View file

@ -1040,6 +1040,18 @@ const ModelDashboard: React.FC<ModelDashboardProps> = ({
userRole={userRole}
setEditModalVisible={setEditModalVisible}
setSelectedModel={setSelectedModel}
onModelUpdate={(updatedModel) => {
// Update the model in the modelData.data array
const updatedModelData = {
...modelData,
data: modelData.data.map((model: any) =>
model.model_info.id === updatedModel.model_info.id ? updatedModel : model
)
};
setModelData(updatedModelData);
// Trigger a refresh to update UI
handleRefreshClick();
}}
/>
) : (
<TabGroup className="gap-2 p-8 h-[75vh] w-full mt-2">

View file

@ -32,6 +32,7 @@ interface ModelInfoViewProps {
editModel: boolean;
setEditModalVisible: (visible: boolean) => void;
setSelectedModel: (model: any) => void;
onModelUpdate?: (updatedModel: any) => void;
}
export default function ModelInfoView({
@ -43,7 +44,8 @@ export default function ModelInfoView({
userRole,
editModel,
setEditModalVisible,
setSelectedModel
setSelectedModel,
onModelUpdate
}: ModelInfoViewProps) {
const [form] = Form.useForm();
const [localModelData, setLocalModelData] = useState(modelData);
@ -78,17 +80,22 @@ export default function ModelInfoView({
model_info: {
id: modelId,
}
};
await modelUpdateCall(accessToken, updateData);
setLocalModelData({
const updatedModelData = {
...localModelData,
model_name: values.model_name,
litellm_model_name: values.litellm_model_name,
litellm_params: updateData.litellm_params
});
};
setLocalModelData(updatedModelData);
if (onModelUpdate) {
onModelUpdate(updatedModelData);
}
message.success("Model settings updated successfully");
setIsDirty(false);