"""LLM Function Calling 工具定义

定义 Chat Agent 可使用的工具 schema（OpenAI function calling 格式）。
"""

CHAT_TOOLS = [
    {
        "type": "function",
        "function": {
            "name": "get_strategy_board",
            "description": "获取今日作战结论，包括市场状态、今日打法、建议仓位、重点板块和规避规则",
            "parameters": {
                "type": "object",
                "properties": {},
                "required": [],
            },
        },
    },
    {
        "type": "function",
        "function": {
            "name": "get_market_temperature",
            "description": "获取当前市场温度数据，包括涨跌家数、涨停数、连板高度、炸板率等市场情绪指标",
            "parameters": {
                "type": "object",
                "properties": {},
                "required": [],
            },
        },
    },
    {
        "type": "function",
        "function": {
            "name": "get_hot_sectors",
            "description": "获取当前热门板块排名，包括板块涨跌幅、资金流入、涨停数、热度评分",
            "parameters": {
                "type": "object",
                "properties": {
                    "limit": {
                        "type": "integer",
                        "description": "返回板块数量，默认 10",
                    },
                },
                "required": [],
            },
        },
    },
    {
        "type": "function",
        "function": {
            "name": "get_latest_recommendations",
            "description": "获取最新的股票推荐列表，包括综合评分、各维度得分、买卖信号、参考价格",
            "parameters": {
                "type": "object",
                "properties": {},
                "required": [],
            },
        },
    },
    {
        "type": "function",
        "function": {
            "name": "get_user_watchlist_snapshot",
            "description": "获取当前用户自选股概览，包括分组、最新结论、建议、触发条件和摘要",
            "parameters": {
                "type": "object",
                "properties": {},
                "required": [],
            },
        },
    },
    {
        "type": "function",
        "function": {
            "name": "get_stock_kline",
            "description": "获取个股K线数据（含技术指标 MA/MACD/RSI/BOLL），用于分析个股走势",
            "parameters": {
                "type": "object",
                "properties": {
                    "ts_code": {
                        "type": "string",
                        "description": "股票代码，如 '000001.SZ'",
                    },
                    "days": {
                        "type": "integer",
                        "description": "获取天数，默认 60",
                    },
                },
                "required": ["ts_code"],
            },
        },
    },
    {
        "type": "function",
        "function": {
            "name": "get_stock_capital_flow",
            "description": "获取个股资金流向数据，包括主力净流入金额",
            "parameters": {
                "type": "object",
                "properties": {
                    "ts_code": {
                        "type": "string",
                        "description": "股票代码，如 '000001.SZ'",
                    },
                    "days": {
                        "type": "integer",
                        "description": "获取天数，默认 10",
                    },
                },
                "required": ["ts_code"],
            },
        },
    },
    {
        "type": "function",
        "function": {
            "name": "search_stock",
            "description": "通过名称或代码模糊搜索股票",
            "parameters": {
                "type": "object",
                "properties": {
                    "keyword": {
                        "type": "string",
                        "description": "搜索关键词（股票名称或代码）",
                    },
                },
                "required": ["keyword"],
            },
        },
    },
    {
        "type": "function",
        "function": {
            "name": "get_stock_technical_signal",
            "description": "获取个股技术面信号详情，包括均线排列、MACD状态、RSI、布林带位置、量价关系、支撑压力位、位置安全评估",
            "parameters": {
                "type": "object",
                "properties": {
                    "ts_code": {
                        "type": "string",
                        "description": "股票代码，如 '000001.SZ'",
                    },
                },
                "required": ["ts_code"],
            },
        },
    },
    {
        "type": "function",
        "function": {
            "name": "diagnose_stock",
            "description": "对单只 A 股做系统化个股会诊，优先分析资金流向、主线板块、量价关系、价格行为和位置边界，技术指标只做节奏与风控备注，输出结论、触发条件、失效条件和风险边界",
            "parameters": {
                "type": "object",
                "properties": {
                    "ts_code": {
                        "type": "string",
                        "description": "股票代码，如 '000001.SZ'。如果用户只给名称，应先调用 search_stock 找到 ts_code",
                    },
                    "mode": {
                        "type": "string",
                        "enum": ["entry", "holding", "review", "tracking"],
                        "description": "诊断模式：entry 建仓前；holding 持仓复核；review 回撤复盘；tracking 继续跟踪。默认 entry",
                    },
                },
                "required": ["ts_code"],
            },
        },
    },
    {
        "type": "function",
        "function": {
            "name": "get_sector_performance",
            "description": "获取板块近期涨跌表现、资金流入、涨停数等综合数据，用于判断板块强弱和轮动方向",
            "parameters": {
                "type": "object",
                "properties": {
                    "sector_name": {
                        "type": "string",
                        "description": "板块名称关键词，如 '半导体'、'新能源'",
                    },
                },
                "required": ["sector_name"],
            },
        },
    },
    {
        "type": "function",
        "function": {
            "name": "get_realtime_indices",
            "description": "获取指数实时行情数据，包括上证指数、深证成指、创业板指的实时涨跌幅和成交数据。盘中时段返回实时数据，盘后返回当日收盘数据",
            "parameters": {
                "type": "object",
                "properties": {},
                "required": [],
            },
        },
    },
]