From 10e47a2d80bddfdaab488e50c9081f7af2cd365e Mon Sep 17 00:00:00 2001 From: Rafal Wojdyla Date: Wed, 3 May 2023 14:44:37 +0100 Subject: [PATCH] Embeddings util - remove unnecessary sort by index --- openai/embeddings_utils.py | 2 -- 1 file changed, 2 deletions(-) diff --git a/openai/embeddings_utils.py b/openai/embeddings_utils.py index 1b65e7c8e9..f1d438c9c0 100644 --- a/openai/embeddings_utils.py +++ b/openai/embeddings_utils.py @@ -46,7 +46,6 @@ def get_embeddings( list_of_text = [text.replace("\n", " ") for text in list_of_text] data = openai.Embedding.create(input=list_of_text, engine=engine, **kwargs).data - data = sorted(data, key=lambda x: x["index"]) # maintain the same order as input. return [d["embedding"] for d in data] @@ -60,7 +59,6 @@ async def aget_embeddings( list_of_text = [text.replace("\n", " ") for text in list_of_text] data = (await openai.Embedding.acreate(input=list_of_text, engine=engine, **kwargs)).data - data = sorted(data, key=lambda x: x["index"]) # maintain the same order as input. return [d["embedding"] for d in data]