diff --git a/deepeval/models/openai_embedding_model.py b/deepeval/models/openai_embedding_model.py index 0435fc8d3..d419de390 100644 --- a/deepeval/models/openai_embedding_model.py +++ b/deepeval/models/openai_embedding_model.py @@ -69,11 +69,8 @@ def load_model(self): self.model_name = KEY_FILE_HANDLER.fetch_data( KeyValues.LOCAL_EMBEDDING_MODEL_NAME ) - if self.should_use_local_ollama(): - return OpenAI( - base_url=base_url, - api_key=api_key - ) + if self.should_use_local_ollama(): + return OpenAI(base_url=base_url, api_key=api_key) else: return OpenAIEmbeddings( model=self.model_name, @@ -141,11 +138,11 @@ def should_use_local_ollama(self): KeyValues.LOCAL_EMBEDDING_BASE_URL ) return base_url == "http://localhost:11434/v1/" - + def get_model_name(self): if self.should_use_azure_openai(): return "azure openai" elif self.should_use_local_embeddings(): return "local embeddings" elif self.model_name: - return self.model_name \ No newline at end of file + return self.model_name