github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-06-25 08:03:39 +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") cache_dir = values.get("cache_dir")
threads = values.get("threads") threads = values.get("threads")
gpu = values.get("gpu") gpu = values.get("gpu")
pkg_to_import = "fastembed-gpu" if gpu else "fastembed" pkg_to_install = "fastembed-gpu" if gpu else "fastembed"
try: try:
fastembed = importlib.import_module(pkg_to_import) fastembed = importlib.import_module("fastembed")
except ModuleNotFoundError: except ModuleNotFoundError:
raise ImportError( raise ImportError(
f"Could not import '{pkg_to_import}' Python package. " "Could not import 'fastembed' Python package. "
f"Please install it with `pip install {pkg_to_import}`." 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( raise ImportError(
f"FastEmbedEmbeddings requires " 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( values["model"] = fastembed.TextEmbedding(