diff --git a/client/src/data-provider/api-endpoints.ts b/client/src/data-provider/api-endpoints.ts index 74568e55a6..2ae4fc55f0 100644 --- a/client/src/data-provider/api-endpoints.ts +++ b/client/src/data-provider/api-endpoints.ts @@ -6,6 +6,10 @@ export const messages = (id: string) => { return `/api/messages/${id}`; }; +export const abortRequest = (endpoint: string) => { + return `/api/ask/${endpoint}/abort`; +}; + export const conversations = (pageNumber: string) => { return `/api/convos?pageNumber=${pageNumber}`; }; diff --git a/client/src/data-provider/data-service.ts b/client/src/data-provider/data-service.ts index 1b717dc888..3a8a6a4ab6 100644 --- a/client/src/data-provider/data-service.ts +++ b/client/src/data-provider/data-service.ts @@ -6,6 +6,10 @@ export function getConversations(pageNumber: string): Promise { + return request.post(endpoints.abortRequest(endpoint), { arg: {abortKey, message} }); +} + export function deleteConversation(payload: t.TDeleteConversationRequest) { //todo: this should be a DELETE request return request.post(endpoints.deleteConversation(), {arg: payload}); diff --git a/client/src/data-provider/react-query-service.ts b/client/src/data-provider/react-query-service.ts index c85660214c..8c8c406a75 100644 --- a/client/src/data-provider/react-query-service.ts +++ b/client/src/data-provider/react-query-service.ts @@ -21,6 +21,10 @@ export enum QueryKeys { tokenCount = "tokenCount", } +export const useAbortRequestWithMessage = (): UseMutationResult => { + return useMutation(({ endpoint, abortKey, message }) => dataService.abortRequestWithMessage(endpoint, abortKey, message)); +}; + export const useGetUserQuery = (): QueryObserverResult => { return useQuery([QueryKeys.user], () => dataService.getUser(), { refetchOnWindowFocus: false,