From 44ef2340027d897932c9688b47926b9b878bd530 Mon Sep 17 00:00:00 2001
From: MdNazishArmanShorthillsAI <nazish@shorthills.ai>
Date: Mon, 17 Feb 2025 12:43:51 +0530
Subject: [PATCH] Improved variable assignment to use your own azure open ai
 embedding model

---
 lightrag/llm/azure_openai.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/lightrag/llm/azure_openai.py b/lightrag/llm/azure_openai.py
index 1070af7f..4d15a674 100644
--- a/lightrag/llm/azure_openai.py
+++ b/lightrag/llm/azure_openai.py
@@ -165,7 +165,7 @@ async def azure_openai_complete(
 )
 async def azure_openai_embed(
     texts: list[str],
-    model: str = "text-embedding-3-small",
+    model: str = os.getenv("EMBEDDING_MODEL", "text-embedding-3-small"),
     base_url: str = None,
     api_key: str = None,
     api_version: str = None,