From f4f7a058f44ea23cd572bbcf791baf6a184cef9b Mon Sep 17 00:00:00 2001
From: Ishaan Jaff <ishaanjaffer0324@gmail.com>
Date: Tue, 4 Jun 2024 21:11:55 -0700
Subject: [PATCH] ui - fix show internal_user -> Internal User on UI

---
 ui/litellm-dashboard/src/app/page.tsx                  | 4 ++++
 ui/litellm-dashboard/src/components/user_dashboard.tsx | 4 ++++
 2 files changed, 8 insertions(+)

diff --git a/ui/litellm-dashboard/src/app/page.tsx b/ui/litellm-dashboard/src/app/page.tsx
index c218a8aad..522289ef3 100644
--- a/ui/litellm-dashboard/src/app/page.tsx
+++ b/ui/litellm-dashboard/src/app/page.tsx
@@ -35,6 +35,10 @@ function formatUserRole(userRole: string) {
       return "Admin";
     case "proxy_admin_viewer":
       return "Admin Viewer";
+    case "internal_user":
+      return "Internal User";
+    case "internal_viewer":
+      return "Internal Viewer";
     case "app_user":
       return "App User";
     default:
diff --git a/ui/litellm-dashboard/src/components/user_dashboard.tsx b/ui/litellm-dashboard/src/components/user_dashboard.tsx
index 587719b0a..1a2439055 100644
--- a/ui/litellm-dashboard/src/components/user_dashboard.tsx
+++ b/ui/litellm-dashboard/src/components/user_dashboard.tsx
@@ -99,6 +99,10 @@ const UserDashboard: React.FC<UserDashboardProps> = ({
         return "Admin Viewer";
       case "app_user":
         return "App User";
+      case "internal_user":
+        return "Internal User";
+      case "internal_viewer":
+        return "Internal Viewer";
       default:
         return "Unknown Role";
     }