Merge branch 'main' into main

2025-01-20 12:26:01 +08:00
parent 70425b0357 617eb1bfaa
commit f8ba76f7b8
4 changed files with 4 additions and 9 deletions
--- a/lightrag/llm.py
+++ b/lightrag/llm.py
@@ -626,16 +626,11 @@ async def nvidia_openai_complete(


 async def azure_openai_complete(
-    model: str = "gpt-4o-mini",
-    prompt="",
-    system_prompt=None,
-    history_messages=[],
-    keyword_extraction=False,
-    **kwargs,
+    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(
-        model,
+        os.getenv("LLM_MODEL", "gpt-4o-mini"),
        prompt,
        system_prompt=system_prompt,
        history_messages=history_messages,