Merge branch 'main' into ollama-emulation

2025-01-19 13:32:07 +08:00
parent b008c586ea 476e31f1e5
commit b50fdeccaa
1 changed files with 2 additions and 2 deletions
--- 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,