mirror of
https://github.com/HKUDS/nanobot.git
synced 2026-04-06 19:23:39 +00:00
e7798a28ee
Refactor Tool methods and type handling; introduce JSON Schema support for tool parameters (schema module, validation tests). Made-with: Cursor
57 lines
1.9 KiB
Python
57 lines
1.9 KiB
Python
"""Spawn tool for creating background subagents."""
|
|
|
|
from typing import TYPE_CHECKING, Any
|
|
|
|
from nanobot.agent.tools.base import Tool, tool_parameters
|
|
from nanobot.agent.tools.schema import StringSchema, tool_parameters_schema
|
|
|
|
if TYPE_CHECKING:
|
|
from nanobot.agent.subagent import SubagentManager
|
|
|
|
|
|
@tool_parameters(
|
|
tool_parameters_schema(
|
|
task=StringSchema("The task for the subagent to complete"),
|
|
label=StringSchema("Optional short label for the task (for display)"),
|
|
required=["task"],
|
|
)
|
|
)
|
|
class SpawnTool(Tool):
|
|
"""Tool to spawn a subagent for background task execution."""
|
|
|
|
def __init__(self, manager: "SubagentManager"):
|
|
self._manager = manager
|
|
self._origin_channel = "cli"
|
|
self._origin_chat_id = "direct"
|
|
self._session_key = "cli:direct"
|
|
|
|
def set_context(self, channel: str, chat_id: str) -> None:
|
|
"""Set the origin context for subagent announcements."""
|
|
self._origin_channel = channel
|
|
self._origin_chat_id = chat_id
|
|
self._session_key = f"{channel}:{chat_id}"
|
|
|
|
@property
|
|
def name(self) -> str:
|
|
return "spawn"
|
|
|
|
@property
|
|
def description(self) -> str:
|
|
return (
|
|
"Spawn a subagent to handle a task in the background. "
|
|
"Use this for complex or time-consuming tasks that can run independently. "
|
|
"The subagent will complete the task and report back when done. "
|
|
"For deliverables or existing projects, inspect the workspace first "
|
|
"and use a dedicated subdirectory when helpful."
|
|
)
|
|
|
|
async def execute(self, task: str, label: str | None = None, **kwargs: Any) -> str:
|
|
"""Spawn a subagent to execute the given task."""
|
|
return await self._manager.spawn(
|
|
task=task,
|
|
label=label,
|
|
origin_channel=self._origin_channel,
|
|
origin_chat_id=self._origin_chat_id,
|
|
session_key=self._session_key,
|
|
)
|