diff --git a/lightrag/llm.py b/lightrag/llm.py
index 89fb89e0..1f52d4ae 100644
--- a/lightrag/llm.py
+++ b/lightrag/llm.py
@@ -622,11 +622,11 @@ async def nvidia_openai_complete(
 
 
 async def azure_openai_complete(
-    prompt, system_prompt=None, history_messages=[], keyword_extraction=False, **kwargs
+    model: str = "gpt-4o-mini", prompt, system_prompt=None, history_messages=[], keyword_extraction=False, **kwargs
 ) -> str:
     keyword_extraction = kwargs.pop("keyword_extraction", None)
     result = await azure_openai_complete_if_cache(
-        "conversation-4o-mini",
+        model,
         prompt,
         system_prompt=system_prompt,
         history_messages=history_messages,