Fix: unexpected keyword argument error.

2024-12-05 14:11:43 +08:00
parent 6d274019dd
commit e99832cc13
8 changed files with 8 additions and 8 deletions
--- a/examples/lightrag_lmdeploy_demo.py
+++ b/examples/lightrag_lmdeploy_demo.py
@@ -12,7 +12,7 @@ if not os.path.exists(WORKING_DIR):


 async def lmdeploy_model_complete(
-    prompt=None, system_prompt=None, history_messages=[], **kwargs
+    prompt=None, system_prompt=None, history_messages=[], keyword_extraction=False, **kwargs
 ) -> str:
    model_name = kwargs["hashing_kv"].global_config["llm_model_name"]
    return await lmdeploy_model_if_cache(