From 86d64c3447c298bf2559c74f8ae36cb73bea9385 Mon Sep 17 00:00:00 2001
From: Krrish Dholakia <krrishdholakia@gmail.com>
Date: Thu, 13 Jun 2024 14:51:45 -0700
Subject: [PATCH] build(ui/teams.tsx): allow resetting teams budget

---
 .../src/components/create_key_button.tsx      |  1 +
 ui/litellm-dashboard/src/components/teams.tsx | 22 +++++++++++++++++++
 2 files changed, 23 insertions(+)

diff --git a/ui/litellm-dashboard/src/components/create_key_button.tsx b/ui/litellm-dashboard/src/components/create_key_button.tsx
index 83e8d92647..75edb2ef3b 100644
--- a/ui/litellm-dashboard/src/components/create_key_button.tsx
+++ b/ui/litellm-dashboard/src/components/create_key_button.tsx
@@ -261,6 +261,7 @@ const CreateKey: React.FC<CreateKeyProps> = ({
                 >
                   <Select defaultValue={null} placeholder="n/a">
                     <Select.Option value="24h">daily</Select.Option>
+                    <Select.Option value="7d">weekly</Select.Option>
                     <Select.Option value="30d">monthly</Select.Option>
                   </Select>
                 </Form.Item>
diff --git a/ui/litellm-dashboard/src/components/teams.tsx b/ui/litellm-dashboard/src/components/teams.tsx
index 56d4f5e0d6..a9b8ffb68b 100644
--- a/ui/litellm-dashboard/src/components/teams.tsx
+++ b/ui/litellm-dashboard/src/components/teams.tsx
@@ -151,6 +151,17 @@ const Team: React.FC<TeamProps> = ({
             <Form.Item label="Max Budget (USD)" name="max_budget">
               <InputNumber step={0.01} precision={2} width={200} />
             </Form.Item>
+            <Form.Item
+                  className="mt-8"
+                  label="Reset Budget"
+                  name="budget_duration"
+                >
+                  <Select2 defaultValue={null} placeholder="n/a">
+                    <Select2.Option value="24h">daily</Select2.Option>
+                    <Select2.Option value="7d">weekly</Select2.Option>
+                    <Select2.Option value="30d">monthly</Select2.Option>
+                  </Select2>
+                </Form.Item>
             <Form.Item label="Tokens per minute Limit (TPM)" name="tpm_limit">
               <InputNumber step={1} width={400} />
             </Form.Item>
@@ -635,6 +646,17 @@ const Team: React.FC<TeamProps> = ({
                 <Form.Item label="Max Budget (USD)" name="max_budget">
                   <InputNumber step={0.01} precision={2} width={200} />
                 </Form.Item>
+                <Form.Item
+                  className="mt-8"
+                  label="Reset Budget"
+                  name="budget_duration"
+                >
+                  <Select2 defaultValue={null} placeholder="n/a">
+                    <Select2.Option value="24h">daily</Select2.Option>
+                    <Select2.Option value="7d">weekly</Select2.Option>
+                    <Select2.Option value="30d">monthly</Select2.Option>
+                  </Select2>
+                </Form.Item>
                 <Form.Item
                   label="Tokens per minute Limit (TPM)"
                   name="tpm_limit"