diff --git a/apps/web/components/company-brain-header.tsx b/apps/web/components/company-brain-header.tsx index 99cd11162..e7c9258a1 100644 --- a/apps/web/components/company-brain-header.tsx +++ b/apps/web/components/company-brain-header.tsx @@ -21,6 +21,7 @@ import { LifeBuoy, LayoutGrid, MenuIcon, + Plus, SearchIcon, Settings, Settings2, @@ -55,6 +56,7 @@ const BACKEND = type SlackStatus = { connected: boolean; teamName: string | null } interface CompanyBrainHeaderProps { + onAddMemory?: () => void onOpenSearch?: () => void } @@ -108,7 +110,10 @@ function useSlackStatus() { }) } -export function CompanyBrainHeader({ onOpenSearch }: CompanyBrainHeaderProps) { +export function CompanyBrainHeader({ + onAddMemory, + onOpenSearch, +}: CompanyBrainHeaderProps) { const { user, org, organizations, setActiveOrg } = useAuth() const autumn = useCustomer() const { currentPlan } = useTokenUsage(autumn) @@ -412,6 +417,18 @@ export function CompanyBrainHeader({ onOpenSearch }: CompanyBrainHeaderProps) { "linear-gradient(180deg, #0A0E14 0%, #05070A 100%)", }} > + {onAddMemory && ( + <> + + + Add memory + + + + )} + {onAddMemory && ( + + + + + + Add memory (C) + + + )} {canInvite && ( diff --git a/apps/web/components/header.tsx b/apps/web/components/header.tsx index c30f84c4d..20f76f192 100644 --- a/apps/web/components/header.tsx +++ b/apps/web/components/header.tsx @@ -70,7 +70,12 @@ const brainTileClass = (active: boolean) => export function Header(props: HeaderProps) { const hasCompanyBrain = useHasCompanyBrain() if (hasCompanyBrain) { - return + return ( + + ) } return }