AI/dify
1
0
Fork 0
mirror of https://git.mirrors.martin98.com/https://github.com/langgenius/dify.git synced 2025-08-14 05:36:02 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix reranking model field error (#6654)

Browse Source
This commit is contained in:
Jyong 2024-07-25 10:07:55 +08:00 committed by GitHub
parent 4ec9a87e46
commit 91fd8521c3
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
api/core/workflow/nodes/knowledge_retrieval/entities.py
View File

@ -46,8 +46,8 @@ class MultipleRetrievalConfig(BaseModel):
score_threshold: Optional[float] = None score_threshold: Optional[float] = None
reranking_mode: str = 'reranking_model' reranking_mode: str = 'reranking_model'
reranking_enable: bool = True reranking_enable: bool = True
reranking_model: RerankingModelConfig reranking_model: Optional[RerankingModelConfig] = None
weights: WeightedScoreConfig weights: Optional[WeightedScoreConfig] = None
class ModelConfig(BaseModel): class ModelConfig(BaseModel):

4
api/core/workflow/nodes/knowledge_retrieval/knowledge_retrieval_node.py
View File

@ -140,8 +140,8 @@ class KnowledgeRetrievalNode(BaseNode):
elif node_data.retrieval_mode == DatasetRetrieveConfigEntity.RetrieveStrategy.MULTIPLE.value: elif node_data.retrieval_mode == DatasetRetrieveConfigEntity.RetrieveStrategy.MULTIPLE.value:
if node_data.multiple_retrieval_config.reranking_mode == 'reranking_model': if node_data.multiple_retrieval_config.reranking_mode == 'reranking_model':
reranking_model = { reranking_model = {
'reranking_provider_name': node_data.multiple_retrieval_config.reranking_model['provider'], 'reranking_provider_name': node_data.multiple_retrieval_config.reranking_model.provider,
'reranking_model_name': node_data.multiple_retrieval_config.reranking_model['name'] 'reranking_model_name': node_data.multiple_retrieval_config.reranking_model.model
} }
weights = None weights = None
elif node_data.multiple_retrieval_config.reranking_mode == 'weighted_score': elif node_data.multiple_retrieval_config.reranking_mode == 'weighted_score':