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

feat(model_dashboard.tsx): allow user to edit input cost per token for model on ui

Browse source 
also contains fixes for `/model/update`
This commit is contained in:
Krrish Dholakia 2024-05-07 20:57:21 -07:00
parent 1882ee1c4c
commit b9ec7acb08
30 changed files with 145 additions and 133 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -332,8 +332,15 @@ const ModelDashboard: React.FC<ModelDashboardProps> = ({
<InputNumber min={0} step={1} />
</Form.Item>
<Form.Item
label="input_cost_per_token"
name="input_cost_per_token"
tooltip="float (optional) - Input cost per token"
>
<InputNumber min={0} step={0.0001} />
</Form.Item>
<Form.Item
label="model_id"
@ -394,19 +401,16 @@ const handleEditSubmit = async (formValues: Record<string, any>) => {
console.log("handleEditSubmit payload:", payload);
let newModelValue = await modelUpdateCall(accessToken, payload);
try {
let newModelValue = await modelUpdateCall(accessToken, payload);
message.success("Model updated successfully, restart server to see updates");
// Update the teams state with the updated team data
// if (teams) {
// const updatedTeams = teams.map((team) =>
// team.team_id === teamId ? newTeamValues.data : team
// );
// setTeams(updatedTeams);
// }
message.success("Model updated successfully, restart server to see updates");
setEditModalVisible(false);
setSelectedModel(null);
} catch (error) {
console.log(`Error occurred`)
}
setEditModalVisible(false);
setSelectedModel(null);
};
@ -956,9 +960,8 @@ const handleEditSubmit = async (formValues: Record<string, any>) => {
</Accordion>
</TableCell>
<TableCell>{model.input_cost}</TableCell>
<TableCell>{model.output_cost}</TableCell>
<TableCell>{model.input_cost || model.litellm_params.input_cost_per_token || null}</TableCell>
<TableCell>{model.output_cost || model.litellm_params.output_cost_per_token || null}</TableCell>
<TableCell>{model.max_tokens}</TableCell>
<TableCell>
{