+
)}
-
;
+ {/*
; */}
+ {!isSearching && (
+
+
+
+ )}
+ {isSearching && (
+
+
+
+ )}
);
}
diff --git a/web/src/pages/next-searches/hooks.ts b/web/src/pages/next-searches/hooks.ts
index ac20d6541..01dac13f2 100644
--- a/web/src/pages/next-searches/hooks.ts
+++ b/web/src/pages/next-searches/hooks.ts
@@ -6,7 +6,6 @@ import { useMutation, useQuery, useQueryClient } from '@tanstack/react-query';
import { useCallback, useState } from 'react';
import { useTranslation } from 'react-i18next';
import { useParams, useSearchParams } from 'umi';
-
interface CreateSearchProps {
name: string;
description?: string;
@@ -122,40 +121,6 @@ interface DeleteSearchResponse {
message: string;
}
-export const useDeleteSearch = () => {
- const { t } = useTranslation();
-
- const {
- data,
- isError,
- mutateAsync: deleteSearchMutation,
- } = useMutation
({
- mutationKey: ['deleteSearch'],
- mutationFn: async (props) => {
- const response = await searchService.deleteSearch(props);
- if (response.code !== 0) {
- throw new Error(response.message || 'Failed to delete search');
- }
- return response;
- },
- onSuccess: () => {
- message.success(t('message.deleted'));
- },
- onError: (error) => {
- message.error(t('message.error', { error: error.message }));
- },
- });
-
- const deleteSearch = useCallback(
- (props: DeleteSearchProps) => {
- return deleteSearchMutation(props);
- },
- [deleteSearchMutation],
- );
-
- return { data, isError, deleteSearch };
-};
-
export interface IllmSettingProps {
llm_id: string;
parameter: string;
@@ -237,6 +202,42 @@ export const useFetchSearchDetail = (tenantId?: string) => {
return { data: data?.data, isLoading, isError };
};
+export const useDeleteSearch = () => {
+ const { t } = useTranslation();
+ const queryClient = useQueryClient();
+ const {
+ data,
+ isError,
+ mutateAsync: deleteSearchMutation,
+ } = useMutation({
+ mutationKey: ['deleteSearch'],
+ mutationFn: async (props) => {
+ const { data: response } = await searchService.deleteSearch(props);
+ if (response.code !== 0) {
+ throw new Error(response.message || 'Failed to delete search');
+ }
+
+ queryClient.invalidateQueries({ queryKey: ['searchList'] });
+ return response;
+ },
+ onSuccess: () => {
+ message.success(t('message.deleted'));
+ },
+ onError: (error) => {
+ message.error(t('message.error', { error: error.message }));
+ },
+ });
+
+ const deleteSearch = useCallback(
+ (props: DeleteSearchProps) => {
+ return deleteSearchMutation(props);
+ },
+ [deleteSearchMutation],
+ );
+
+ return { data, isError, deleteSearch };
+};
+
export type IUpdateSearchProps = Omit & {
search_id: string;
};
diff --git a/web/src/pages/search/hooks.ts b/web/src/pages/search/hooks.ts
index 8c7607972..abfecd698 100644
--- a/web/src/pages/search/hooks.ts
+++ b/web/src/pages/search/hooks.ts
@@ -217,7 +217,11 @@ export const useTestRetrieval = (
};
};
-export const useShowMindMapDrawer = (kbIds: string[], question: string) => {
+export const useShowMindMapDrawer = (
+ kbIds: string[],
+ question: string,
+ searchId = '',
+) => {
const { visible, showModal, hideModal } = useSetModalState();
const ref = useRef();
@@ -228,7 +232,7 @@ export const useShowMindMapDrawer = (kbIds: string[], question: string) => {
} = useSearchFetchMindMap();
const handleShowModal = useCallback(() => {
- const searchParams = { question: trim(question), kb_ids: kbIds };
+ const searchParams = { question: trim(question), kb_ids: kbIds, searchId };
if (
!isEmpty(searchParams.question) &&
!isEqual(searchParams, ref.current)
@@ -237,7 +241,7 @@ export const useShowMindMapDrawer = (kbIds: string[], question: string) => {
fetchMindMap(searchParams);
}
showModal();
- }, [fetchMindMap, showModal, question, kbIds]);
+ }, [fetchMindMap, showModal, question, kbIds, searchId]);
return {
mindMap,