39 lines
1.4 KiB
Python
39 lines
1.4 KiB
Python
"""金融 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,
|
|
)
|