agent_config.py 8.0 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201
  1. from qwen_agent.sub_agent.gis.gis_geometry_agent import GisGeometryAgent
  2. from qwen_agent.sub_agent.gis.gis_geocoder_agent import GisGeocoderAgent
  3. from qwen_agent.sub_agent.gis.gis_surrounding_facilities_query_agent import GisSurroundingFacilitiesQueryAgent
  4. from qwen_agent.sub_agent.gis.spatial_analysis_agent import SpatialAnalysisAgent
  5. from qwen_agent.sub_agent.sql.trade_sql_agent import TradeSqlAgent
  6. from qwen_agent.sub_agent.summary_agent import SummaryAgent
  7. from qwen_agent.sub_agent.ChartAgent import ChartAgent
  8. from qwen_agent.sub_agent.sql.land_site_selection_sql_agent import LandSiteSelectionSqlAgent
  9. from qwen_agent.sub_agent.sql.land_find_sql_agent import LandFindSqlAgent
  10. from qwen_agent.sub_agent.sql.land_supply_sql_agent import LandSupplySqlAgent
  11. from qwen_agent.sub_agent.sql.land_use_sql_agent import LandUseSqlAgent
  12. from qwen_agent.sub_agent.sql.land_approval_sql_agent import LandApprovalSqlAgent
  13. from qwen_agent.sub_agent.sql.kfq_eval_sql_agent import KfqEvalSqlAgent
  14. from qwen_agent.sub_agent.ReportAgent import ReportAgent
  15. from qwen_agent.sub_agent.doc.notice_agent import NoticeAgent
  16. from qwen_agent.sub_agent.doc.work_summary_agent import WorkSummaryAgent
  17. from qwen_agent.sub_agent.doc.share_agent import ShareAgent
  18. from qwen_agent.sub_agent.doc.leader_agent import LeaderAgent
  19. from qwen_agent.sub_agent.doc.personal_agent import PersonalAgent
  20. from qwen_agent.sub_agent.doc.work_plan_agent import WorkPLanAgent
  21. from qwen_agent.sub_agent.doc.search_report_agent import SearchReportAgent
  22. from qwen_agent.sub_agent.doc.material_agent import MaterialAgent
  23. from qwen_agent.sub_agent.doc.polish_agent import PolishAgent
  24. from qwen_agent.sub_agent.doc.expand_write_agent import ExpandWriteAgent
  25. from qwen_agent.sub_agent.doc.continue_write_agent import ContinueWriteAgent
  26. from qwen_agent.sub_agent.doc.error_check_agent import ErrorCheckAgent
  27. from qwen_agent.sub_agent.doc.sensitive_check_agent import SensitiveCheckAgent
  28. ActionDict = {
  29. "generate_chart": ChartAgent,
  30. "summary": SummaryAgent,
  31. "LandSiteSelectionSqlAgent": LandSiteSelectionSqlAgent,
  32. "LandFindSqlAgent": LandFindSqlAgent,
  33. "LandApprovalSqlAgent": LandApprovalSqlAgent,
  34. "LandSupplySqlAgent": LandSupplySqlAgent,
  35. "LandUseSqlAgent": LandUseSqlAgent,
  36. "KfqEvalSqlAgent": KfqEvalSqlAgent,
  37. "TradeSqlAgent": TradeSqlAgent,
  38. "ReportAgent": ReportAgent,
  39. "SpatialAnalysisAgent": SpatialAnalysisAgent,
  40. "GisGeometryAgent": GisGeometryAgent,
  41. "GisGeocoderAgent": GisGeocoderAgent,
  42. "GisSurroundingFacilitiesQueryAgent": GisSurroundingFacilitiesQueryAgent,
  43. "NoticeAgent": NoticeAgent,
  44. "WorkSummaryAgent": WorkSummaryAgent,
  45. "ShareAgent": ShareAgent,
  46. "LeaderAgent": LeaderAgent,
  47. "PersonalAgent": PersonalAgent,
  48. "WorkPLanAgent": WorkPLanAgent,
  49. "SearchReportAgent": SearchReportAgent,
  50. "MaterialAgent": MaterialAgent,
  51. "PolishAgent": PolishAgent,
  52. "ExpandWriteAgent": ExpandWriteAgent,
  53. "ContinueWriteAgent": ContinueWriteAgent,
  54. "ErrorCheckAgent": ErrorCheckAgent,
  55. "SensitiveCheckAgent": SensitiveCheckAgent
  56. }
  57. AgentCNNameDict = {
  58. "generate_chart": "图表生成",
  59. "gis_planner": "空间分析",
  60. "SpatialAnalysisAgent": "空间分析和图层查询",
  61. "GisGeometryAgent": "Gis 基础操作",
  62. "GisGeocoderAgent": "Gis 地理编码",
  63. "GisSurroundingFacilitiesQueryAgent": "周边设施查询",
  64. "summary": "总结摘要",
  65. "plan_dispatcher": "任务分发",
  66. "Chat": "聊天机器人",
  67. "KnowledgeChat": "知识库问答",
  68. "LandSiteSelectionSqlAgent": "智能选址",
  69. "LandFindSqlAgent": "找图找数",
  70. "KfqEvalSqlAgent": "园区及开发区评价查询",
  71. "LandSupplySqlAgent": "供地合同信息查询",
  72. "LandUseSqlAgent": "土地利用现状信息查询",
  73. "LandApprovalSqlAgent": "土地报批项目信息查询",
  74. "ReportAgent": "分析报告生成",
  75. "land_site_selection_planner": "地块智能选址plan",
  76. "land_find_planner": "找图找数plan",
  77. "land_supply_planner": "企业信息查询plan",
  78. "land_use_planner": "地利用现状信息查询plan",
  79. "land_approval_planner": "土地报批项目查询plan",
  80. "kfq_eval_planner": "园区及开发区评价查询plan",
  81. "TradeSqlAgent": "土地公开交易信息查询",
  82. "landtrade_planner": "土地公开交易信息查询plan",
  83. "report_planner": "分析报告生成plan",
  84. "doc_write_planner": "用于公文生成,并可以对文字进行润色、扩写、续写,还能检查文章的内容是否有错误、是否包含敏感词的plan",
  85. "NoticeAgent": "用于生成通知报告的Agent",
  86. "WorkSummaryAgent": "用于生成工作总结的Agent",
  87. "ShareAgent": "用于生成心得体会的Agent",
  88. "LeaderAgent": "用于生成领导讲话的Agent",
  89. "PersonalAgent": "用于生成个人讲话的Agent",
  90. "WorkPLanAgent": "用于生成工作方案的Agent",
  91. "SearchReportAgent": "用于生成调研报告的Agent",
  92. "MaterialAgent": "用于生成宣传材料的Agent",
  93. "PolishAgent": "用于文章润色的Agent",
  94. "ExpandWriteAgent": "用于文章扩写的Agent",
  95. "ContinueWriteAgent": "用于文章续写的Agent",
  96. "ErrorCheckAgent": "用于检查文章内容是否有错误的Agent",
  97. "SensitiveCheckAgent": "用于文章内敏感词检查的Agent"
  98. }
  99. LLMDict_Qwen72 = {
  100. "plan_dispatcher": 'qwen-72b-chat',
  101. "generate_chart": 'qwen-72b-chat',
  102. "summary": 'qwen-72b-chat',
  103. "TenderResultSqlAgent": 'qwen-72b-chat',
  104. "planner": "qwen-72b-chat",
  105. "show_case": "qwen-72b-chat",
  106. "Chat": "qwen-72b-chat",
  107. }
  108. LLMDict_Qwen_72B_1211 = {
  109. "plan_dispatcher": 'Qwen_72B_1211',
  110. "generate_chart": 'Qwen_72B_1211',
  111. "summary": 'Qwen_72B_1211',
  112. "TenderResultSqlAgent": 'Qwen_72B_1211',
  113. "planner": "Qwen_72B_1211",
  114. "show_case": "Qwen_72B_1211",
  115. "Chat": "Qwen_72B_1211",
  116. }
  117. LLMDict = {
  118. "plan_dispatcher": 'qwen-plus',
  119. "generate_chart": 'Code2',
  120. "summary": 'qwen-plus',
  121. "TenderResultSqlAgent": 'Code',
  122. "planner": "Code2",
  123. "show_case": "Code2",
  124. "Chat": "qwen-plus",
  125. }
  126. LLMDict_Code = {
  127. "plan_dispatcher": 'Code2',
  128. "generate_chart": 'Code2',
  129. "summary": 'Code2',
  130. "TenderResultSqlAgent": 'Code2',
  131. "planner": "Code2",
  132. "show_case": "Code2",
  133. "Chat": "Code2",
  134. }
  135. LLMDict_GPT4_TURBO = {
  136. "plan_dispatcher": 'gpt-4-turbo',
  137. "generate_chart": 'gpt-4-turbo',
  138. "summary": 'gpt-4-turbo',
  139. "TenderResultSqlAgent": 'gpt-4-turbo',
  140. "planner": "gpt-4-turbo",
  141. "show_case": "gpt-4-turbo",
  142. "Chat": "gpt-4-turbo",
  143. }
  144. LLMDict_GPT4 = {
  145. "generate_chart": 'gpt-4',
  146. "summary": 'gpt-4',
  147. "TenderResultSqlAgent": 'gpt-4',
  148. "planner": "gpt-4",
  149. "ProductRecallAgent": "gpt-4",
  150. "show_case": "gpt-4",
  151. "Chat": "gpt-4",
  152. }
  153. LLMDict_Qwen_72B_1211 = {
  154. "plan_dispatcher": 'qwen2.5-instruct',
  155. "generate_chart": 'qwen2.5-instruct',
  156. "summary": 'qwen2.5-instruct',
  157. "planner": "qwen2.5-instruct",
  158. "show_case": "qwen2.5-instruct",
  159. "Chat": "qwen2.5-instruct",
  160. "LandSupplySqlAgent": "qwen2.5-instruct",
  161. "TradeSqlAgent": "qwen2.5-instruct",
  162. "LandUseSqlAgent": "qwen2.5-instruct",
  163. "KfqEvalSqlAgent": "qwen2.5-instruct",
  164. "LandApprovalSqlAgent": "qwen2.5-instruct",
  165. "ReportAgent": "qwen2.5-instruct",
  166. # 智能选址
  167. "LandSiteSelectionSqlAgent": "qwen2.5-instruct",
  168. # 找图找数
  169. "LandFindSqlAgent": "qwen2.5-instruct",
  170. # 空间分析
  171. "SpatialAnalysisAgent": "qwen2.5-instruct",
  172. "GisGeometryAgent": "qwen2.5-instruct",
  173. "GisGeocoderAgent": "qwen2.5-instruct",
  174. "GisSurroundingFacilitiesQueryAgent": "qwen2.5-instruct",
  175. "NoticeAgent": "qwen2.5-instruct",
  176. "WorkSummaryAgent": "qwen2.5-instruct",
  177. "ShareAgent": "qwen2.5-instruct",
  178. "LeaderAgent": "qwen2.5-instruct",
  179. "PersonalAgent": "qwen2.5-instruct",
  180. "WorkPLanAgent": "qwen2.5-instruct",
  181. "SearchReportAgent": "qwen2.5-instruct",
  182. "MaterialAgent": "qwen2.5-instruct",
  183. "PolishAgent": "qwen2.5-instruct",
  184. "ExpandWriteAgent": "qwen2.5-instruct",
  185. "ContinueWriteAgent": "qwen2.5-instruct",
  186. "ErrorCheckAgent": "qwen2.5-instruct",
  187. "SensitiveCheckAgent": "qwen2.5-instruct"
  188. }