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
ca4aa340
Unverified
Commit
ca4aa340
authored
Feb 19, 2024
by
Yeuoly
Committed by
GitHub
Feb 19, 2024
Browse files
Options
Browse Files
Download
Email Patches
Plain Diff
fix: Add model_uid validation for model_uid in Xinference models (#2468)
parent
767d8a4b
Changes
3
Show whitespace changes
Inline
Side-by-side
Showing
3 changed files
with
9 additions
and
0 deletions
+9
-0
llm.py
api/core/model_runtime/model_providers/xinference/llm/llm.py
+3
-0
rerank.py
...model_runtime/model_providers/xinference/rerank/rerank.py
+3
-0
text_embedding.py
...del_providers/xinference/text_embedding/text_embedding.py
+3
-0
No files found.
api/core/model_runtime/model_providers/xinference/llm/llm.py
View file @
ca4aa340
...
@@ -97,6 +97,9 @@ class XinferenceAILargeLanguageModel(LargeLanguageModel):
...
@@ -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'
]
...
...
api/core/model_runtime/model_providers/xinference/rerank/rerank.py
View file @
ca4aa340
...
@@ -92,6 +92,9 @@ class XinferenceRerankModel(RerankModel):
...
@@ -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
,
...
...
api/core/model_runtime/model_providers/xinference/text_embedding/text_embedding.py
View file @
ca4aa340
...
@@ -113,6 +113,9 @@ class XinferenceTextEmbeddingModel(TextEmbeddingModel):
...
@@ -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
)
...
...
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