zlei6/ragflow
1
0
Fork 0
mirror of https://github.com/infiniflow/ragflow.git synced 2026-08-23 16:46:36 +08:00
Code Issues Packages Projects Releases Wiki Activity

Refactor: do not include compliation chunks in retrieval testing (#18364)

Browse Source 
Retrieval Testing: do not include compilation result.
Search: include compilation result.
This commit is contained in:
Wang Qi
2026-08-17 16:19:05 +08:00
committed by GitHub
parent 3e38b6cacb
commit 57b907cd1e
11 changed files with 47 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
internal/handler/dataset_search_test.go
View File

@@ -157,7 +157,7 @@ func TestDatasetsHandlerSearchDatasetsSuccess(t *testing.T) {
h := &DatasetsHandler{searchDatasetsService: fake}
rec := httptest.NewRecorder()
req := httptest.NewRequest(http.MethodPost, "/api/v1/datasets/search", strings.NewReader(`{"question":" hello ","dataset_ids":["ds-1"],"top_k":7}`))
req := httptest.NewRequest(http.MethodPost, "/api/v1/datasets/search", strings.NewReader(`{"question":" hello ","dataset_ids":["ds-1"],"top_k":7,"include_knowledge_compilation":false}`))
req.Header.Set("Content-Type", "application/json")
c, _ := gin.CreateTestContext(rec)
c.Request = req
@@ -171,6 +171,9 @@ func TestDatasetsHandlerSearchDatasetsSuccess(t *testing.T) {
if fake.userID != "user-1" || fake.req == nil || fake.req.Question != "hello" || len(fake.req.DatasetIDs) != 1 || fake.req.DatasetIDs[0] != "ds-1" {
t.Fatalf("call args userID=%q req=%#v", fake.userID, fake.req)
}
if fake.req.IncludeCompiledChunks == nil || *fake.req.IncludeCompiledChunks {
t.Fatalf("include_knowledge_compilation=%v want false", fake.req.IncludeCompiledChunks)
}
body := decodeSearchResponse(t, rec)
if body["code"] != float64(common.CodeSuccess) {
t.Fatalf("code=%v want=%d body=%s", body["code"], common.CodeSuccess, rec.Body.String())