From 05aefe18ce74630ca4f0b0fccf78ad97c00ce95d Mon Sep 17 00:00:00 2001
From: Ishaan Jaff <ishaanjaffer0324@gmail.com>
Date: Wed, 10 Apr 2024 13:33:32 -0700
Subject: [PATCH] ui - set admin when setting up sso

---
 ui/litellm-dashboard/src/components/admins.tsx | 17 +++++++++++++----
 1 file changed, 13 insertions(+), 4 deletions(-)

diff --git a/ui/litellm-dashboard/src/components/admins.tsx b/ui/litellm-dashboard/src/components/admins.tsx
index 926df9b87..cff4ca2b8 100644
--- a/ui/litellm-dashboard/src/components/admins.tsx
+++ b/ui/litellm-dashboard/src/components/admins.tsx
@@ -61,6 +61,7 @@ const AdminPanel: React.FC<AdminPanelProps> = ({
   const [isInstructionsModalVisible, setIsInstructionsModalVisible] = useState(false);
 
 const handleAddSSOOk = () => {
+  
   setIsAddSSOModalVisible(false);
   form.resetFields();
 };
@@ -71,10 +72,11 @@ const handleAddSSOCancel = () => {
 };
 
 const handleShowInstructions = (formValues: Record<string, any>) => {
+  handleAdminCreate(formValues);
+  handleSSOUpdate(formValues);
   setIsAddSSOModalVisible(false);
   setIsInstructionsModalVisible(true);
   // Optionally, you can call handleSSOUpdate here with the formValues
-  // handleSSOUpdate(formValues);
 };
 
 const handleInstructionsOk = () => {
@@ -439,10 +441,17 @@ const handleInstructionsCancel = () => {
           labelAlign="left"
         >
           <>
+          <Form.Item
+              label="Admin Email"
+              name="user_email"
+              rules={[{ required: true, message: "Please enter the email of the proxy admin" }]}
+            >
+              <Input />
+            </Form.Item>
             <Form.Item
               label="PROXY BASE URL"
               name="proxy_base_url"
-              rules={[{ required: true, message: "Please enter the public key" }]}
+              rules={[{ required: true, message: "Please enter the proxy base url" }]}
             >
               <Input />
             </Form.Item>
@@ -450,7 +459,7 @@ const handleInstructionsCancel = () => {
             <Form.Item
               label="GOOGLE CLIENT ID"
               name="google_client_id"
-              rules={[{ required: true, message: "Please enter the public key" }]}
+              rules={[{ required: true, message: "Please enter the google client id" }]}
             >
               <Input.Password />
             </Form.Item>
@@ -458,7 +467,7 @@ const handleInstructionsCancel = () => {
             <Form.Item
               label="GOOGLE CLIENT SECRET"
               name="google_client_secret"
-              rules={[{ required: true, message: "Please enter the private key" }]}
+              rules={[{ required: true, message: "Please enter the google client secret" }]}
             >
               <Input.Password />
             </Form.Item>