openai

type ChatCompletionRequest struct {
	Model            string                         `json:"model"`
	Messages         []ChatCompletionRequestMessage `json:"messages"`
	Temperature      float64                        `json:"temperature,omitempty"`
	TopP             float64                        `json:"top_p,omitempty"`
	N                int                            `json:"n,omitempty"`
	Stream           bool                           `json:"stream,omitempty"`
	Stop             []string                       `json:"stop,omitempty"`
	MaxTokens        int                            `json:"max_tokens,omitempty"`
	PresencePenalty  float64                        `json:"presence_penalty,omitempty"`
	FrequencyPenalty float64                        `json:"frequency_penalty,omitempty"`
	LogitBias        map[string]int                 `json:"logit_bias,omitempty"`
	User             string                         `json:"user,omitempty"`
}

type ChatCompletionRequestMessage struct {
	Role    ChatRole `json:"role"`
	Content string   `json:"content"`
	Name    string   `json:"name,omitempty"`
}

type ChatCompletionResponse struct {
	ID      string `json:"id"`
	Object  string `json:"object"`
	Created int    `json:"created"`
	Model   string `json:"model"`
	Choices []struct {
		Index        int                           `json:"index"`
		Message      ChatCompletionResponseMessage `json:"message"`
		FinishReason string                        `json:"finish_reason"`
	} `json:"choices"`
	Usage struct {
		PromptTokens     int `json:"prompt_tokens"`
		CompletionTokens int `json:"completion_tokens"`
		TotalTokens      int `json:"total_tokens"`
	} `json:"usage"`
}

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

type ChatRole string

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

type ClientOption func(*Client)

type CompletionRequest struct {
	Model            string         `json:"model"`
	Prompt           []string       `json:"prompt,omitempty"`
	Suffix           string         `json:"suffix,omitempty"`
	MaxTokens        int            `json:"max_tokens,omitempty"`
	Temperature      float64        `json:"temperature,omitempty"`
	TopP             float64        `json:"top_p,omitempty"`
	N                int            `json:"n,omitempty"`
	Stream           bool           `json:"stream,omitempty"`
	Logprobs         int            `json:"logprobs,omitempty"`
	Echo             bool           `json:"echo,omitempty"`
	Stop             []string       `json:"stop,omitempty"`
	PresencePenalty  float64        `json:"presence_penalty,omitempty"`
	FrequencyPenalty float64        `json:"frequency_penalty,omitempty"`
	BestOf           int            `json:"best_of,omitempty"`
	LogitBias        map[string]int `json:"logit_bias,omitempty"`
	User             string         `json:"user,omitempty"`
}

type CompletionResponse struct {
	ID      string `json:"id"`
	Object  string `json:"object"`
	Created int    `json:"created"`
	Model   string `json:"model"`
	Choices []struct {
		Text     string `json:"text"`
		Index    int    `json:"index"`
		Logprobs struct {
			Tokens        []string             `json:"tokens"`
			TokenLogprobs []float64            `json:"token_logprobs"`
			TopLogprobs   []map[string]float64 `json:"top_logprobs"`
			TextOffset    []int                `json:"text_offset"`
		} `json:"logprobs"`
		FinishReason string `json:"finish_reason"`
	} `json:"choices"`
	Usage struct {
		PromptTokens     int `json:"prompt_tokens"`
		CompletionTokens int `json:"completion_tokens"`
		TotalTokens      int `json:"total_tokens"`
	} `json:"usage"`
}

type DeleteFileResponse struct {
	ID      string `json:"id"`
	Object  string `json:"object"`
	Deleted bool   `json:"deleted"`
}

type DeleteModelResponse struct {
	ID      string `json:"id"`
	Object  string `json:"object"`
	Deleted bool   `json:"deleted"`
}

type EditRequest struct {
	Model       string  `json:"model"`
	Input       string  `json:"input,omitempty"`
	Instruction string  `json:"instruction"`
	N           int     `json:"n,omitempty"`
	Temperature float64 `json:"temperature,omitempty"`
	TopP        float64 `json:"top_p,omitempty"`
}

type EditResponse struct {
	ID      string `json:"id"`
	Object  string `json:"object"`
	Created int    `json:"created"`
	Model   string `json:"model"`
	Choices []struct {
		Text     string `json:"text"`
		Index    int    `json:"index"`
		Logprobs struct {
			Tokens        []string             `json:"tokens"`
			TokenLogprobs []float64            `json:"token_logprobs"`
			TopLogprobs   []map[string]float64 `json:"top_logprobs"`
			TextOffset    []int                `json:"text_offset"`
		} `json:"logprobs"`
		FinishReason string `json:"finish_reason"`
	} `json:"choices"`
	Usage struct {
		PromptTokens     int `json:"prompt_tokens"`
		CompletionTokens int `json:"completion_tokens"`
		TotalTokens      int `json:"total_tokens"`
	} `json:"usage"`
}

type EmbeddingRequest struct {
	Model string   `json:"model"`
	Input []string `json:"input"`
	User  string   `json:"user,omitempty"`
}

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

type Error struct {
	StatusCode int    `json:"-"`
	Message    string `json:"message"`
	Type       string `json:"type"`
}

type File struct {
	ID            string         `json:"id"`
	Object        string         `json:"object"`
	Bytes         int            `json:"bytes"`
	CreatedAt     int            `json:"created_at"`
	Filename      string         `json:"filename"`
	Purpose       string         `json:"purpose"`
	Status        string         `json:"status"`
	StatusDetails map[string]any `json:"status_details"`
}

type FilesResponse struct {
	Object string `json:"object"`
	Data   []File `json:"data"`
}

type FineTune struct {
	ID              string          `json:"id"`
	Object          string          `json:"object"`
	CreatedAt       int             `json:"created_at"`
	UpdatedAt       int             `json:"updated_at"`
	Model           string          `json:"model"`
	FineTunedModel  string          `json:"fine_tuned_model"`
	OrganizationID  string          `json:"organization_id"`
	Status          string          `json:"status"`
	Hyperparams     map[string]any  `json:"hyperparams"`
	TrainingFiles   []File          `json:"training_files"`
	ValidationFiles []File          `json:"validation_files"`
	ResultFiles     []File          `json:"result_files"`
	Events          []FineTuneEvent `json:"events"`
}

type FineTuneEvent struct {
	Object    string `json:"object"`
	CreatedAt int    `json:"created_at"`
	Level     string `json:"level"`
	Message   string `json:"message"`
}

type FineTuneEventsResponse struct {
	Object string          `json:"object"`
	Data   []FineTuneEvent `json:"data"`
}

type FineTuneRequest struct {
	TrainingFile                 string    `json:"training_file"`
	ValidationFile               string    `json:"validation_file,omitempty"`
	Model                        string    `json:"model,omitempty"`
	NEpochs                      int       `json:"n_epochs,omitempty"`
	BatchSize                    int       `json:"batch_size,omitempty"`
	LearningRateMultiplier       float64   `json:"learning_rate_multiplier,omitempty"`
	PromptLossWeight             float64   `json:"prompt_loss_weight,omitempty"`
	ComputeClassificationMetrics bool      `json:"compute_classification_metrics,omitempty"`
	ClassificationNClasses       int       `json:"classification_n_classes,omitempty"`
	ClassificationPositiveClass  string    `json:"classification_positive_class,omitempty"`
	ClassificationBetas          []float64 `json:"classification_betas,omitempty"`
	Suffix                       string    `json:"suffix,omitempty"`
}

type FineTunesResponse struct {
	Object string     `json:"object"`
	Data   []FineTune `json:"data"`
}

type ImageEditRequest struct {
	Image          formdata.File       `form:"image"`
	Mask           formdata.File       `form:"mask,omitempty"`
	Prompt         string              `form:"prompt"`
	N              int                 `form:"n,omitempty"`
	Size           ImageSize           `form:"size,omitempty"`
	ResponseFormat ImageResponseFormat `form:"response_format,omitempty"`
	User           string              `form:"user,omitempty"`
}

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

type ImageRequest struct {
	Prompt         string              `json:"prompt"`
	N              int                 `json:"n,omitempty"`
	Size           ImageSize           `json:"size,omitempty"`
	ResponseFormat ImageResponseFormat `json:"response_format,omitempty"`
	User           string              `json:"user,omitempty"`
}

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

type ImageResponseFormat string

type ImageSize string

type ImageVariationRequest struct {
	Image          formdata.File       `form:"image"`
	N              int                 `form:"n,omitempty"`
	Size           ImageSize           `form:"size,omitempty"`
	ResponseFormat ImageResponseFormat `form:"response_format,omitempty"`
	User           string              `form:"user,omitempty"`
}

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

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

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

type ModerationRequest struct {
	Input []string `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 struct {
			Hate            bool `json:"hate"`
			HateThreatening bool `json:"hate/threatening"`
			SelfHarm        bool `json:"self-harm"`
			Sexual          bool `json:"sexual"`
			SexualMinors    bool `json:"sexual/minors"`
			Violence        bool `json:"violence"`
			ViolenceGraphic bool `json:"violence/graphic"`
		} `json:"categories"`
		CategoryScores struct {
			Hate            float64 `json:"hate"`
			HateThreatening float64 `json:"hate/threatening"`
			SelfHarm        float64 `json:"self-harm"`
			Sexual          float64 `json:"sexual"`
			SexualMinors    float64 `json:"sexual/minors"`
			Violence        float64 `json:"violence"`
			ViolenceGraphic float64 `json:"violence/graphic"`
		} `json:"category_scores"`
	} `json:"results"`
}

type TranscriptionRequest struct {
	File           formdata.File               `form:"file"`
	Model          string                      `json:"model"`
	Prompt         string                      `json:"prompt,omitempty"`
	ResponseFormat TranscriptionResponseFormat `json:"response_format,omitempty"`
	Temperature    float64                     `json:"temperature,omitempty"`
	Language       string                      `json:"language,omitempty"`
}

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

type TranscriptionResponseFormat string

type TranslationRequest struct {
	File           formdata.File             `form:"file"`
	Model          string                    `json:"model"`
	Prompt         string                    `json:"prompt,omitempty"`
	ResponseFormat TranslationResponseFormat `json:"response_format,omitempty"`
	Temperature    float64                   `json:"temperature,omitempty"`
}

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

type TranslationResponseFormat string

type UploadFileRequest struct {
	File    formdata.File `form:"file"`
	Purpose string        `form:"purpose"`
}