mirror of
https://github.com/songquanpeng/one-api.git
synced 2025-09-29 22:56:39 +08:00
41 lines
1007 B
Go
41 lines
1007 B
Go
package types
|
|
|
|
type Usage struct {
|
|
PromptTokens int `json:"prompt_tokens"`
|
|
CompletionTokens int `json:"completion_tokens"`
|
|
TotalTokens int `json:"total_tokens"`
|
|
}
|
|
|
|
type OpenAIError struct {
|
|
Code any `json:"code,omitempty"`
|
|
Message string `json:"message"`
|
|
Param string `json:"param,omitempty"`
|
|
Type string `json:"type"`
|
|
InnerError any `json:"innererror,omitempty"`
|
|
}
|
|
|
|
type OpenAIErrorWithStatusCode struct {
|
|
OpenAIError
|
|
StatusCode int `json:"status_code"`
|
|
}
|
|
|
|
type OpenAIErrorResponse struct {
|
|
Error OpenAIError `json:"error,omitempty"`
|
|
}
|
|
|
|
func ErrorWrapper(err error, code string, statusCode int) *OpenAIErrorWithStatusCode {
|
|
openAIError := OpenAIError{
|
|
Message: err.Error(),
|
|
Type: "one_api_error",
|
|
Code: code,
|
|
}
|
|
return &OpenAIErrorWithStatusCode{
|
|
OpenAIError: openAIError,
|
|
StatusCode: statusCode,
|
|
}
|
|
}
|
|
|
|
// type GeneralErrorHandling interface {
|
|
// HandleError(resp *http.Response) (openAIErrorWithStatusCode *OpenAIErrorWithStatusCode)
|
|
// }
|