From ebf05c10a9cdbd8e259f87cd4eda51ace56b43dc Mon Sep 17 00:00:00 2001
From: Ishaan Jaff <ishaanjaffer0324@gmail.com>
Date: Sat, 31 May 2025 12:21:40 -0700
Subject: [PATCH] (ui) fix view

---
 ui/litellm-dashboard/src/components/key_edit_view.tsx  | 4 +---
 ui/litellm-dashboard/src/components/team/team_info.tsx | 5 +----
 2 files changed, 2 insertions(+), 7 deletions(-)

diff --git a/ui/litellm-dashboard/src/components/key_edit_view.tsx b/ui/litellm-dashboard/src/components/key_edit_view.tsx
index 7c196b0f2d..4ec3abdc99 100644
--- a/ui/litellm-dashboard/src/components/key_edit_view.tsx
+++ b/ui/litellm-dashboard/src/components/key_edit_view.tsx
@@ -6,7 +6,6 @@ import { fetchTeamModels } from "../components/create_key_button";
 import { modelAvailableCall } from "./networking";
 import NumericalInput from "./shared/numerical_input";
 import VectorStoreSelector from "./vector_store_management/VectorStoreSelector";
-import PremiumVectorStoreSelector from "./common_components/PremiumVectorStoreSelector";
 
 interface KeyEditViewProps {
   keyData: KeyResponse;
@@ -173,12 +172,11 @@ export function KeyEditView({
       </Form.Item>
 
       <Form.Item label="Vector Stores" name="vector_stores">
-        <PremiumVectorStoreSelector
+        <VectorStoreSelector
           onChange={(values) => form.setFieldValue('vector_stores', values)}
           value={form.getFieldValue('vector_stores')}
           accessToken={accessToken || ""}
           placeholder="Select vector stores"
-          premiumUser={premiumUser}
         />
       </Form.Item>
 
diff --git a/ui/litellm-dashboard/src/components/team/team_info.tsx b/ui/litellm-dashboard/src/components/team/team_info.tsx
index b75721507d..eebb4b8801 100644
--- a/ui/litellm-dashboard/src/components/team/team_info.tsx
+++ b/ui/litellm-dashboard/src/components/team/team_info.tsx
@@ -329,7 +329,6 @@ const TeamInfoView: React.FC<TeamInfoProps> = ({
                 objectPermission={info.object_permission} 
                 variant="card"
                 accessToken={accessToken}
-                premiumUser={premiumUser}
               />
             </Grid>
           </TabPanel>
@@ -460,12 +459,11 @@ const TeamInfoView: React.FC<TeamInfoProps> = ({
                   </Form.Item>
 
                   <Form.Item label="Vector Stores" name="vector_stores">
-                    <PremiumVectorStoreSelector
+                    <VectorStoreSelector
                       onChange={(values) => form.setFieldValue('vector_stores', values)}
                       value={form.getFieldValue('vector_stores')}
                       accessToken={accessToken || ""}
                       placeholder="Select vector stores"
-                      premiumUser={premiumUser}
                     />
                   </Form.Item>
                   
@@ -537,7 +535,6 @@ const TeamInfoView: React.FC<TeamInfoProps> = ({
                     variant="inline"
                     className="pt-4 border-t border-gray-200"
                     accessToken={accessToken}
-                    premiumUser={premiumUser}
                   />
                 </div>
               )}