Просмотр исходного кода

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

tags/0.5.6
Yeuoly 1 год назад
Родитель
Сommit
ca4aa340f6
Аккаунт пользователя с таким Email не найден

+ 3
- 0
api/core/model_runtime/model_providers/xinference/llm/llm.py Просмотреть файл

@@ -97,6 +97,9 @@ class XinferenceAILargeLanguageModel(LargeLanguageModel):
}
"""
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(
server_url=credentials['server_url'],
model_uid=credentials['model_uid']

+ 3
- 0
api/core/model_runtime/model_providers/xinference/rerank/rerank.py Просмотреть файл

@@ -92,6 +92,9 @@ class XinferenceRerankModel(RerankModel):
:return:
"""
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(
model=model,
credentials=credentials,

+ 3
- 0
api/core/model_runtime/model_providers/xinference/text_embedding/text_embedding.py Просмотреть файл

@@ -113,6 +113,9 @@ class XinferenceTextEmbeddingModel(TextEmbeddingModel):
:return:
"""
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']
model_uid = credentials['model_uid']
extra_args = XinferenceHelper.get_xinference_extra_parameter(server_url=server_url, model_uid=model_uid)

Загрузка…
Отмена
Сохранить