feat: send question with retrieval api #2247 (#2272)

### What problem does this PR solve?
feat: send question with retrieval api #2247

### Type of change


- [x] New Feature (non-breaking change which adds functionality)
This commit is contained in:
balibabu
2024-09-05 19:32:55 +08:00
committed by GitHub
parent 9377192859
commit 6ae0da92cb
17 changed files with 264 additions and 78 deletions

View File

@ -5,7 +5,10 @@ import {
IStats,
IToken,
} from '@/interfaces/database/chat';
import { IFeedbackRequestBody } from '@/interfaces/request/chat';
import {
IAskRequestBody,
IFeedbackRequestBody,
} from '@/interfaces/request/chat';
import i18n from '@/locales/config';
import { IClientConversation } from '@/pages/chat/interface';
import chatService from '@/services/chat-service';
@ -477,3 +480,23 @@ export const useFetchNextSharedConversation = (conversationId: string) => {
};
//#endregion
//#region search page
export const useFetchMindMap = () => {
const {
data,
isPending: loading,
mutateAsync,
} = useMutation({
mutationKey: ['fetchMindMap'],
mutationFn: async (params: IAskRequestBody) => {
const { data } = await chatService.getMindMap(params);
return data;
},
});
return { data, loading, fetchMindMap: mutateAsync };
};
//#endregion

View File

@ -209,7 +209,7 @@ export const useTestChunkRetrieval = (): ResponsePostType<ITestingResult> & {
mutationFn: async (values: any) => {
const { data } = await kbService.retrieval_test({
...values,
kb_id: knowledgeBaseId,
kb_id: values.kb_id ?? knowledgeBaseId,
page,
size: pageSize,
});

View File

@ -243,6 +243,10 @@ export const useSendMessageWithSse = (
const x = await reader?.read();
if (x) {
const { done, value } = x;
if (done) {
console.info('done');
break;
}
try {
const val = JSON.parse(value?.data || '');
const d = val?.data;
@ -256,10 +260,6 @@ export const useSendMessageWithSse = (
} catch (e) {
console.warn(e);
}
if (done) {
console.info('done');
break;
}
}
}
console.info('done?');