gmakstutis/LightRAG
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Merge pull request #1801 from Chida82/main

Browse source 
fix Azure deployment
This commit is contained in:
Daniel.y 2025-07-18 16:00:09 +08:00 • committed by GitHub
commit 801bdc520e
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 6 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

12
lightrag/llm/azure_openai.py
View file

@ -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,
)