import edward

search/search_manager.py

@@ -0,0 +1,44 @@
+from database.database_manager import DatabaseManager
+from search.search_from_doubao import DouBao
+from search.search_from_kimi import Kimi
+from search.search_from_tongyi import TongYi
+
+
+def send_dispatch_content_inner(content: str, category: str, title: str):
+    """send dispatch content"""
+    db_manager = DatabaseManager()
+    db_manager.send_dispatch_content(category=category, title=title, content=content, ai_generate=1)
+
+
+def search(input_text: str):
+    """search messages"""
+    kimi = Kimi()
+    kimi_search_result = kimi.search(input_text)
+    kimi.quit()
+    tong_yi = TongYi()
+    tong_yi_search_result = tong_yi.search(input_text)
+    tong_yi.quit()
+    dou_bao = DouBao()
+    dou_bao_search_result = dou_bao.search(input_text)
+    dou_bao.quit()
+    return f"{kimi_search_result}\n{tong_yi_search_result}\n{dou_bao_search_result}"
+
+
+def summarize(text: str) -> str:
+    """Summarizes the given text."""
+    prompt = ("请根据提供的草稿，生成一篇完整的文章。\n"
+              "要求去重和格式化。\n"
+              "整篇文章全部采用并且只使用简洁的有序列表格式，不需要小标题，例如：\n1、今天天气不错\n2、张三捉住一只猫\n"
+              "\n以下是草稿内容：\n{}").format(text)
+    kimi = Kimi()
+    kimi_inference_result = kimi.search(prompt)
+    kimi.quit()
+    print(kimi_inference_result)
+    return kimi_inference_result
+
+
+def editor(subject: str, category: str, title: str):
+    """这是一个编辑工具，根据提供的主题，进行素材检索和文章生成"""
+    draft = search(subject)
+    kimi_inference_result = summarize(draft)
+    send_dispatch_content_inner(kimi_inference_result, category, title)