Magicyuan commited on
Commit
28b7304
·
1 Parent(s): 451706a

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

Browse files

].func # ["func"]
^^^^
AttributeError: 'dict' object has no attribute 'func'

Files changed (1) hide show
  1. lightrag/utils.py +1 -3
lightrag/utils.py CHANGED
@@ -479,9 +479,7 @@ async def handle_cache(hashing_kv, args_hash, prompt, mode="default"):
479
  quantized = min_val = max_val = None
480
  if is_embedding_cache_enabled:
481
  # Use embedding cache
482
- embedding_model_func = hashing_kv.global_config[
483
- "embedding_func"
484
- ].func # ["func"]
485
  llm_model_func = hashing_kv.global_config.get("llm_model_func")
486
 
487
  current_embedding = await embedding_model_func([prompt])
 
479
  quantized = min_val = max_val = None
480
  if is_embedding_cache_enabled:
481
  # Use embedding cache
482
+ embedding_model_func = hashing_kv.global_config["embedding_func"]["func"]
 
 
483
  llm_model_func = hashing_kv.global_config.get("llm_model_func")
484
 
485
  current_embedding = await embedding_model_func([prompt])