From 3bc7c4d8f11c7951777298a15d5203b469b2d264 Mon Sep 17 00:00:00 2001
From: yangdx <yangdx@znipower.com>
Date: Sat, 1 Feb 2025 22:18:59 +0800
Subject: [PATCH] Save cache_type to llm_response_cache

---
 lightrag/utils.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/lightrag/utils.py b/lightrag/utils.py
index e868d32e..9316f908 100644
--- a/lightrag/utils.py
+++ b/lightrag/utils.py
@@ -568,6 +568,7 @@ async def save_to_cache(hashing_kv, cache_data: CacheData):
 
     mode_cache[cache_data.args_hash] = {
         "return": cache_data.content,
+        "cache_type": cache_data.cache_data,
         "embedding": cache_data.quantized.tobytes().hex()
         if cache_data.quantized is not None
         else None,