chore: allow LLM to handle failed tool calls

ibuildthecloud · ibuildthecloud · commit 18a74f0fba35 · 2024-06-03T11:25:03.000-07:00
diff --git a/pkg/engine/cmd.go b/pkg/engine/cmd.go
@@ -80,6 +80,9 @@ func (e *Engine) runCommand(ctx Context, tool types.Tool, input string, toolCate
 	}
 
 	if err := cmd.Run(); err != nil {
+		if toolCategory == NoCategory {
+			return fmt.Sprintf("ERROR: got (%v) while running tool, OUTPUT: %s", err, all), nil
+		}
 		_, _ = os.Stderr.Write(output.Bytes())
 		log.Errorf("failed to run tool [%s] cmd %v: %v", tool.Parameters.Name, cmd.Args, err)
 		return "", fmt.Errorf("ERROR: %s: %w", all, err)

Original file line number	Diff line number	Diff line change
`@@ -80,6 +80,9 @@ func (e *Engine) runCommand(ctx Context, tool types.Tool, input string, toolCate`
`80`	`80`	`}`
`81`	`81`
`82`	`82`	`if err := cmd.Run(); err != nil {`
	`83`	`+ if toolCategory == NoCategory {`
	`84`	`+ return fmt.Sprintf("ERROR: got (%v) while running tool, OUTPUT: %s", err, all), nil`
	`85`	`+ }`
`83`	`86`	`_, _ = os.Stderr.Write(output.Bytes())`
`84`	`87`	`log.Errorf("failed to run tool [%s] cmd %v: %v", tool.Parameters.Name, cmd.Args, err)`
`85`	`88`	`return "", fmt.Errorf("ERROR: %s: %w", all, err)`