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(
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: