diff --git a/components/ai_chat/resources/page/components/model_menu_item/model_menu_item.tsx b/components/ai_chat/resources/page/components/model_menu_item/model_menu_item.tsx index 065ef260127..5284a45b75c 100644 --- a/components/ai_chat/resources/page/components/model_menu_item/model_menu_item.tsx +++ b/components/ai_chat/resources/page/components/model_menu_item/model_menu_item.tsx @@ -39,6 +39,17 @@ const ModelContent = (props: ModelContentProps) => { ) } + if (model.isNearModel) { + return ( + + ) + } if ( model.options.leoModelOptions?.access === Mojom.ModelAccess.PREMIUM && showPremiumLabel diff --git a/components/ai_chat/resources/page/components/model_selector/index.test.tsx b/components/ai_chat/resources/page/components/model_selector/index.test.tsx index a6d3c21129f..c6bb8248689 100644 --- a/components/ai_chat/resources/page/components/model_selector/index.test.tsx +++ b/components/ai_chat/resources/page/components/model_selector/index.test.tsx @@ -310,7 +310,7 @@ describe('ModelSelector', () => { expect(alert).toHaveTextContent('CHAT_UI_AGENT_MODE_MODEL_INFO') }, ) - it('should display near icon for near models', async () => { + it('should display near icon and beta label for near models', async () => { render() // Make sure the anchor button is visible @@ -327,11 +327,13 @@ describe('ModelSelector', () => { // Make sure the default menu items are visible const menuItems = document.querySelectorAll('leo-menu-item') - expect(menuItems[2]).toHaveTextContent('Premium Model') + const nearMenuItem = menuItems[2] + + // Check that the beta label is visible + expect(nearMenuItem).toHaveTextContent('CHAT_UI_MODEL_BETA_LABEL') // Check that the near icon is present in the Premium Model menu item - const premiumMenuItem = within(menuItems[2]) - const nearIcon = premiumMenuItem.getByRole('img') + const nearIcon = within(nearMenuItem).getByRole('img') expect(nearIcon).toHaveClass('nearIcon') }) }) diff --git a/components/resources/ai_chat_ui_strings.grdp b/components/resources/ai_chat_ui_strings.grdp index 19996701da8..e8d16c75a25 100644 --- a/components/resources/ai_chat_ui_strings.grdp +++ b/components/resources/ai_chat_ui_strings.grdp @@ -147,6 +147,9 @@ Local + + Beta + Recommended models