51fecc67d9
Co-authored-by: Quentin Torroba <quentin.torroba@mistral.ai> Co-authored-by: Michel Thomazo <michel.thomazo@mistral.ai> Co-authored-by: Clément Drouin <clement.drouin@mistral.ai> Co-authored-by: Vincent Guilloux <vincent.guilloux@mistral.ai> Co-authored-by: Clément Siriex <clement.sirieix@mistral.ai> Co-authored-by: Kim-Adeline Miguel <kimadeline.miguel@mistral.ai> Co-authored-by: Nicolas Karolak <nicolas@karolak.fr>
80 lines
3 KiB
Python
80 lines
3 KiB
Python
from __future__ import annotations
|
|
|
|
import pytest
|
|
|
|
from tests.conftest import build_test_vibe_config
|
|
from vibe.core.agents.manager import AgentManager
|
|
from vibe.core.agents.models import BUILTIN_AGENTS, EXPLORE, AgentSafety, AgentType
|
|
|
|
|
|
class TestAgentProfile:
|
|
def test_explore_agent_is_subagent(self) -> None:
|
|
"""Test that EXPLORE agent has SUBAGENT type."""
|
|
assert EXPLORE.agent_type == AgentType.SUBAGENT
|
|
|
|
def test_explore_agent_has_safe_safety(self) -> None:
|
|
"""Test that EXPLORE agent has SAFE safety level."""
|
|
assert EXPLORE.safety == AgentSafety.SAFE
|
|
|
|
def test_explore_agent_has_enabled_tools(self) -> None:
|
|
"""Test that EXPLORE agent has expected enabled tools."""
|
|
enabled_tools = EXPLORE.overrides.get("enabled_tools", [])
|
|
assert "grep" in enabled_tools
|
|
assert "read_file" in enabled_tools
|
|
|
|
def test_builtin_agents_contains_explore(self) -> None:
|
|
"""Test that BUILTIN_AGENTS includes explore."""
|
|
assert "explore" in BUILTIN_AGENTS
|
|
assert BUILTIN_AGENTS["explore"] is EXPLORE
|
|
|
|
|
|
class TestAgentManager:
|
|
@pytest.fixture
|
|
def manager(self) -> AgentManager:
|
|
config = build_test_vibe_config(
|
|
include_project_context=False, include_prompt_detail=False
|
|
)
|
|
return AgentManager(lambda: config)
|
|
|
|
def test_get_subagents_returns_only_subagents(self, manager: AgentManager) -> None:
|
|
"""Test that only SUBAGENT type agents are returned."""
|
|
subagents = manager.get_subagents()
|
|
|
|
for agent in subagents:
|
|
assert agent.agent_type == AgentType.SUBAGENT
|
|
|
|
def test_get_subagents_includes_explore(self, manager: AgentManager) -> None:
|
|
"""Test that EXPLORE is included in subagents."""
|
|
subagents = manager.get_subagents()
|
|
names = [a.name for a in subagents]
|
|
|
|
assert "explore" in names
|
|
|
|
def test_get_subagents_excludes_agents(self, manager: AgentManager) -> None:
|
|
"""Test that AGENT type agents are not returned."""
|
|
subagents = manager.get_subagents()
|
|
names = [a.name for a in subagents]
|
|
|
|
# These are AGENT type
|
|
assert "default" not in names
|
|
assert "plan" not in names
|
|
assert "auto-approve" not in names
|
|
|
|
def test_get_builtin_agent(self, manager: AgentManager) -> None:
|
|
"""Test getting a builtin agent by name."""
|
|
agent = manager.get_agent("explore")
|
|
|
|
assert agent is EXPLORE
|
|
assert agent.agent_type == AgentType.SUBAGENT
|
|
|
|
def test_get_nonexistent_agent_raises(self, manager: AgentManager) -> None:
|
|
"""Test that getting a nonexistent agent raises ValueError."""
|
|
with pytest.raises(ValueError, match="not found"):
|
|
manager.get_agent("nonexistent-agent")
|
|
|
|
def test_get_default_agent(self, manager: AgentManager) -> None:
|
|
"""Test getting the default agent."""
|
|
agent = manager.get_agent("default")
|
|
|
|
assert agent.name == "default"
|
|
assert agent.agent_type == AgentType.AGENT
|