From 30be5efd3ab3eff8054bfecd4fe36ef080fcb20a Mon Sep 17 00:00:00 2001 From: QiyiJiang <3442721524@qq.com> Date: Sun, 14 Jul 2024 13:29:58 +0000 Subject: [PATCH] =?UTF-8?q?jsonl=20=E6=A0=BC=E5=BC=8F?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- src/plugins/oneke.py | 18 +++--------------- 1 file changed, 3 insertions(+), 15 deletions(-) diff --git a/src/plugins/oneke.py b/src/plugins/oneke.py index 97d87a8b..0f1f3757 100644 --- a/src/plugins/oneke.py +++ b/src/plugins/oneke.py @@ -136,22 +136,10 @@ class OneKE: task = "KG" output = self.predict(text=text, schema=schema, task=task, language="zh") formatted_output = parse_and_format_output(output=output, task_type=task) - if os.path.exists(output_path): - with open(output_path, 'r', encoding='utf-8') as f: - existing_data = json.load(f) - else: - existing_data = [] - # 将新的数据添加到现有数据之后 - existing_data.extend(formatted_output) - - # 将合并后的数据写回 JSON 文件 - with open(output_path, 'w', encoding='utf-8') as f: - json.dump(existing_data, f, ensure_ascii=False, indent=4) - - # with open(output_path, 'a+', encoding='utf-8') as f: - # for entry in formatted_output: - # f.write(json.dumps(entry, ensure_ascii=False) + '\n') + with open(output_path, 'a+', encoding='utf-8') as f: + for entry in formatted_output: + f.write(json.dumps(entry, ensure_ascii=False) + '\n') print(f"预测结果已添加到 {output_path} 文件中。") return output_path