yujunjun/haystack
1
0
Fork 0
mirror of https://github.com/deepset-ai/haystack.git synced 2025-11-29 16:36:34 +00:00
Code Issues Packages Projects Releases Wiki Activity

update sentence transformers import error message (#7906)

Browse Source
This commit is contained in:
Stefano Fiorucci 2024-06-20 18:15:01 +02:00 committed by GitHub
parent 4c59000c21
commit d80e01492b
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
5 changed files with 9 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
haystack/components/embedders/backends/sentence_transformers_backend.py
View File

@ -9,7 +9,7 @@ import numpy as np
from haystack.lazy_imports import LazyImport
from haystack.utils.auth import Secret
with LazyImport(message="Run 'pip install \"sentence-transformers>=2.2.0\"'") as sentence_transformers_import:
with LazyImport(message="Run 'pip install \"sentence-transformers>=2.3.0\"'") as sentence_transformers_import:
from sentence_transformers import SentenceTransformer

2
haystack/components/evaluators/sas_evaluator.py
View File

@ -11,7 +11,7 @@ from haystack.lazy_imports import LazyImport
from haystack.utils import ComponentDevice, expit
from haystack.utils.auth import Secret, deserialize_secrets_inplace
with LazyImport(message="Run 'pip install scikit-learn \"sentence-transformers>=2.2.0\"'") as sas_import:
with LazyImport(message="Run 'pip install \"sentence-transformers>=2.3.0\"'") as sas_import:
from sentence_transformers import CrossEncoder, SentenceTransformer, util
from transformers import AutoConfig

2
haystack/components/rankers/sentence_transformers_diversity.py
View File

@ -11,7 +11,7 @@ from haystack.utils import ComponentDevice, Secret, deserialize_secrets_inplace
logger = logging.getLogger(__name__)
with LazyImport(message="Run 'pip install \"sentence-transformers>=2.2.0\"'") as torch_and_sentence_transformers_import:
with LazyImport(message="Run 'pip install \"sentence-transformers>=2.3.0\"'") as torch_and_sentence_transformers_import:
import torch
from sentence_transformers import SentenceTransformer

2
pyproject.toml
View File

@ -79,7 +79,7 @@ format-check = "ruff format --check {args}"
extra-dependencies = [
"transformers[torch,sentencepiece]==4.41.2", # ExtractiveReader, TransformersSimilarityRanker, LocalWhisperTranscriber, HFGenerators...
"huggingface_hub>=0.23.0", # Hugging Face API Generators and Embedders
"sentence-transformers>=2.2.0", # SentenceTransformersTextEmbedder and SentenceTransformersDocumentEmbedder
"sentence-transformers>=2.3.0", # SentenceTransformersTextEmbedder and SentenceTransformersDocumentEmbedder
"langdetect", # TextLanguageRouter and DocumentLanguageClassifier
"openai-whisper>=20231106", # LocalWhisperTranscriber

5
releasenotes/notes/update-sentence-transformers-import-error-msg-895b8ed2355ae2fe.yaml Normal file
View File

@ -0,0 +1,5 @@
---
enhancements:
- |
Update the error message when the `sentence-transformers` library is not installed
and the used component requires it.