Bläddra i källkod

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

tags/0.5.6
Yeuoly 1 år sedan
förälder
incheckning
ca4aa340f6
Inget konto är kopplat till bidragsgivarens mejladress

+ 3
- 0
api/core/model_runtime/model_providers/xinference/llm/llm.py Visa fil

@@ -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 Visa fil

@@ -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 Visa fil

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

Laddar…
Avbryt
Spara