From 961278fe1a6398bbd4a849ea6b0da83838f1b07a Mon Sep 17 00:00:00 2001 From: Kevin Rohling Date: Thu, 14 Dec 2023 18:18:27 -0800 Subject: [PATCH] bug fix for openai embedding model --- bondai/models/openai/openai_embedding_model.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/bondai/models/openai/openai_embedding_model.py b/bondai/models/openai/openai_embedding_model.py index 0b061e5..581ca5d 100644 --- a/bondai/models/openai/openai_embedding_model.py +++ b/bondai/models/openai/openai_embedding_model.py @@ -11,7 +11,7 @@ def __init__( model: OpenAIModelNames = OpenAIModelNames.TEXT_EMBEDDING_ADA_002, connection_params: Dict = None, ): - self._model = model.value + self._model = model.value if isinstance(model, OpenAIModelNames) else model self._connection_params = connection_params if ModelConfig[self._model]["model_type"] != OpenAIModelType.EMBEDDING: raise Exception(f"Model {model} is not an embedding model.")