Unverified Commit 53be4d27 authored by WangBooth's avatar WangBooth Committed by GitHub

fix #1637 (#1638)

Co-authored-by: 's avatarbaomi.wbm <baomi.wbm@dtwave-inc.com>
parent 3c37fd37
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)
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment