修复缓存处理的运行时错误
Browse files].func # ["func"]
^^^^
AttributeError: 'dict' object has no attribute 'func'
- 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])
|