feature@修改引用格式

gomate-community · Sep 20, 2024 · d0f3723 · d0f3723
1 parent 6a7edfc
commit d0f3723
Showing 1 changed file with 4 additions and 4 deletions.
diff --git 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,