584 lines
21 KiB
Python
584 lines
21 KiB
Python
"""Marketplace E2E 集成测试 - 多 Agent 市场架构端到端流程"""
|
||
|
||
from __future__ import annotations
|
||
|
||
import pytest
|
||
from unittest.mock import AsyncMock, MagicMock
|
||
|
||
from agentkit.chat.skill_routing import CostAwareRouter, SkillRoutingResult
|
||
from agentkit.org.context import OrganizationContext, AgentProfile
|
||
from agentkit.quality.alignment import AlignmentGuard, AlignmentConfig, CascadeAlert, ConstraintInjector
|
||
from agentkit.marketplace.auction import AuctionHouse, Bid, AuctionResult
|
||
from agentkit.marketplace.wealth import WealthTracker
|
||
|
||
|
||
# ---------------------------------------------------------------------------
|
||
# Fixtures
|
||
# ---------------------------------------------------------------------------
|
||
|
||
|
||
@pytest.fixture
|
||
def mock_llm_gateway():
|
||
"""Mock LLMGateway for CostAwareRouter Layer 1 classification."""
|
||
gw = AsyncMock()
|
||
response = MagicMock()
|
||
response.content = '{"complexity": 0.5}'
|
||
gw.chat = AsyncMock(return_value=response)
|
||
return gw
|
||
|
||
|
||
@pytest.fixture
|
||
def mock_skill_registry():
|
||
"""Mock SkillRegistry with no skills by default."""
|
||
registry = MagicMock()
|
||
registry.list_skills.return_value = []
|
||
registry.get.side_effect = KeyError("not found")
|
||
return registry
|
||
|
||
|
||
@pytest.fixture
|
||
def mock_intent_router():
|
||
"""Mock IntentRouter that returns no match by default."""
|
||
router = AsyncMock()
|
||
router.route = AsyncMock(return_value=None)
|
||
return router
|
||
|
||
|
||
# ---------------------------------------------------------------------------
|
||
# Test 1: Simple chat routes to default agent (Layer 0)
|
||
# ---------------------------------------------------------------------------
|
||
|
||
|
||
class TestSimpleChatRoutesToDefault:
|
||
"""简单对话走 Layer 0 规则匹配，路由到默认 Agent"""
|
||
|
||
@pytest.mark.asyncio
|
||
async def test_greeting_routes_to_default(self, mock_skill_registry, mock_intent_router):
|
||
router = CostAwareRouter(llm_gateway=None, org_context=None)
|
||
result = await router.route(
|
||
content="你好",
|
||
skill_registry=mock_skill_registry,
|
||
intent_router=mock_intent_router,
|
||
default_tools=[],
|
||
default_system_prompt="You are helpful",
|
||
default_model="default",
|
||
default_agent_name="default",
|
||
)
|
||
assert result.match_method == "greeting"
|
||
assert result.agent_name == "default"
|
||
assert result.complexity == 0.0
|
||
assert result.matched is False
|
||
|
||
@pytest.mark.asyncio
|
||
async def test_chat_mode_routes_to_default(self, mock_skill_registry, mock_intent_router):
|
||
router = CostAwareRouter(llm_gateway=None, org_context=None)
|
||
result = await router.route(
|
||
content="谢谢",
|
||
skill_registry=mock_skill_registry,
|
||
intent_router=mock_intent_router,
|
||
default_tools=[],
|
||
default_system_prompt="You are helpful",
|
||
default_model="default",
|
||
default_agent_name="default",
|
||
)
|
||
assert result.match_method == "chat_mode"
|
||
assert result.agent_name == "default"
|
||
assert result.complexity == 0.0
|
||
|
||
|
||
# ---------------------------------------------------------------------------
|
||
# Test 2: Complex task routes via capability matching
|
||
# ---------------------------------------------------------------------------
|
||
|
||
|
||
class TestCapabilityMatching:
|
||
"""高复杂度任务通过 OrganizationContext 能力匹配路由"""
|
||
|
||
@pytest.mark.asyncio
|
||
async def test_complex_task_routes_via_capability(self, mock_llm_gateway, mock_skill_registry, mock_intent_router):
|
||
# Set up LLM to return high complexity
|
||
high_response = MagicMock()
|
||
high_response.content = '{"complexity": 0.9}'
|
||
mock_llm_gateway.chat = AsyncMock(return_value=high_response)
|
||
|
||
# Set up org_context with a capable agent
|
||
org_context = OrganizationContext()
|
||
org_context.register_agent(AgentProfile(
|
||
name="research_agent",
|
||
agent_type="react",
|
||
capabilities=["research", "analysis"],
|
||
skills=["research"],
|
||
))
|
||
|
||
# Mock find_best_agent to return the research agent
|
||
org_context.find_best_agent = MagicMock(
|
||
return_value=org_context.get_agent_profile("research_agent")
|
||
)
|
||
|
||
router = CostAwareRouter(
|
||
llm_gateway=mock_llm_gateway,
|
||
org_context=org_context,
|
||
)
|
||
result = await router.route(
|
||
content="请对市场趋势进行深度分析并给出投资建议",
|
||
skill_registry=mock_skill_registry,
|
||
intent_router=mock_intent_router,
|
||
default_tools=[],
|
||
default_system_prompt="You are helpful",
|
||
default_model="default",
|
||
default_agent_name="default",
|
||
)
|
||
assert result.matched is True
|
||
assert result.match_method == "capability"
|
||
assert result.agent_name == "research_agent"
|
||
assert result.complexity >= 0.7
|
||
|
||
|
||
# ---------------------------------------------------------------------------
|
||
# Test 3: Alignment guard detects cascade risk
|
||
# ---------------------------------------------------------------------------
|
||
|
||
|
||
class TestAlignmentCascadeDetection:
|
||
"""AlignmentGuard 检测级联故障风险"""
|
||
|
||
def test_cascade_alert_on_excessive_interactions(self):
|
||
config = AlignmentConfig(cascade_max_interactions=3)
|
||
guard = AlignmentGuard(config=config)
|
||
|
||
# Record interactions below threshold
|
||
for _ in range(3):
|
||
alert = guard.record_interaction("session-1")
|
||
assert alert is None
|
||
|
||
# Next interaction should trigger alert
|
||
alert = guard.record_interaction("session-1")
|
||
assert alert is not None
|
||
assert isinstance(alert, CascadeAlert)
|
||
assert alert.alert_type == "interaction_limit"
|
||
assert alert.current_value == 4
|
||
assert alert.threshold == 3
|
||
|
||
def test_cascade_alert_on_loop_depth(self):
|
||
config = AlignmentConfig(cascade_max_depth=2)
|
||
guard = AlignmentGuard(config=config)
|
||
|
||
# Depth within threshold
|
||
alert = guard.record_loop_depth("session-1", 2)
|
||
assert alert is None
|
||
|
||
# Depth exceeds threshold
|
||
alert = guard.record_loop_depth("session-1", 3)
|
||
assert alert is not None
|
||
assert alert.alert_type == "loop_depth"
|
||
assert alert.current_value == 3
|
||
assert alert.threshold == 2
|
||
|
||
def test_reset_session_clears_counts(self):
|
||
config = AlignmentConfig(cascade_max_interactions=2)
|
||
guard = AlignmentGuard(config=config)
|
||
|
||
guard.record_interaction("session-1")
|
||
guard.record_interaction("session-1")
|
||
guard.record_interaction("session-1") # triggers alert
|
||
assert guard.get_interaction_count("session-1") == 3
|
||
|
||
guard.reset_session("session-1")
|
||
assert guard.get_interaction_count("session-1") == 0
|
||
|
||
|
||
# ---------------------------------------------------------------------------
|
||
# Test 4: Transparency TRACE mode returns execution trace
|
||
# ---------------------------------------------------------------------------
|
||
|
||
|
||
class TestTransparencyTraceMode:
|
||
"""TRACE 透明度模式返回执行追踪"""
|
||
|
||
@pytest.mark.asyncio
|
||
async def test_trace_mode_populates_execution_trace(self, mock_skill_registry, mock_intent_router):
|
||
router = CostAwareRouter(llm_gateway=None, org_context=None)
|
||
result = await router.route(
|
||
content="你好",
|
||
skill_registry=mock_skill_registry,
|
||
intent_router=mock_intent_router,
|
||
default_tools=[],
|
||
default_system_prompt="You are helpful",
|
||
default_model="default",
|
||
default_agent_name="default",
|
||
transparency="TRACE",
|
||
)
|
||
assert result.transparency_level == "TRACE"
|
||
assert len(result.execution_trace) > 0
|
||
assert result.execution_trace[0]["layer"] == 0
|
||
|
||
@pytest.mark.asyncio
|
||
async def test_silent_mode_no_trace(self, mock_skill_registry, mock_intent_router):
|
||
router = CostAwareRouter(llm_gateway=None, org_context=None)
|
||
result = await router.route(
|
||
content="你好",
|
||
skill_registry=mock_skill_registry,
|
||
intent_router=mock_intent_router,
|
||
default_tools=[],
|
||
default_system_prompt="You are helpful",
|
||
default_model="default",
|
||
default_agent_name="default",
|
||
transparency="SILENT",
|
||
)
|
||
assert result.transparency_level == "SILENT"
|
||
assert result.execution_trace == []
|
||
|
||
|
||
# ---------------------------------------------------------------------------
|
||
# Test 5: Auction mode routes via auction
|
||
# ---------------------------------------------------------------------------
|
||
|
||
|
||
class TestAuctionMode:
|
||
"""拍卖模式通过 AuctionHouse 选择 Agent"""
|
||
|
||
@pytest.mark.asyncio
|
||
async def test_auction_selects_best_bidder(self):
|
||
wealth_tracker = WealthTracker(initial_wealth=100.0)
|
||
wealth_tracker.reward("agent_a", 50.0) # agent_a is richer
|
||
|
||
auction_house = AuctionHouse(wealth_tracker=wealth_tracker)
|
||
|
||
bids = [
|
||
Bid(
|
||
agent_name="agent_a",
|
||
architecture="react",
|
||
estimated_steps=3,
|
||
estimated_cost=0.5,
|
||
confidence=0.9,
|
||
payment_offer=1.0,
|
||
capabilities=["research"],
|
||
),
|
||
Bid(
|
||
agent_name="agent_b",
|
||
architecture="rewoo",
|
||
estimated_steps=5,
|
||
estimated_cost=0.8,
|
||
confidence=0.7,
|
||
payment_offer=0.5,
|
||
capabilities=["research"],
|
||
),
|
||
]
|
||
|
||
result = await auction_house.run_auction("research task", bids)
|
||
assert result.winner is not None
|
||
assert result.winner.agent_name == "agent_a"
|
||
assert result.total_bidders == 2
|
||
|
||
@pytest.mark.asyncio
|
||
async def test_auction_no_bidders(self):
|
||
auction_house = AuctionHouse()
|
||
result = await auction_house.run_auction("task", [])
|
||
assert result.winner is None
|
||
assert result.total_bidders == 0
|
||
|
||
@pytest.mark.asyncio
|
||
async def test_bankrupt_agent_excluded(self):
|
||
wealth_tracker = WealthTracker(initial_wealth=-150.0)
|
||
auction_house = AuctionHouse(wealth_tracker=wealth_tracker)
|
||
|
||
bids = [
|
||
Bid(
|
||
agent_name="bankrupt_agent",
|
||
architecture="react",
|
||
estimated_steps=1,
|
||
estimated_cost=0.1,
|
||
confidence=0.9,
|
||
payment_offer=1.0,
|
||
),
|
||
]
|
||
|
||
result = await auction_house.run_auction("task", bids)
|
||
assert result.winner is None
|
||
assert "bankrupt" in result.selection_reason.lower()
|
||
|
||
|
||
# ---------------------------------------------------------------------------
|
||
# Test 6: Constraint injection works end-to-end
|
||
# ---------------------------------------------------------------------------
|
||
|
||
|
||
class TestConstraintInjection:
|
||
"""约束注入端到端测试"""
|
||
|
||
def test_inject_constraints_into_input_data(self):
|
||
config = AlignmentConfig(constraints=["不得泄露用户隐私", "禁止生成有害内容"])
|
||
guard = AlignmentGuard(config=config)
|
||
|
||
input_data = {"content": "请帮我写一篇文章"}
|
||
injected = guard.inject_constraints(input_data)
|
||
|
||
assert "alignment_constraints" in injected
|
||
assert "不得泄露用户隐私" in injected["alignment_constraints"]
|
||
assert "禁止生成有害内容" in injected["alignment_constraints"]
|
||
# Original data preserved
|
||
assert injected["content"] == "请帮我写一篇文章"
|
||
|
||
def test_inject_does_not_mutate_original(self):
|
||
config = AlignmentConfig(constraints=["constraint_1"])
|
||
guard = AlignmentGuard(config=config)
|
||
|
||
input_data = {"key": "value"}
|
||
injected = guard.inject_constraints(input_data)
|
||
|
||
assert "alignment_constraints" not in input_data
|
||
assert "alignment_constraints" in injected
|
||
|
||
|
||
# ---------------------------------------------------------------------------
|
||
# Test 7: OrganizationContext builds from AgentPool
|
||
# ---------------------------------------------------------------------------
|
||
|
||
|
||
class TestOrganizationContextFromAgentPool:
|
||
"""OrganizationContext 从 AgentPool 构建"""
|
||
|
||
def test_build_from_agent_pool_with_skills(self):
|
||
# Mock AgentPool
|
||
agent_pool = MagicMock()
|
||
agent_pool.list_agents.return_value = [
|
||
{"name": "writer", "agent_type": "react"},
|
||
{"name": "analyst", "agent_type": "plan_exec"},
|
||
]
|
||
|
||
# Mock SkillRegistry — writer has a skill, analyst does not
|
||
skill_registry = MagicMock()
|
||
|
||
writer_skill = MagicMock()
|
||
writer_config = MagicMock()
|
||
writer_config.capabilities = [MagicMock(tag="writing"), MagicMock(tag="creative")]
|
||
writer_config.execution_mode = "react"
|
||
writer_config.llm = {"model": "gpt-4"}
|
||
writer_config.max_concurrency = 2
|
||
writer_skill.config = writer_config
|
||
|
||
def get_skill(name):
|
||
if name == "writer":
|
||
return writer_skill
|
||
raise KeyError(name)
|
||
|
||
skill_registry.get = MagicMock(side_effect=get_skill)
|
||
|
||
org_context = OrganizationContext.from_agent_pool(
|
||
agent_pool=agent_pool,
|
||
skill_registry=skill_registry,
|
||
)
|
||
|
||
profiles = org_context.list_agents()
|
||
assert len(profiles) == 2
|
||
|
||
writer_profile = org_context.get_agent_profile("writer")
|
||
assert writer_profile is not None
|
||
assert writer_profile.agent_type == "react"
|
||
assert "writing" in writer_profile.capabilities
|
||
assert "creative" in writer_profile.capabilities
|
||
assert writer_profile.model == "gpt-4"
|
||
assert writer_profile.max_concurrency == 2
|
||
|
||
analyst_profile = org_context.get_agent_profile("analyst")
|
||
assert analyst_profile is not None
|
||
assert analyst_profile.agent_type == "plan_exec"
|
||
# No skill found → default values
|
||
assert analyst_profile.capabilities == []
|
||
assert analyst_profile.model == "default"
|
||
|
||
def test_build_from_empty_agent_pool(self):
|
||
agent_pool = MagicMock()
|
||
agent_pool.list_agents.return_value = []
|
||
skill_registry = MagicMock()
|
||
|
||
org_context = OrganizationContext.from_agent_pool(
|
||
agent_pool=agent_pool,
|
||
skill_registry=skill_registry,
|
||
)
|
||
|
||
assert org_context.list_agents() == []
|
||
|
||
def test_find_best_agent_by_capability(self):
|
||
org_context = OrganizationContext()
|
||
org_context.register_agent(AgentProfile(
|
||
name="researcher",
|
||
agent_type="react",
|
||
capabilities=["research", "analysis"],
|
||
skills=["research"],
|
||
current_load=0,
|
||
))
|
||
org_context.register_agent(AgentProfile(
|
||
name="writer",
|
||
agent_type="react",
|
||
capabilities=["writing", "creative"],
|
||
skills=["writing"],
|
||
current_load=2,
|
||
))
|
||
|
||
# Find agent with research capability
|
||
best = org_context.find_best_agent(["research"])
|
||
assert best is not None
|
||
assert best.name == "researcher"
|
||
|
||
# Find agent with both research and analysis
|
||
best = org_context.find_best_agent(["research", "analysis"])
|
||
assert best is not None
|
||
assert best.name == "researcher"
|
||
|
||
# No agent with unknown capability
|
||
best = org_context.find_best_agent(["coding"])
|
||
assert best is None
|
||
|
||
|
||
# ---------------------------------------------------------------------------
|
||
# Test 8: Full pipeline: Chat → Router → Agent → AlignmentGuard
|
||
# ---------------------------------------------------------------------------
|
||
|
||
|
||
class TestFullPipeline:
|
||
"""完整流水线: 用户消息 → CostAwareRouter → 技能匹配 → 约束注入 → 对齐检查"""
|
||
|
||
@pytest.mark.asyncio
|
||
async def test_full_pipeline_greeting(self):
|
||
"""简单问候走完整流水线"""
|
||
# Setup
|
||
org_context = OrganizationContext()
|
||
alignment_config = AlignmentConfig(
|
||
constraints=["不得包含敏感信息"],
|
||
cascade_max_interactions=10,
|
||
)
|
||
guard = AlignmentGuard(config=alignment_config)
|
||
router = CostAwareRouter(llm_gateway=None, org_context=org_context)
|
||
|
||
mock_skill_registry = MagicMock()
|
||
mock_skill_registry.list_skills.return_value = []
|
||
mock_intent_router = AsyncMock()
|
||
|
||
# Step 1: Route the message
|
||
result = await router.route(
|
||
content="你好",
|
||
skill_registry=mock_skill_registry,
|
||
intent_router=mock_intent_router,
|
||
default_tools=[],
|
||
default_system_prompt="You are helpful",
|
||
default_model="default",
|
||
default_agent_name="default",
|
||
)
|
||
assert result.match_method == "greeting"
|
||
assert result.agent_name == "default"
|
||
|
||
# Step 2: Inject constraints
|
||
input_data = {"content": result.clean_content}
|
||
injected = guard.inject_constraints(input_data)
|
||
assert "alignment_constraints" in injected
|
||
|
||
# Step 3: Check alignment on simulated output
|
||
output = {"result": "你好！有什么我可以帮助你的吗？"}
|
||
check_result = await guard.check_output(output)
|
||
assert check_result.passed is True
|
||
|
||
# Step 4: Record interaction (no cascade)
|
||
alert = guard.record_interaction("session-1")
|
||
assert alert is None
|
||
|
||
@pytest.mark.asyncio
|
||
async def test_full_pipeline_with_constraint_violation(self):
|
||
"""输出违反约束时被检测到"""
|
||
alignment_config = AlignmentConfig(
|
||
constraints=["password", "secret_key"],
|
||
)
|
||
guard = AlignmentGuard(config=alignment_config)
|
||
|
||
# Output containing a constraint keyword
|
||
output = {"result": "Your password is 123456"}
|
||
check_result = await guard.check_output(output)
|
||
assert check_result.passed is False
|
||
assert len(check_result.violations) > 0
|
||
assert check_result.checked_by == "rule"
|
||
|
||
@pytest.mark.asyncio
|
||
async def test_full_pipeline_complex_task_with_alignment(self):
|
||
"""复杂任务走完整流水线：路由 → 能力匹配 → 约束注入 → 对齐检查"""
|
||
# Setup LLM gateway returning high complexity
|
||
mock_llm = AsyncMock()
|
||
high_response = MagicMock()
|
||
high_response.content = '{"complexity": 0.85}'
|
||
mock_llm.chat = AsyncMock(return_value=high_response)
|
||
|
||
# Setup org context with capable agent
|
||
org_context = OrganizationContext()
|
||
org_context.register_agent(AgentProfile(
|
||
name="analyst",
|
||
agent_type="react",
|
||
capabilities=["analysis", "market_research"],
|
||
skills=["market_analysis"],
|
||
current_load=0,
|
||
))
|
||
org_context.find_best_agent = MagicMock(
|
||
return_value=org_context.get_agent_profile("analyst")
|
||
)
|
||
|
||
alignment_config = AlignmentConfig(
|
||
constraints=["不得提供具体投资建议"],
|
||
cascade_max_interactions=5,
|
||
)
|
||
guard = AlignmentGuard(config=alignment_config, llm_gateway=mock_llm)
|
||
|
||
router = CostAwareRouter(
|
||
llm_gateway=mock_llm,
|
||
org_context=org_context,
|
||
)
|
||
|
||
mock_skill_registry = MagicMock()
|
||
mock_skill_registry.list_skills.return_value = []
|
||
mock_intent_router = AsyncMock()
|
||
|
||
# Step 1: Route complex task
|
||
result = await router.route(
|
||
content="请分析当前AI行业的市场趋势",
|
||
skill_registry=mock_skill_registry,
|
||
intent_router=mock_intent_router,
|
||
default_tools=[],
|
||
default_system_prompt="You are a market analyst",
|
||
default_model="default",
|
||
default_agent_name="default",
|
||
transparency="TRACE",
|
||
)
|
||
assert result.matched is True
|
||
assert result.match_method == "capability"
|
||
assert result.agent_name == "analyst"
|
||
assert result.complexity >= 0.7
|
||
assert len(result.execution_trace) > 0
|
||
|
||
# Step 2: Inject constraints
|
||
input_data = {"content": result.clean_content}
|
||
injected = guard.inject_constraints(input_data)
|
||
assert "alignment_constraints" in injected
|
||
|
||
# Step 3: Simulate agent output and check alignment
|
||
safe_output = {"result": "AI行业目前呈现稳步增长趋势，主要驱动力来自大模型技术的突破。"}
|
||
check_result = await guard.check_output(safe_output)
|
||
assert check_result.passed is True
|
||
|
||
# Step 4: Record interaction
|
||
alert = guard.record_interaction("session-complex")
|
||
assert alert is None # Under threshold
|
||
|
||
@pytest.mark.asyncio
|
||
async def test_full_pipeline_cascade_alert(self):
|
||
"""级联故障检测在完整流水线中触发"""
|
||
alignment_config = AlignmentConfig(
|
||
cascade_max_interactions=2,
|
||
)
|
||
guard = AlignmentGuard(config=alignment_config)
|
||
|
||
# Simulate multiple interactions
|
||
guard.record_interaction("session-cascade")
|
||
guard.record_interaction("session-cascade")
|
||
alert = guard.record_interaction("session-cascade")
|
||
|
||
assert alert is not None
|
||
assert alert.alert_type == "interaction_limit"
|
||
assert alert.current_value == 3
|