openai

type AssistantMessage struct {
	Content   string     `json:"content,omitempty"`
	Name      string     `json:"name,omitempty"`
	ToolCalls []ToolCall `json:"tool_calls,omitempty"`
}

type ChatChunk struct {
	ID      string   `json:"id"`
	Object  string   `json:"object"`
	Created int64    `json:"created"`
	Model   string   `json:"model"`
	Choices []Choice `json:"choices"`
}

type ChatRequest struct {
	Messages         []Message     `json:"messages"`
	Model            LanguageModel `json:"model"`
	FrequencyPenalty float32       `json:"frequency_penalty,omitempty"`
	LogitBias        float32       `json:"logit_bias,omitempty"`
	LogProbs         bool          `json:"logprobs,omitempty"`
	TopLogProbs      int           `json:"top_logprobs,omitempty"`
	MaxTokens        int           `json:"max_tokens,omitempty"`
	N                int           `json:"n,omitempty"`
	PresencePenalty  float32       `json:"presence_penalty,omitempty"`
	ResponseFormat   string        `json:"response_format,omitempty"` // TODO
	Seed             int           `json:"seed,omitempty"`
	Stop             []string      `json:"stop,omitempty"`
	Stream           bool          `json:"stream,omitempty"`
	Temperature      float32       `json:"temperature,omitempty"`
	TopP             float32       `json:"top_p,omitempty"`
	Tools            []Tool        `json:"tools,omitempty"`
	ToolChoices      []ToolChoice  `json:"tool_choices,omitempty"`
	User             string        `json:"user,omitempty"`
}

type ChatResponse struct {
	ID      string `json:"id"`
	Choices []struct {
		FinishReason string `json:"finish_reason"`
		Index        int    `json:"index"`
		Message      struct {
			Content   string     `json:"content"`
			ToolCalls []ToolCall `json:"tool_calls"`
			Role      string     `json:"role"`
		} `json:"message"`
		LogProbs []LogProb `json:"logprobs"`
	} `json:"choices"`
	Created           int    `json:"created"`
	Model             string `json:"model"`
	SystemFingerprint string `json:"system_fingerprint"`
	Object            string `json:"object"`
	Usage             struct {
		CompletionTokens int `json:"completion_tokens"`
		PromptTokens     int `json:"prompt_tokens"`
		TotalTokens      int `json:"total_tokens"`
	} `json:"usage"`
}

type Choice struct {
	Delta        Delta  `json:"delta"`
	FinishReason string `json:"finish_reason"`
	Index        int    `json:"index"`
}

type Client struct {
	// contains filtered or unexported fields
}

type Content interface {
	Type() string
}

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

type CreateSpeachRequest struct {
	Model          SpeechModel  `json:"model"`
	Input          string       `json:"input"`
	Voice          SpeechVoice  `json:"voice"`
	ResponseFormat SpeechFormat `json:"response_format,omitempty"`
	Speed          float32      `json:"speed,omitempty"`
}

type CreateTranscriptionRequest struct {
	File                   string             `json:"file"`
	Model                  TranscriptionModel `json:"model"`
	Language               string             `json:"language,omitempty"`
	Prompt                 string             `json:"prompt,omitempty"`
	ResponseFormat         TranscriptFormat   `json:"response_format,omitempty"`
	Temperature            float32            `json:"temperature,omitempty"`
	TimestampGranularities []string           `json:"timestamp_granularities,omitempty"`
}

type CreateTranslationRequest struct {
	File           string             `json:"file"`
	Model          TranscriptionModel `json:"model"`
	Prompt         string             `json:"prompt,omitempty"`
	ResponseFormat TranscriptFormat   `json:"response_format,omitempty"`
	Temperature    float32            `json:"temperature,omitempty"`
}

type Delta struct {
	Content string `json:"content"`
	Role    string `json:"role"`
}

type EditImageRequest struct {
	Image          string      `json:"image"`
	Prompt         string      `json:"prompt"`
	Mask           string      `json:"mask,omitempty"`
	Model          ImageModel  `json:"model,omitempty"`
	N              int         `json:"n,omitempty"`
	Size           ImageSize   `json:"size,omitempty"`
	ResponseFormat ImageFormat `json:"response_format,omitempty"`
	User           string      `json:"user,omitempty"`
}

type EmbedRequest struct {
	Input          []string       `json:"input"`
	Model          EmbeddingModel `json:"model"`
	EncodingFormat EncodingFormat `json:"encoding_format,omitempty"`
	Dimensions     int            `json:"dimensions,omitempty"`
	User           string         `json:"user,omitempty"`
}

type EmbedResponse struct {
	Object string      `json:"object"`
	Data   []Embedding `json:"data"`
	Model  string      `json:"model"`
	Usage  struct {
		PromptTokens int `json:"prompt_tokens"`
		TotalTokens  int `json:"total_tokens"`
	} `json:"usage"`
}

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

type EmbeddingModel string

type EncodingFormat string

type ErrorResponse struct {
	Error struct {
		Type    string `json:"type"`
		Code    string `json:"code"`
		Message string `json:"message"`
		Param   string `json:"param"`
	} `json:"error"`
}

type ImageContent struct {
	URL string `json:"url"`
}

type ImageFormat string

type ImageModel string

type ImageResponse struct {
	Created int `json:"created"`
	Data    []struct {
		B64JSON       string `json:"b64_json"`
		URL           string `json:"url"`
		RevisedPrompt string `json:"revised_prompt"`
	} `json:"data"`
}

type ImageSize string

type LanguageModel string

type ListModelsResponse struct {
	Object string  `json:"object"`
	Data   []Model `json:"data"`
}

type LogProb struct {
	Token       string  `json:"token"`
	LogProb     float32 `json:"logprob"`
	Bytes       []byte  `json:"bytes"`
	TopLogProbs []struct {
		Token   string  `json:"token"`
		LogProb float32 `json:"logprob"`
		Bytes   []byte  `json:"bytes"`
	} `json:"top_logprobs"`
}

type Message interface {
	Role() Role
}

type Model struct {
	ID      string `json:"id"`
	Object  string `json:"object"`
	Created int    `json:"created"`
	OwnedBy string `json:"owned_by"`
}

type MultipartFormDataRequest interface {
	AddFields(writer *multipart.Writer) error
}

type Role string

type SpeechFormat string

type SpeechModel string

type SpeechResponse struct {
	Format  SpeechFormat `json:"format"`
	Content []byte       `json:"content"`
}

type SpeechVoice string

type StreamCallback func(ctx context.Context, chunk *ChatChunk)

type SystemMessage struct {
	Content string `json:"content"`
	Name    string `json:"name,omitempty"`
}

type TextContent struct {
	Text string `json:"text"`
}

type Tool struct {
	Name        string         `json:"name"`
	Description string         `json:"description"`
	Parameters  map[string]any `json:"parameters"`
}

type ToolCall struct {
	ID       string `json:"id"`
	Type     string `json:"type"`
	Function struct {
		Name      string `json:"name"`
		Arguments string `json:"arguments"`
	} `json:"function"`
}

type ToolChoice struct {
	Type     string `json:"type"`
	Function struct {
		Name string `json:"name"`
	} `json:"function"`
}

type ToolMessage struct {
	Content    string `json:"content"`
	ToolCallID string `json:"tool_call_id"`
}

type TranscriptFormat string

type TranscriptionModel string

type TranscriptionResponse struct {
	Task     string  `json:"task"`
	Language string  `json:"language"`
	Duration float32 `json:"duration"`
	Text     string  `json:"text"`
	Words    []struct {
		Word  string  `json:"word"`
		Start float32 `json:"start"`
		End   float32 `json:"end"`
	} `json:"words"`
	Segments []struct {
		ID               int     `json:"id"`
		Seek             int     `json:"seek"`
		Start            float32 `json:"start"`
		End              float32 `json:"end"`
		Text             string  `json:"text"`
		Tokens           []int   `json:"tokens"`
		Temperature      float32 `json:"temperature"`
		AvgLogProb       float32 `json:"avg_log_prob"`
		CompressionRatio float32 `json:"compression_ratio"`
		NoSpeechProb     float32 `json:"no_speech_prob"`
	} `json:"segments"`
}

type TranslationResponse struct {
	Text string `json:"text"`
}

type UserMessage struct {
	Content []Content `json:"content"`
	Name    string    `json:"name,omitempty"`
}