delete weight_type in knowledge retrieval node (#6892)

2 years ago · 2ab9af3b38
parent 24a89f7753
commit 2ab9af3b38
2 changed files with 0 additions and 3 deletions
--- 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.
--- 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 @@ class KnowledgeRetrievalNode(BaseNode):
            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,