Update lightrag_azure_openai_demo.py

2024-11-21 18:49:29 +08:00
parent 2e8768b301
commit 9c612833a8
1 changed files with 3 additions and 3 deletions
--- 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 = []