bug fix for openai embedding model

krohling · Dec 15, 2023 · 961278f · 961278f
1 parent 77ea942
commit 961278f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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.")