gmakstutis/LightRAG
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

update llm.py

Browse source
This commit is contained in:
LarFii 2024-10-15 20:06:59 +08:00
parent 70baa4b572
commit a92f7bfd61
1 changed files with 2 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
lightrag/llm.py
View file

@ -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,