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 2fa750589401e..11e550c65fda6 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 @@ async def _aget_query_embedding(self, query: str) -> List[float]: 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 @@ async def _aget_text_embedding(self, text: str) -> List[float]: 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]: """