Files
group_fqcd_jr/tests/unit/service/test_model_agent_entrypoint.py
T
张胜宇 e239eb778b docs: 品牌全量口径统一为「南方基金」+ 作废文档清理
1) 客服 Agent 四份交付文档 + 构建脚手架:品牌由包装占位 XX科技 / 旧名 南方财富
   统一为南方基金(热线 400-889-8899 / 官网 nffund.com),系统名改为「智能服务系统」;
   同步追加 §0.4 修订记录行,工程记录行保留原占位字面以支撑硬编码扫描验收。
2) 开发文档:清理 28 份已作废/残留文档(14 份移出归档 + 14 份仓库副本),
   新增《文档规整方案与开发前待决事项-2026-09-17》。
3) 客服agent 四份交付文档首次纳入本分支。
2026-09-17 15:15:22 +08:00

41 lines
1.7 KiB
Python

import pytest
from app.core.contracts import AgentDefinition, CoreResult, RequestContext
from app.core.errors import RecoverableAgentError
from app.service.agent.base import BaseAgent
from app.service.agent.factory import AgentFactory
from app.service.model_gateway import ModelDispatchService, ModelGenerationService
class DemoAgent(BaseAgent):
async def handle(self, request, context):
return CoreResult(text="ok")
class Gateway:
async def generate(self, *, endpoint_code, prompt, timeout_ms):
return "model answer"
@pytest.mark.asyncio
async def test_factory_injects_model_generation_entrypoint(governance) -> None:
definition = AgentDefinition(agent_type="demo", version="1", allowed_roles=("customer",),
allowed_portals=("api",))
factory = AgentFactory(governance, ModelGenerationService(ModelDispatchService(Gateway())))
factory.register(definition, lambda _: DemoAgent(definition))
context = RequestContext(user_id="1", trace_id="t", roles=("customer",),
permissions=("agent:run",))
agent = factory.create("demo", context)
endpoint = type("Endpoint", (), {"endpoint_code": "primary", "timeout_ms": 1000})()
result = await agent.generate_with_model([endpoint], "classify this")
assert result.text == "model answer"
@pytest.mark.asyncio
async def test_agent_without_model_dependency_fails_closed() -> None:
definition = AgentDefinition(agent_type="demo", version="1", allowed_roles=("customer",),
allowed_portals=("api",))
agent = DemoAgent(definition)
with pytest.raises(RecoverableAgentError, match="模型服务"):
await agent.generate_with_model([], "classify this")