diff --git a/components/ai_chat/resources/page/state/conversation_context.tsx b/components/ai_chat/resources/page/state/conversation_context.tsx index 9681de83a993..095c00fef599 100644 --- a/components/ai_chat/resources/page/state/conversation_context.tsx +++ b/components/ai_chat/resources/page/state/conversation_context.tsx @@ -4,12 +4,10 @@ // You can obtain one at https://mozilla.org/MPL/2.0/. import * as React from 'react' - import * as mojom from 'gen/brave/components/ai_chat/core/common/mojom/ai_chat.mojom.m.js' -import usePromise from '$web-common/usePromise' import * as API from '../api/' -import { useAIChat } from './ai_chat_context' import { isLeoModel } from '../model_utils' +import { useAIChat } from './ai_chat_context' const MAX_INPUT_CHAR = 2000 const CHAR_LIMIT_THRESHOLD = MAX_INPUT_CHAR * 0.8 @@ -125,10 +123,8 @@ export const getFirstValidAction = (actionList: mojom.ActionGroup[]) => export function useActionMenu( filter: string, - getActions: () => Promise + actionList: mojom.ActionGroup[] ) { - const { result: actionList = [] } = usePromise(getActions, []) - return React.useMemo(() => { const reg = new RegExp(/^\/\w+/) @@ -314,8 +310,7 @@ export function ConversationContextProvider( }) }, [context.conversationUuid, context.faviconCacheKey]) - const actionList = useActionMenu(context.inputText, () => - Promise.resolve(aiChatContext.allActions) + const actionList = useActionMenu(context.inputText, aiChatContext.allActions ) const shouldShowLongConversationInfo = React.useMemo(() => { diff --git a/components/ai_rewriter/resources/page/Context.tsx b/components/ai_rewriter/resources/page/Context.tsx index 5ab6e31d0442..a36fc792826b 100644 --- a/components/ai_rewriter/resources/page/Context.tsx +++ b/components/ai_rewriter/resources/page/Context.tsx @@ -111,15 +111,21 @@ export default function Context(props: React.PropsWithChildren) { const [forwardHistory, setForwardHistory] = React.useState([]) const [backHistory, setBackHistory] = React.useState([]) const [isGenerating, setIsGenerating] = React.useState(false) + const [allActions, setAllActions] = React.useState([]) - const actionList = useActionMenu(instructionsText, () => getRewriterPageHandler().getActionMenuList().then(({ actionList }) => actionList)) + const actionList = useActionMenu(instructionsText, allActions) const charCountContext = useCharCountInfo(instructionsText) React.useEffect(() => { - getRewriterPageHandler() + const rewriterAPI = getRewriterPageHandler() + rewriterAPI .getInitialText() .then(({ initialText }) => setInitialText(initialText)) + rewriterAPI + .getActionMenuList() + .then(({ actionList }) => setAllActions(actionList)) + const callbackRouter = getCallbackRouter() callbackRouter.onUpdatedGeneratedText.addListener(setGeneratedText) return () => {