From 862ef32fdc2cf14cd5cd07155ac62b0dd3fc6fd8 Mon Sep 17 00:00:00 2001 From: Djordje Date: Fri, 30 Aug 2024 20:10:34 +0400 Subject: [PATCH] community: Fixed infinity embeddings async request (#25882) **Description:** Fix async infinity embeddings **Issue:** #24942 @baskaryan, @ccurme --- libs/community/langchain_community/embeddings/infinity.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/libs/community/langchain_community/embeddings/infinity.py b/libs/community/langchain_community/embeddings/infinity.py index 73da1cd6053..f9032427d5d 100644 --- a/libs/community/langchain_community/embeddings/infinity.py +++ b/libs/community/langchain_community/embeddings/infinity.py @@ -287,7 +287,7 @@ class TinyAsyncOpenAIInfinityEmbeddingClient: #: :meta private: f"Infinity returned an unexpected response with status " f"{response.status}: {response.text}" ) - embedding = (await response.json())["embeddings"] + embedding = (await response.json())["data"] return [e["embedding"] for e in embedding] async def aembed(self, model: str, texts: List[str]) -> List[List[float]]: @@ -313,7 +313,7 @@ class TinyAsyncOpenAIInfinityEmbeddingClient: #: :meta private: *[ self._async_request( session=session, - **self._kwargs_post_request(model=model, texts=t), + kwargs=self._kwargs_post_request(model=model, texts=t), ) for t in perm_texts_batched ]