fix: embedding init err (#956)

This commit is contained in:
takatost 2023-08-22 17:43:59 +08:00 • committed by GitHub
commit 78d3aa5fcd
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 24 additions and 2 deletions

View file

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