zl-q
80ad5141a6
Visibility mask refactoring: - Replace dead UI_REALTIME bit with CONTEXT_ASSEMBLY (bit 1) - Remove visibility_consumer_bit from SystemAgentLLMConfig and system_agents.yaml - Simplify _resolve_user_message_visibility_mask: chat->UI_HISTORY|CONTEXT_ASSEMBLY, automation->0 - Simplify _resolve_stage_visibility_mask: memory->UI_HISTORY, router/worker->UI_HISTORY|CONTEXT_ASSEMBLY - Remove stage_visibility_bit_map from store.py Task queue renaming: - Replace default_broker/bulk_broker/critical_broker with worker_agent_broker/worker_automation_broker - Queue names: 'default'/'bulk'/'critical' -> 'agent'/'automation' - Rename run_command_task -> run_command_task_agent/run_command_task_automation - AgentService derives queue from runtime_mode: chat->agent, automation->automation Architecture cleanup: - Move context_service.py from runtime/ to agentscope/services/ - Add MemoryService in v1/memory/ following repository/service pattern - Move consumer_registry.py and pipeline_spec.py from schemas/agent to agentscope/schemas/ - Delete dead code: registry_builder.py, VisibilityBitRef - Delete superseded plan docs
37 lines
1.1 KiB
Python
37 lines
1.1 KiB
Python
from __future__ import annotations
|
|
|
|
import pytest
|
|
|
|
from core.agentscope.schemas.consumer_registry import (
|
|
AgentConsumerBinding,
|
|
ConsumerRegistry,
|
|
)
|
|
from core.agentscope.schemas.pipeline_spec import ExecutorKind, PipelineSpec, StageSpec
|
|
from schemas.agent.system_agent import AgentType
|
|
|
|
|
|
def test_consumer_registry_rejects_duplicate_bits() -> None:
|
|
with pytest.raises(ValueError, match="duplicate visibility bit"):
|
|
ConsumerRegistry(
|
|
bindings=[
|
|
AgentConsumerBinding(agent_type="router", bit=16),
|
|
AgentConsumerBinding(agent_type="worker", bit=16),
|
|
]
|
|
)
|
|
|
|
|
|
def test_pipeline_spec_requires_non_empty_stages() -> None:
|
|
with pytest.raises(ValueError, match="at least 1 item"):
|
|
PipelineSpec(mode="worker", stages=[])
|
|
|
|
|
|
def test_stage_spec_normalizes_stage_name() -> None:
|
|
spec = StageSpec(
|
|
stage_name=" Worker ",
|
|
agent_type=AgentType.WORKER,
|
|
executor_kind=ExecutorKind.REACT,
|
|
)
|
|
|
|
assert spec.stage_name == "worker"
|
|
assert spec.agent_type == AgentType.WORKER
|