瀏覽代碼

fix Weight rerank mode info missed when create dataset (#16190)

tags/1.1.1
Jyong 7 月之前
父節點
當前提交
e7572066a4
No account linked to committer's email address
共有 1 個文件被更改,包括 17 次插入0 次删除
  1. 17
    0
      api/services/entities/knowledge_entities/knowledge_entities.py

+ 17
- 0
api/services/entities/knowledge_entities/knowledge_entities.py 查看文件

@@ -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):

Loading…
取消
儲存