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

修复缓存处理的运行时错误

Browse Source 
].func  # ["func"]
 ^^^^
AttributeError: 'dict' object has no attribute 'func'
This commit is contained in:
Magic_yuan 2025-01-24 11:01:32 +08:00
parent 443aab2882
commit f6d29e1793
1 changed files with 1 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
lightrag/utils.py
View File

@ -479,9 +479,7 @@ async def handle_cache(hashing_kv, args_hash, prompt, mode="default"):
quantized = min_val = max_val = None
if is_embedding_cache_enabled:
# Use embedding cache
embedding_model_func = hashing_kv.global_config[
"embedding_func"
].func # ["func"]
embedding_model_func = hashing_kv.global_config["embedding_func"]["func"]
llm_model_func = hashing_kv.global_config.get("llm_model_func")
current_embedding = await embedding_model_func([prompt])