update hf_model_complete

2024-10-14 20:33:46 +08:00
parent c89fdc4d63
commit d47dd465ce
3 changed files with 6 additions and 7 deletions
--- a/lightrag/operate.py
+++ b/lightrag/operate.py
@@ -959,7 +959,6 @@ async def naive_query(
    global_config: dict,
 ):
    use_model_func = global_config["llm_model_func"]
-    use_model_name = global_config['llm_model_name']
    results = await chunks_vdb.query(query, top_k=query_param.top_k)
    if not len(results):
        return PROMPTS["fail_response"]
@@ -982,7 +981,6 @@ async def naive_query(
    response = await use_model_func(
        query,
        system_prompt=sys_prompt,
-        model_name = use_model_name
    )

    if len(response)>len(sys_prompt):