Integrate BACKBEAT SDK and resolve KACHING license validation
Major integrations and fixes: - Added BACKBEAT SDK integration for P2P operation timing - Implemented beat-aware status tracking for distributed operations - Added Docker secrets support for secure license management - Resolved KACHING license validation via HTTPS/TLS - Updated docker-compose configuration for clean stack deployment - Disabled rollback policies to prevent deployment failures - Added license credential storage (CHORUS-DEV-MULTI-001) Technical improvements: - BACKBEAT P2P operation tracking with phase management - Enhanced configuration system with file-based secrets - Improved error handling for license validation - Clean separation of KACHING and CHORUS deployment stacks 🤖 Generated with [Claude Code](https://claude.ai/code) Co-Authored-By: Claude <noreply@anthropic.com>
This commit is contained in:
119
vendor/github.com/sashabaranov/go-openai/stream_reader.go
generated
vendored
Normal file
119
vendor/github.com/sashabaranov/go-openai/stream_reader.go
generated
vendored
Normal file
@@ -0,0 +1,119 @@
|
||||
package openai
|
||||
|
||||
import (
|
||||
"bufio"
|
||||
"bytes"
|
||||
"fmt"
|
||||
"io"
|
||||
"net/http"
|
||||
"regexp"
|
||||
|
||||
utils "github.com/sashabaranov/go-openai/internal"
|
||||
)
|
||||
|
||||
var (
|
||||
headerData = regexp.MustCompile(`^data:\s*`)
|
||||
errorPrefix = regexp.MustCompile(`^data:\s*{"error":`)
|
||||
)
|
||||
|
||||
type streamable interface {
|
||||
ChatCompletionStreamResponse | CompletionResponse
|
||||
}
|
||||
|
||||
type streamReader[T streamable] struct {
|
||||
emptyMessagesLimit uint
|
||||
isFinished bool
|
||||
|
||||
reader *bufio.Reader
|
||||
response *http.Response
|
||||
errAccumulator utils.ErrorAccumulator
|
||||
unmarshaler utils.Unmarshaler
|
||||
|
||||
httpHeader
|
||||
}
|
||||
|
||||
func (stream *streamReader[T]) Recv() (response T, err error) {
|
||||
rawLine, err := stream.RecvRaw()
|
||||
if err != nil {
|
||||
return
|
||||
}
|
||||
|
||||
err = stream.unmarshaler.Unmarshal(rawLine, &response)
|
||||
if err != nil {
|
||||
return
|
||||
}
|
||||
return response, nil
|
||||
}
|
||||
|
||||
func (stream *streamReader[T]) RecvRaw() ([]byte, error) {
|
||||
if stream.isFinished {
|
||||
return nil, io.EOF
|
||||
}
|
||||
|
||||
return stream.processLines()
|
||||
}
|
||||
|
||||
//nolint:gocognit
|
||||
func (stream *streamReader[T]) processLines() ([]byte, error) {
|
||||
var (
|
||||
emptyMessagesCount uint
|
||||
hasErrorPrefix bool
|
||||
)
|
||||
|
||||
for {
|
||||
rawLine, readErr := stream.reader.ReadBytes('\n')
|
||||
if readErr != nil || hasErrorPrefix {
|
||||
respErr := stream.unmarshalError()
|
||||
if respErr != nil {
|
||||
return nil, fmt.Errorf("error, %w", respErr.Error)
|
||||
}
|
||||
return nil, readErr
|
||||
}
|
||||
|
||||
noSpaceLine := bytes.TrimSpace(rawLine)
|
||||
if errorPrefix.Match(noSpaceLine) {
|
||||
hasErrorPrefix = true
|
||||
}
|
||||
if !headerData.Match(noSpaceLine) || hasErrorPrefix {
|
||||
if hasErrorPrefix {
|
||||
noSpaceLine = headerData.ReplaceAll(noSpaceLine, nil)
|
||||
}
|
||||
writeErr := stream.errAccumulator.Write(noSpaceLine)
|
||||
if writeErr != nil {
|
||||
return nil, writeErr
|
||||
}
|
||||
emptyMessagesCount++
|
||||
if emptyMessagesCount > stream.emptyMessagesLimit {
|
||||
return nil, ErrTooManyEmptyStreamMessages
|
||||
}
|
||||
|
||||
continue
|
||||
}
|
||||
|
||||
noPrefixLine := headerData.ReplaceAll(noSpaceLine, nil)
|
||||
if string(noPrefixLine) == "[DONE]" {
|
||||
stream.isFinished = true
|
||||
return nil, io.EOF
|
||||
}
|
||||
|
||||
return noPrefixLine, nil
|
||||
}
|
||||
}
|
||||
|
||||
func (stream *streamReader[T]) unmarshalError() (errResp *ErrorResponse) {
|
||||
errBytes := stream.errAccumulator.Bytes()
|
||||
if len(errBytes) == 0 {
|
||||
return
|
||||
}
|
||||
|
||||
err := stream.unmarshaler.Unmarshal(errBytes, &errResp)
|
||||
if err != nil {
|
||||
errResp = nil
|
||||
}
|
||||
|
||||
return
|
||||
}
|
||||
|
||||
func (stream *streamReader[T]) Close() error {
|
||||
return stream.response.Body.Close()
|
||||
}
|
||||
Reference in New Issue
Block a user