From 74aa5c5f5fec2643fd42b71e9a192ae3aca0fddc Mon Sep 17 00:00:00 2001 From: Christian Rocha Date: Tue, 21 Oct 2025 08:18:18 -0400 Subject: [PATCH] docs(examples): tidy up stream example --- examples/stream/main.go | 13 ++++++------- 1 file changed, 6 insertions(+), 7 deletions(-) diff --git a/examples/stream/main.go b/examples/stream/main.go index 7ed38a6bc41521c846dc354bc0abb7ac8ef3e354..d06b10ce6a1fe4f255d00672ed2a8fe46caf7c89 100644 --- a/examples/stream/main.go +++ b/examples/stream/main.go @@ -3,7 +3,6 @@ package main import ( "context" - "encoding/json" "fmt" "os" @@ -15,13 +14,13 @@ func main() { provider := openai.New(openai.WithAPIKey(os.Getenv("OPENAI_API_KEY"))) model, err := provider.LanguageModel("gpt-4o") if err != nil { - fmt.Println(err) + fmt.Fprint(os.Stderr, err) os.Exit(1) } stream, err := model.Stream(context.Background(), fantasy.Call{ Prompt: fantasy.Prompt{ - fantasy.NewUserMessage("Whats the weather in pristina."), + fantasy.NewUserMessage("Whats the weather in Pristina, Kosovo?"), }, Temperature: fantasy.Opt(0.7), Tools: []fantasy.Tool{ @@ -44,16 +43,16 @@ func main() { }, }) if err != nil { - fmt.Println(err) + fmt.Fprint(os.Stderr, err) os.Exit(1) } for chunk := range stream { - data, err := json.Marshal(chunk) if err != nil { - fmt.Println(err) + fmt.Fprint(os.Stderr, err) continue } - fmt.Println(string(data)) + fmt.Print(chunk.Delta) } + fmt.Print("\n") }