Fix: unexpected keyword argument error.

2024-12-05 14:11:43 +08:00
parent c352eb6d84
commit c31a2654f9
8 changed files with 8 additions and 8 deletions
--- a/examples/lightrag_azure_openai_demo.py
+++ b/examples/lightrag_azure_openai_demo.py
@@ -30,7 +30,7 @@ os.mkdir(WORKING_DIR)


 async def llm_model_func(
-    prompt, system_prompt=None, history_messages=[], **kwargs
+    prompt, system_prompt=None, history_messages=[], keyword_extraction=False, **kwargs
 ) -> str:
    client = AzureOpenAI(
        api_key=AZURE_OPENAI_API_KEY,