From d0f3723f995b98aa0741b47c6222afe22eabbed3 Mon Sep 17 00:00:00 2001 From: yanqiangmiffy Date: Fri, 20 Sep 2024 16:26:40 +0800 Subject: [PATCH] =?UTF-8?q?feature@=E4=BF=AE=E6=94=B9=E5=BC=95=E7=94=A8?= =?UTF-8?q?=E6=A0=BC=E5=BC=8F?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- gomate/modules/citation/match_citation.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/gomate/modules/citation/match_citation.py b/gomate/modules/citation/match_citation.py index e999424..2a167c5 100644 --- a/gomate/modules/citation/match_citation.py +++ b/gomate/modules/citation/match_citation.py @@ -96,14 +96,14 @@ def ground_response( highlighted_start_end = self.highlight_common_substrings(sentence, evidence_sentence,evidence) quote_list.append( { - "doc_id": 90564, # 文件id + "doc_id": selected_docs[i]["newsinfo"]["id"], # 文件id "chk_id": best_idx, # 切片索引（从0开始） # 非文内溯源知识集合无需返回 - "doc_source": "新闻来源", + "doc_source": selected_docs[i]["newsinfo"]["id"], # 新闻时间, 非文内溯源知识集合无需返回 - "doc_date": "2021-10-19", + "doc_date": selected_docs[i]["newsinfo"]["date"], # 非文内溯源知识集合无需返回 - "doc_title": "新闻标题", + "doc_title": selected_docs[i]["newsinfo"]["title"], # 非文内溯源知识集合无需返回 "chk_content": evidence, "best_ratio": best_ratio,