From 5a67df606cf8ea152184d35a6477a48331e21780 Mon Sep 17 00:00:00 2001 From: Chris Hallacy Date: Fri, 26 Aug 2022 07:59:04 -0700 Subject: [PATCH] Add an = to <= check in embedding_utils --- openai/embeddings_utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/openai/embeddings_utils.py b/openai/embeddings_utils.py index 2db5514c64..47a04e6582 100644 --- a/openai/embeddings_utils.py +++ b/openai/embeddings_utils.py @@ -27,7 +27,7 @@ def get_embedding(text: str, engine="text-similarity-davinci-001") -> List[float def get_embeddings( list_of_text: List[str], engine="text-similarity-babbage-001" ) -> List[List[float]]: - assert len(list_of_text) < 2048, "The batch size should not be larger than 2048." + assert len(list_of_text) <= 2048, "The batch size should not be larger than 2048." # replace newlines, which can negatively affect performance. list_of_text = [text.replace("\n", " ") for text in list_of_text]