From f2e8732c4308ad41f878f6c9f00b8b879801a8c0 Mon Sep 17 00:00:00 2001
From: LarFii <834462287@qq.com>
Date: Tue, 15 Oct 2024 20:06:59 +0800
Subject: [PATCH] update llm.py

---
 lightrag/llm.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/lightrag/llm.py b/lightrag/llm.py
index bc2ac1f3..87b156c5 100644
--- a/lightrag/llm.py
+++ b/lightrag/llm.py
@@ -118,9 +118,9 @@ async def gpt_4o_mini_complete(
 async def hf_model_complete(
     prompt, system_prompt=None, history_messages=[], **kwargs
 ) -> str:
-    input_string = kwargs['hashing_kv'].global_config['llm_model_name']
+    model_name = kwargs['hashing_kv'].global_config['llm_model_name']
     return await hf_model_if_cache(
-        input_string,
+        model_name,
         prompt,
         system_prompt=system_prompt,
         history_messages=history_messages,