fix wrong using of RetrievalMethod Enum (#6345)

2024-07-16 19:09:04 +08:00
parent ed9e692263
commit 0de224b153
10 changed files with 24 additions and 24 deletions
--- a/api/services/hit_testing_service.py
+++ b/api/services/hit_testing_service.py
@@ -9,7 +9,7 @@ from models.account import Account
 from models.dataset import Dataset, DatasetQuery, DocumentSegment

 default_retrieval_model = {
-    'search_method': RetrievalMethod.SEMANTIC_SEARCH,
+    'search_method': RetrievalMethod.SEMANTIC_SEARCH.value,
    'reranking_enable': False,
    'reranking_model': {
        'reranking_provider_name': '',