From bdb7c4a8b3034232cbed925b04a4d84fdbd2250b Mon Sep 17 00:00:00 2001 From: ccurme Date: Tue, 29 Apr 2025 14:45:04 -0400 Subject: [PATCH] huggingface: fix embeddings return type (#31072) Integration tests failing cc @hanouticelina --- .../langchain_huggingface/embeddings/huggingface_endpoint.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/libs/partners/huggingface/langchain_huggingface/embeddings/huggingface_endpoint.py b/libs/partners/huggingface/langchain_huggingface/embeddings/huggingface_endpoint.py index 067b02434e9..afd19f9a8c0 100644 --- a/libs/partners/huggingface/langchain_huggingface/embeddings/huggingface_endpoint.py +++ b/libs/partners/huggingface/langchain_huggingface/embeddings/huggingface_endpoint.py @@ -115,7 +115,7 @@ class HuggingFaceEndpointEmbeddings(BaseModel, Embeddings): _model_kwargs = self.model_kwargs or {} # api doc: https://huggingface.github.io/text-embeddings-inference/#/Text%20Embeddings%20Inference/embed responses = self.client.feature_extraction(text=texts, **_model_kwargs) - return responses + return responses.tolist() async def aembed_documents(self, texts: list[str]) -> list[list[float]]: """Async Call to HuggingFaceHub's embedding endpoint for embedding search docs. @@ -132,7 +132,7 @@ class HuggingFaceEndpointEmbeddings(BaseModel, Embeddings): responses = await self.async_client.feature_extraction( text=texts, **_model_kwargs ) - return responses + return responses.tolist() def embed_query(self, text: str) -> list[float]: """Call out to HuggingFaceHub's embedding endpoint for embedding query text.