update example_group

qingzhong1 · Jan 9, 2024 · 21ca0fa · 21ca0fa
1 parent 44807fe
commit 21ca0fa
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 5 deletions.
diff --git a/erniebot-agent/applications/erniebot_researcher/requirements.txt b/erniebot-agent/applications/erniebot_researcher/requirements.txt
@@ -1,4 +1,6 @@
 gradio
 jsonlines
 md2pdf
-langchain
+langchain
+md2pdf
+scikit-learn
diff --git a/erniebot-agent/applications/erniebot_researcher/tools/semantic_citation_tool.py b/erniebot-agent/applications/erniebot_researcher/tools/semantic_citation_tool.py
@@ -1,6 +1,6 @@
 import logging
 import string
-from typing import Optional
+from typing import Dict, List, Optional
 
 from erniebot_agent.tools.base import Tool
 
@@ -35,9 +35,9 @@ def __init__(self, theta_min=0.4, theta_max=0.95, citation_num=5) -> None:
         self.theta_min = theta_min
         self.theta_max = theta_max
         self.citation_num = citation_num
-        self.recoder_cite_dict: dict = {}
-        self.recoder_cite_list: list = []
-        self.recoder_cite_title: list = []
+        self.recoder_cite_dict: Dict = {}
+        self.recoder_cite_list: List = []
+        self.recoder_cite_title: List = []
 
     def add_url_sentences(self, sententces: str, citation_faiss_research):
         sentence_splits = sententces.split("。")