update report

qingzhong1 · Jan 10, 2024 · b8c34b7 · b8c34b7
1 parent 494b020
commit b8c34b7
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/erniebot-agent/applications/erniebot_researcher/sample_group_agent.py b/erniebot-agent/applications/erniebot_researcher/sample_group_agent.py
@@ -109,8 +109,8 @@ def get_agents(retriever_sets, tool_sets, llm, llm_long, dir_path, target_path):
         system_message=SystemMessage("你是一个报告生成助手。你可以根据用户的指定内容生成一份报告手稿"),
         dir_path=dir_path,
         report_type=args.report_type,
-        retriever_abstract_tool=retriever_sets["abstract"],
-        retriever_tool=retriever_sets["full_text"],
+        retriever_abstract_db=retriever_sets["abstract"],
+        retriever_fulltext_db=retriever_sets["full_text"],
         intent_detection_tool=tool_sets["intent_detection"],
         task_planning_tool=tool_sets["task_planning"],
         report_writing_tool=tool_sets["report_writing"],
@@ -130,7 +130,7 @@ def get_agents(retriever_sets, tool_sets, llm, llm_long, dir_path, target_path):
         name="polish",
         llm=llm,
         llm_long=llm_long,
-        faiss_name_citation=args.index_name_citation,
+        citation_index_name=args.index_name_citation,
         embeddings=retriever_sets["embeddings"],
         dir_path=target_path,
         report_type=args.report_type,