stream

type Builder struct {
	// contains filtered or unexported fields
}

type CallbackWriter interface {
	// WriteText sends a text content message as the assistant role.
	WriteText(content string)
	// WriteToolCall sends a tool call message with the given id, function name, and JSON arguments.
	WriteToolCall(id, name, arguments string)
	// WriteToolResult sends a tool result message for the specified tool call.
	WriteToolResult(toolCallID, content string)
	// WriteReasoning sends a reasoning/thinking message as the assistant role.
	WriteReasoning(reasoning string)
	// WriteData sends a custom data message with the specified type and payload.
	WriteData(dataType string, data any)
	// WriteMessage sends an arbitrary pre-built message to the stream.
	WriteMessage(msg Message)
}

type Chunk map[string]any

type ChunkType string

type Message struct {
	Role       Role
	Content    string
	ToolCalls  []ToolCall
	ToolCallID string
	Reasoning  string
	Data       map[string]any
}

type MessageSource interface {
	// Recv receives the next message from the stream. Returns io.EOF when no more messages are available.
	Recv() (Message, error)
	// Close releases resources associated with this message source.
	Close() error
}

type Options struct {
	SendReasoning bool
	SendSources   bool
	SendStart     bool
	SendFinish    bool
	OnError       func(err error) string
	OnFinish      func(content string)
	GenerateID    func(prefix string) string
}

type ResponseWriter interface {
	io.Writer
}

type Role string

type Source struct {
	Type      string
	ID        string
	URL       string
	Title     string
	MediaType string
}

type StreamWriter interface {
	// WriteChunk writes a single chunk to the stream output.
	WriteChunk(chunk Chunk) error
	// Flush flushes any buffered data to the underlying writer.
	Flush() error
}

type ToolCall struct {
	ID        string
	Name      string
	Arguments string
}