estel/dify
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

feat: xinference rerank model support (#1615)

Browse Source
This commit is contained in:
takatost
2023-11-25 03:56:00 +08:00
committed by GitHub
parent ea35f1dce1
commit 0e627c920f
9 changed files with 215 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
api/controllers/console/workspace/model_providers.py
View File

@@ -115,7 +115,7 @@ class ModelProviderModelValidateApi(Resource):
parser = reqparse.RequestParser()
parser.add_argument('model_name', type=str, required=True, nullable=False, location='json')
parser.add_argument('model_type', type=str, required=True, nullable=False,
choices=['text-generation', 'embeddings', 'speech2text'], location='json')
choices=['text-generation', 'embeddings', 'speech2text', 'reranking'], location='json')
parser.add_argument('config', type=dict, required=True, nullable=False, location='json')
args = parser.parse_args()
@@ -155,7 +155,7 @@ class ModelProviderModelUpdateApi(Resource):
parser = reqparse.RequestParser()
parser.add_argument('model_name', type=str, required=True, nullable=False, location='json')
parser.add_argument('model_type', type=str, required=True, nullable=False,
choices=['text-generation', 'embeddings', 'speech2text'], location='json')
choices=['text-generation', 'embeddings', 'speech2text', 'reranking'], location='json')
parser.add_argument('config', type=dict, required=True, nullable=False, location='json')
args = parser.parse_args()
@@ -184,7 +184,7 @@ class ModelProviderModelUpdateApi(Resource):
parser = reqparse.RequestParser()
parser.add_argument('model_name', type=str, required=True, nullable=False, location='args')
parser.add_argument('model_type', type=str, required=True, nullable=False,
choices=['text-generation', 'embeddings', 'speech2text'], location='args')
choices=['text-generation', 'embeddings', 'speech2text', 'reranking'], location='args')
args = parser.parse_args()
provider_service = ProviderService()