| @@ -84,6 +84,22 @@ class RerankingModel(BaseModel): | |||
| reranking_model_name: Optional[str] = None | |||
| class WeightVectorSetting(BaseModel): | |||
| vector_weight: float | |||
| embedding_provider_name: str | |||
| embedding_model_name: str | |||
| class WeightKeywordSetting(BaseModel): | |||
| keyword_weight: float | |||
| class WeightModel(BaseModel): | |||
| weight_type: str | |||
| vector_setting: Optional[WeightVectorSetting] = None | |||
| keyword_setting: Optional[WeightKeywordSetting] = None | |||
| class RetrievalModel(BaseModel): | |||
| search_method: Literal["hybrid_search", "semantic_search", "full_text_search"] | |||
| reranking_enable: bool | |||
| @@ -92,6 +108,7 @@ class RetrievalModel(BaseModel): | |||
| top_k: int | |||
| score_threshold_enabled: bool | |||
| score_threshold: Optional[float] = None | |||
| weights: Optional[WeightModel] = None | |||
| class MetaDataConfig(BaseModel): | |||