fix: xinference reranker return_documents (#6888)

2024-08-01 19:57:53 +08:00
parent 093f902335
commit cc4785f094
2 changed files with 3 additions and 2 deletions
--- a/api/core/model_runtime/model_providers/xinference/rerank/rerank.py
+++ b/api/core/model_runtime/model_providers/xinference/rerank/rerank.py
@@ -57,6 +57,7 @@ class XinferenceRerankModel(RerankModel):
                documents=docs,
                query=query,
                top_n=top_n,
+                return_documents=True
            )
        except RuntimeError as e:
            raise InvokeServerUnavailableError(str(e))
@@ -66,7 +67,7 @@ class XinferenceRerankModel(RerankModel):
        for idx, result in enumerate(response['results']):
            # format document
            index = result['index']
-            page_content = result['document']
+            page_content = result['document'] if isinstance(result['document'], str) else result['document']['text']
            rerank_document = RerankDocument(
                index=index,
                text=page_content,