From 5b2823390383685f6cfc0a782db6a003800a847a Mon Sep 17 00:00:00 2001 From: Dario Chini Date: Thu, 17 Jul 2025 23:11:07 +0200 Subject: [PATCH] fix Azure deployment --- lightrag/llm/azure_openai.py | 12 ++++++------ 1 file changed, 6 insertions(+), 6 deletions(-) diff --git a/lightrag/llm/azure_openai.py b/lightrag/llm/azure_openai.py index 6354ef18..e2fcedbb 100644 --- a/lightrag/llm/azure_openai.py +++ b/lightrag/llm/azure_openai.py @@ -47,7 +47,7 @@ async def azure_openai_complete_if_cache( api_version: str | None = None, **kwargs, ): - model = model or os.getenv("AZURE_OPENAI_DEPLOYMENT") or os.getenv("LLM_MODEL") + deployment = os.getenv("AZURE_OPENAI_DEPLOYMENT") or model or os.getenv("LLM_MODEL") base_url = ( base_url or os.getenv("AZURE_OPENAI_ENDPOINT") or os.getenv("LLM_BINDING_HOST") ) @@ -62,7 +62,7 @@ async def azure_openai_complete_if_cache( openai_async_client = AsyncAzureOpenAI( azure_endpoint=base_url, - azure_deployment=model, + azure_deployment=deployment, api_key=api_key, api_version=api_version, ) @@ -136,9 +136,9 @@ async def azure_openai_embed( api_key: str | None = None, api_version: str | None = None, ) -> np.ndarray: - model = ( - model - or os.getenv("AZURE_EMBEDDING_DEPLOYMENT") + deployment = ( + os.getenv("AZURE_EMBEDDING_DEPLOYMENT") + or model or os.getenv("EMBEDDING_MODEL", "text-embedding-3-small") ) base_url = ( @@ -159,7 +159,7 @@ async def azure_openai_embed( openai_async_client = AsyncAzureOpenAI( azure_endpoint=base_url, - azure_deployment=model, + azure_deployment=deployment, api_key=api_key, api_version=api_version, )