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

2024-10-30 11:17:39 +08:00
parent 5ad5d0cff4
commit 9ebd453b87
2 changed files with 32 additions and 2 deletions
--- a/api/core/rag/rerank/rerank_type.py
+++ b/api/core/rag/rerank/rerank_type.py
@@ -1,6 +1,6 @@
 from enum import Enum


-class RerankMode(Enum):
+class RerankMode(str, Enum):
    RERANKING_MODEL = "reranking_model"
    WEIGHTED_SCORE = "weighted_score"