Add embedding_func to llm_response_cache

This commit is contained in:
yangdx
2025-02-01 22:19:16 +08:00
parent 3bc7c4d8f1
commit b87703aea6

View File

@@ -231,7 +231,7 @@ class LightRAG:
self.llm_response_cache = self.key_string_value_json_storage_cls( self.llm_response_cache = self.key_string_value_json_storage_cls(
namespace="llm_response_cache", namespace="llm_response_cache",
embedding_func=None, embedding_func=self.embedding_func,
) )
#### ####
@@ -275,7 +275,7 @@ class LightRAG:
else: else:
hashing_kv = self.key_string_value_json_storage_cls( hashing_kv = self.key_string_value_json_storage_cls(
namespace="llm_response_cache", 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)( self.llm_model_func = limit_async_func_call(self.llm_model_max_async)(