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, api_version: str | None = None,
**kwargs, **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 = (
base_url or os.getenv("AZURE_OPENAI_ENDPOINT") or os.getenv("LLM_BINDING_HOST") 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( openai_async_client = AsyncAzureOpenAI(
azure_endpoint=base_url, azure_endpoint=base_url,
azure_deployment=model, azure_deployment=deployment,
api_key=api_key, api_key=api_key,
api_version=api_version, api_version=api_version,
) )
@ -136,9 +136,9 @@ async def azure_openai_embed(
api_key: str | None = None, api_key: str | None = None,
api_version: str | None = None, api_version: str | None = None,
) -> np.ndarray: ) -> np.ndarray:
model = ( deployment = (
model os.getenv("AZURE_EMBEDDING_DEPLOYMENT")
or os.getenv("AZURE_EMBEDDING_DEPLOYMENT") or model
or os.getenv("EMBEDDING_MODEL", "text-embedding-3-small") or os.getenv("EMBEDDING_MODEL", "text-embedding-3-small")
) )
base_url = ( base_url = (
@ -159,7 +159,7 @@ async def azure_openai_embed(
openai_async_client = AsyncAzureOpenAI( openai_async_client = AsyncAzureOpenAI(
azure_endpoint=base_url, azure_endpoint=base_url,
azure_deployment=model, azure_deployment=deployment,
api_key=api_key, api_key=api_key,
api_version=api_version, api_version=api_version,
) )