|
|
@ -43,14 +43,15 @@ class HuggingFaceEmbeddings(BaseModel, Embeddings):
|
|
|
|
try:
|
|
|
|
try:
|
|
|
|
import sentence_transformers
|
|
|
|
import sentence_transformers
|
|
|
|
|
|
|
|
|
|
|
|
self.client = sentence_transformers.SentenceTransformer(
|
|
|
|
except ImportError as exc:
|
|
|
|
self.model_name, cache_folder=self.cache_folder, **self.model_kwargs
|
|
|
|
|
|
|
|
)
|
|
|
|
|
|
|
|
except ImportError:
|
|
|
|
|
|
|
|
raise ValueError(
|
|
|
|
raise ValueError(
|
|
|
|
"Could not import sentence_transformers python package. "
|
|
|
|
"Could not import sentence_transformers python package. "
|
|
|
|
"Please install it with `pip install sentence_transformers`."
|
|
|
|
"Please install it with `pip install sentence_transformers`."
|
|
|
|
)
|
|
|
|
) from exc
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
self.client = sentence_transformers.SentenceTransformer(
|
|
|
|
|
|
|
|
self.model_name, cache_folder=self.cache_folder, **self.model_kwargs
|
|
|
|
|
|
|
|
)
|
|
|
|
|
|
|
|
|
|
|
|
class Config:
|
|
|
|
class Config:
|
|
|
|
"""Configuration for this pydantic object."""
|
|
|
|
"""Configuration for this pydantic object."""
|
|
|
|