utils

type BatchTask struct {
	Model      string
	DoneChan   chan error
	TaskFunc   func() (BatchTaskResult, error)
	RetryTimes int
	Context    context.Context
}

type BatchTaskResult struct {
	RequeueAfter int64
}

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

type BotChatCompletionStreamReader struct {
	ChatCompletionStreamReader
}

type Breaker struct {
	LastTime time.Time
}

type ChatCompletionStreamReader struct {
	EmptyMessagesLimit uint
	IsFinished         bool

	Reader         *bufio.Reader
	Response       *http.Response
	ErrAccumulator ErrorAccumulator
	Unmarshaler    Unmarshaler

	model.HttpHeader
}

type DefaultErrorAccumulator struct {
	Buffer errorBuffer
}

type ErrorAccumulator interface {
	Write(p []byte) error
	Bytes() []byte
}

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

type JSONMarshaller struct{}

type JSONUnmarshaler struct{}

type Marshaller interface {
	Marshal(value interface{}) ([]byte, error)
}

type RequestBuilder interface {
	Build(ctx context.Context, method, url string, body interface{}, header http.Header) (*http.Request, error)
}

type RetryPolicy struct {
	MaxAttempts    int
	InitialBackoff time.Duration
	MaxBackoff     time.Duration
}

type Unmarshaler interface {
	Unmarshal(data []byte, v interface{}) error
}