package main

import (
    "context"
    "fmt"
    . "github.com/rehiy/one-llm/baidu"
)

func main() {
    client := NewClient("xxxx", "yyyy", true)
    stream, err := client.CreateChatCompletionStream(context.Background(), ChatCompletionRequest{
        Messages: []ChatCompletionMessage{
            {Content: "Hello!", Role: ChatMessageRoleUser},
        },
        Stream: true,
    })

    defer stream.Close()

    fmt.Println("Stream response: ")
    for {
        response, err := stream.Recv()
        if errors.Is(err, io.EOF) {
            fmt.Printf("\nStream finished: %d %s\n", response.ErrorCode, response.ErrorMsg)
            return
        }
        if err != nil {
            fmt.Printf("\nStream error: %v\n", err)
            return
        }
        fmt.Printf("error: %s\n", response.ErrorMsg)
        fmt.Printf("resp: %s\n", response.Result)
    }
}