45 lines
1.7 KiB
Python
45 lines
1.7 KiB
Python
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)
|