main.go RawHistoryFolder

1// In-process Go embedding of the zot agent runtime via the sdk package.
2// Compare to examples/rpc/go which spawns `zot rpc` as a subprocess.
3package main
4
5import (
6	"context"
7	"fmt"
8	"os"
9	"strings"
10
11	"github.com/patriceckhart/zot/packages/agent/sdk"
12)
13
14func main() {
15	if len(os.Args) < 2 {
16		fmt.Fprintln(os.Stderr, "usage: sdkdemo <prompt>")
17		os.Exit(2)
18	}
19	prompt := strings.Join(os.Args[1:], " ")
20
21	rt, err := sdk.New(sdk.Config{})
22	if err != nil {
23		fmt.Fprintln(os.Stderr, err)
24		os.Exit(1)
25	}
26	defer rt.Close()
27
28	events, err := rt.Prompt(context.Background(), prompt, nil)
29	if err != nil {
30		fmt.Fprintln(os.Stderr, err)
31		os.Exit(1)
32	}
33	for ev := range events {
34		switch ev.Type {
35		case "text_delta":
36			fmt.Print(ev.Delta)
37		case "tool_call":
38			fmt.Fprintf(os.Stderr, "\n[tool] %s\n", ev.Name)
39		case "error":
40			fmt.Fprintf(os.Stderr, "\n[error] %s\n", ev.Error)
41		}
42	}
43	fmt.Println()
44	fmt.Fprintf(os.Stderr, "cost so far: $%.4f\n", rt.Cost().CostUSD)
45}