reformat

confident-ai · Jan 29, 2025 · b8e3da5 · b8e3da5
1 parent 106dad9
commit b8e3da5
Showing 1 changed file with 4 additions and 7 deletions.
diff --git 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
+            return self.model_name