Explorar o código

delete weight_type (#6865)

Jyong hai 8 meses
pai
achega
8157fccf6d

+ 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),
 }