fix wrong using of RetrievalMethod Enum (#6345)

2025-12-09 02:46:52 +08:00 · 2024-07-16 19:09:04 +08:00
parent ed9e692263
commit 0de224b153
10 changed files with 24 additions and 24 deletions
--- a/api/models/dataset.py
+++ b/api/models/dataset.py
@@ -117,7 +117,7 @@ class Dataset(db.Model):
    @property
    def retrieval_model_dict(self):
        default_retrieval_model = {
-            'search_method': RetrievalMethod.SEMANTIC_SEARCH,
+            'search_method': RetrievalMethod.SEMANTIC_SEARCH.value,
            'reranking_enable': False,
            'reranking_model': {
                'reranking_provider_name': '',