yujunjun/LightRAG
1
0
Fork 0
mirror of https://github.com/HKUDS/LightRAG.git synced 2025-12-14 00:15:53 +00:00
Code Issues Packages Projects Releases Wiki Activity

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( async def hf_model_complete(
prompt, system_prompt=None, history_messages=[], **kwargs prompt, system_prompt=None, history_messages=[], **kwargs
) -> str: ) -> 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( return await hf_model_if_cache(
input_string, model_name,
prompt, prompt,
system_prompt=system_prompt, system_prompt=system_prompt,
history_messages=history_messages, history_messages=history_messages,