From 8157fccf6d385ab537be9e2f6a6ab57c96bab92c Mon Sep 17 00:00:00 2001
From: Jyong <76649700+JohnJyong@users.noreply.github.com>
Date: Thu, 1 Aug 2024 13:02:33 +0800
Subject: [PATCH] delete weight_type (#6865)

---
 api/core/rag/data_post_processor/data_post_processor.py | 1 -
 api/core/rag/rerank/entity/weight.py                    | 2 --
 api/fields/dataset_fields.py                            | 1 -
 3 files changed, 4 deletions(-)

diff --git a/api/core/rag/data_post_processor/data_post_processor.py b/api/core/rag/data_post_processor/data_post_processor.py
index 2ed6d7418..ad9ee4f7c 100644
--- 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'],
diff --git a/api/core/rag/rerank/entity/weight.py b/api/core/rag/rerank/entity/weight.py
index 36afc89a2..6dbbad2f8 100644
--- 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
diff --git a/api/fields/dataset_fields.py b/api/fields/dataset_fields.py
index 120b66a92..a9f79b5c6 100644
--- 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),
 }