Keep each schedule with its owning agent

[onechat] / agent.go

diff --git a/agent.go b/agent.go
index 0d97cdf4af176d51fcde931e38d7288f003be1ca..274806be60c2f5d772f5d4d38de1693ef4f1bb76 100644 (file)
--- a/agent.go
+++ b/agent.go
@@ -10,6 +10,7 @@ import (
        "io"
        "os"
        "path/filepath"
+       "slices"
        "strings"
 )
 
@@ -96,9 +97,85 @@ func (a *Agent) Prompt() (string, error) {
                return "", fmt.Errorf("find instructions path: %w", err)
        }
 
+       if a.config.Parent != "" {
+               return promptChild(a.config.Name, a.config.Parent, a.config.Responsibility, path, instructions), nil
+       }
+
        return prompt(a.config.Name, path, instructions), nil
 }
 
+// Tools returns the tools shared by the turns and compactions of this agent. A
+// child has no WhatsApp 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 tools
+}
+
+// Children reads the direct children in directory order. Each child needs an
+// existing config with its directory name and this agent as parent. Children
+// cannot have children. A missing children directory means no children.
+func (a *Agent) Children() ([]*Agent, error) {
+       dir := filepath.Join(a.dir, "children")
+       entries, err := os.ReadDir(dir)
+       if errors.Is(err, os.ErrNotExist) {
+               return nil, nil
+       }
+
+       if err != nil {
+               return nil, fmt.Errorf("read children: %w", err)
+       }
+
+       if a.config.Parent != "" && len(entries) != 0 {
+               return nil, errors.New("children cannot have children")
+       }
+
+       var children []*Agent
+
+       keys := map[string]bool{a.CacheKey(): true}
+
+       for _, entry := range entries {
+               name := entry.Name()
+               if !entry.IsDir() {
+                       return nil, fmt.Errorf("child %q must be a directory", name)
+               }
+
+               path := filepath.Join(dir, name)
+               _, err = os.Stat(filepath.Join(path, "config.json"))
+               if err != nil {
+                       return nil, fmt.Errorf("child %q config: %w", name, err)
+               }
+
+               child, err := openAgent(path)
+               if err != nil {
+                       return nil, fmt.Errorf("open child %q: %w", name, err)
+               }
+
+               if child.config.Name != name || child.config.Parent != a.config.Name {
+                       return nil, fmt.Errorf("child %q needs name %q and parent %q", name, name, a.config.Name)
+               }
+
+               if child.config.Name == a.config.Name || keys[child.CacheKey()] {
+                       return nil, fmt.Errorf("child %q must have its own name and ID", name)
+               }
+
+               keys[child.CacheKey()] = true
+               _, err = child.Children()
+               if err != nil {
+                       return nil, fmt.Errorf("read child %q: %w", name, err)
+               }
+
+               children = append(children, child)
+       }
+
+       return children, 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) {
@@ -125,13 +202,15 @@ func (a *Agent) instructions() (string, error) {
 // 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.
+// 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"`
+       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.
@@ -149,6 +228,14 @@ func (c AgentConfig) validate() error {
                return errors.New("agent needs a model and an effort")
        }
 
+       if (strings.TrimSpace(c.Responsibility) == "") != (c.Parent == "") {
+               return errors.New("a child needs both responsibility and parent")
+       }
+
+       if strings.ContainsAny(c.Parent, "\r\n") {
+               return errors.New("parent must be one line of text")
+       }
+
        if !filepath.IsAbs(c.Workdir) {
                return fmt.Errorf("workdir %q is not an absolute path", c.Workdir)
        }