fix:hard-coded top-k fallback issue. (#24879)

2025-09-01 15:46:37 +08:00
parent d41d4deaac
commit 5b3cc560d5
11 changed files with 16 additions and 16 deletions
--- a/api/services/dataset_service.py
+++ b/api/services/dataset_service.py
@@ -1149,7 +1149,7 @@ class DocumentService:
                        "search_method": RetrievalMethod.SEMANTIC_SEARCH.value,
                        "reranking_enable": False,
                        "reranking_model": {"reranking_provider_name": "", "reranking_model_name": ""},
-                        "top_k": 2,
+                        "top_k": 4,
                        "score_threshold_enabled": False,
                    }

@@ -1612,7 +1612,7 @@ class DocumentService:
                search_method=RetrievalMethod.SEMANTIC_SEARCH.value,
                reranking_enable=False,
                reranking_model=RerankingModel(reranking_provider_name="", reranking_model_name=""),
-                top_k=2,
+                top_k=4,
                score_threshold_enabled=False,
            )
        # save dataset