28 lines
1.2 KiB
Go
28 lines
1.2 KiB
Go
package xai
|
|
|
|
import "one-api/dto"
|
|
|
|
// ChatCompletionResponse represents the response from XAI chat completion API
|
|
type ChatCompletionResponse struct {
|
|
Id string `json:"id"`
|
|
Object string `json:"object"`
|
|
Created int64 `json:"created"`
|
|
Model string `json:"model"`
|
|
Choices []dto.OpenAITextResponseChoice `json:"choices"`
|
|
Usage *dto.Usage `json:"usage"`
|
|
SystemFingerprint string `json:"system_fingerprint"`
|
|
}
|
|
|
|
// quality, size or style are not supported by xAI API at the moment.
|
|
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"`
|
|
// ExtraFields json.RawMessage `json:"extra_fields,omitempty"`
|
|
}
|