Skip to content

Commit f24d193

Browse files
authored
Embeddings util - remove unnecessary sort by index (openai#430)
1 parent c975bce commit f24d193

File tree

1 file changed

+0
-2
lines changed

1 file changed

+0
-2
lines changed

openai/embeddings_utils.py

-2
Original file line numberDiff line numberDiff line change
@@ -46,7 +46,6 @@ def get_embeddings(
4646
list_of_text = [text.replace("\n", " ") for text in list_of_text]
4747

4848
data = openai.Embedding.create(input=list_of_text, engine=engine, **kwargs).data
49-
data = sorted(data, key=lambda x: x["index"]) # maintain the same order as input.
5049
return [d["embedding"] for d in data]
5150

5251

@@ -60,7 +59,6 @@ async def aget_embeddings(
6059
list_of_text = [text.replace("\n", " ") for text in list_of_text]
6160

6261
data = (await openai.Embedding.acreate(input=list_of_text, engine=engine, **kwargs)).data
63-
data = sorted(data, key=lambda x: x["index"]) # maintain the same order as input.
6462
return [d["embedding"] for d in data]
6563

6664

0 commit comments

Comments
 (0)