diff --git a/web/src/hooks/chat-hooks.ts b/web/src/hooks/chat-hooks.ts index e5e29dac6..91054c75b 100644 --- a/web/src/hooks/chat-hooks.ts +++ b/web/src/hooks/chat-hooks.ts @@ -95,7 +95,12 @@ export const useSetNextDialog = () => { mutationFn: async (params: IDialog) => { const { data } = await chatService.setDialog(params); if (data.retcode === 0) { - queryClient.invalidateQueries({ queryKey: ['fetchDialogList'] }); + queryClient.invalidateQueries({ + queryKey: ['fetchDialogList'], + }); + queryClient.invalidateQueries({ + queryKey: ['fetchDialog'], + }); message.success( i18n.t(`message.${params.dialog_id ? 'modified' : 'created'}`), ); @@ -110,7 +115,11 @@ export const useSetNextDialog = () => { export const useFetchNextDialog = () => { const { dialogId } = useGetChatSearchParams(); - const { data, isFetching: loading } = useQuery({ + const { + data, + isFetching: loading, + refetch, + } = useQuery({ queryKey: ['fetchDialog', dialogId], gcTime: 0, initialData: {} as IDialog, @@ -123,7 +132,7 @@ export const useFetchNextDialog = () => { }, }); - return { data, loading }; + return { data, loading, refetch }; }; export const useFetchManualDialog = () => { diff --git a/web/src/pages/chat/hooks.ts b/web/src/pages/chat/hooks.ts index 6be9e0382..b11b9e098 100644 --- a/web/src/pages/chat/hooks.ts +++ b/web/src/pages/chat/hooks.ts @@ -429,7 +429,7 @@ export const useSendNextMessage = () => { if ( answer.answer && (answer?.conversationId === conversationId || - (!done && conversationId === '')) + ((!done || (done && answer.audio_binary)) && conversationId === '')) ) { addNewestAnswer(answer); }