From 9ae6eb1388508756d769a7056561a9cd12125bc8 Mon Sep 17 00:00:00 2001
From: Ishaan Jaff <ishaanjaffer0324@gmail.com>
Date: Wed, 29 May 2024 14:19:31 -0700
Subject: [PATCH] feat - implement logout sso

---
 ui/litellm-dashboard/src/components/navbar.tsx | 10 ++++++++++
 1 file changed, 10 insertions(+)

diff --git a/ui/litellm-dashboard/src/components/navbar.tsx b/ui/litellm-dashboard/src/components/navbar.tsx
index f37dcb63e..dfa17add4 100644
--- a/ui/litellm-dashboard/src/components/navbar.tsx
+++ b/ui/litellm-dashboard/src/components/navbar.tsx
@@ -39,7 +39,9 @@ const Navbar: React.FC<NavbarProps> = ({
 
   // const userColors = require('./ui_colors.json') || {};
   const isLocal = process.env.NODE_ENV === "development";
+  const proxyBaseUrl = isLocal ? "http://localhost:4000" : null;
   const imageUrl = isLocal ? "http://localhost:4000/get_image" : "/get_image";
+  const logoutUrl = proxyBaseUrl ? `${proxyBaseUrl}` : `/`;
 
   const items: MenuProps["items"] = [
     {
@@ -52,6 +54,14 @@ const Navbar: React.FC<NavbarProps> = ({
         </>
       ),
     },
+    {
+      key: "2",
+      label: (
+        <Link href={logoutUrl}>
+          <p>Logout</p>
+        </Link>
+      ),
+    }
   ];
 
   return (