Underground-Digital/Workflow-Engine

1import os2from time import sleep3 4from core.model_runtime.entities.rerank_entities import RerankResult5from core.model_runtime.model_providers.wenxin.rerank.rerank import WenxinRerankModel6 7 8def test_invoke_bce_reranker_base_v1():9    sleep(3)10    model = WenxinRerankModel()11 12    response = model.invoke(13        model="bce-reranker-base_v1",14        credentials={"api_key": os.environ.get("WENXIN_API_KEY"), "secret_key": os.environ.get("WENXIN_SECRET_KEY")},15        query="What is Deep Learning?",16        docs=["Deep Learning is ...", "My Book is ..."],17        user="abc-123",18    )19 20    assert isinstance(response, RerankResult)21    assert len(response.docs) == 222