gardel/lightrag
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Set embedding_func in all llm_response_cache

Browse Source
This commit is contained in:
yangdx
2025-02-02 03:14:07 +08:00
parent bed5a97ae2
commit fdc9017ded
1 changed files with 7 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
lightrag/lightrag.py
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: