From 53be4d27127fd355b6d089d096d2f9a1cfb57ba6 Mon Sep 17 00:00:00 2001 From: WangBooth Date: Tue, 28 Nov 2023 20:05:50 +0800 Subject: [PATCH] fix #1637 (#1638) Co-authored-by: baomi.wbm --- .../models/reranking/xinference_reranking.py | 9 ++++----- 1 file changed, 4 insertions(+), 5 deletions(-) diff --git a/api/core/model_providers/models/reranking/xinference_reranking.py b/api/core/model_providers/models/reranking/xinference_reranking.py index 0efcf189f0..47c1c6cd01 100644 --- a/api/core/model_providers/models/reranking/xinference_reranking.py +++ b/api/core/model_providers/models/reranking/xinference_reranking.py @@ -1,12 +1,11 @@ import logging -from typing import Optional, List - -from langchain.schema import Document -from xinference_client.client.restful.restful_client import Client +from typing import List, Optional from core.model_providers.error import LLMBadRequestError from core.model_providers.models.reranking.base import BaseReranking from core.model_providers.providers.base import BaseModelProvider +from langchain.schema import Document +from xinference_client.client.restful.restful_client import Client class XinferenceReranking(BaseReranking): @@ -48,7 +47,7 @@ class XinferenceReranking(BaseReranking): ) # score threshold check if score_threshold is not None: - if result.relevance_score >= score_threshold: + if result['relevance_score'] >= score_threshold: rerank_documents.append(rerank_document) else: rerank_documents.append(rerank_document)