Selaa lähdekoodia

fix reranking model field error (#6654)

Jyong 9 kuukautta sitten
vanhempi
commit
91fd8521c3

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

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

+ 2 - 2
api/core/workflow/nodes/knowledge_retrieval/knowledge_retrieval_node.py

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