chat_completion

type AssistantChatCompletionMessageUnion struct {
	Refusal      *string                         `json:"refusal,omitempty"`
	Name         *string                         `json:"name,omitempty"`
	Audio        AssistantMessageAudio           `json:"audio,omitempty"`
	Content      AssistantMessageContentUnion    `json:"content,omitempty"`
	FunctionCall AssistantMessageFunctionCall    `json:"function_call,omitempty"`
	ToolCalls    []AssistantMessageToolCallUnion `json:"tool_calls,omitempty"`
	Role         *string                         `json:"role,omitempty"` // "assistant"
}

type AssistantMessageAudio struct {
	ID string `json:"id"`
}

type AssistantMessageContentPartUnion struct {
	OfText    *TextPart    `json:",omitempty"`
	OfRefusal *RefusalPart `json:",omitempty"`
}

type AssistantMessageContentUnion struct {
	OfString *string                            `json:",omitempty"`
	OfList   []AssistantMessageContentPartUnion `json:",omitempty"`
}

type AssistantMessageCustomToolCall struct {
	Type   string                              `json:"type"` // "custom"
	ID     string                              `json:"id,omitempty"`
	Custom AssistantMessageCustomToolCallParam `json:"custom"`
}

type AssistantMessageCustomToolCallParam struct {
	Name  string `json:"name"`
	Input string `json:"input"`
}

type AssistantMessageFunctionCall struct {
	Name      string `json:"name"`
	Arguments string `json:"arguments,omitempty"`
}

type AssistantMessageFunctionToolCall struct {
	Type     string                                `json:"type"` // "function"
	ID       string                                `json:"id,omitempty"`
	Function AssistantMessageFunctionToolCallParam `json:"function"`
}

type AssistantMessageFunctionToolCallParam struct {
	Name      string `json:"name"`
	Arguments string `json:"input"`
}

type AssistantMessageToolCallUnion struct {
	OfFunction AssistantMessageFunctionToolCall `json:",omitempty"`
	OfCustom   AssistantMessageCustomToolCall   `json:",omitempty"`
}

type AudioParam struct {
	Voice  string `json:"voice"`
	Format string `json:"format"`
}

type AudioPart struct {
	Type       string     `json:"type,omitempty"` // "input_audio"
	InputAudio InputAudio `json:"input_audio,omitempty"`
}

type ChatCompletionChunk struct {
	Id                string                      `json:"id"`
	Object            string                      `json:"object"`
	Created           int                         `json:"created"`
	Model             string                      `json:"model"`
	ServiceTier       string                      `json:"service_tier"`
	SystemFingerprint string                      `json:"system_fingerprint"`
	Choices           []ChatCompletionChunkChoice `json:"choices"`
	Obfuscation       string                      `json:"obfuscation"`
	Usage             *Usage                      `json:"usage,omitempty"`
}

type ChatCompletionChunkChoice struct {
	Index        int                            `json:"index"`
	Delta        ChatCompletionChunkChoiceDelta `json:"delta"`
	FinishReason string                         `json:"finish_reason"`
	Obfuscation  *string                        `json:"obfuscation,omitempty"`
}

type ChatCompletionChunkChoiceDelta struct {
	Role    constants.Role `json:"role"`
	Content string         `json:"content"`
	Refusal string         `json:"refusal"`
}

type ChatCompletionMessageUnion struct {
	OfDeveloper *DeveloperChatCompletionMessageUnion `json:",omitzero"`
	OfSystem    *SystemChatCompletionMessageUnion    `json:",omitzero"`
	OfUser      *UserChatCompletionMessageUnion      `json:",omitzero"`
	OfAssistant *AssistantChatCompletionMessageUnion `json:",omitzero"`
	OfTool      *ToolChatCompletionMessageUnion      `json:",omitzero"`
	OfFunction  *FunctionChatCompletionMessageUnion  `json:",omitzero"`
}

type Choice struct {
	FinishReason string         `json:"finish_reason"` // "stop", "length", "content_filter"
	Index        int64          `json:"index"`
	Logprobs     ChoiceLogprobs `json:"logprobs"`
	Message      OutputMessage  `json:"message"`
}

type ChoiceLogprobs struct {
	TextOffset    []int64              `json:"text_offset"`
	TokenLogprobs []float64            `json:"token_logprobs"`
	Tokens        []string             `json:"tokens"`
	TopLogprobs   []map[string]float64 `json:"top_logprobs"`
}

type CompletionUsageCompletionTokensDetails struct {
	// When using Predicted Outputs, the number of tokens in the prediction that
	// appeared in the completion.
	AcceptedPredictionTokens int64 `json:"accepted_prediction_tokens"`
	// Audio input tokens generated by the model.
	AudioTokens int64 `json:"audio_tokens"`
	// Tokens generated by the model for reasoning.
	ReasoningTokens int64 `json:"reasoning_tokens"`
	// When using Predicted Outputs, the number of tokens in the prediction that did
	// not appear in the completion. However, like reasoning tokens, these tokens are
	// still counted in the total completion tokens for purposes of billing, output,
	// and context window limits.
	RejectedPredictionTokens int64 `json:"rejected_prediction_tokens"`
}

type CompletionUsagePromptTokensDetails struct {
	// Audio input tokens present in the prompt.
	AudioTokens int64 `json:"audio_tokens"`
	// Cached tokens present in the prompt.
	CachedTokens int64 `json:"cached_tokens"`
}

type ContentTypeFile string

type ContentTypeImageUrl string

type ContentTypeInputAudio string

type ContentTypeRefusal string

type ContentTypeText string

type DeveloperChatCompletionMessageUnion struct {
	Name    *string                      `json:"name,omitempty"`
	Role    constants.Role               `json:"role,omitempty"` // "developer"
	Content DeveloperMessageContentUnion `json:"content,omitempty"`
}

type DeveloperMessageContentUnion struct {
	OfString *string    `json:",omitempty"`
	OfList   []TextPart `json:",omitempty"`
}

type File struct {
	FileID   *string `json:"file_id,omitempty"`
	Filename *string `json:"filename,omitempty"`
	FileData *string `json:"file_data,omitempty"`
}

type FilePart struct {
	Type string `json:"type,omitempty"`
	File File   `json:"file,omitempty"`
}

type FunctionCallOptionParam struct {
	Name string `json:"name"`
}

type FunctionCallParam struct {
	OfFunctionCallMode   *string                  `json:",omitempty"`
	OfFunctionCallOption *FunctionCallOptionParam `json:",omitempty"`
}

type FunctionChatCompletionMessageUnion struct {
	Name    *string        `json:"name,omitempty"`
	Role    constants.Role `json:"role,omitempty"` //
	Content *string        `json:"content,omitempty"`
}

type FunctionsParam struct {
	Name        string         `json:"name"`
	Description *string        `json:"description,omitempty"`
	Parameters  map[string]any `json:"parameters,omitempty"`
}

type ImagePart struct {
	Type     string   `json:"type,omitempty"` // "image_url"
	ImageUrl ImageUrl `json:"image_url,omitempty"`
}

type ImageUrl struct {
	Url    string `json:"url"`
	Detail string `json:"detail"` // "auto", "low", "high"
}

type InputAudio struct {
	Format string `json:"format,omitzero"` // "wav", "mp3"
	Data   string `json:"data"`
}

type OutputMessage struct {
	Role         constants.Role                   `json:"role"`
	Refusal      string                           `json:"refusal"`
	Content      string                           `json:"content"`
	FunctionCall AssistantMessageFunctionCall     `json:"function_call"`
	ToolCalls    []ToolChatCompletionMessageUnion `json:"tool_calls"`
	Audio        OutputMessageAudio               `json:"audio"`
}

type OutputMessageAudio struct {
}

type RefusalPart struct {
	Type    string `json:"type"` // "refusal"
	Refusal string `json:"refusal,omitempty"`
}

type Request struct {
	Messages            []ChatCompletionMessageUnion `json:"messages"`
	Model               string                       `json:"model"`
	FrequencyPenalty    *float64                     `json:"frequency_penalty,omitempty"`
	Logprobs            *bool                        `json:"logprobs,omitempty"`
	MaxCompletionTokens *int64                       `json:"max_completion_tokens,omitempty"`
	MaxTokens           *int64                       `json:"max_tokens,omitempty"` // Deprecated in favour of `MaxCompletionTokens`
	N                   *int64                       `json:"n,omitempty"`          // How many choices needs to be generated
	PresencePenalty     *float64                     `json:"presence_penalty,omitempty"`
	Seed                *int64                       `json:"seed,omitempty"`
	Store               *bool                        `json:"store,omitempty"`
	Temperature         *float64                     `json:"temperature,omitempty"`
	TopLogprobs         *int64                       `json:"top_logprobs,omitempty"`
	TopP                *float64                     `json:"top_p,omitempty"`
	ParallelToolCalls   *bool                        `json:"parallel_tool_calls,omitempty"`
	PromptCacheKey      *string                      `json:"prompt_cache_key,omitempty"`
	SafetyIdentifier    *string                      `json:"safety_identifier,omitempty"`
	User                *string                      `json:"user,omitempty"`
	Audio               *AudioParam                  `json:"audio,omitempty"`
	LogitBias           map[string]int64             `json:"logit_bias,omitempty"`
	Metadata            map[string]string            `json:"metadata,omitempty"`
	Modalities          []string                     `json:"modalities,omitempty"`       // "text", "audio"
	ReasoningEffort     *string                      `json:"reasoning_effort,omitempty"` // "minimal", "low", "medium", "high"
	ServiceTier         *string                      `json:"service_tier,omitempty"`     // "auto", "default", "flex", "scale", "priority"
	Stop                *StopParam                   `json:"stop,omitempty"`
	Stream              *bool                        `json:"stream,omitempty"`
	StreamOptions       *StreamOptionParam           `json:"stream_options,omitempty"` // Set only when setting stream=true
	Verbosity           *string                      `json:"verbosity,omitempty"`      // "low", "medium", "high"
	FunctionCall        *FunctionCallParam           `json:"function_call,omitempty"`  // Deprecated in favour of `tool_choice`
	Functions           []FunctionsParam             `json:"functions,omitempty"`      // Deprecated in favour of tools
	Prediction          any                          `json:"prediction,omitempty"`
	ResponseFormat      any                          `json:"response_format,omitempty"`
	ToolChoice          *string                      `json:"tool_choice,omitempty"`
	Tools               any                          `json:"tools,omitempty"`
	WebSearchOptions    any                          `json:"web_search_options,omitempty"`
}

type Response struct {
	ID                string   `json:"id"`
	Choices           []Choice `json:"choices"`
	Created           int64    `json:"created"`
	Model             string   `json:"model"`
	Object            string   `json:"object"` // "text_completion"
	SystemFingerprint string   `json:"system_fingerprint"`
	Usage             Usage    `json:"usage"`
}

type ResponseChunk struct {
	OfChatCompletionChunk *ChatCompletionChunk `json:",omitempty"`
}

type StopParam struct {
	OfString *string  `json:",omitempty"`
	OfList   []string `json:",omitempty"`
}

type StreamOptionParam struct {
	IncludeObfuscation *bool `json:"include_obfuscation,omitempty"`
	IncludeUsage       *bool `json:"include_usage,omitempty"`
}

type StringConstant interface {
	Value() string
}

type SystemChatCompletionMessageUnion struct {
	Name    *string                   `json:"name,omitempty"`
	Role    constants.Role            `json:"role,omitempty"` // system
	Content SystemMessageContentUnion `json:"content,omitempty"`
}

type SystemMessageContentUnion struct {
	OfString *string    `json:",omitempty"`
	OfList   []TextPart `json:",omitempty"`
}

type TextPart struct {
	Type string `json:"type,omitempty"` // "text"
	Text string `json:"text,omitempty"`
}

type ToolCallTypeCustom string

type ToolCallTypeFunction string

type ToolChatCompletionMessageUnion struct {
	Role       constants.Role          `json:"role,omitempty"` // tool
	Content    ToolMessageContentUnion `json:"content,omitempty"`
	ToolCallID string                  `json:"tool_call_id,omitempty"`
}

type ToolMessageContentUnion struct {
	OfString *string    `json:",omitempty"`
	OfList   []TextPart `json:",omitempty"`
}

type Usage struct {
	// Number of tokens in the generated completion.
	CompletionTokens int64 `json:"completion_tokens,required"`
	// Number of tokens in the prompt.
	PromptTokens int64 `json:"prompt_tokens,required"`
	// Total number of tokens used in the request (prompt + completion).
	TotalTokens int64 `json:"total_tokens,required"`
	// Breakdown of tokens used in a completion.
	CompletionTokensDetails CompletionUsageCompletionTokensDetails `json:"completion_tokens_details"`
	// Breakdown of tokens used in the prompt.
	PromptTokensDetails CompletionUsagePromptTokensDetails `json:"prompt_tokens_details"`
}

type UserChatCompletionMessageUnion struct {
	Name    *string                 `json:"name,omitempty"`
	Role    constants.Role          `json:"role,omitempty"` // user
	Content UserMessageContentUnion `json:"content,omitempty"`
}

type UserMessageContentPartUnion struct {
	OfText       *TextPart  `json:",omitempty"`
	OfImageUrl   *ImagePart `json:",omitempty"`
	OfInputAudio *AudioPart `json:",omitempty"`
	OfFile       *FilePart  `json:",omitempty"`
}

type UserMessageContentUnion struct {
	OfString *string                       `json:",omitempty"`
	OfList   []UserMessageContentPartUnion `json:",omitempty"`
}