print.go RawHistoryFolder

1// Package modes implements zot's three run modes: print, json, interactive.
2package modes
3
4import (
5	"context"
6	"fmt"
7	"io"
8	"strings"
9
10	"github.com/patriceckhart/zot/packages/core"
11	"github.com/patriceckhart/zot/packages/provider"
12)
13
14// RunPrint runs the agent to completion and writes only the final
15// assistant text block to out. Exit code comes from the caller.
16func RunPrint(ctx context.Context, ag *core.Agent, prompt string, images []provider.ImageBlock, out io.Writer) error {
17	var finalText strings.Builder
18	var lastAssistant string
19	var runErr error
20
21	sink := func(ev core.AgentEvent) {
22		switch e := ev.(type) {
23		case core.EvAssistantMessage:
24			// Keep the most recent assistant text block; by the end it's the final answer.
25			var sb strings.Builder
26			for _, c := range e.Message.Content {
27				if tb, ok := c.(provider.TextBlock); ok {
28					if sb.Len() > 0 {
29						sb.WriteString("\n")
30					}
31					sb.WriteString(tb.Text)
32				}
33			}
34			if sb.Len() > 0 {
35				lastAssistant = sb.String()
36			}
37		case core.EvTurnEnd:
38			if e.Err != nil {
39				runErr = e.Err
40			}
41		}
42	}
43
44	if err := ag.Prompt(ctx, prompt, images, sink); err != nil {
45		return err
46	}
47	if runErr != nil {
48		return runErr
49	}
50
51	finalText.WriteString(lastAssistant)
52	if finalText.Len() > 0 && !strings.HasSuffix(finalText.String(), "\n") {
53		finalText.WriteString("\n")
54	}
55	_, err := fmt.Fprint(out, finalText.String())
56	return err
57}