fix score threshold limit be None (#6900)

2024-08-02 12:10:51 +08:00
parent 56af1a0adf
commit 44801df8f8
4 changed files with 10 additions and 5 deletions
--- a/api/tests/unit_tests/services/workflow/test_workflow_converter.py
+++ b/api/tests/unit_tests/services/workflow/test_workflow_converter.py
@@ -208,7 +208,8 @@ def test__convert_to_knowledge_retrieval_node_for_chatbot():
            reranking_model={
                'reranking_provider_name': 'cohere',
                'reranking_model_name': 'rerank-english-v2.0'
-            }
+            },
+            reranking_enabled=True
        )
    )

@@ -251,7 +252,8 @@ def test__convert_to_knowledge_retrieval_node_for_workflow_app():
            reranking_model={
                'reranking_provider_name': 'cohere',
                'reranking_model_name': 'rerank-english-v2.0'
-            }
+            },
+            reranking_enabled=True
        )
    )