"path/filepath"
"slices"
"strings"
+
+ "golang.org/x/sys/unix"
)
const (
agentIDBytes = 16
)
+// Agent is one agent and its directory. The directory holds config.json, the
+// user's instructions in instructions.md, the chat, and the usage file.
+type Agent struct {
+ dir string
+ config AgentConfig
+}
+
+// AgentConfig holds the settings of the config.json of an agent. ID is random,
+// made once when the file is made, and never changes. The cache key comes from
+// the ID, so a rename keeps the prompt cache and two agents with one name do
+// not share a key. Workdir is the directory where the commands of the agent
+// run. Responsibility and Parent are set together for a child agent.
+type AgentConfig struct {
+ Name string `json:"name"`
+ ID string `json:"id"`
+ Model string `json:"model"`
+ Effort string `json:"effort"`
+ Workdir string `json:"workdir"`
+ Responsibility string `json:"responsibility,omitempty"`
+ Parent string `json:"parent,omitempty"`
+}
+
// openAgent reads the config.json of the agent in dir. When the file does not
// exist, it makes the directory and writes the file with the defaults and a new
// ID. A missing field of an existing file takes the default, but the ID must be
}, nil
}
-// Agent is one agent and its directory. The directory holds config.json, the
-// user's instructions in instructions.md, the chat, and the usage file.
-type Agent struct {
- dir string
- config AgentConfig
+// Images returns the shared data-directory image store, also for a child.
+func (a *Agent) Images() *ImageStore {
+ root := a.dir
+ if a.config.Parent != "" {
+ root = filepath.Dir(filepath.Dir(root))
+ }
+
+ return &ImageStore{dir: filepath.Join(filepath.Dir(root), imageDirName)}
+}
+
+// Voices returns the shared data-directory audio store, also for a child.
+func (a *Agent) Voices() *VoiceStore {
+ root := a.dir
+ if a.config.Parent != "" {
+ root = filepath.Dir(filepath.Dir(root))
+ }
+
+ return &VoiceStore{dir: filepath.Join(filepath.Dir(root), voiceDirName)}
}
// CacheKey returns the prompt cache key of every model call of the agent.
}
// Tools returns the tools shared by the turns and compactions of this agent. A
-// child has no WhatsApp tools.
+// child has no image output, WhatsApp, child creation, or route tools.
func (a *Agent) Tools() []ModelTool {
tools := promptTools()
if a.config.Parent != "" {
tools = slices.DeleteFunc(tools, func(tool ModelTool) bool {
- return strings.HasPrefix(tool.Name, "whatsapp_")
+ return strings.HasPrefix(tool.Name, "whatsapp_") || tool.Name == "image_show" || tool.Name == "agent_create" || tool.Name == "route" || tool.Name == "unroute" || tool.Name == "routes"
})
}
return children, nil
}
+// CreateChild writes a new direct child with its own ID and the model, effort,
+// and working directory of the root. Child and subagent names share a
+// namespace. The config is complete before an atomic rename publishes the
+// child. Preparation stays outside children, so an interrupted write cannot
+// stop startup.
+func (a *Agent) CreateChild(name, responsibility string, subagents *Subagents) (_ *Agent, err error) {
+ if a.config.Parent != "" {
+ return nil, errors.New("only the root can create children")
+ }
+
+ if !agentChildNameValid(name) {
+ return nil, errors.New("name must have 2 to 32 ASCII letters and start with a capital letter")
+ }
+
+ if name == a.config.Name {
+ return nil, errors.New("the name is already used by the root")
+ }
+
+ responsibility = strings.TrimSpace(responsibility)
+ if responsibility == "" {
+ return nil, errors.New("responsibility must not be empty")
+ }
+
+ err = subagents.Reserve(name)
+ if err != nil {
+ return nil, fmt.Errorf("reserve child name: %w", err)
+ }
+
+ published := false
+ defer func() {
+ if !published {
+ subagents.Release(name)
+ }
+ }()
+
+ staged, err := os.MkdirTemp(a.dir, ".child-*")
+ if err != nil {
+ return nil, fmt.Errorf("prepare child directory: %w", err)
+ }
+
+ defer func() {
+ cleanupErr := os.RemoveAll(staged)
+ if cleanupErr != nil {
+ err = errors.Join(err, fmt.Errorf("remove prepared child: %w", cleanupErr))
+ }
+ }()
+
+ config := a.config
+ config.Name = name
+ config.ID = ""
+ config.Parent = a.config.Name
+ config.Responsibility = responsibility
+ child, err := writeAgent(staged, config)
+ if err != nil {
+ return nil, fmt.Errorf("write child: %w", err)
+ }
+
+ dir := filepath.Join(a.dir, "children")
+ err = os.MkdirAll(dir, 0o700)
+ if err != nil {
+ return nil, fmt.Errorf("create children directory: %w", err)
+ }
+
+ dir = filepath.Join(dir, name)
+ err = unix.Renameat2(unix.AT_FDCWD, staged, unix.AT_FDCWD, dir, unix.RENAME_NOREPLACE)
+ if err != nil {
+ return nil, fmt.Errorf("publish child directory: %w", err)
+ }
+
+ published = true
+ child.dir = dir
+
+ return child, nil
+}
+
// SubagentPrompt returns the system prompt of a subagent of the agent, with the
// user's instructions as Prompt reads them.
func (a *Agent) SubagentPrompt() (string, error) {
return string(data), nil
}
-// AgentConfig holds the settings of the config.json of an agent. ID is random,
-// made once when the file is made, and never changes. The cache key comes from
-// the ID, so a rename keeps the prompt cache and two agents with one name do
-// not share a key. Workdir is the directory where the commands of the agent
-// run. Responsibility and Parent are set together for a child agent.
-type AgentConfig struct {
- Name string `json:"name"`
- ID string `json:"id"`
- Model string `json:"model"`
- Effort string `json:"effort"`
- Workdir string `json:"workdir"`
- Responsibility string `json:"responsibility,omitempty"`
- Parent string `json:"parent,omitempty"`
-}
-
// validate makes sure that each setting has a valid value.
func (c AgentConfig) validate() error {
if c.Name == "" || strings.ContainsAny(c.Name, "\r\n") {
return nil
}
+// agentChildNameValid accepts one word of 2 to 32 ASCII letters with an initial
+// capital letter. Child names describe their area.
+func agentChildNameValid(name string) bool {
+ if len(name) < 2 || len(name) > 32 || name[0] < 'A' || name[0] > 'Z' {
+ return false
+ }
+
+ for i := 1; i < len(name); i++ {
+ letter := name[i]
+ if (letter < 'A' || letter > 'Z') && (letter < 'a' || letter > 'z') {
+ return false
+ }
+ }
+
+ return true
+}
+
// createAgent writes the config.json of a new agent in dir with the defaults
// and a new ID.
func createAgent(dir string) (*Agent, error) {
return nil, err
}
+ return writeAgent(dir, config)
+}
+
+// writeAgent gives config a new ID and writes it in dir.
+func writeAgent(dir string, config AgentConfig) (*Agent, error) {
id := make([]byte, agentIDBytes)
- _, err = rand.Read(id)
+ _, err := rand.Read(id)
if err != nil {
return nil, fmt.Errorf("make agent ID: %w", err)
}
projects / onechat / blobdiff