From b75115264dbfe451fbfce1d431c883b8dc876204 Mon Sep 17 00:00:00 2001 From: balibabu Date: Tue, 13 Aug 2024 10:51:51 +0800 Subject: [PATCH] =?UTF-8?q?fix:=20Error=20on=20chat=20api=EF=BC=8C=20#1918=20(#1923)?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit ### What problem does this PR solve? fix: Error on chat api, #1918 ### Type of change - [x] Bug Fix (non-breaking change which fixes an issue) --- web/src/components/api-service/hooks.ts | 6 ++++-- web/src/hooks/chat-hooks.ts | 2 +- 2 files changed, 5 insertions(+), 3 deletions(-) diff --git a/web/src/components/api-service/hooks.ts b/web/src/components/api-service/hooks.ts index 4ca1b37a9..d93eab2db 100644 --- a/web/src/components/api-service/hooks.ts +++ b/web/src/components/api-service/hooks.ts @@ -1,6 +1,5 @@ import { useCreateNextToken, - useFetchNextStats, useFetchTokenList, useRemoveNextToken, } from '@/hooks/chat-hooks'; @@ -10,6 +9,7 @@ import { useTranslate, } from '@/hooks/common-hooks'; import { IStats } from '@/interfaces/database/chat'; +import { useQueryClient } from '@tanstack/react-query'; import { message } from 'antd'; import { useCallback } from 'react'; @@ -46,7 +46,9 @@ type ChartStatsType = { }; export const useSelectChartStatsList = (): ChartStatsType => { - const { data: stats } = useFetchNextStats(); + const queryClient = useQueryClient(); + const data = queryClient.getQueriesData({ queryKey: ['fetchStats'] }); + const stats: IStats = data[0][1] as IStats; return Object.keys(stats).reduce((pre, cur) => { const item = stats[cur as keyof IStats]; diff --git a/web/src/hooks/chat-hooks.ts b/web/src/hooks/chat-hooks.ts index 84089f0ea..93a5f6304 100644 --- a/web/src/hooks/chat-hooks.ts +++ b/web/src/hooks/chat-hooks.ts @@ -260,8 +260,8 @@ const getDay = (date?: Dayjs) => date?.format('YYYY-MM-DD'); export const useFetchNextStats = () => { const [pickerValue, setPickerValue] = useState([ - dayjs(), dayjs().subtract(7, 'day'), + dayjs(), ]); const { data, isFetching: loading } = useQuery({ queryKey: ['fetchStats', pickerValue],