chore: use getenv optional param when getting default embeddings model

defenseunicorns · Aug 1, 2024 · 5b1eed3 · 5b1eed3
1 parent 65a3c40
commit 5b1eed3
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/src/leapfrogai_api/backend/rag/leapfrogai_embeddings.py b/src/leapfrogai_api/backend/rag/leapfrogai_embeddings.py
@@ -46,8 +46,7 @@ async def aembed_query(self, text: str) -> list[float]:
         return list_of_embeddings[0]
 
     async def _get_model(
-        self,
-        model_name: str = os.getenv("DEFAULT_EMBEDDINGS_MODEL") or "text-embeddings",
+        self, model_name: str = os.getenv("DEFAULT_EMBEDDINGS_MODEL", "text-embeddings")
     ):
         """Gets the embeddings model.