from qwen_agent.sub_agent.gis.gis_geocoder_agent import GisGeocoderAgent
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_supply_sql_agent import LandSupplySqlAgent
from qwen_agent.sub_agent.sql.land_use_sql_agent import LandUseSqlAgent
from qwen_agent.sub_agent.sql.land_find_sql_agent import LandFindSqlAgent
from qwen_agent.sub_agent.gis.gis_layer_operation import GisLayerOperationAgent
ActionDict = {
    "generate_chart": ChartAgent,
    "summary": SummaryAgent,
    "LandFindSqlAgent": LandFindSqlAgent,
    "LandSiteSelectionSqlAgent": LandSiteSelectionSqlAgent,
    "LandSupplySqlAgent": LandSupplySqlAgent,
    "LandUseSqlAgent": LandUseSqlAgent,
    "GisGeocoderAgent": GisGeocoderAgent,
    "GisLayerOperationAgent":GisLayerOperationAgent
}

AgentCNNameDict = {
    "generate_chart": "图表生成",
    "GisGeocoderAgent": "Gis 地理编码",
    "summary": "总结摘要",
    "plan_dispatcher": "任务分发",
    "Chat": "聊天机器人",
    "LandSiteSelectionSqlAgent": "智能选址",
    "LandFindSqlAgent": "找图找数",
    "LandSupplySqlAgent": "供地合同信息查询",
    "land_site_selection_planner": "地块智能选址plan",
    "land_find_planner": "找图找数plan",
    "land_supply_planner": "企业信息查询plan",
    "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_DS_R1_21B = {
    "plan_dispatcher": 'DeepSeek-R1-Distill-Qwen-32B',
    "generate_chart": 'DeepSeek-R1-Distill-Qwen-32B',
    "summary": 'DeepSeek-R1-Distill-Qwen-32B',
    "planner": "DeepSeek-R1-Distill-Qwen-32B",
    "show_case": "DeepSeek-R1-Distill-Qwen-32B",
    "Chat": "DeepSeek-R1-Distill-Qwen-32B",
    "LandSupplySqlAgent": "DeepSeek-R1-Distill-Qwen-32B",
    # 智能选址
    "LandSiteSelectionSqlAgent": "DeepSeek-R1-Distill-Qwen-32B",
    # 找图找数
    "LandFindSqlAgent": "DeepSeek-R1-Distill-Qwen-32B",
    "GisGeocoderAgent": "DeepSeek-R1-Distill-Qwen-32B",
    "GisLayerOperationAgent": "DeepSeek-R1-Distill-Qwen-32B"
}

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",
    # 智能选址
    "LandSiteSelectionSqlAgent": "qwen2.5-instruct",
    # 找图找数
    "LandFindSqlAgent": "qwen2.5-instruct",
    # 空间分析
    "SpatialAnalysisAgent": "qwen2.5-instruct",
    "GisGeocoderAgent": "qwen2.5-instruct",
    "GisLayerOperationAgent": "qwen2.5-instruct"
}