浏览代码

delete weight_type in knowledge retrieval node (#6892)

tags/0.6.16
Jyong 1年前
父节点
当前提交
2ab9af3b38
没有帐户链接到提交者的电子邮件

+ 0
- 2
api/core/workflow/nodes/knowledge_retrieval/entities.py 查看文件

""" """
Weighted score Config. Weighted score Config.
""" """
weight_type: str
vector_setting: VectorSetting vector_setting: VectorSetting
keyword_setting: KeywordSetting keyword_setting: KeywordSetting


reranking_model: Optional[RerankingModelConfig] = None reranking_model: Optional[RerankingModelConfig] = None
weights: Optional[WeightedScoreConfig] = None weights: Optional[WeightedScoreConfig] = None



class ModelConfig(BaseModel): class ModelConfig(BaseModel):
""" """
Model Config. Model Config.

+ 0
- 1
api/core/workflow/nodes/knowledge_retrieval/knowledge_retrieval_node.py 查看文件

elif node_data.multiple_retrieval_config.reranking_mode == 'weighted_score': elif node_data.multiple_retrieval_config.reranking_mode == 'weighted_score':
reranking_model = None reranking_model = None
weights = { weights = {
'weight_type': node_data.multiple_retrieval_config.weights.weight_type,
'vector_setting': { 'vector_setting': {
"vector_weight": node_data.multiple_retrieval_config.weights.vector_setting.vector_weight, "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, "embedding_provider_name": node_data.multiple_retrieval_config.weights.vector_setting.embedding_provider_name,

正在加载...
取消
保存