fix tongyi embedding endpoint return None output (#10857)

2024-11-19 21:04:17 +08:00 · 2024-11-19 21:04:17 +08:00 · bd05df5cc5
commit bd05df5cc5
parent ee1f14621a
1 changed files with 2 additions and 0 deletions
--- a/api/core/model_runtime/model_providers/tongyi/rerank/rerank.py
+++ b/api/core/model_runtime/model_providers/tongyi/rerank/rerank.py
@ -65,6 +65,8 @@ class GTERerankModel(RerankModel):
        )

        rerank_documents = []
+        if not response.output:
+            return RerankResult(model=model, docs=rerank_documents)
        for _, result in enumerate(response.output.results):
            # format document
            rerank_document = RerankDocument(