fix: embedding init err (#956)
This commit is contained in:
parent
a7c78d2cd2
commit
78d3aa5fcd
2 changed files with 24 additions and 2 deletions
|
|
@ -1,4 +1,4 @@
|
|||
from langchain.embeddings import XinferenceEmbeddings
|
||||
from core.third_party.langchain.embeddings.xinference_embedding import XinferenceEmbedding as XinferenceEmbeddings
|
||||
from replicate.exceptions import ModelError, ReplicateError
|
||||
|
||||
from core.model_providers.error import LLMBadRequestError
|
||||
|
|
@ -14,7 +14,8 @@ class XinferenceEmbedding(BaseEmbedding):
|
|||
)
|
||||
|
||||
client = XinferenceEmbeddings(
|
||||
**credentials,
|
||||
server_url=credentials['server_url'],
|
||||
model_uid=credentials['model_uid'],
|
||||
)
|
||||
|
||||
super().__init__(model_provider, client, name)
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue