X-Git-Url: https://git.giacomolli.uk/onechat/blobdiff_plain/448c931a776c3e3f65ca2eab11da0c062d597818..bd7da3557dc7a17020931f5a49ed5007fa81a645:/subagent_run.go?ds=inline diff --git a/subagent_run.go b/subagent_run.go index 24e4f82..e96975c 100644 --- a/subagent_run.go +++ b/subagent_run.go @@ -132,7 +132,7 @@ func (r *SubagentRuns) Spawn(ctx context.Context, tasks []string, effort *string view, err := r.layout.Split(r.memory.ViewLines(nil)) if err != nil { - return "", err + return "", fmt.Errorf("split worker view: %w", err) } names := make([]string, 0, len(tasks)) @@ -143,7 +143,7 @@ func (r *SubagentRuns) Spawn(ctx context.Context, tasks []string, effort *string } if err != nil { - return "", err + return "", fmt.Errorf("create worker: %w", err) } names = append(names, name) @@ -157,19 +157,19 @@ func (r *SubagentRuns) Spawn(ctx context.Context, tasks []string, effort *string func (r *SubagentRuns) create(ctx context.Context, note, task string, view []string, effort string) (string, error) { subagent, err := r.subagents.Create(ctx) if err != nil { - return "", err + return "", fmt.Errorf("create worker log: %w", err) } for _, message := range [][2]string{{"note", note}, {"user", task}} { _, err = subagent.Append(message[0], message[1]) if err != nil { - return "", err + return "", fmt.Errorf("append worker message: %w", err) } } err = subagent.Flush(ctx) if err != nil { - return "", err + return "", fmt.Errorf("flush worker log: %w", err) } err = os.MkdirAll(subagent.WorkDir(), 0o700) @@ -210,7 +210,11 @@ func (r *SubagentRuns) start(ctx context.Context, run *SubagentRun) error { r.mutex.Unlock() run.cancel() - return err + if err != nil { + return fmt.Errorf("save running workers: %w", err) + } + + return nil } r.workers.Add(1) @@ -329,12 +333,12 @@ func (r *SubagentRuns) workPath() string { func (r *SubagentRuns) chain(ctx context.Context, run *SubagentRun) (string, error) { instructions, err := r.agent.SubagentPrompt() if err != nil { - return "", err + return "", fmt.Errorf("read worker instructions: %w", err) } log, err := run.subagent.Show() if err != nil { - return "", err + return "", fmt.Errorf("read worker log: %w", err) } input := make([]any, 0, len(run.view)+2) @@ -410,13 +414,17 @@ func (r *SubagentRuns) Stop(name string) (bool, error) { func (r *SubagentRuns) Resume(ctx context.Context) error { names, err := r.load() if err != nil || len(names) == 0 { - return err + if err != nil { + return fmt.Errorf("load interrupted workers: %w", err) + } + + return nil } text := "The daemon restarted, cutting " + strings.Join(names, ", ") + "; nothing was lost: go on." err = r.note(text) if err != nil { - return err + return fmt.Errorf("record interrupted workers: %w", err) } for _, name := range names { @@ -468,7 +476,7 @@ func (r *SubagentRuns) Tell(ctx context.Context, name, message string) (string, told, err := run.subagent.Append("user", message) if err != nil { - return "", err + return "", fmt.Errorf("append running worker message: %w", err) } run.pending = append(run.pending, told) @@ -485,12 +493,12 @@ func (r *SubagentRuns) Tell(ctx context.Context, name, message string) (string, subagent, err := r.subagents.Get(ctx, name) if err != nil { - return "", err + return "", fmt.Errorf("open worker log: %w", err) } effort, err := r.effort(subagent) if err != nil { - return "", err + return "", fmt.Errorf("read worker effort: %w", err) } err = r.memory.Settle(ctx) @@ -500,7 +508,7 @@ func (r *SubagentRuns) Tell(ctx context.Context, name, message string) (string, view, err := r.layout.Split(r.memory.ViewLines(nil)) if err != nil { - return "", err + return "", fmt.Errorf("split worker view: %w", err) } _, err = subagent.Append("user", message) @@ -509,7 +517,7 @@ func (r *SubagentRuns) Tell(ctx context.Context, name, message string) (string, } if err != nil { - return "", err + return "", fmt.Errorf("save worker message: %w", err) } err = r.start(ctx, &SubagentRun{ @@ -518,7 +526,7 @@ func (r *SubagentRuns) Tell(ctx context.Context, name, message string) (string, effort: effort, }) if err != nil { - return "", err + return "", fmt.Errorf("start told worker: %w", err) } return "Sent. " + name + "'s reply will reach you as a message.", nil @@ -529,7 +537,7 @@ func (r *SubagentRuns) Tell(ctx context.Context, name, message string) (string, func (r *SubagentRuns) effort(subagent *Subagent) (string, error) { note, err := subagent.First() if err != nil { - return "", err + return "", fmt.Errorf("read worker note: %w", err) } effort, own := strings.CutPrefix(note.Text, subagentNote+subagentNoteEffort)