youngkingdom/dify
1
0
Fork 0
mirror of https://github.com/langgenius/dify.git synced 2026-03-27 01:00:13 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix(web): keep hit-testing payload

Browse Source
This commit is contained in:
yyh
2026-01-02 19:03:56 +08:00
parent 6d7026255e
commit bde4b82695
2 changed files with 4 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
web/service/datasets.ts
View File

@ -28,6 +28,7 @@ import type {
FetchDatasetsParams,
FileIndexingEstimateResponse,
HitTestingRecordsResponse,
HitTestingRequest,
HitTestingResponse,
IndexingEstimateParams,
IndexingEstimateResponse,
@ -187,8 +188,8 @@ export const modifyDocMetadata = ({ datasetId, documentId, body }: CommonDocReq
}
// hit testing
export const hitTesting = ({ datasetId, queryText, retrieval_model }: { datasetId: string, queryText: string, retrieval_model: RetrievalConfig }): Promise<HitTestingResponse> => {
return post<HitTestingResponse>(`/datasets/${datasetId}/hit-testing`, { body: { query: queryText, retrieval_model } })
export const hitTesting = ({ datasetId, query, retrieval_model, attachment_ids }: { datasetId: string } & HitTestingRequest): Promise<HitTestingResponse> => {
return post<HitTestingResponse>(`/datasets/${datasetId}/hit-testing`, { body: { query, retrieval_model, attachment_ids } })
}
export const externalKnowledgeBaseHitTesting = ({ datasetId, query, external_retrieval_model }: { datasetId: string, query: string, external_retrieval_model: { top_k: number, score_threshold: number, score_threshold_enabled: boolean } }): Promise<ExternalKnowledgeBaseHitTestingResponse> => {

2
web/service/knowledge/use-hit-testing.ts
View File

@ -26,7 +26,7 @@ export const useInvalidateHitTestingRecords = (datasetId: string) => {
export const useHitTesting = (datasetId: string) => {
return useMutation({
mutationKey: [NAME_SPACE, 'hit-testing', datasetId],
mutationFn: (params: HitTestingRequest) => hitTesting({ datasetId, queryText: params.query, retrieval_model: params.retrieval_model }),
mutationFn: (params: HitTestingRequest) => hitTesting({ datasetId, ...params }),
})
}