瀏覽代碼

delete weight_type (#6865)

tags/0.6.16
Jyong 1 年之前
父節點
當前提交
8157fccf6d
No account linked to committer's email address

+ 0
- 1
api/core/rag/data_post_processor/data_post_processor.py 查看文件

@@ -37,7 +37,6 @@ class DataPostProcessor:
return WeightRerankRunner(
tenant_id,
Weights(
weight_type=weights['weight_type'],
vector_setting=VectorSetting(
vector_weight=weights['vector_setting']['vector_weight'],
embedding_provider_name=weights['vector_setting']['embedding_provider_name'],

+ 0
- 2
api/core/rag/rerank/entity/weight.py 查看文件

@@ -16,8 +16,6 @@ class KeywordSetting(BaseModel):
class Weights(BaseModel):
"""Model for weighted rerank."""

weight_type: str

vector_setting: VectorSetting

keyword_setting: KeywordSetting

+ 0
- 1
api/fields/dataset_fields.py 查看文件

@@ -29,7 +29,6 @@ vector_setting_fields = {
}

weighted_score_fields = {
'weight_type': fields.String,
'keyword_setting': fields.Nested(keyword_setting_fields),
'vector_setting': fields.Nested(vector_setting_fields),
}

Loading…
取消
儲存