袁聪的最后一次完善更新

tests/contract/test_financial_nl2sql_agent_contract.py

@@ -0,0 +1,84 @@
+"""金融 NL2SQL Agent 的注册、声明和只读工具调用契约。"""
+
+from typing import Any, cast
+
+import pytest
+
+from app.core.contracts import (
+    AgentDefinition,
+    AgentRequest,
+    CoreResult,
+    RequestContext,
+    ResolvedAgentConfig,
+    ToolCallRecord,
+)
+from app.service.agent.base import BaseAgent
+from app.service.agent.factory import AgentFactory
+from app.service.agent.implementations.financial_nl2sql import FinancialNL2SQLAgent
+from app.service.tool_executor import ToolExecution
+
+
+class StubGovernance:
+    async def resolve(
+        self, definition: AgentDefinition, context: RequestContext
+    ) -> ResolvedAgentConfig:
+        del definition, context
+        return ResolvedAgentConfig(
+            config_version="test",
+            prompt_version="test",
+            model_endpoint="",
+            allowed_tools_by_intent={
+                "financial_query": ("query_financial_data",),
+                "general": ("query_financial_data",),
+            },
+        )
+
+    async def recall(self, context: RequestContext) -> tuple[Any, ...]:
+        del context
+        return ()
+
+    async def review(self, result: Any, context: RequestContext, config: ResolvedAgentConfig,
+                     memories: tuple[Any, ...], *, agent_type: str = "") -> Any:
+        del context, config, memories, agent_type
+        return result
+
+
+class StubExecutor:
+    def __init__(self) -> None:
+        self.calls: list[tuple[str, str, dict[str, Any]]] = []
+
+    async def execute(
+        self, *, name: str, arguments: dict[str, Any], intent: str,
+        configured_tools: dict[str, tuple[str, ...]], context: RequestContext,
+    ) -> ToolExecution:
+        del context
+        assert name in configured_tools[intent]
+        self.calls.append((name, intent, arguments))
+        return ToolExecution(
+            output={
+                "message": "查询成功",
+                "data": {"total": 1, "rows": [{"nav": "1.250000"}]},
+                "sql": "SELECT n.nav AS nav FROM fin_nav n WHERE 1=1 LIMIT 50",
+            },
+            record=ToolCallRecord(tool_name=name, status="succeeded"),
+        )
+
+
+@pytest.mark.asyncio
+async def test_financial_agent_is_registered_and_calls_read_only_tool() -> None:
+    executor = StubExecutor()
+    factory = AgentFactory(cast(Any, StubGovernance()), tool_executor=cast(Any, executor))
+    factory.register(FinancialNL2SQLAgent.definition,
+                     lambda _context: FinancialNL2SQLAgent(FinancialNL2SQLAgent.definition))
+    context = RequestContext(user_id="10001", trace_id="financial-contract", roles=("operator",),
+                             permissions=("agent:run", "financial:nl2sql:read"), data_scope="all")
+    agent = factory.create("financial_nl2sql", context)
+    assert isinstance(agent, BaseAgent)
+    await agent.resolve_config(context)
+    request = AgentRequest(agent_type="financial_nl2sql", message="查询15911最新净值",
+                           session_id="financial-session", idempotency_key="financial-key-123456")
+    result = await agent.handle(request, context)
+    assert result.text == "查询成功"
+    assert result.data == {"total": 1, "rows": [{"nav": "1.250000"}]}
+    assert result.sql == "SELECT n.nav AS nav FROM fin_nav n WHERE 1=1 LIMIT 50"
+    assert executor.calls[0][0:2] == ("query_financial_data", "financial_query")