fix tongyi embedding endpoint return None output (#10857)

langgenius · Nov 19, 2024 · bd05df5 · bd05df5
1 parent ee1f146
commit bd05df5
Showing 1 changed file with 2 additions and 0 deletions.
diff --git 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 @@ def _invoke(
         )
 
         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(