diff --git a/openai/api_resources/embedding.py b/openai/api_resources/embedding.py index e937636404..b85d99f594 100644 --- a/openai/api_resources/embedding.py +++ b/openai/api_resources/embedding.py @@ -79,6 +79,7 @@ async def acreate(cls, *args, **kwargs): # If an engine isn't using this optimization, don't do anything if type(data["embedding"]) == str: + assert_has_numpy() data["embedding"] = np.frombuffer( base64.b64decode(data["embedding"]), dtype="float32" ).tolist()