diff --git a/llama-index-integrations/embeddings/llama-index-embeddings-huggingface/llama_index/embeddings/huggingface/base.py b/llama-index-integrations/embeddings/llama-index-embeddings-huggingface/llama_index/embeddings/huggingface/base.py index 2fa750589401e036645249f97a73f56a4ebd1c3e..11e550c65fda699b1fe4da06705f2b9c467cdcc3 100644 --- a/llama-index-integrations/embeddings/llama-index-embeddings-huggingface/llama_index/embeddings/huggingface/base.py +++ b/llama-index-integrations/embeddings/llama-index-embeddings-huggingface/llama_index/embeddings/huggingface/base.py @@ -279,7 +279,7 @@ class HuggingFaceEmbedding(MultiModalEmbedding): Returns: List[float]: numpy array of embeddings """ - return self._get_query_embedding(query) + return await asyncio.to_thread(self._get_query_embedding, query) async def _aget_text_embedding(self, text: str) -> List[float]: """ @@ -291,7 +291,7 @@ class HuggingFaceEmbedding(MultiModalEmbedding): Returns: List[float]: numpy array of embeddings """ - return self._get_text_embedding(text) + return await asyncio.to_thread(self._get_text_embedding, text) def _get_text_embedding(self, text: str) -> List[float]: """ diff --git a/llama-index-integrations/embeddings/llama-index-embeddings-huggingface/pyproject.toml b/llama-index-integrations/embeddings/llama-index-embeddings-huggingface/pyproject.toml index 3142c011924b582cacd59966da22f558f2f80457..5cb4609792025c930ce57a70f3887ffbf319f21f 100644 --- a/llama-index-integrations/embeddings/llama-index-embeddings-huggingface/pyproject.toml +++ b/llama-index-integrations/embeddings/llama-index-embeddings-huggingface/pyproject.toml @@ -28,7 +28,7 @@ exclude = ["**/BUILD"] license = "MIT" name = "llama-index-embeddings-huggingface" readme = "README.md" -version = "0.5.1" +version = "0.5.2" [tool.poetry.dependencies] python = ">=3.9,<4.0"