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

fix: Add model_uid validation for model_uid in Xinference models (#2468)

Browse Source
This commit is contained in:
Yeuoly
2024-02-19 10:43:25 +08:00
committed by GitHub
parent 767d8a4b05
commit ca4aa340f6
3 changed files with 9 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -97,6 +97,9 @@ class XinferenceAILargeLanguageModel(LargeLanguageModel):
} }
""" """
try: try:
if "/" in credentials['model_uid'] or "?" in credentials['model_uid'] or "#" in credentials['model_uid']:
raise CredentialsValidateFailedError("model_uid should not contain /, ?, or #")
extra_param = XinferenceHelper.get_xinference_extra_parameter( extra_param = XinferenceHelper.get_xinference_extra_parameter(
server_url=credentials['server_url'], server_url=credentials['server_url'],
model_uid=credentials['model_uid'] model_uid=credentials['model_uid']

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

@@ -92,6 +92,9 @@ class XinferenceRerankModel(RerankModel):
:return: :return:
""" """
try: try:
if "/" in credentials['model_uid'] or "?" in credentials['model_uid'] or "#" in credentials['model_uid']:
raise CredentialsValidateFailedError("model_uid should not contain /, ?, or #")
self.invoke( self.invoke(
model=model, model=model,
credentials=credentials, credentials=credentials,

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

@@ -113,6 +113,9 @@ class XinferenceTextEmbeddingModel(TextEmbeddingModel):
:return: :return:
""" """
try: try:
if "/" in credentials['model_uid'] or "?" in credentials['model_uid'] or "#" in credentials['model_uid']:
raise CredentialsValidateFailedError("model_uid should not contain /, ?, or #")
server_url = credentials['server_url'] server_url = credentials['server_url']
model_uid = credentials['model_uid'] model_uid = credentials['model_uid']
extra_args = XinferenceHelper.get_xinference_extra_parameter(server_url=server_url, model_uid=model_uid) extra_args = XinferenceHelper.get_xinference_extra_parameter(server_url=server_url, model_uid=model_uid)