yangdx
commited on
Commit
·
0bf83a8
1
Parent(s):
71fd78c
Set embedding_func in all llm_response_cache
Browse files- lightrag/lightrag.py +7 -7
lightrag/lightrag.py
CHANGED
|
@@ -916,7 +916,7 @@ class LightRAG:
|
|
| 916 |
else self.key_string_value_json_storage_cls(
|
| 917 |
namespace="llm_response_cache",
|
| 918 |
global_config=asdict(self),
|
| 919 |
-
embedding_func=
|
| 920 |
),
|
| 921 |
prompt=prompt,
|
| 922 |
)
|
|
@@ -933,7 +933,7 @@ class LightRAG:
|
|
| 933 |
else self.key_string_value_json_storage_cls(
|
| 934 |
namespace="llm_response_cache",
|
| 935 |
global_config=asdict(self),
|
| 936 |
-
embedding_func=
|
| 937 |
),
|
| 938 |
)
|
| 939 |
elif param.mode == "mix":
|
|
@@ -952,7 +952,7 @@ class LightRAG:
|
|
| 952 |
else self.key_string_value_json_storage_cls(
|
| 953 |
namespace="llm_response_cache",
|
| 954 |
global_config=asdict(self),
|
| 955 |
-
embedding_func=
|
| 956 |
),
|
| 957 |
)
|
| 958 |
else:
|
|
@@ -993,7 +993,7 @@ class LightRAG:
|
|
| 993 |
or self.key_string_value_json_storage_cls(
|
| 994 |
namespace="llm_response_cache",
|
| 995 |
global_config=asdict(self),
|
| 996 |
-
embedding_func=
|
| 997 |
),
|
| 998 |
)
|
| 999 |
|
|
@@ -1024,7 +1024,7 @@ class LightRAG:
|
|
| 1024 |
else self.key_string_value_json_storage_cls(
|
| 1025 |
namespace="llm_response_cache",
|
| 1026 |
global_config=asdict(self),
|
| 1027 |
-
embedding_func=
|
| 1028 |
),
|
| 1029 |
)
|
| 1030 |
elif param.mode == "naive":
|
|
@@ -1040,7 +1040,7 @@ class LightRAG:
|
|
| 1040 |
else self.key_string_value_json_storage_cls(
|
| 1041 |
namespace="llm_response_cache",
|
| 1042 |
global_config=asdict(self),
|
| 1043 |
-
embedding_func=
|
| 1044 |
),
|
| 1045 |
)
|
| 1046 |
elif param.mode == "mix":
|
|
@@ -1059,7 +1059,7 @@ class LightRAG:
|
|
| 1059 |
else self.key_string_value_json_storage_cls(
|
| 1060 |
namespace="llm_response_cache",
|
| 1061 |
global_config=asdict(self),
|
| 1062 |
-
embedding_func=
|
| 1063 |
),
|
| 1064 |
)
|
| 1065 |
else:
|
|
|
|
| 916 |
else self.key_string_value_json_storage_cls(
|
| 917 |
namespace="llm_response_cache",
|
| 918 |
global_config=asdict(self),
|
| 919 |
+
embedding_func=self.embedding_func,
|
| 920 |
),
|
| 921 |
prompt=prompt,
|
| 922 |
)
|
|
|
|
| 933 |
else self.key_string_value_json_storage_cls(
|
| 934 |
namespace="llm_response_cache",
|
| 935 |
global_config=asdict(self),
|
| 936 |
+
embedding_func=self.embedding_func,
|
| 937 |
),
|
| 938 |
)
|
| 939 |
elif param.mode == "mix":
|
|
|
|
| 952 |
else self.key_string_value_json_storage_cls(
|
| 953 |
namespace="llm_response_cache",
|
| 954 |
global_config=asdict(self),
|
| 955 |
+
embedding_func=self.embedding_func,
|
| 956 |
),
|
| 957 |
)
|
| 958 |
else:
|
|
|
|
| 993 |
or self.key_string_value_json_storage_cls(
|
| 994 |
namespace="llm_response_cache",
|
| 995 |
global_config=asdict(self),
|
| 996 |
+
embedding_func=self.embedding_func,
|
| 997 |
),
|
| 998 |
)
|
| 999 |
|
|
|
|
| 1024 |
else self.key_string_value_json_storage_cls(
|
| 1025 |
namespace="llm_response_cache",
|
| 1026 |
global_config=asdict(self),
|
| 1027 |
+
embedding_func=self.embedding_funcne,
|
| 1028 |
),
|
| 1029 |
)
|
| 1030 |
elif param.mode == "naive":
|
|
|
|
| 1040 |
else self.key_string_value_json_storage_cls(
|
| 1041 |
namespace="llm_response_cache",
|
| 1042 |
global_config=asdict(self),
|
| 1043 |
+
embedding_func=self.embedding_func,
|
| 1044 |
),
|
| 1045 |
)
|
| 1046 |
elif param.mode == "mix":
|
|
|
|
| 1059 |
else self.key_string_value_json_storage_cls(
|
| 1060 |
namespace="llm_response_cache",
|
| 1061 |
global_config=asdict(self),
|
| 1062 |
+
embedding_func=self.embedding_func,
|
| 1063 |
),
|
| 1064 |
)
|
| 1065 |
else:
|