Sfoglia il codice sorgente

delete weight_type in knowledge retrieval node (#6892)

Jyong 8 mesi fa
parent
commit
2ab9af3b38

+ 0 - 2
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.

+ 0 - 1
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,