delete weight_type (#6865)

2024-08-01 13:02:33 +08:00 · 2024-08-01 13:02:33 +08:00 · 8157fccf6d
commit 8157fccf6d
parent cbf7f21ade
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