mirror of
https://github.com/songquanpeng/one-api.git
synced 2025-09-18 17:46:37 +08:00
50bab08496
- Refactor constant definitions and organization - Clean up package level variables and functions - Introduce new `relaymode` and `apitype` packages for constant definitions - Refactor and simplify code in several packages including `openai`, `relay/channel/baidu`, `relay/util`, `relay/controller`, `relay/channeltype` - Add helper functions in `relay/channeltype` package to convert channel type constants to corresponding API type constants - Remove deprecated functions such as `ResponseText2Usage` from `relay/channel/openai/helper.go` - Modify code in `relay/util/validation.go` and related files to use new `validator.ValidateTextRequest` function - Rename `util` package to `relaymode` and update related imports in several packages
146 lines
4.4 KiB
Go
146 lines
4.4 KiB
Go
package openai
|
|
|
|
import "github.com/songquanpeng/one-api/relay/model"
|
|
|
|
type TextContent struct {
|
|
Type string `json:"type,omitempty"`
|
|
Text string `json:"text,omitempty"`
|
|
}
|
|
|
|
type ImageContent struct {
|
|
Type string `json:"type,omitempty"`
|
|
ImageURL *model.ImageURL `json:"image_url,omitempty"`
|
|
}
|
|
|
|
type ChatRequest struct {
|
|
Model string `json:"model"`
|
|
Messages []model.Message `json:"messages"`
|
|
MaxTokens int `json:"max_tokens"`
|
|
}
|
|
|
|
type TextRequest struct {
|
|
Model string `json:"model"`
|
|
Messages []model.Message `json:"messages"`
|
|
Prompt string `json:"prompt"`
|
|
MaxTokens int `json:"max_tokens"`
|
|
//Stream bool `json:"stream"`
|
|
}
|
|
|
|
// ImageRequest docs: https://platform.openai.com/docs/api-reference/images/create
|
|
type ImageRequest struct {
|
|
Model string `json:"model"`
|
|
Prompt string `json:"prompt" binding:"required"`
|
|
N int `json:"n,omitempty"`
|
|
Size string `json:"size,omitempty"`
|
|
Quality string `json:"quality,omitempty"`
|
|
ResponseFormat string `json:"response_format,omitempty"`
|
|
Style string `json:"style,omitempty"`
|
|
User string `json:"user,omitempty"`
|
|
}
|
|
|
|
type WhisperJSONResponse struct {
|
|
Text string `json:"text,omitempty"`
|
|
}
|
|
|
|
type WhisperVerboseJSONResponse struct {
|
|
Task string `json:"task,omitempty"`
|
|
Language string `json:"language,omitempty"`
|
|
Duration float64 `json:"duration,omitempty"`
|
|
Text string `json:"text,omitempty"`
|
|
Segments []Segment `json:"segments,omitempty"`
|
|
}
|
|
|
|
type Segment struct {
|
|
Id int `json:"id"`
|
|
Seek int `json:"seek"`
|
|
Start float64 `json:"start"`
|
|
End float64 `json:"end"`
|
|
Text string `json:"text"`
|
|
Tokens []int `json:"tokens"`
|
|
Temperature float64 `json:"temperature"`
|
|
AvgLogprob float64 `json:"avg_logprob"`
|
|
CompressionRatio float64 `json:"compression_ratio"`
|
|
NoSpeechProb float64 `json:"no_speech_prob"`
|
|
}
|
|
|
|
type TextToSpeechRequest struct {
|
|
Model string `json:"model" binding:"required"`
|
|
Input string `json:"input" binding:"required"`
|
|
Voice string `json:"voice" binding:"required"`
|
|
Speed float64 `json:"speed"`
|
|
ResponseFormat string `json:"response_format"`
|
|
}
|
|
|
|
type UsageOrResponseText struct {
|
|
*model.Usage
|
|
ResponseText string
|
|
}
|
|
|
|
type SlimTextResponse struct {
|
|
Choices []TextResponseChoice `json:"choices"`
|
|
model.Usage `json:"usage"`
|
|
Error model.Error `json:"error"`
|
|
}
|
|
|
|
type TextResponseChoice struct {
|
|
Index int `json:"index"`
|
|
model.Message `json:"message"`
|
|
FinishReason string `json:"finish_reason"`
|
|
}
|
|
|
|
type TextResponse struct {
|
|
Id string `json:"id"`
|
|
Model string `json:"model,omitempty"`
|
|
Object string `json:"object"`
|
|
Created int64 `json:"created"`
|
|
Choices []TextResponseChoice `json:"choices"`
|
|
model.Usage `json:"usage"`
|
|
}
|
|
|
|
type EmbeddingResponseItem struct {
|
|
Object string `json:"object"`
|
|
Index int `json:"index"`
|
|
Embedding []float64 `json:"embedding"`
|
|
}
|
|
|
|
type EmbeddingResponse struct {
|
|
Object string `json:"object"`
|
|
Data []EmbeddingResponseItem `json:"data"`
|
|
Model string `json:"model"`
|
|
model.Usage `json:"usage"`
|
|
}
|
|
|
|
type ImageData struct {
|
|
Url string `json:"url,omitempty"`
|
|
B64Json string `json:"b64_json,omitempty"`
|
|
RevisedPrompt string `json:"revised_prompt,omitempty"`
|
|
}
|
|
|
|
type ImageResponse struct {
|
|
Created int64 `json:"created"`
|
|
Data []ImageData `json:"data"`
|
|
//model.Usage `json:"usage"`
|
|
}
|
|
|
|
type ChatCompletionsStreamResponseChoice struct {
|
|
Index int `json:"index"`
|
|
Delta model.Message `json:"delta"`
|
|
FinishReason *string `json:"finish_reason,omitempty"`
|
|
}
|
|
|
|
type ChatCompletionsStreamResponse struct {
|
|
Id string `json:"id"`
|
|
Object string `json:"object"`
|
|
Created int64 `json:"created"`
|
|
Model string `json:"model"`
|
|
Choices []ChatCompletionsStreamResponseChoice `json:"choices"`
|
|
Usage *model.Usage `json:"usage"`
|
|
}
|
|
|
|
type CompletionsStreamResponse struct {
|
|
Choices []struct {
|
|
Text string `json:"text"`
|
|
FinishReason string `json:"finish_reason"`
|
|
} `json:"choices"`
|
|
}
|