github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-06-24 23:54:14 +00:00
Code Issues Packages Projects Releases Wiki Activity

community: Fixed infinity embeddings async request (#25882)

Browse Source 
**Description:** Fix async infinity embeddings
**Issue:** #24942  

@baskaryan, @ccurme
This commit is contained in:
Djordje 2024-08-30 20:10:34 +04:00 committed by GitHub
parent f75d5621e2
commit 862ef32fdc
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
libs/community/langchain_community/embeddings/infinity.py
View File

@ -287,7 +287,7 @@ class TinyAsyncOpenAIInfinityEmbeddingClient: #: :meta private:
f"Infinity returned an unexpected response with status " f"Infinity returned an unexpected response with status "
f"{response.status}: {response.text}" f"{response.status}: {response.text}"
) )
embedding = (await response.json())["embeddings"] embedding = (await response.json())["data"]
return [e["embedding"] for e in embedding] return [e["embedding"] for e in embedding]
async def aembed(self, model: str, texts: List[str]) -> List[List[float]]: async def aembed(self, model: str, texts: List[str]) -> List[List[float]]:
@ -313,7 +313,7 @@ class TinyAsyncOpenAIInfinityEmbeddingClient: #: :meta private:
*[ *[
self._async_request( self._async_request(
session=session, 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 for t in perm_texts_batched
] ]