From 5025e1687da621f929d946c74b6352dff027060b Mon Sep 17 00:00:00 2001 From: Izel Levy Date: Thu, 1 Feb 2024 15:43:50 +0200 Subject: [PATCH] Fix --- pinecone_text/dense/openai_encoder.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/pinecone_text/dense/openai_encoder.py b/pinecone_text/dense/openai_encoder.py index 90c2893..a7674fc 100644 --- a/pinecone_text/dense/openai_encoder.py +++ b/pinecone_text/dense/openai_encoder.py @@ -49,6 +49,8 @@ def __init__( "`pip install pinecone-text[openai]" ) super().__init__() + + assert dimension > 0, "dimension must be a positive integer" self._model_name = model_name self._dimension = dimension self._client = self._create_client(**kwargs) @@ -85,7 +87,6 @@ def _encode( model=self._model_name, ) if self._dimension is not None: - assert self._dimension > 0, "dimension must be a positive integer" params["dimensions"] = self._dimension response = self._client.embeddings.create(**params) except OpenAIError as e: