Ver código fonte

delete weight_type in knowledge retrieval node (#6892)

tags/0.6.16
Jyong 1 ano atrás
pai
commit
2ab9af3b38
Nenhuma conta vinculada ao e-mail do autor do commit

+ 0
- 2
api/core/workflow/nodes/knowledge_retrieval/entities.py Ver arquivo

@@ -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 Ver arquivo

@@ -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,

Carregando…
Cancelar
Salvar