From 9c612833a881ce493cf76245347cc37bc503d3a8 Mon Sep 17 00:00:00 2001
From: Magic_yuan <72277650+magicyuan876@users.noreply.github.com>
Date: Thu, 21 Nov 2024 18:49:29 +0800
Subject: [PATCH] Update lightrag_azure_openai_demo.py

---
 examples/lightrag_azure_openai_demo.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/examples/lightrag_azure_openai_demo.py b/examples/lightrag_azure_openai_demo.py
index 4e134830..7073e3cb 100644
--- a/examples/lightrag_azure_openai_demo.py
+++ b/examples/lightrag_azure_openai_demo.py
@@ -34,9 +34,9 @@ async def llm_model_func(
         prompt, system_prompt=None, history_messages=[], **kwargs
 ) -> str:
     client = AzureOpenAI(
-        api_key=LLM_AZURE_OPENAI_KEY,
-        api_version=LLM_AZURE_OPENAI_VERSION,
-        azure_endpoint=LLM_AZURE_OPENAI_API
+        api_key=AZURE_OPENAI_API_KEY,
+        api_version=AZURE_OPENAI_API_VERSION,
+        azure_endpoint=AZURE_OPENAI_DEPLOYMENT
     )
 
     messages = []