diff --git a/web/service/datasets.ts b/web/service/datasets.ts
index aa42baab6b..e91fb18af1 100644
--- a/web/service/datasets.ts
+++ b/web/service/datasets.ts
@@ -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> => {
diff --git a/web/service/knowledge/use-hit-testing.ts b/web/service/knowledge/use-hit-testing.ts
index d84caff6e0..0395c3652b 100644
--- a/web/service/knowledge/use-hit-testing.ts
+++ b/web/service/knowledge/use-hit-testing.ts
@@ -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 }),
   })
 }