mirror of
https://github.com/sipeed/picoclaw.git
synced 2026-06-12 18:08:54 +00:00
xiaoen
484ef399f1
delegate(agent_id, task) hands off a task to a named agent and blocks until the result is ready. The target agent runs with its own config via the TargetAgentID mechanism in SubTurnConfig. Key behaviors: - Self-delegation explicitly rejected - Permission gated by subagents.allow_agents (D4) - Spawner errors preserve the underlying error via WithError - Nil result from spawner handled gracefully - Response attributed with target agent ID Ref: #2148
102 lines
2.7 KiB
Go
102 lines
2.7 KiB
Go
package tools
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"strings"
|
|
)
|
|
|
|
// DelegateTool delegates a task to a specific named agent and waits for
|
|
// the result. Unlike spawn (async, fire-and-forget) or subagent (sync but
|
|
// generic), delegate targets a named agent and runs the task using that
|
|
// agent's own workspace, model, and tools.
|
|
type DelegateTool struct {
|
|
spawner SubTurnSpawner
|
|
allowlistCheck func(targetAgentID string) bool
|
|
selfAgentID string
|
|
}
|
|
|
|
func NewDelegateTool() *DelegateTool {
|
|
return &DelegateTool{}
|
|
}
|
|
|
|
func (t *DelegateTool) SetSpawner(spawner SubTurnSpawner) {
|
|
t.spawner = spawner
|
|
}
|
|
|
|
func (t *DelegateTool) SetAllowlistChecker(check func(targetAgentID string) bool) {
|
|
t.allowlistCheck = check
|
|
}
|
|
|
|
func (t *DelegateTool) SetSelfAgentID(id string) {
|
|
t.selfAgentID = id
|
|
}
|
|
|
|
func (t *DelegateTool) Name() string {
|
|
return "delegate"
|
|
}
|
|
|
|
func (t *DelegateTool) Description() string {
|
|
return "Delegate a task to another agent and wait for the result. " +
|
|
"Use this when another agent is better suited to handle a specific task " +
|
|
"based on their capabilities. The target agent runs with its own workspace, " +
|
|
"model, and tools."
|
|
}
|
|
|
|
func (t *DelegateTool) Parameters() map[string]any {
|
|
return map[string]any{
|
|
"type": "object",
|
|
"properties": map[string]any{
|
|
"agent_id": map[string]any{
|
|
"type": "string",
|
|
"description": "The ID of the target agent to delegate the task to",
|
|
},
|
|
"task": map[string]any{
|
|
"type": "string",
|
|
"description": "Clear description of the task to delegate",
|
|
},
|
|
},
|
|
"required": []string{"agent_id", "task"},
|
|
}
|
|
}
|
|
|
|
func (t *DelegateTool) Execute(ctx context.Context, args map[string]any) *ToolResult {
|
|
agentID, _ := args["agent_id"].(string)
|
|
if strings.TrimSpace(agentID) == "" {
|
|
return ErrorResult("agent_id is required and must be a non-empty string")
|
|
}
|
|
|
|
task, _ := args["task"].(string)
|
|
if strings.TrimSpace(task) == "" {
|
|
return ErrorResult("task is required and must be a non-empty string")
|
|
}
|
|
|
|
if t.selfAgentID != "" && agentID == t.selfAgentID {
|
|
return ErrorResult("cannot delegate to self")
|
|
}
|
|
|
|
if t.allowlistCheck != nil && !t.allowlistCheck(agentID) {
|
|
return ErrorResult(fmt.Sprintf("not allowed to delegate to agent %q", agentID))
|
|
}
|
|
|
|
if t.spawner == nil {
|
|
return ErrorResult("delegate tool not configured")
|
|
}
|
|
|
|
result, err := t.spawner.SpawnSubTurn(ctx, SubTurnConfig{
|
|
TargetAgentID: agentID,
|
|
SystemPrompt: task,
|
|
Async: false,
|
|
})
|
|
if err != nil {
|
|
return ErrorResult(fmt.Sprintf("delegation to agent %q failed: %v", agentID, err)).WithError(err)
|
|
}
|
|
if result == nil {
|
|
return ErrorResult(fmt.Sprintf("delegation to agent %q returned no result", agentID))
|
|
}
|
|
|
|
result.ForLLM = fmt.Sprintf("[Response from agent %q]\n%s", agentID, result.ForLLM)
|
|
|
|
return result
|
|
}
|