AI/dify
1
0
Fork 0
mirror of https://git.mirrors.martin98.com/https://github.com/langgenius/dify.git synced 2025-08-14 04:25:59 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix: xinference reranker return_documents (#6888)

Browse Source
This commit is contained in:
Weaxs 2024-08-01 19:57:53 +08:00 committed by GitHub
parent 093f902335
commit cc4785f094
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
api/core/model_runtime/model_providers/xinference/rerank/rerank.py
View File

@ -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,

2
api/tests/integration_tests/model_runtime/__mock/xinference.py
View File

@ -106,7 +106,7 @@ class MockXinferenceClass:
def _check_cluster_authenticated(self):
self._cluster_authed = True
def rerank(self: RESTfulRerankModelHandle, documents: list[str], query: str, top_n: int) -> dict:
def rerank(self: RESTfulRerankModelHandle, documents: list[str], query: str, top_n: int, return_documents: bool) -> dict:
# check if self._model_uid is a valid uuid
if not re.match(r'[a-f0-9]{8}-[a-f0-9]{4}-[a-f0-9]{4}-[a-f0-9]{4}-[a-f0-9]{12}', self._model_uid) and \
self._model_uid != 'rerank':