Help us shape the future of Go! Take the 2026 Go Developer Survey

types

package
v0.13.2 Latest Latest
Warning

This package is not in the latest version of its module.

 Go to latest Published: Apr 27, 2026 License: AGPL-3.0 Imports: 9 Imported by: 1

Documentation

Index

Constants

View Source 
const (
	RelayFormatOpenAI                    RelayFormat = "openai"
	RelayFormatClaude                                = "claude"
	RelayFormatGemini                                = "gemini"
	RelayFormatOpenAIResponses                       = "openai_responses"
	RelayFormatOpenAIResponsesCompaction             = "openai_responses_compaction"
	RelayFormatOpenAIAudio                           = "openai_audio"
	RelayFormatOpenAIImage                           = "openai_image"
	RelayFormatOpenAIRealtime                        = "openai_realtime"
	RelayFormatRerank                                = "rerank"
	RelayFormatEmbedding                             = "embedding"

	RelayFormatTask    = "task"
	RelayFormatMjProxy = "mj_proxy"
)

Variables

This section is empty.

Functions

func IsChannelError 

func IsChannelError(err *NewAPIError) bool

func IsRecordErrorLog 

func IsRecordErrorLog(e *NewAPIError) bool

func IsSkipRetryError 

func IsSkipRetryError(err *NewAPIError) bool

func LoadFromJsonString 

func LoadFromJsonString[K comparable, V any](m *RWMap[K, V], jsonStr string) error

func LoadFromJsonStringWithCallback added in v0.10.9 

func LoadFromJsonStringWithCallback[K comparable, V any](m *RWMap[K, V], jsonStr string, onSuccess func()) error

LoadFromJsonStringWithCallback loads a JSON string into the RWMap and calls the callback on success.

Types

type Base64Source added in v0.12.2 

type Base64Source struct {
	Base64Data string
	MimeType   string
	// contains filtered or unexported fields
}

func NewBase64FileSource added in v0.10.8 

func NewBase64FileSource(base64Data string, mimeType string) *Base64Source

func (*Base64Source) ClearCache added in v0.12.2 

func (b *Base64Source) ClearCache()

func (*Base64Source) ClearRawData added in v0.12.2 

func (b *Base64Source) ClearRawData()

func (*Base64Source) GetCache added in v0.12.2 

func (b *Base64Source) GetCache() *CachedFileData

func (*Base64Source) GetIdentifier added in v0.12.2 

func (b *Base64Source) GetIdentifier() string

func (*Base64Source) GetRawData added in v0.12.2 

func (b *Base64Source) GetRawData() string

func (*Base64Source) HasCache added in v0.12.2 

func (b *Base64Source) HasCache() bool

func (*Base64Source) IsRegistered added in v0.12.2 

func (b *Base64Source) IsRegistered() bool

func (*Base64Source) IsURL added in v0.12.2 

func (b *Base64Source) IsURL() bool

func (*Base64Source) Mu added in v0.12.2 

func (b *Base64Source) Mu() *sync.Mutex

func (*Base64Source) SetCache added in v0.12.2 

func (b *Base64Source) SetCache(data *CachedFileData)

func (*Base64Source) SetRegistered added in v0.12.2 

func (b *Base64Source) SetRegistered(registered bool)

type CachedFileData added in v0.10.8 

type CachedFileData struct {
	MimeType    string        // MIME 类型
	Size        int64         // 文件大小（字节）
	DiskSize    int64         // 磁盘缓存实际占用大小（字节，通常是 base64 长度）
	ImageConfig *image.Config // 图片配置（如果是图片）
	ImageFormat string        // 图片格式（如果是图片）

	OnClose func(size int64)
	// contains filtered or unexported fields
}

func NewDiskCachedData added in v0.10.8 

func NewDiskCachedData(diskPath string, mimeType string, size int64) *CachedFileData

func NewMemoryCachedData added in v0.10.8 

func NewMemoryCachedData(base64Data string, mimeType string, size int64) *CachedFileData

func (*CachedFileData) Close added in v0.10.8 

func (c *CachedFileData) Close() error

func (*CachedFileData) GetBase64Data added in v0.10.8 

func (c *CachedFileData) GetBase64Data() (string, error)

func (*CachedFileData) IsDisk added in v0.10.8 

func (c *CachedFileData) IsDisk() bool

func (*CachedFileData) SetBase64Data added in v0.10.8 

func (c *CachedFileData) SetBase64Data(data string)

type ChannelError 

type ChannelError struct {
	ChannelId   int    `json:"channel_id"`
	ChannelType int    `json:"channel_type"`
	ChannelName string `json:"channel_name"`
	IsMultiKey  bool   `json:"is_multi_key"`
	AutoBan     bool   `json:"auto_ban"`
	UsingKey    string `json:"using_key"`
}

func NewChannelError 

func NewChannelError(channelId int, channelType int, channelName string, isMultiKey bool, usingKey string, autoBan bool) *ChannelError

type ClaudeError 

type ClaudeError struct {
	Type    string `json:"type,omitempty"`
	Message string `json:"message,omitempty"`
}

type ErrorCode 

type ErrorCode string
const (
	ErrorCodeInvalidRequest         ErrorCode = "invalid_request"
	ErrorCodeSensitiveWordsDetected ErrorCode = "sensitive_words_detected"
	ErrorCodeViolationFeeGrokCSAM   ErrorCode = "violation_fee.grok.csam"

	// new api error
	ErrorCodeCountTokenFailed   ErrorCode = "count_token_failed"
	ErrorCodeModelPriceError    ErrorCode = "model_price_error"
	ErrorCodeInvalidApiType     ErrorCode = "invalid_api_type"
	ErrorCodeJsonMarshalFailed  ErrorCode = "json_marshal_failed"
	ErrorCodeDoRequestFailed    ErrorCode = "do_request_failed"
	ErrorCodeGetChannelFailed   ErrorCode = "get_channel_failed"
	ErrorCodeGenRelayInfoFailed ErrorCode = "gen_relay_info_failed"

	// channel error
	ErrorCodeChannelNoAvailableKey        ErrorCode = "channel:no_available_key"
	ErrorCodeChannelParamOverrideInvalid  ErrorCode = "channel:param_override_invalid"
	ErrorCodeChannelHeaderOverrideInvalid ErrorCode = "channel:header_override_invalid"
	ErrorCodeChannelModelMappedError      ErrorCode = "channel:model_mapped_error"
	ErrorCodeChannelAwsClientError        ErrorCode = "channel:aws_client_error"
	ErrorCodeChannelInvalidKey            ErrorCode = "channel:invalid_key"
	ErrorCodeChannelResponseTimeExceeded  ErrorCode = "channel:response_time_exceeded"

	// client request error
	ErrorCodeReadRequestBodyFailed ErrorCode = "read_request_body_failed"
	ErrorCodeConvertRequestFailed  ErrorCode = "convert_request_failed"
	ErrorCodeAccessDenied          ErrorCode = "access_denied"

	// request error
	ErrorCodeBadRequestBody ErrorCode = "bad_request_body"

	// response error
	ErrorCodeReadResponseBodyFailed ErrorCode = "read_response_body_failed"
	ErrorCodeBadResponseStatusCode  ErrorCode = "bad_response_status_code"
	ErrorCodeBadResponse            ErrorCode = "bad_response"
	ErrorCodeBadResponseBody        ErrorCode = "bad_response_body"
	ErrorCodeEmptyResponse          ErrorCode = "empty_response"
	ErrorCodeAwsInvokeError         ErrorCode = "aws_invoke_error"
	ErrorCodeModelNotFound          ErrorCode = "model_not_found"
	ErrorCodePromptBlocked          ErrorCode = "prompt_blocked"

	// sql error
	ErrorCodeQueryDataError  ErrorCode = "query_data_error"
	ErrorCodeUpdateDataError ErrorCode = "update_data_error"

	// quota error
	ErrorCodeInsufficientUserQuota      ErrorCode = "insufficient_user_quota"
	ErrorCodePreConsumeTokenQuotaFailed ErrorCode = "pre_consume_token_quota_failed"
)

type ErrorType 

type ErrorType string
const (
	ErrorTypeNewAPIError     ErrorType = "new_api_error"
	ErrorTypeOpenAIError     ErrorType = "openai_error"
	ErrorTypeClaudeError     ErrorType = "claude_error"
	ErrorTypeMidjourneyError ErrorType = "midjourney_error"
	ErrorTypeGeminiError     ErrorType = "gemini_error"
	ErrorTypeRerankError     ErrorType = "rerank_error"
	ErrorTypeUpstreamError   ErrorType = "upstream_error"
)

type FileMeta 

type FileMeta struct {
	FileType
	Source FileSource // 统一的文件来源（URL 或 base64）
	Detail string     // 图片细节级别（low/high/auto）
}

func NewFileMeta added in v0.10.8 

func NewFileMeta(fileType FileType, source FileSource) *FileMeta

NewFileMeta 创建新的 FileMeta

func NewImageFileMeta added in v0.10.8 

func NewImageFileMeta(source FileSource, detail string) *FileMeta

NewImageFileMeta 创建图片类型的 FileMeta

func (*FileMeta) GetIdentifier added in v0.10.8 

func (f *FileMeta) GetIdentifier() string

GetIdentifier 获取文件标识符（用于日志）

func (*FileMeta) GetRawData added in v0.10.8 

func (f *FileMeta) GetRawData() string

GetRawData 获取原始数据（兼容旧代码） Deprecated: 请使用 Source.GetRawData()

func (*FileMeta) IsURL added in v0.10.8 

func (f *FileMeta) IsURL() bool

IsURL 判断是否是 URL 来源

type FileSource added in v0.10.8 

type FileSource interface {
	IsURL() bool
	GetIdentifier() string
	GetRawData() string
	ClearRawData()

	SetCache(data *CachedFileData)
	GetCache() *CachedFileData
	HasCache() bool
	ClearCache()

	IsRegistered() bool
	SetRegistered(registered bool)
	Mu() *sync.Mutex
}

FileSource 统一的文件来源抽象接口 支持 URL 和 base64 两种来源，提供懒加载和缓存机制

func NewFileSourceFromData added in v0.12.2 

func NewFileSourceFromData(data string, mimeType string) FileSource

type FileType 

type FileType string
const (
	FileTypeImage FileType = "image" // Image file type
	FileTypeAudio FileType = "audio" // Audio file type
	FileTypeVideo FileType = "video" // Video file type
	FileTypeFile  FileType = "file"  // Generic file type
)

type GroupRatioInfo 

type GroupRatioInfo struct {
	GroupRatio        float64
	GroupSpecialRatio float64
	HasSpecialRatio   bool
}

type LocalFileData 

type LocalFileData struct {
	MimeType   string
	Base64Data string
	Url        string
	Size       int64
}

type NewAPIError 

type NewAPIError struct {
	Err        error
	RelayError any

	StatusCode int
	Metadata   json.RawMessage
	// contains filtered or unexported fields
}

func InitOpenAIError 

func InitOpenAIError(errorCode ErrorCode, statusCode int, ops ...NewAPIErrorOptions) *NewAPIError

func NewError 

func NewError(err error, errorCode ErrorCode, ops ...NewAPIErrorOptions) *NewAPIError

func NewErrorWithStatusCode 

func NewErrorWithStatusCode(err error, errorCode ErrorCode, statusCode int, ops ...NewAPIErrorOptions) *NewAPIError

func NewOpenAIError 

func NewOpenAIError(err error, errorCode ErrorCode, statusCode int, ops ...NewAPIErrorOptions) *NewAPIError

func WithClaudeError 

func WithClaudeError(claudeError ClaudeError, statusCode int, ops ...NewAPIErrorOptions) *NewAPIError

func WithOpenAIError 

func WithOpenAIError(openAIError OpenAIError, statusCode int, ops ...NewAPIErrorOptions) *NewAPIError

func (*NewAPIError) Error 

func (e *NewAPIError) Error() string

func (*NewAPIError) ErrorWithStatusCode 

func (e *NewAPIError) ErrorWithStatusCode() string

func (*NewAPIError) GetErrorCode 

func (e *NewAPIError) GetErrorCode() ErrorCode

func (*NewAPIError) GetErrorType 

func (e *NewAPIError) GetErrorType() ErrorType

func (*NewAPIError) MaskSensitiveError 

func (e *NewAPIError) MaskSensitiveError() string

func (*NewAPIError) MaskSensitiveErrorWithStatusCode 

func (e *NewAPIError) MaskSensitiveErrorWithStatusCode() string

func (*NewAPIError) SetMessage 

func (e *NewAPIError) SetMessage(message string)

func (*NewAPIError) ToClaudeError 

func (e *NewAPIError) ToClaudeError() ClaudeError

func (*NewAPIError) ToOpenAIError 

func (e *NewAPIError) ToOpenAIError() OpenAIError

func (*NewAPIError) Unwrap 

func (e *NewAPIError) Unwrap() error

Unwrap enables errors.Is / errors.As to work with NewAPIError by exposing the underlying error.

type NewAPIErrorOptions 

type NewAPIErrorOptions func(*NewAPIError)

func ErrOptionWithHideErrMsg 

func ErrOptionWithHideErrMsg(replaceStr string) NewAPIErrorOptions

func ErrOptionWithNoRecordErrorLog 

func ErrOptionWithNoRecordErrorLog() NewAPIErrorOptions

func ErrOptionWithSkipRetry 

func ErrOptionWithSkipRetry() NewAPIErrorOptions

func ErrOptionWithStatusCode added in v0.12.7 

func ErrOptionWithStatusCode(statusCode int) NewAPIErrorOptions

type OpenAIError 

type OpenAIError struct {
	Message  string          `json:"message"`
	Type     string          `json:"type"`
	Param    string          `json:"param"`
	Code     any             `json:"code"`
	Metadata json.RawMessage `json:"metadata,omitempty"`
}

type PriceData 

type PriceData struct {
	FreeModel            bool
	ModelPrice           float64
	ModelRatio           float64
	CompletionRatio      float64
	CacheRatio           float64
	CacheCreationRatio   float64
	CacheCreation5mRatio float64
	CacheCreation1hRatio float64
	ImageRatio           float64
	AudioRatio           float64
	AudioCompletionRatio float64
	OtherRatios          map[string]float64
	UsePrice             bool
	Quota                int // 按次计费的最终额度（MJ / Task）
	QuotaToPreConsume    int // 按量计费的预消耗额度
	GroupRatioInfo       GroupRatioInfo
}

func (*PriceData) AddOtherRatio 

func (p *PriceData) AddOtherRatio(key string, ratio float64)

func (*PriceData) ToSetting 

func (p *PriceData) ToSetting() string

type RWMap 

type RWMap[K comparable, V any] struct {
	// contains filtered or unexported fields
}

func NewRWMap 

func NewRWMap[K comparable, V any]() *RWMap[K, V]

func (*RWMap[K, V]) AddAll 

func (m *RWMap[K, V]) AddAll(other map[K]V)

func (*RWMap[K, V]) Clear 

func (m *RWMap[K, V]) Clear()

func (*RWMap[K, V]) Get 

func (m *RWMap[K, V]) Get(key K) (V, bool)

func (*RWMap[K, V]) Len 

func (m *RWMap[K, V]) Len() int

func (*RWMap[K, V]) MarshalJSON 

func (m *RWMap[K, V]) MarshalJSON() ([]byte, error)

func (*RWMap[K, V]) MarshalJSONString added in v0.10.9 

func (m *RWMap[K, V]) MarshalJSONString() string

MarshalJSONString returns the JSON string representation of the RWMap.

func (*RWMap[K, V]) ReadAll 

func (m *RWMap[K, V]) ReadAll() map[K]V

ReadAll returns a copy of the entire map.

func (*RWMap[K, V]) Set 

func (m *RWMap[K, V]) Set(key K, value V)

func (*RWMap[K, V]) UnmarshalJSON 

func (m *RWMap[K, V]) UnmarshalJSON(b []byte) error

type RelayFormat 

type RelayFormat string

type RequestMeta 

type RequestMeta struct {
	OriginalModelName string `json:"original_model_name"`
	UserUsingGroup    string `json:"user_using_group"`
	PromptTokens      int    `json:"prompt_tokens"`
	PreConsumedQuota  int    `json:"pre_consumed_quota"`
}

type Set 

type Set[T comparable] struct {
	// contains filtered or unexported fields
}

func NewSet 

func NewSet[T comparable]() *Set[T]

NewSet 创建并返回一个新的 Set

func (*Set[T]) Add 

func (s *Set[T]) Add(item T)

func (*Set[T]) Contains 

func (s *Set[T]) Contains(item T) bool

Contains 检查 Set 是否包含某个元素

func (*Set[T]) Items 

func (s *Set[T]) Items() []T

Items 返回 Set 中所有元素组成的切片 注意：由于 map 的无序性，返回的切片元素顺序是随机的

func (*Set[T]) Len 

func (s *Set[T]) Len() int

Len 返回 Set 中元素的数量

func (*Set[T]) Remove 

func (s *Set[T]) Remove(item T)

Remove 从 Set 中移除一个元素

type TokenCountMeta 

type TokenCountMeta struct {
	TokenType     TokenType   `json:"token_type,omitempty"`     // Type of tokens used in the request
	CombineText   string      `json:"combine_text,omitempty"`   // Combined text from all messages
	ToolsCount    int         `json:"tools_count,omitempty"`    // Number of tools used
	NameCount     int         `json:"name_count,omitempty"`     // Number of names in the request
	MessagesCount int         `json:"messages_count,omitempty"` // Number of messages in the request
	Files         []*FileMeta `json:"files,omitempty"`          // List of files, each with type and content
	MaxTokens     int         `json:"max_tokens,omitempty"`     // Maximum tokens allowed in the request

	ImagePriceRatio float64 `json:"image_ratio,omitempty"` // Ratio for image size, if applicable

}

type TokenType 

type TokenType string
const (
	TokenTypeTextNumber TokenType = "text_number" // Text or number tokens
	TokenTypeTokenizer  TokenType = "tokenizer"   // Tokenizer tokens
	TokenTypeImage      TokenType = "image"       // Image tokens
)

type URLSource added in v0.12.2 

type URLSource struct {
	URL string
	// contains filtered or unexported fields
}

func NewURLFileSource added in v0.10.8 

func NewURLFileSource(url string) *URLSource

func (*URLSource) ClearCache added in v0.12.2 

func (b *URLSource) ClearCache()

func (*URLSource) ClearRawData added in v0.12.2 

func (u *URLSource) ClearRawData()

func (*URLSource) GetCache added in v0.12.2 

func (b *URLSource) GetCache() *CachedFileData

func (*URLSource) GetIdentifier added in v0.12.2 

func (u *URLSource) GetIdentifier() string

func (*URLSource) GetRawData added in v0.12.2 

func (u *URLSource) GetRawData() string

func (*URLSource) HasCache added in v0.12.2 

func (b *URLSource) HasCache() bool

func (*URLSource) IsRegistered added in v0.12.2 

func (b *URLSource) IsRegistered() bool

func (*URLSource) IsURL added in v0.12.2 

func (u *URLSource) IsURL() bool

func (*URLSource) Mu added in v0.12.2 

func (b *URLSource) Mu() *sync.Mutex

func (*URLSource) SetCache added in v0.12.2 

func (b *URLSource) SetCache(data *CachedFileData)

func (*URLSource) SetRegistered added in v0.12.2 

func (b *URLSource) SetRegistered(registered bool)

Source Files

View all Source files

Jump to

Keyboard shortcuts

? : This menu
/ : Search site
f or F : Jump to
y or Y : Canonical URL
go.dev uses cookies from Google to deliver and enhance the quality of its services and to analyze traffic. Learn more.