Fix bugs

2025-02-11 11:42:46 +08:00
parent 87e6bc5c42
commit 0c3b754108
2 changed files with 3 additions and 2 deletions
--- a/lightrag/operate.py
+++ b/lightrag/operate.py
@@ -1504,7 +1504,7 @@ async def naive_query(
    use_model_func = global_config["llm_model_func"]
    args_hash = compute_args_hash(query_param.mode, query, cache_type="query")
    cached_response, quantized, min_val, max_val = await handle_cache(
-        hashing_kv, args_hash, query, "default", cache_type="query"
+        hashing_kv, args_hash, query, query_param.mode, cache_type="query"
    )
    if cached_response is not None:
        return cached_response