From d37490adc35de0afe63bcee432a0fd7f7a8ec1d1 Mon Sep 17 00:00:00 2001
From: Jyong <76649700+JohnJyong@users.noreply.github.com>
Date: Wed, 12 Mar 2025 18:44:10 +0800
Subject: [PATCH] fix dataset reranking mode miss (#15643)

---
 api/services/entities/knowledge_entities/knowledge_entities.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/api/services/entities/knowledge_entities/knowledge_entities.py b/api/services/entities/knowledge_entities/knowledge_entities.py
index f14c5b513a..b84339fd5f 100644
--- a/api/services/entities/knowledge_entities/knowledge_entities.py
+++ b/api/services/entities/knowledge_entities/knowledge_entities.py
@@ -88,6 +88,7 @@ class RetrievalModel(BaseModel):
     search_method: Literal["hybrid_search", "semantic_search", "full_text_search"]
     reranking_enable: bool
     reranking_model: Optional[RerankingModel] = None
+    reranking_mode: Optional[str] = None
     top_k: int
     score_threshold_enabled: bool
     score_threshold: Optional[float] = None