mirror of
https://github.com/songquanpeng/one-api.git
synced 2025-11-15 04:33:42 +08:00
🎨 结构中继控制器
This commit is contained in:
@@ -3,16 +3,16 @@ package types
|
||||
import "mime/multipart"
|
||||
|
||||
type SpeechAudioRequest struct {
|
||||
Model string `json:"model"`
|
||||
Input string `json:"input"`
|
||||
Voice string `json:"voice"`
|
||||
Model string `json:"model" binding:"required"`
|
||||
Input string `json:"input" binding:"required"`
|
||||
Voice string `json:"voice" binding:"required"`
|
||||
ResponseFormat string `json:"response_format,omitempty"`
|
||||
Speed float64 `json:"speed,omitempty"`
|
||||
}
|
||||
|
||||
type AudioRequest struct {
|
||||
File *multipart.FileHeader `form:"file"`
|
||||
Model string `form:"model"`
|
||||
File *multipart.FileHeader `form:"file" binding:"required"`
|
||||
Model string `form:"model" binding:"required"`
|
||||
Language string `form:"language"`
|
||||
Prompt string `form:"prompt"`
|
||||
ResponseFormat string `form:"response_format"`
|
||||
|
||||
@@ -49,8 +49,8 @@ type ChatCompletionResponseFormat struct {
|
||||
}
|
||||
|
||||
type ChatCompletionRequest struct {
|
||||
Model string `json:"model"`
|
||||
Messages []ChatCompletionMessage `json:"messages"`
|
||||
Model string `json:"model" binding:"required"`
|
||||
Messages []ChatCompletionMessage `json:"messages" binding:"required"`
|
||||
MaxTokens int `json:"max_tokens,omitempty"`
|
||||
Temperature float64 `json:"temperature,omitempty"`
|
||||
TopP float64 `json:"top_p,omitempty"`
|
||||
|
||||
@@ -22,19 +22,3 @@ type OpenAIErrorWithStatusCode struct {
|
||||
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)
|
||||
// }
|
||||
|
||||
@@ -1,8 +1,8 @@
|
||||
package types
|
||||
|
||||
type CompletionRequest struct {
|
||||
Model string `json:"model"`
|
||||
Prompt any `json:"prompt,omitempty"`
|
||||
Model string `json:"model" binding:"required"`
|
||||
Prompt any `json:"prompt" binding:"required"`
|
||||
Suffix string `json:"suffix,omitempty"`
|
||||
MaxTokens int `json:"max_tokens,omitempty"`
|
||||
Temperature float32 `json:"temperature,omitempty"`
|
||||
|
||||
@@ -1,8 +1,8 @@
|
||||
package types
|
||||
|
||||
type EmbeddingRequest struct {
|
||||
Model string `json:"model"`
|
||||
Input any `json:"input"`
|
||||
Model string `json:"model" binding:"required"`
|
||||
Input any `json:"input" binding:"required"`
|
||||
EncodingFormat string `json:"encoding_format,omitempty"`
|
||||
User string `json:"user,omitempty"`
|
||||
}
|
||||
|
||||
@@ -3,7 +3,7 @@ package types
|
||||
import "mime/multipart"
|
||||
|
||||
type ImageRequest struct {
|
||||
Prompt string `json:"prompt,omitempty"`
|
||||
Prompt string `json:"prompt,omitempty" binding:"required"`
|
||||
Model string `json:"model,omitempty"`
|
||||
N int `json:"n,omitempty"`
|
||||
Quality string `json:"quality,omitempty"`
|
||||
@@ -25,7 +25,7 @@ type ImageResponseDataInner struct {
|
||||
}
|
||||
|
||||
type ImageEditRequest struct {
|
||||
Image *multipart.FileHeader `form:"image"`
|
||||
Image *multipart.FileHeader `form:"image" binding:"required"`
|
||||
Mask *multipart.FileHeader `form:"mask"`
|
||||
Model string `form:"model"`
|
||||
Prompt string `form:"prompt"`
|
||||
|
||||
@@ -1,7 +1,7 @@
|
||||
package types
|
||||
|
||||
type ModerationRequest struct {
|
||||
Input string `json:"input,omitempty"`
|
||||
Input string `json:"input,omitempty" binding:"required"`
|
||||
Model string `json:"model,omitempty"`
|
||||
}
|
||||
|
||||
|
||||
Reference in New Issue
Block a user