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,