From 6200a2a00e5d54cece5b754d20f6c3bd423ef71f Mon Sep 17 00:00:00 2001 From: Harrison Chase Date: Sat, 22 Apr 2023 09:06:36 -0700 Subject: [PATCH] Harrison/error hf (#3348) Co-authored-by: Rui Melo <44201826+rufimelo99@users.noreply.github.com> --- langchain/embeddings/huggingface.py | 11 ++++++----- 1 file changed, 6 insertions(+), 5 deletions(-) diff --git a/langchain/embeddings/huggingface.py b/langchain/embeddings/huggingface.py index b0bd03e9..24256227 100644 --- a/langchain/embeddings/huggingface.py +++ b/langchain/embeddings/huggingface.py @@ -43,14 +43,15 @@ class HuggingFaceEmbeddings(BaseModel, Embeddings): try: import sentence_transformers - self.client = sentence_transformers.SentenceTransformer( - self.model_name, cache_folder=self.cache_folder, **self.model_kwargs - ) - except ImportError: + except ImportError as exc: raise ValueError( "Could not import sentence_transformers python package. " "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: """Configuration for this pydantic object."""