fix: Remove kAModel #1306 (#1713)

### What problem does this PR solve?

fix: Remove kAModel #1306
### Type of change

- [x] Bug Fix (non-breaking change which fixes an issue)
This commit is contained in:
balibabu
2024-07-26 10:23:51 +08:00
committed by GitHub
parent c943517932
commit 3f495b2d22
4 changed files with 25 additions and 39 deletions

View File

@ -236,4 +236,22 @@ export const useTestChunkRetrieval = () => {
return testChunk;
};
export const useTestNextChunkRetrieval = () => {
const {
data,
isPending: loading,
mutateAsync,
} = useMutation({
mutationKey: ['testChunk'],
mutationFn: async (canvasIds: string[]) => {
const { data } = await kbService.retrieval_test({ canvasIds });
if (data.retcode === 0) {
}
return data?.data ?? [];
},
});
return { data, loading, testChunk: mutateAsync };
};
//#endregion