yujunjun/LightRAG
1
0
Fork 0
mirror of https://github.com/HKUDS/LightRAG.git synced 2025-12-24 13:22:21 +00:00
Code Issues Packages Projects Releases Wiki Activity

add api_version to azure_openai_complete_if_cache

Browse Source
This commit is contained in:
Ahmad Hatahet 2024-11-30 17:47:33 +01:00
parent 9b92d425f6
commit 137315ec18
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
lightrag/llm.py
View File

@ -92,12 +92,15 @@ async def azure_openai_complete_if_cache(
history_messages=[],
base_url=None,
api_key=None,
api_version=None,
**kwargs,
):
if api_key:
os.environ["AZURE_OPENAI_API_KEY"] = api_key
if base_url:
os.environ["AZURE_OPENAI_ENDPOINT"] = base_url
if api_version:
os.environ["AZURE_OPENAI_API_VERSION"] = api_version
openai_async_client = AsyncAzureOpenAI(
azure_endpoint=os.getenv("AZURE_OPENAI_ENDPOINT"),