mirror of
https://github.com/HKUDS/nanobot.git
synced 2026-05-19 16:12:30 +00:00
043f0e67f7
This commit implements a progressive refactoring of the tool system to support plugin discovery, scoped loading, and protocol-driven runtime context injection. Key changes: - Add Tool ABC metadata (tool_name, _scopes) and ToolContext dataclass for dependency injection. - Introduce ToolLoader with pkgutil-based builtin discovery and entry_points-based third-party plugin loading. - Add scope filtering (core/subagent/memory) so different contexts load appropriate tool sets. - Introduce ContextAware protocol and RequestContext dataclass to replace hardcoded per-tool context injection in AgentLoop. - Add RuntimeState / MutableRuntimeState protocols to decouple MyTool from AgentLoop. - Migrate all built-in tools to declare scopes and implement create()/enabled() hooks. - Migrate MessageTool, SpawnTool, CronTool, and MyTool to ContextAware. - Refactor AgentLoop to use ToolLoader and protocol-driven context injection. - Refactor SubagentManager to use ToolLoader(scope="subagent") with per-run FileStates isolation. - Register all built-in tools via pyproject.toml entry_points. - Add comprehensive tests for loader scopes, entry_points, ContextAware, subagent tools, and runtime state sync.
24 lines
613 B
Python
24 lines
613 B
Python
from __future__ import annotations
|
|
|
|
from nanobot.agent.tools.context import ContextAware, RequestContext
|
|
|
|
|
|
class _ContextTool:
|
|
def __init__(self):
|
|
self.last_ctx = None
|
|
|
|
def set_context(self, ctx: RequestContext) -> None:
|
|
self.last_ctx = ctx
|
|
|
|
|
|
def test_context_aware_sets_request_context():
|
|
tool = _ContextTool()
|
|
ctx = RequestContext(channel="test", chat_id="123", session_key="test:123")
|
|
tool.set_context(ctx)
|
|
assert tool.last_ctx.channel == "test"
|
|
|
|
|
|
def test_context_tool_is_instance_of_context_aware():
|
|
tool = _ContextTool()
|
|
assert isinstance(tool, ContextAware)
|