mirror of
https://github.com/HKUDS/nanobot.git
synced 2026-05-19 16:12:30 +00:00
fe90edd71f
GlobTool is redundant — GrepTool already supports glob-based file filtering via its `glob` parameter, making a standalone glob-only tool unnecessary. Removing it simplifies the tool surface and reduces LLM confusion between glob and grep.
54 lines
1.7 KiB
Python
54 lines
1.7 KiB
Python
"""Tests for SubagentManager."""
|
|
|
|
from pathlib import Path
|
|
from unittest.mock import MagicMock
|
|
|
|
import pytest
|
|
|
|
from nanobot.agent.subagent import SubagentManager
|
|
from nanobot.bus.queue import MessageBus
|
|
from nanobot.providers.base import LLMProvider
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_subagent_uses_tool_loader():
|
|
"""Verify subagent registers tools via ToolLoader, not hard-coded imports."""
|
|
provider = MagicMock(spec=LLMProvider)
|
|
provider.get_default_model.return_value = "test"
|
|
sm = SubagentManager(
|
|
provider=provider,
|
|
workspace=Path("/tmp"),
|
|
bus=MessageBus(),
|
|
model="test",
|
|
max_tool_result_chars=16_000,
|
|
)
|
|
tools = sm._build_tools()
|
|
assert tools.has("read_file")
|
|
assert tools.has("write_file")
|
|
assert not tools.has("message")
|
|
assert not tools.has("spawn")
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_subagent_build_tools_isolates_file_read_state(tmp_path):
|
|
"""Each spawned subagent needs a fresh file-state cache."""
|
|
(tmp_path / "note.txt").write_text("hello\n", encoding="utf-8")
|
|
provider = MagicMock(spec=LLMProvider)
|
|
provider.get_default_model.return_value = "test"
|
|
sm = SubagentManager(
|
|
provider=provider,
|
|
workspace=tmp_path,
|
|
bus=MessageBus(),
|
|
model="test",
|
|
max_tool_result_chars=16_000,
|
|
)
|
|
|
|
first_read = sm._build_tools().get("read_file")
|
|
second_read = sm._build_tools().get("read_file")
|
|
|
|
assert first_read is not second_read
|
|
assert (await first_read.execute(path="note.txt")).startswith("1| hello")
|
|
second_result = await second_read.execute(path="note.txt")
|
|
assert second_result.startswith("1| hello")
|
|
assert "File unchanged" not in second_result
|