Set embedding_func in all llm_response_cache

This commit is contained in:
yangdx
2025-02-02 03:14:07 +08:00
parent bed5a97ae2
commit fdc9017ded

View File

@@ -916,7 +916,7 @@ class LightRAG:
else self.key_string_value_json_storage_cls( else self.key_string_value_json_storage_cls(
namespace="llm_response_cache", namespace="llm_response_cache",
global_config=asdict(self), global_config=asdict(self),
embedding_func=None, embedding_func=self.embedding_func,
), ),
prompt=prompt, prompt=prompt,
) )
@@ -933,7 +933,7 @@ class LightRAG:
else self.key_string_value_json_storage_cls( else self.key_string_value_json_storage_cls(
namespace="llm_response_cache", namespace="llm_response_cache",
global_config=asdict(self), global_config=asdict(self),
embedding_func=None, embedding_func=self.embedding_func,
), ),
) )
elif param.mode == "mix": elif param.mode == "mix":
@@ -952,7 +952,7 @@ class LightRAG:
else self.key_string_value_json_storage_cls( else self.key_string_value_json_storage_cls(
namespace="llm_response_cache", namespace="llm_response_cache",
global_config=asdict(self), global_config=asdict(self),
embedding_func=None, embedding_func=self.embedding_func,
), ),
) )
else: else:
@@ -993,7 +993,7 @@ class LightRAG:
or self.key_string_value_json_storage_cls( or self.key_string_value_json_storage_cls(
namespace="llm_response_cache", namespace="llm_response_cache",
global_config=asdict(self), 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( else self.key_string_value_json_storage_cls(
namespace="llm_response_cache", namespace="llm_response_cache",
global_config=asdict(self), global_config=asdict(self),
embedding_func=None, embedding_func=self.embedding_funcne,
), ),
) )
elif param.mode == "naive": elif param.mode == "naive":
@@ -1040,7 +1040,7 @@ class LightRAG:
else self.key_string_value_json_storage_cls( else self.key_string_value_json_storage_cls(
namespace="llm_response_cache", namespace="llm_response_cache",
global_config=asdict(self), global_config=asdict(self),
embedding_func=None, embedding_func=self.embedding_func,
), ),
) )
elif param.mode == "mix": elif param.mode == "mix":
@@ -1059,7 +1059,7 @@ class LightRAG:
else self.key_string_value_json_storage_cls( else self.key_string_value_json_storage_cls(
namespace="llm_response_cache", namespace="llm_response_cache",
global_config=asdict(self), global_config=asdict(self),
embedding_func=None, embedding_func=self.embedding_func,
), ),
) )
else: else: