one-api/types/image.go

package types

import "mime/multipart"

type ImageRequest struct {
	Prompt         string `json:"prompt,omitempty"`
	Model          string `json:"model,omitempty"`
	N              int    `json:"n,omitempty"`
	Quality        string `json:"quality,omitempty"`
	Size           string `json:"size,omitempty"`
	Style          string `json:"style,omitempty"`
	ResponseFormat string `json:"response_format,omitempty"`
	User           string `json:"user,omitempty"`
}

type ImageResponse struct {
	Created int64                    `json:"created,omitempty"`
	Data    []ImageResponseDataInner `json:"data,omitempty"`
}

type ImageResponseDataInner struct {
	URL           string `json:"url,omitempty"`
	B64JSON       string `json:"b64_json,omitempty"`
	RevisedPrompt string `json:"revised_prompt,omitempty"`
}

type ImageEditRequest struct {
	Image          *multipart.FileHeader `form:"image"`
	Mask           *multipart.FileHeader `form:"mask"`
	Model          string                `form:"model"`
	Prompt         string                `form:"prompt"`
	N              int                   `form:"n"`
	Size           string                `form:"size"`
	ResponseFormat string                `form:"response_format"`
	User           string                `form:"user"`
}