jyapayne/dify
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

add rerank check when doing mutil-retrieval (#9998)

Browse source
This commit is contained in:
Jyong 2024-10-30 11:17:39 +08:00 • committed by GitHub
commit 9ebd453b87
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 32 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

2
api/core/rag/rerank/rerank_type.py
View file

@ -1,6 +1,6 @@
from enum import Enum
class RerankMode(Enum):
class RerankMode(str, Enum):
RERANKING_MODEL = "reranking_model"
WEIGHTED_SCORE = "weighted_score"