From 43b50b7151f313828ff9050309d75b6dc460b3fe Mon Sep 17 00:00:00 2001
From: Ishaan Jaff <ishaanjaffer0324@gmail.com>
Date: Sat, 4 May 2024 21:39:23 -0700
Subject: [PATCH] ui - show retry policy for errors

---
 .../src/components/model_dashboard.tsx        | 21 +++++++++++++------
 1 file changed, 15 insertions(+), 6 deletions(-)

diff --git a/ui/litellm-dashboard/src/components/model_dashboard.tsx b/ui/litellm-dashboard/src/components/model_dashboard.tsx
index 9b8b3c345..744d1ec21 100644
--- a/ui/litellm-dashboard/src/components/model_dashboard.tsx
+++ b/ui/litellm-dashboard/src/components/model_dashboard.tsx
@@ -1254,12 +1254,21 @@ const handleEditSubmit = async (formValues: Record<string, any>) => {
               </Title>
               <Text className="mb-6">How many retries should be attempted based on the Exception</Text>
               {retry_policy_map &&
-  Object.keys(retry_policy_map).map((key, idx) => (
-    <div key={idx} className="flex justify-start">
-      <Text>{key}</Text>
-      <InputNumber className="mr-5" />
-    </div>
-  ))}
+  <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" />
+          </td>
+        </tr>
+      ))}
+    </tbody>
+  </table>
+}
 
             </TabPanel>