Skip to content
Projects
Groups
Snippets
Help
Loading...
Help
Submit feedback
Contribute to GitLab
Sign in
Toggle navigation
D
dify
Project
Project
Details
Activity
Releases
Cycle Analytics
Repository
Repository
Files
Commits
Branches
Tags
Contributors
Graph
Compare
Charts
Issues
0
Issues
0
List
Board
Labels
Milestones
Merge Requests
0
Merge Requests
0
CI / CD
CI / CD
Pipelines
Jobs
Schedules
Charts
Wiki
Wiki
Snippets
Snippets
Members
Members
Collapse sidebar
Close sidebar
Activity
Graph
Charts
Create a new issue
Jobs
Commits
Issue Boards
Open sidebar
ai-tech
dify
Commits
53be4d27
Unverified
Commit
53be4d27
authored
Nov 28, 2023
by
WangBooth
Committed by
GitHub
Nov 28, 2023
Browse files
Options
Browse Files
Download
Email Patches
Plain Diff
fix #1637 (#1638)
Co-authored-by:
baomi.wbm
<
baomi.wbm@dtwave-inc.com
>
parent
3c37fd37
Changes
1
Show whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
4 additions
and
5 deletions
+4
-5
xinference_reranking.py
.../model_providers/models/reranking/xinference_reranking.py
+4
-5
No files found.
api/core/model_providers/models/reranking/xinference_reranking.py
View file @
53be4d27
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
)
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment