delete weight_type (#6865)

2 years ago · 8157fccf6d
parent cbf7f21ade
commit 8157fccf6d
3 changed files with 0 additions and 4 deletions
--- a/api/core/rag/data_post_processor/data_post_processor.py
+++ b/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'],
--- a/api/core/rag/rerank/entity/weight.py
+++ b/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
--- a/api/fields/dataset_fields.py
+++ b/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),
 }