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

update utils.py

Browse Source
This commit is contained in:
LarFii
2024-10-11 11:22:17 +08:00
parent 8ff64dfe2c
commit cb0ba0b447
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
lightrag/utils.py
View File

@@ -94,11 +94,11 @@ def wrap_embedding_func_with_attrs(**kwargs):
def load_json(file_name): def load_json(file_name):
if not os.path.exists(file_name): if not os.path.exists(file_name):
return None return None
with open(file_name) as f: with open(file_name, encoding="utf-8") as f:
return json.load(f) return json.load(f)
def write_json(json_obj, file_name): def write_json(json_obj, file_name):
with open(file_name, "w") as f: with open(file_name, "w", encoding="utf-8") as f:
json.dump(json_obj, f, indent=2, ensure_ascii=False) json.dump(json_obj, f, indent=2, ensure_ascii=False)
def encode_string_by_tiktoken(content: str, model_name: str = "gpt-4o"): def encode_string_by_tiktoken(content: str, model_name: str = "gpt-4o"):