初始化导入 new-api 源码

relay/channel/xai/dto.go

@@ -0,0 +1,27 @@
+package xai
+
+import "github.com/QuantumNous/new-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"`
+}