X-Git-Url: https://git.giacomolli.uk/onechat/blobdiff_plain/723f6e18adc6482ed913e1618c8a76e981720c2b..d28e3b2858db322899a81b59a30400c1acad2b42:/tools.go?ds=sidebyside

diff --git a/tools.go b/tools.go
index a75514f..96cbb97 100644
--- a/tools.go
+++ b/tools.go
@@ -68,13 +68,15 @@ type Toolbox struct {
 	messages  *DaemonMessages
 	children  *DaemonChildren
 	routes    *Routes
+	images    *ImageStore
 }
 
 // Call logs the call of the tool name with its JSON arguments, makes the log
-// durable, runs the tool, and logs its clipped output, which it returns. A
-// failed tool gives its error as the output. An error of Call itself is an
-// error of the log. When ctx ends before the flush, the tool does not run.
-func (b *Toolbox) Call(ctx context.Context, name, arguments string) (string, error) {
+// durable, runs the tool, and logs its clipped text. It returns that text or
+// the text and image parts of a single-message zoom. Pixels never enter the
+// log. A failed tool gives its error as the output. An error of Call itself is
+// an error of the log. When ctx ends before the flush, the tool does not run.
+func (b *Toolbox) Call(ctx context.Context, name, arguments string) (any, error) {
 	_, err := b.log.Append("tool", toolCallText(name, arguments))
 	if err != nil {
 		return "", fmt.Errorf("append tool call: %w", err)
@@ -90,12 +92,26 @@ func (b *Toolbox) Call(ctx context.Context, name, arguments string) (string, err
 		output = toolClip(b.run(ctx, name, arguments))
 	}
 
+	var (
+		result    any = output
+		resultErr error
+	)
+
+	if err == nil {
+		result, resultErr = imageToolOutput(b, name, arguments, output)
+	}
+
+	if resultErr != nil {
+		output = "error: " + resultErr.Error()
+		result = output
+	}
+
 	_, err = b.log.Append("echo", output)
 	if err != nil {
 		return "", fmt.Errorf("append tool result: %w", err)
 	}
 
-	return output, nil
+	return result, nil
 }
 
 // run runs one tool and returns its output, or its error as text.