diff --git a/extension/app/src/pages/MainPage.tsx b/extension/app/src/pages/MainPage.tsx index 039ec3e4b217..a48a91ec8894 100644 --- a/extension/app/src/pages/MainPage.tsx +++ b/extension/app/src/pages/MainPage.tsx @@ -1,5 +1,6 @@ import { Avatar, + Button, DropdownMenu, DropdownMenuContent, DropdownMenuItem, @@ -7,6 +8,7 @@ import { LogoHorizontalColorLogo, LogoutIcon, } from "@dust-tt/sparkle"; +import { useAuth } from "@extension/components/auth/AuthProvider"; import type { ProtectedRouteChildrenProps } from "@extension/components/auth/ProtectedRoute"; import { ConversationContainer } from "@extension/components/conversation/ConversationContainer"; import { ConversationsListButton } from "@extension/components/conversation/ConversationsListButton"; @@ -18,13 +20,40 @@ export const MainPage = ({ workspace, handleLogout, }: ProtectedRouteChildrenProps) => { + const { handleSelectWorkspace } = useAuth(); + return ( <>
-
+
+ {user.workspaces.length > 1 && ( +
+

Workspace:

+ + +
+ )}