openai

type BadRequestError struct {
	PromptTokens int
	TotalTokens  int
	TokenLimit   int
}

type ClientError struct {
	Status     string
	StatusCode int
}

type ImageRequest struct {
	Model  string `json:"model"`
	Prompt string `json:"prompt"`
	N      int    `json:"n"`
	Size   string `json:"size"`
}

type ImageResponse struct {
	Created int `json:"created"`
	Data    []struct {
		Url string `json:"url"`
	} `json:"data"`
}

type Message interface {
	GetFormat() string
}

type Messages []Message

type Prompt interface {
	GetPurpose() string
	GetHistory() ([]string, []string)
	GetQuestion() string
	GetReferences() [][]byte
}

type RateLimitError struct {
	RetryAfter time.Duration
}

type TTSReqOptions func(TextToSpeechRequestBody) TextToSpeechRequestBody

type TextCompletionRequest struct {
	Model    string   `json:"model"`
	Messages Messages `json:"messages"`
}

type TextCompletionResponse struct {
	Choices []struct {
		Message TextMessage `json:"message"`
	} `json:"choices"`
}

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

type TextToSpeechRequestBody struct {
	Model string `json:"model"`
	Input string `json:"input"`
	Voice Voice  `json:"voice"`
}

type VisionCompletionResponse struct {
	Choices []struct {
		Message struct {
			Content string `json:"content"`
		} `json:"message"`
	} `json:"choices"`
}

type VisionImageURL struct {
	Type     string `json:"type"`
	ImageURL struct {
		URL string `json:"url"`
	} `json:"image_url"`
}

type VisionMessage struct {
	Role    string           `json:"role"`
	Content []VisionImageURL `json:"content"`
}

type VisionRequest struct {
	Model     string   `json:"model"`
	Messages  Messages `json:"messages"`
	MaxTokens int      `json:"max_tokens"`
}

type Voice string