package main

import (
	"context"
	"log"
	"strings"

	"github.com/zhouyangtingwen/dify-sdk-go"
)

func main() {
	var (
		ctx = context.Background()
		c = dify.NewClient("your-dify-server-host", "your-api-key-here")

		req = &dify.ChatMessageRequest{
			Query: "your-question",
			User: "your-user",
		}

		ch chan dify.ChatMessageStreamChannelResponse
		err error
	)

	if ch, err = c.Api().ChatMessagesStream(ctx, req); err != nil {
		return
	}

	var strBuilder strings.Builder

	for {
		select {
		case <-ctx.Done():
			return
		case streamData, isOpen := <-ch:
			if err = streamData.Err; err != nil {
				log.Println(err.Error())
				return
			}
			if !isOpen {
				log.Println(strBuilder.String())
				return
			}

			strBuilder.WriteString(streamData.Answer)
		}
	}
}