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

ui - show retry policy per model group

Browse source
This commit is contained in:
Ishaan Jaff 2024-05-06 17:23:51 -07:00
parent cc83d8ec1a
commit e06285ebf4
1 changed files with 34 additions and 18 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -1275,24 +1275,40 @@ const handleEditSubmit = async (formValues: Record<string, any>) => {
<Text className="mb-6">How many retries should be attempted based on the Exception</Text>
{retry_policy_map &&
<table>
<tbody>
{Object.keys(retry_policy_map).map((key, idx) => (
<tr key={idx} className="flex justify-between items-center mt-2">
<td>
<Text>{key}</Text>
</td>
<td>
<InputNumber
className="ml-5"
defaultValue={defaultRetry}
min={0}
step={1}
/>
</td>
</tr>
))}
</tbody>
</table>
<tbody>
{Object.entries(retry_policy_map).map(([exceptionType, retryPolicyKey], idx) => {
let retryCount = (modelGroupRetryPolicy[selectedModelGroup] ?? {})[retryPolicyKey];
if (retryCount == null) {
retryCount = defaultRetry;
}
return (
<tr key={idx} className="flex justify-between items-center mt-2">
<td>
<Text>{exceptionType}</Text>
</td>
<td>
<InputNumber
className="ml-5"
value={retryCount}
min={0}
step={1}
onChange={(value) => {
setModelGroupRetryPolicy({
...modelGroupRetryPolicy,
[selectedModelGroup]: {
...modelGroupRetryPolicy[selectedModelGroup],
[retryPolicyKey]: value,
},
});
}}
/>
</td>
</tr>
);
})}
</tbody>
</table>
}
<Button className="mt-6 mr-8">
Save