agent_config.py 8.3 KB

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