diff --git a/lightrag/lightrag.py b/lightrag/lightrag.py index 108b4dd6..5ae0a47f 100644 --- a/lightrag/lightrag.py +++ b/lightrag/lightrag.py @@ -916,7 +916,7 @@ class LightRAG: else self.key_string_value_json_storage_cls( namespace="llm_response_cache", global_config=asdict(self), - embedding_func=None, + embedding_func=self.embedding_func, ), prompt=prompt, ) @@ -933,7 +933,7 @@ class LightRAG: else self.key_string_value_json_storage_cls( namespace="llm_response_cache", global_config=asdict(self), - embedding_func=None, + embedding_func=self.embedding_func, ), ) elif param.mode == "mix": @@ -952,7 +952,7 @@ class LightRAG: else self.key_string_value_json_storage_cls( namespace="llm_response_cache", global_config=asdict(self), - embedding_func=None, + embedding_func=self.embedding_func, ), ) else: @@ -993,7 +993,7 @@ class LightRAG: or self.key_string_value_json_storage_cls( namespace="llm_response_cache", global_config=asdict(self), - embedding_func=None, + embedding_func=self.embedding_func, ), ) @@ -1024,7 +1024,7 @@ class LightRAG: else self.key_string_value_json_storage_cls( namespace="llm_response_cache", global_config=asdict(self), - embedding_func=None, + embedding_func=self.embedding_funcne, ), ) elif param.mode == "naive": @@ -1040,7 +1040,7 @@ class LightRAG: else self.key_string_value_json_storage_cls( namespace="llm_response_cache", global_config=asdict(self), - embedding_func=None, + embedding_func=self.embedding_func, ), ) elif param.mode == "mix": @@ -1059,7 +1059,7 @@ class LightRAG: else self.key_string_value_json_storage_cls( namespace="llm_response_cache", global_config=asdict(self), - embedding_func=None, + embedding_func=self.embedding_func, ), ) else: