fix wrong using of RetrievalMethod Enum (#6345)

This commit is contained in:
Jyong 2024-07-16 19:09:04 +08:00 committed by GitHub
commit 0de224b153
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
10 changed files with 24 additions and 24 deletions

View file

@ -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': '',