github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-06-24 15:43:54 +00:00
Code Issues Packages Projects Releases Wiki Activity

community: fix typo in the module imported when using GPU with FastEmbedEmbeddings (#29631)

Browse Source 
Made a mistake in the module to import (the module stay the same only
the installed package changes), fixed it and tested it

https://github.com/langchain-ai/langchain/pull/29627

@ccurme
This commit is contained in:
Vincent Emonet 2025-02-06 16:26:08 +01:00 committed by GitHub
parent f8fd65dea2
commit db8201d4da
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 6 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
libs/community/langchain_community/embeddings/fastembed.py
View File

@ -84,21 +84,21 @@ class FastEmbedEmbeddings(BaseModel, Embeddings):
cache_dir = values.get("cache_dir")
threads = values.get("threads")
gpu = values.get("gpu")
pkg_to_import = "fastembed-gpu" if gpu else "fastembed"
pkg_to_install = "fastembed-gpu" if gpu else "fastembed"
try:
fastembed = importlib.import_module(pkg_to_import)
fastembed = importlib.import_module("fastembed")
except ModuleNotFoundError:
raise ImportError(
f"Could not import '{pkg_to_import}' Python package. "
f"Please install it with `pip install {pkg_to_import}`."
"Could not import 'fastembed' Python package. "
f"Please install it with `pip install {pkg_to_install}`."
)
if importlib.metadata.version(pkg_to_import) < MIN_VERSION:
if importlib.metadata.version(pkg_to_install) < MIN_VERSION:
raise ImportError(
f"FastEmbedEmbeddings requires "
f'`pip install -U "{pkg_to_import}>={MIN_VERSION}"`.'
f'`pip install -U "{pkg_to_install}>={MIN_VERSION}"`.'
)
values["model"] = fastembed.TextEmbedding(