delete weight_type in knowledge retrieval node (#6892)

langgenius · Aug 1, 2024 · 2ab9af3 · 2ab9af3
1 parent 24a89f7
commit 2ab9af3
Show file tree

Hide file tree

Showing 2 changed files with 0 additions and 3 deletions.
diff --git a/api/core/workflow/nodes/knowledge_retrieval/entities.py b/api/core/workflow/nodes/knowledge_retrieval/entities.py
@@ -33,7 +33,6 @@ class WeightedScoreConfig(BaseModel):
     """
     Weighted score Config.
     """
-    weight_type: str
     vector_setting: VectorSetting
     keyword_setting: KeywordSetting
 
@@ -49,7 +48,6 @@ class MultipleRetrievalConfig(BaseModel):
     reranking_model: Optional[RerankingModelConfig] = None
     weights: Optional[WeightedScoreConfig] = None
 
-
 class ModelConfig(BaseModel):
     """
      Model Config.

diff --git a/api/core/workflow/nodes/knowledge_retrieval/knowledge_retrieval_node.py b/api/core/workflow/nodes/knowledge_retrieval/knowledge_retrieval_node.py
@@ -147,7 +147,6 @@ def _fetch_dataset_retriever(self, node_data: KnowledgeRetrievalNodeData, query:
             elif node_data.multiple_retrieval_config.reranking_mode == 'weighted_score':
                 reranking_model = None
                 weights = {
-                    'weight_type': node_data.multiple_retrieval_config.weights.weight_type,
                     'vector_setting': {
                         "vector_weight": node_data.multiple_retrieval_config.weights.vector_setting.vector_weight,
                         "embedding_provider_name": node_data.multiple_retrieval_config.weights.vector_setting.embedding_provider_name,