From a67b39c87e55809bb3be65b5bcac8d4815f3fbe7 Mon Sep 17 00:00:00 2001 From: shaohuzhang1 <80892890+shaohuzhang1@users.noreply.github.com> Date: Tue, 15 Jul 2025 16:00:22 +0800 Subject: [PATCH] fix: reranker prefix (#3605) --- .../impl/local_model_provider/model/reranker.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/apps/models_provider/impl/local_model_provider/model/reranker.py b/apps/models_provider/impl/local_model_provider/model/reranker.py index 7b1402359..5ac2a4ca8 100644 --- a/apps/models_provider/impl/local_model_provider/model/reranker.py +++ b/apps/models_provider/impl/local_model_provider/model/reranker.py @@ -52,9 +52,10 @@ class WebLocalBaseReranker(MaxKBBaseModel, BaseDocumentCompressor): Sequence[Document]: if documents is None or len(documents) == 0: return [] + prefix = CONFIG.get_admin_path() bind = f'{CONFIG.get("LOCAL_MODEL_HOST")}:{CONFIG.get("LOCAL_MODEL_PORT")}' res = requests.post( - f'{CONFIG.get("LOCAL_MODEL_PROTOCOL")}://{bind}/api/model/{self.model_id}/compress_documents', + f'{CONFIG.get("LOCAL_MODEL_PROTOCOL")}://{bind}{prefix}/api/model/{self.model_id}/compress_documents', json={'documents': [{'page_content': document.page_content, 'metadata': document.metadata} for document in documents], 'query': query}, headers={'Content-Type': 'application/json'}) result = res.json()