rony-llm-agent/pkg/agent/subagent.go

package agent

import (
	"context"

	"github.com/VictorVargas/rony-llm-agent/pkg/llm"
	"github.com/VictorVargas/rony-llm-agent/pkg/persona"
	"github.com/VictorVargas/rony-llm-agent/pkg/tools"
)

// SubAgent describes a specialized agent invocable via a delegate tool (see
// docs/phase2.md §5). The caller (the harness) is responsible for building
// Persona and Tools — the boundary is the same as for the main Loop: this
// package orchestrates, it doesn't decide personas or wire concrete tools.
type SubAgent struct {
	Name          string
	Description   string
	Persona       persona.Persona
	Tools         tools.Registry
	MaxIterations int
}

// Run executes the sub-agent's task to completion using llmClient and the
// given AGENTS.md content, and returns its final response.
func (s SubAgent) Run(ctx context.Context, llmClient llm.LLMClient, agentsMD string, task string) (Response, error) {
	cfg := Config{
		LLM:      llmClient,
		Persona:  s.Persona,
		Tools:    s.Tools,
		MaxIters: s.MaxIterations,
		AgentsMD: agentsMD,
	}
	if cfg.MaxIters == 0 {
		cfg.MaxIters = DefaultMaxIterations
	}
	return New(cfg).Run(ctx, llm.Message{Role: llm.RoleUser, Content: task})
}

// SubAgentRegistry looks up SubAgents by name for the delegate tool.
type SubAgentRegistry map[string]SubAgent

// Get returns the named sub-agent, if registered.
func (r SubAgentRegistry) Get(name string) (SubAgent, bool) {
	s, ok := r[name]
	return s, ok
}