1package agent
 2
 3import (
 4	"context"
 5	_ "embed"
 6	"errors"
 7
 8	"charm.land/fantasy"
 9
10	"github.com/charmbracelet/crush/internal/agent/prompt"
11	"github.com/charmbracelet/crush/internal/agent/tools"
12	"github.com/charmbracelet/crush/internal/config"
13)
14
15//go:embed templates/agent_tool.md
16var agentToolDescription []byte
17
18type AgentParams struct {
19	Prompt string `json:"prompt" description:"The task for the agent to perform"`
20}
21
22const (
23	AgentToolName = "agent"
24)
25
26func (c *coordinator) agentTool(ctx context.Context) (fantasy.AgentTool, error) {
27	agentCfg, ok := c.cfg.Config().Agents[config.AgentTask]
28	if !ok {
29		return nil, errors.New("task agent not configured")
30	}
31	prompt, err := taskPrompt(prompt.WithWorkingDir(c.cfg.WorkingDir()))
32	if err != nil {
33		return nil, err
34	}
35
36	agent, err := c.buildAgent(ctx, prompt, agentCfg, true)
37	if err != nil {
38		return nil, err
39	}
40	return fantasy.NewParallelAgentTool(
41		AgentToolName,
42		string(agentToolDescription),
43		func(ctx context.Context, params AgentParams, call fantasy.ToolCall) (fantasy.ToolResponse, error) {
44			if params.Prompt == "" {
45				return fantasy.NewTextErrorResponse("prompt is required"), nil
46			}
47
48			sessionID := tools.GetSessionFromContext(ctx)
49			if sessionID == "" {
50				return fantasy.ToolResponse{}, errors.New("session id missing from context")
51			}
52
53			agentMessageID := tools.GetMessageFromContext(ctx)
54			if agentMessageID == "" {
55				return fantasy.ToolResponse{}, errors.New("agent message id missing from context")
56			}
57
58			return c.runSubAgent(ctx, subAgentParams{
59				Agent:          agent,
60				SessionID:      sessionID,
61				AgentMessageID: agentMessageID,
62				ToolCallID:     call.ID,
63				Prompt:         params.Prompt,
64				SessionTitle:   "New Agent Session",
65			})
66		}), nil
67}