huggingface[fix]: model parameter passing error when async embed

langchain-ai · Nov 17, 2024 · fe5126c · fe5126c
1 parent 0a57fc0
commit fe5126c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/libs/partners/huggingface/langchain_huggingface/embeddings/huggingface_endpoint.py b/libs/partners/huggingface/langchain_huggingface/embeddings/huggingface_endpoint.py
@@ -127,7 +127,7 @@ async def aembed_documents(self, texts: List[str]) -> List[List[float]]:
         texts = [text.replace("\n", " ") for text in texts]
         _model_kwargs = self.model_kwargs or {}
         responses = await self.async_client.post(
-            json={"inputs": texts, "parameters": _model_kwargs}, task=self.task
+            json={"inputs": texts, **_model_kwargs}, task=self.task
         )
         return json.loads(responses.decode())