zhaoqingang
2024-12-06 9e51ec752c93d850db15cd61a7a3463d8fe94344
app/task/fetch_agent.py
@@ -108,17 +108,21 @@
    try:
        count = db.query(AgentModel).count()
        if count > 0:
            return
            # 删除表中的所有数据
            result = db.query(AgentModel).delete()
            db.commit()  # 提交事务
        initial_agents = [
            ('80ee430a-e396-48c4-a12c-7c7cdf5eda51', 1, '报告生成', 'BISHENG', 'report'),
            ('basic_excel_merge', 2, '报表合并', 'BASIC', 'excelMerge'),
            ('bfd090d589d811efb3630242ac190006', 4, '文档智能', 'BISHENG', 'report'),
            ('da3451da89d911efb9490242ac190006', 3, '知识问答', 'RAGFLOW', 'knowledgeQA'),
            ('e96eb7a589db11ef87d20242ac190006', 5, '智能问答', 'RAGFLOW', 'chat'),
            ('basic_excel_talk', 6, '智能数据', 'BASIC', 'excelTalk'),
            ('basic_question_talk', 7, '出题组卷', 'BASIC', 'questionTalk'),
            ('9d75142a-66eb-4e23-b7d4-03efe4584915', 8, '小数绘图', 'DIFY', 'imageTalk'),
            ('basic_paper_talk', 8, '文档出卷', 'BASIC', 'paperTalk')
            ('basic_report_clean', 6, '文档报告', 'DIFY', 'reportWorkflow')
            # ('basic_excel_talk', 6, '智能数据', 'BASIC', 'excelTalk'),
            # ('basic_question_talk', 7, '出题组卷', 'BASIC', 'questionTalk'),
            # ('9d75142a-66eb-4e23-b7d4-03efe4584915', 8, '小数绘图', 'DIFY', 'imageTalk'),
            # ('basic_paper_talk', 8, '文档出卷', 'BASIC', 'paperTalk')
        ]
        for agent in initial_agents: