diff --git a/packages/agents-manager/src/hooks/use-sources-action.ts b/packages/agents-manager/src/hooks/use-sources-action.ts index 13dd7270d29c..418bc2abf85f 100644 --- a/packages/agents-manager/src/hooks/use-sources-action.ts +++ b/packages/agents-manager/src/hooks/use-sources-action.ts @@ -47,7 +47,7 @@ export default function useSourcesAction( registerMessageActions: RegisterMessag return [ { - type: 'component' as const, + type: 'component', id: 'sources', label: 'Sources', component: SourcesDisplay, diff --git a/packages/agents-manager/src/hooks/use-zoom-action.ts b/packages/agents-manager/src/hooks/use-zoom-action.ts index 61111e09dc52..c11e4534e808 100644 --- a/packages/agents-manager/src/hooks/use-zoom-action.ts +++ b/packages/agents-manager/src/hooks/use-zoom-action.ts @@ -1,5 +1,6 @@ import { useEffect } from '@wordpress/element'; import ZoomActionButton from '../components/zoom-action-button'; +import { isEditorPage } from '../utils/is-editor-page'; import type { UseAgentChatReturn, UIMessage } from '@automattic/agenttic-client'; type RegisterMessageActions = UseAgentChatReturn[ 'registerMessageActions' ]; @@ -12,7 +13,7 @@ export default function useZoomAction( registerMessageActions: RegisterMessageAc registerMessageActions( { id: 'agents-manager-zoom', actions: ( message: UIMessage ) => { - if ( message.role !== 'agent' ) { + if ( message.role !== 'agent' || ! isEditorPage() ) { return []; } @@ -30,7 +31,7 @@ export default function useZoomAction( registerMessageActions: RegisterMessageAc return [ { type: 'component', - id: 'zoom-toggle', + id: 'zoom', component: ZoomActionButton, order: 5, },