41 lines
1.7 KiB
Python
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")
|