from qwen_agent.sub_agent.gis.gis_geometry_agent import GisGeometryAgent
from qwen_agent.sub_agent.gis.gis_geocoder_agent import GisGeocoderAgent
from qwen_agent.sub_agent.gis.gis_surrounding_facilities_query_agent import GisSurroundingFacilitiesQueryAgent
from qwen_agent.sub_agent.gis.spatial_analysis_agent import SpatialAnalysisAgent
from qwen_agent.sub_agent.sql.trade_sql_agent import TradeSqlAgent
from qwen_agent.sub_agent.summary_agent import SummaryAgent
from qwen_agent.sub_agent.ChartAgent import ChartAgent
from qwen_agent.sub_agent.sql.land_site_selection_sql_agent import LandSiteSelectionSqlAgent
from qwen_agent.sub_agent.sql.land_find_sql_agent import LandFindSqlAgent
from qwen_agent.sub_agent.sql.land_supply_sql_agent import LandSupplySqlAgent
from qwen_agent.sub_agent.sql.land_use_sql_agent import LandUseSqlAgent
from qwen_agent.sub_agent.sql.land_approval_sql_agent import LandApprovalSqlAgent
from qwen_agent.sub_agent.sql.kfq_eval_sql_agent import KfqEvalSqlAgent
from qwen_agent.sub_agent.ReportAgent import ReportAgent
from qwen_agent.sub_agent.doc.notice_agent import NoticeAgent
from qwen_agent.sub_agent.doc.work_summary_agent import WorkSummaryAgent
from qwen_agent.sub_agent.doc.share_agent import ShareAgent
from qwen_agent.sub_agent.doc.leader_agent import LeaderAgent
from qwen_agent.sub_agent.doc.personal_agent import PersonalAgent
from qwen_agent.sub_agent.doc.work_plan_agent import WorkPLanAgent
from qwen_agent.sub_agent.doc.search_report_agent import SearchReportAgent
from qwen_agent.sub_agent.doc.material_agent import MaterialAgent
from qwen_agent.sub_agent.doc.polish_agent import PolishAgent
from qwen_agent.sub_agent.doc.expand_write_agent import ExpandWriteAgent
from qwen_agent.sub_agent.doc.continue_write_agent import ContinueWriteAgent
from qwen_agent.sub_agent.doc.error_check_agent import ErrorCheckAgent
from qwen_agent.sub_agent.doc.sensitive_check_agent import SensitiveCheckAgent
from qwen_agent.sub_agent.gis.gis_layer_operation import GisLayerOperationAgent
ActionDict = {
    "generate_chart": ChartAgent,
    "summary": SummaryAgent,
    "LandSiteSelectionSqlAgent": LandSiteSelectionSqlAgent,
    "LandFindSqlAgent": LandFindSqlAgent,
    "LandApprovalSqlAgent": LandApprovalSqlAgent,
    "LandSupplySqlAgent": LandSupplySqlAgent,
    "LandUseSqlAgent": LandUseSqlAgent,
    "KfqEvalSqlAgent": KfqEvalSqlAgent,
    "TradeSqlAgent": TradeSqlAgent,
    "ReportAgent": ReportAgent,
    "SpatialAnalysisAgent": SpatialAnalysisAgent,
    "GisGeometryAgent": GisGeometryAgent,
    "GisGeocoderAgent": GisGeocoderAgent,
    "GisSurroundingFacilitiesQueryAgent": GisSurroundingFacilitiesQueryAgent,
    "NoticeAgent": NoticeAgent,
    "WorkSummaryAgent": WorkSummaryAgent,
    "ShareAgent": ShareAgent,
    "LeaderAgent": LeaderAgent,
    "PersonalAgent": PersonalAgent,
    "WorkPLanAgent": WorkPLanAgent,
    "SearchReportAgent": SearchReportAgent,
    "MaterialAgent": MaterialAgent,
    "PolishAgent": PolishAgent,
    "ExpandWriteAgent": ExpandWriteAgent,
    "ContinueWriteAgent": ContinueWriteAgent,
    "ErrorCheckAgent": ErrorCheckAgent,
    "SensitiveCheckAgent": SensitiveCheckAgent,
    "GisLayerOperationAgent":GisLayerOperationAgent
}

AgentCNNameDict = {
    "generate_chart": "图表生成",
    "gis_planner": "空间分析",
    "SpatialAnalysisAgent": "空间分析和图层查询",
    "GisGeometryAgent": "Gis 基础操作",
    "GisGeocoderAgent": "Gis 地理编码",
    "GisSurroundingFacilitiesQueryAgent": "周边设施查询",
    "summary": "总结摘要",
    "plan_dispatcher": "任务分发",
    "Chat": "聊天机器人",
    "KnowledgeChat": "知识库问答",
    "LandSiteSelectionSqlAgent": "智能选址",
    "LandFindSqlAgent": "找图找数",
    "KfqEvalSqlAgent": "园区及开发区评价查询",
    "LandSupplySqlAgent": "供地合同信息查询",
    "LandUseSqlAgent": "土地利用现状信息查询",
    "LandApprovalSqlAgent": "土地报批项目信息查询",
    "ReportAgent": "分析报告生成",
    "land_site_selection_planner": "地块智能选址plan",
    "land_find_planner": "找图找数plan",
    "land_supply_planner": "企业信息查询plan",
    "land_use_planner": "地利用现状信息查询plan",
    "land_approval_planner": "土地报批项目查询plan",
    "kfq_eval_planner": "园区及开发区评价查询plan",
    "TradeSqlAgent": "土地公开交易信息查询",
    "landtrade_planner": "土地公开交易信息查询plan",
    "report_planner": "分析报告生成plan",
    "doc_write_planner": "用于公文生成,并可以对文字进行润色、扩写、续写,还能检查文章的内容是否有错误、是否包含敏感词的plan",
    "NoticeAgent": "用于生成通知报告的Agent",
    "WorkSummaryAgent": "用于生成工作总结的Agent",
    "ShareAgent": "用于生成心得体会的Agent",
    "LeaderAgent": "用于生成领导讲话的Agent",
    "PersonalAgent": "用于生成个人讲话的Agent",
    "WorkPLanAgent": "用于生成工作方案的Agent",
    "SearchReportAgent": "用于生成调研报告的Agent",
    "MaterialAgent": "用于生成宣传材料的Agent",
    "PolishAgent": "用于文章润色的Agent",
    "ExpandWriteAgent": "用于文章扩写的Agent",
    "ContinueWriteAgent": "用于文章续写的Agent",
    "ErrorCheckAgent": "用于检查文章内容是否有错误的Agent",
    "SensitiveCheckAgent": "用于文章内敏感词检查的Agent",
    "layer_operation_planner": "用于GIS图层控制系统的Agent",
    "GisLayerOperationAgent": "Gis 图层控制",
}

LLMDict_Qwen72 = {
    "plan_dispatcher": 'qwen-72b-chat',
    "generate_chart": 'qwen-72b-chat',
    "summary": 'qwen-72b-chat',
    "TenderResultSqlAgent": 'qwen-72b-chat',
    "planner": "qwen-72b-chat",
    "show_case": "qwen-72b-chat",
    "Chat": "qwen-72b-chat",
}

LLMDict_Qwen_72B_1211 = {
    "plan_dispatcher": 'Qwen_72B_1211',
    "generate_chart": 'Qwen_72B_1211',
    "summary": 'Qwen_72B_1211',
    "TenderResultSqlAgent": 'Qwen_72B_1211',
    "planner": "Qwen_72B_1211",
    "show_case": "Qwen_72B_1211",
    "Chat": "Qwen_72B_1211",
}

LLMDict = {
    "plan_dispatcher": 'qwen-plus',
    "generate_chart": 'Code2',
    "summary": 'qwen-plus',
    "TenderResultSqlAgent": 'Code',
    "planner": "Code2",
    "show_case": "Code2",
    "Chat": "qwen-plus",
}

LLMDict_Code = {
    "plan_dispatcher": 'Code2',
    "generate_chart": 'Code2',
    "summary": 'Code2',
    "TenderResultSqlAgent": 'Code2',
    "planner": "Code2",
    "show_case": "Code2",
    "Chat": "Code2",
}

LLMDict_GPT4_TURBO = {

    "plan_dispatcher": 'gpt-4-turbo',
    "generate_chart": 'gpt-4-turbo',
    "summary": 'gpt-4-turbo',
    "TenderResultSqlAgent": 'gpt-4-turbo',
    "planner": "gpt-4-turbo",
    "show_case": "gpt-4-turbo",
    "Chat": "gpt-4-turbo",

}

LLMDict_GPT4 = {

    "generate_chart": 'gpt-4',
    "summary": 'gpt-4',
    "TenderResultSqlAgent": 'gpt-4',
    "planner": "gpt-4",
    "ProductRecallAgent": "gpt-4",
    "show_case": "gpt-4",
    "Chat": "gpt-4",

}

LLMDict_Qwen_72B_1211 = {
    "plan_dispatcher": 'qwen2.5-instruct',
    "generate_chart": 'qwen2.5-instruct',
    "summary": 'qwen2.5-instruct',
    "planner": "qwen2.5-instruct",
    "show_case": "qwen2.5-instruct",
    "Chat": "qwen2.5-instruct",
    "LandSupplySqlAgent": "qwen2.5-instruct",
    "TradeSqlAgent": "qwen2.5-instruct",
    "LandUseSqlAgent": "qwen2.5-instruct",
    "KfqEvalSqlAgent": "qwen2.5-instruct",
    "LandApprovalSqlAgent": "qwen2.5-instruct",
    "ReportAgent": "qwen2.5-instruct",
    # 智能选址
    "LandSiteSelectionSqlAgent": "qwen2.5-instruct",
    # 找图找数
    "LandFindSqlAgent": "qwen2.5-instruct",
    # 空间分析
    "SpatialAnalysisAgent": "qwen2.5-instruct",
    "GisGeometryAgent": "qwen2.5-instruct",
    "GisGeocoderAgent": "qwen2.5-instruct",
    "GisSurroundingFacilitiesQueryAgent": "qwen2.5-instruct",
    "NoticeAgent": "qwen2.5-instruct",
    "WorkSummaryAgent": "qwen2.5-instruct",
    "ShareAgent": "qwen2.5-instruct",
    "LeaderAgent": "qwen2.5-instruct",
    "PersonalAgent": "qwen2.5-instruct",
    "WorkPLanAgent": "qwen2.5-instruct",
    "SearchReportAgent": "qwen2.5-instruct",
    "MaterialAgent": "qwen2.5-instruct",
    "PolishAgent": "qwen2.5-instruct",
    "ExpandWriteAgent": "qwen2.5-instruct",
    "ContinueWriteAgent": "qwen2.5-instruct",
    "ErrorCheckAgent": "qwen2.5-instruct",
    "SensitiveCheckAgent": "qwen2.5-instruct",
    "GisLayerOperationAgent": "qwen2.5-instruct"
}