protocol

type ChatCompletion struct {
	GeneralResponse
	Usage Usage `json:"usage,omitempty"`
}

type ChatCompletionChunk struct {
	GeneralResponse
	Usage *Usage `json:"usage,omitempty"`
}

type Completion struct {
	GeneralResponse
	Usage Usage `json:"usage,omitempty"`
}

type CompletionChunk struct {
	GeneralResponse
	Usage *Usage `json:"usage,omitempty"`
}

type EmbedRequest struct {
	Input          any    `json:"input"`
	Model          string `json:"model"`
	EncodingFormat string `json:"encoding_format"`
}

type Embedding struct {
	Object    string    `json:"object"`
	Index     int       `json:"index"`
	Embedding []float32 `json:"embedding"`
}

type EmbeddingResponse struct {
	Object string         `json:"object"`
	Data   []Embedding    `json:"data"`
	Model  string         `json:"model"`
	Usage  EmbeddingUsage `json:"usage"`
}

type EmbeddingUsage struct {
	PromptTokens int `json:"prompt_tokens"`
	TotalTokens  int `json:"total_tokens"`
}

type Error struct {
	Message        string  `json:"message"`
	Type           string  `json:"type"`
	SequenceNumber int     `json:"sequence_number,omitempty"`
	Param          *string `json:"param"`
	Code           *string `json:"code"`
}

type ErrorResponse struct {
	Error Error `json:"error"`
}

type GeneralRequest struct {
	Model         string        `json:"model"`
	Stream        bool          `json:"stream"`
	StreamOptions StreamOptions `json:"stream_options,omitempty"`
}

type GeneralResponse struct {
	Id      string `json:"id"`
	Object  string `json:"object"`
	Created int64  `json:"created"`
	Model   string `json:"model"`
}

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

type ImageResponse struct {
	Background   string        `json:"background,omitempty"`
	Created      int64         `json:"created"`
	Data         []ImageObject `json:"data"`
	OutputFormat string        `json:"output_format,omitempty"`
	Quality      string        `json:"quality,omitempty"`
	Size         string        `json:"size,omitempty"`
	Usage        ImageUsage    `json:"usage,omitempty"`
}

type ImageUsage struct {
	InputTokens        int                          `json:"input_tokens"`
	OutputTokens       int                          `json:"output_tokens"`
	TotalTokens        int                          `json:"total_tokens"`
	InputTokensDetails ImageUsageInputTokensDetails `json:"input_tokens_details,omitempty"`
}

type ImageUsageInputTokensDetails struct {
	ImageTokens int `json:"image_tokens"`
	TextTokens  int `json:"text_tokens"`
}

type StreamOptions struct {
	IncludeUsage *bool `json:"include_usage,omitempty"`
}

type Usage struct {
	PromptTokens     int `json:"prompt_tokens"`
	CompletionTokens int `json:"completion_tokens"`
	TotalTokens      int `json:"total_tokens"`
}