openai

type Base64 string

type Choice struct {
	Index        int             `json:"index"`
	Message      Message[string] `json:"message"`
	Logprobs     interface{}     `json:"logprobs,omitempty"`
	FinishReason string          `json:"finish_reason"`
}

type ChunkTextOpts struct {
	Text      string
	ChunkSize int
}

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

type CompletionRequest[T any] struct {
	Model      string `json:"model"`
	Messages   T      `json:"messages"`
	ToolChoice string `json:"tool_choice,omitempty"`
	Tools      []Tool `json:"tools,omitempty"`
}

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

type DefaultMessages []Message[string]

type Embedding[Encoding []float64 | Base64] struct {
	Object    string   `json:"object"`
	Embedding Encoding `json:"embedding"`
	Index     int      `json:"index"`
}

type EmbeddingRequest[Input string | []string] struct {
	// Input text to embed, encoded as a string or array of tokens.
	Input Input `json:"input"`
	// ID of the model to use.
	Model string `json:"model"`
	// The format to return the embeddings in. Can be either float or base64.
	Encoding string `json:"encoding_format,omitempty"`
	// The number of dimensions the resulting output embeddings should have. Only supported in text-embedding-3 and later models.
	Dimensions int `json:"dimensions,omitempty"`
	// A unique identifier representing your end-user, which can help OpenAI to monitor and detect abuse.
	User string `json:"user,omitempty"`
}

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

type Function struct {
	Name        string             `json:"name"`
	Description string             `json:"description"`
	Parameters  FunctionParameters `json:"parameters"`
}

type FunctionParameters struct {
	Type               string `json:"type"`
	FunctionProperties `json:"properties"`
}

type FunctionPropertie struct {
	Type        string   `json:"type"`
	Description string   `json:"description"`
	Enum        []string `json:"enum,omitempty"`
}

type FunctionProperties goxios.GenericJSON[FunctionPropertie]

type HTTPClient interface {
	Post(string, *goxios.RequestOpts) (*http.Response, error)
	Get(string, *goxios.RequestOpts) (*http.Response, error)
}

type ImagesGenerationsRequestBody struct {
	Model  string `json:"model"`
	Prompt string `json:"prompt"`
	N      int    `json:"n"`
	Size   string `json:"size"`
	Style  string `json:"style,omitempty"` //This param is only supported for dall-e-3.
}

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

type JSONErr struct {
	Message string `json:"message"`
	Type    string `json:"type"`
	Param   string `json:"param,omitempty"`
	Code    string `json:"code,omitempty"`
}

type MediaMessage struct {
	Type     string    `json:"type"`
	Text     string    `json:"text,omitempty"`
	ImageUrl *imageUrl `json:"image_url,omitempty"`
}

type MediaMessages []Message[[]MediaMessage]

type Message[T string | []MediaMessage] struct {
	Role      string     `json:"role"`
	Content   T          `json:"content"`
	ToolCalls []ToolCall `json:"tool_calls,omitempty"`
}

type ModerationRequest[Input string | []string] struct {
	Input Input  `json:"input"`
	Model string `json:"model,omitempty"`
}

type ModerationResponse struct {
	Id      string `json:"id"`
	Model   string `json:"model"`
	Results []struct {
		Flagged    bool                     `json:"flagged"`
		Categories goxios.GenericJSON[bool] `json:"categories"`
	} `json:"results"`
	CategoryScores goxios.GenericJSON[float64] `json:"category_scores"`
}

type OpenAIClient interface {
	Context() context.Context
	ApiKey() string
	BaseURL() string
	AddHeader(goxios.Header)
	Headers() []goxios.Header
}

type OpenAIErr struct {
	Err JSONErr `json:"error"`
	// contains filtered or unexported fields
}

type SpeechRequestBody struct {
	Model string `json:"model"` // The model for speech synthesis.
	Input string `json:"input"` // The input text for synthesis.
	Voice string `json:"voice"` // The voice to be used for synthesis.
}

type Tool struct {
	Type     string   `json:"type"`
	Function Function `json:"function,omitempty"`
}

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

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

type TranscriptionsRequestBody struct {
	Model, Filename, AudioFilePath string
}

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