"""金融 NL2SQL 业务 Agent。"""

from typing import Any, cast

from app.core.contracts import AgentDefinition, AgentRequest, CoreResult, RequestContext
from app.service.agent.base import BaseAgent

TOOL_NAME = "query_financial_data"
INTENTS = ("financial_query", "general")


class FinancialNL2SQLAgent(BaseAgent):
    """把通用金融问题交给公共金融只读查询工具处理。"""

    definition = AgentDefinition(
        agent_type="financial_nl2sql",
        version="1.0.0",
        allowed_roles=("advisor", "operator", "admin", "super_admin"),
        allowed_portals=("api",),
        allowed_tools=(TOOL_NAME,),
        supported_intents=INTENTS,
        # 查询工具自身带有确定性规划与澄清机制，不需要依赖模型意图分类端点。
        requires_model_intent_classification=False,
    )

    async def handle(self, request: AgentRequest, context: RequestContext) -> CoreResult:
        result = cast(dict[str, Any], await self.call_tool(
            TOOL_NAME,
            {"question": request.message, "dry_run": False, "limit": 50},
            intent="financial_query",
            context=context,
        ))
        data = result.get("data")
        return CoreResult(
            text=str(result.get("message", "金融查询未返回结果")),
            data=data if isinstance(data, dict) else {},
            sql=result.get("sql") if isinstance(result.get("sql"), str) else None,
        )