Bladeren bron

fix dataset reranking mode miss (#15643)

Jyong 1 maand geleden
bovenliggende
commit
d37490adc3
1 gewijzigde bestanden met toevoegingen van 1 en 0 verwijderingen
  1. 1 0
      api/services/entities/knowledge_entities/knowledge_entities.py

+ 1 - 0
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"]
     search_method: Literal["hybrid_search", "semantic_search", "full_text_search"]
     reranking_enable: bool
     reranking_enable: bool
     reranking_model: Optional[RerankingModel] = None
     reranking_model: Optional[RerankingModel] = None
+    reranking_mode: Optional[str] = None
     top_k: int
     top_k: int
     score_threshold_enabled: bool
     score_threshold_enabled: bool
     score_threshold: Optional[float] = None
     score_threshold: Optional[float] = None