Add embedding_func to llm_response_cache

2025-02-01 22:19:16 +08:00
parent 3bc7c4d8f1
commit b87703aea6
1 changed files with 2 additions and 2 deletions
--- a/lightrag/lightrag.py
+++ b/lightrag/lightrag.py
@@ -231,7 +231,7 @@ class LightRAG:

        self.llm_response_cache = self.key_string_value_json_storage_cls(
            namespace="llm_response_cache",
-            embedding_func=None,
+            embedding_func=self.embedding_func,
        )

        ####
@@ -275,7 +275,7 @@ class LightRAG:
        else:
            hashing_kv = self.key_string_value_json_storage_cls(
                namespace="llm_response_cache",
-                embedding_func=None,
+                embedding_func=self.embedding_func,
            )

        self.llm_model_func = limit_async_func_call(self.llm_model_max_async)(