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

change process_combine_contexts params type to list[dict[str, str]]

Browse Source
This commit is contained in:
孟超
2025-04-21 12:08:12 +08:00
parent 510a6adfbc
commit 8064a2339f
1 changed files with 9 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
lightrag/utils.py
View File

@@ -385,7 +385,7 @@ def list_of_list_to_json(data: list[list[str]]) -> list[dict[str, str]]:
item = {} item = {}
for i, field_name in enumerate(header): for i, field_name in enumerate(header):
if i < len(row): if i < len(row):
item[field_name] = row[i] item[field_name] = str(row[i])
else: else:
item[field_name] = "" item[field_name] = ""
result.append(item) result.append(item)
@@ -458,19 +458,21 @@ def xml_to_json(xml_file):
return None return None
def process_combine_contexts(hl_context: dict, ll_context: dict): def process_combine_contexts(
hl_context: list[dict[str, str]], ll_context: list[dict[str, str]]
):
seen_content = {} seen_content = {}
combined_data = [] combined_data = []
for item in hl_context + ll_context: for item in hl_context + ll_context:
content_key = {k: v for k, v in item.items() if k != "id"} content_dict = {k: v for k, v in item.items() if k != "id"}
content_key_str = str(content_key) content_key = tuple(sorted(content_dict.items()))
if content_key_str not in seen_content: if content_key not in seen_content:
seen_content[content_key_str] = item seen_content[content_key] = item
combined_data.append(item) combined_data.append(item)
for i, item in enumerate(combined_data): for i, item in enumerate(combined_data):
item["id"] = i item["id"] = str(i)
return combined_data return combined_data