openai

type App struct {
	ApiUrl string
	ApiKey string
}

type AudioService interface {
	Transcriptions(ctx context.Context, req *TranscriptionsRequest) (*TranscriptionsResponse, error)
	Translations(ctx context.Context, req *TranslationsRequest) (*TranslationsResponse, error)
}

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

type ChatCompletion struct {
	Index        int64    `json:"index"`
	Delta        *Delta   `json:"delta"`
	Message      *Message `json:"message"`
	FinishReason string   `json:"finish_reason"`
}

type ChatCreateRequest struct {
	Model            string           `json:"model"`
	Messages         []*Message       `json:"messages,omitempty"`
	Functions        []*Function      `json:"functions,omitempty"`
	FunctionCall     IFunctionCall    `json:"function_call,omitempty"`
	Temperature      float64          `json:"temperature,omitempty"`
	TopP             float64          `json:"top_p,omitempty"`
	N                int64            `json:"n,omitempty"`
	Stream           bool             `json:"stream,omitempty"`
	Stop             []string         `json:"stop,omitempty"`
	MaxTokens        int64            `json:"max_tokens,omitempty"`
	PresencePenalty  float64          `json:"presence_penalty,omitempty"`
	FrequencyPenalty float64          `json:"frequency_penalty,omitempty"`
	LogitBias        map[string]int64 `json:"logit_bias,omitempty"`
	User             string           `json:"user,omitempty"`
}

type ChatCreateResponse struct {
	Id      string            `json:"id"`
	Object  string            `json:"object"`
	Created int64             `json:"created"`
	Choices []*ChatCompletion `json:"choices"`
	Usage   Usage             `json:"usage"`
}

type ChatService interface {
	Create(ctx context.Context, req *ChatCreateRequest) (chan *ChatCreateResponse, error)
}

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

type Client struct {
	Models      ModelService
	Completions CompletionService
	Chat        ChatService
	Edits       EditService
	Images      ImageService
	Embeddings  EmbeddingService
	Audio       AudioService
	Files       FileService
	FineTunes   FineTuneService
	Moderations ModerationService
	// contains filtered or unexported fields
}

type Completion struct {
	Text         string `json:"text"`
	Delta        *Delta `json:"delta"`
	Index        int64  `json:"index"`
	Logprobs     int64  `json:"logprobs"`
	FinishReason string `json:"finish_reason"`
}

type CompletionCreateRequest struct {
	Model            string           `json:"model"`
	Prompt           string           `json:"prompt,omitempty"`
	Suffix           string           `json:"suffix,omitempty"`
	MaxTokens        int64            `json:"max_tokens,omitempty"`
	Temperature      float64          `json:"temperature,omitempty"`
	TopP             float64          `json:"top_p,omitempty"`
	N                int64            `json:"n,omitempty"`
	Stream           bool             `json:"stream,omitempty"`
	Logprobs         int64            `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           int64            `json:"best_of,omitempty"`
	LogitBias        map[string]int64 `json:"logit_bias,omitempty"`
	User             string           `json:"user,omitempty"`
}

type CompletionCreateResponse struct {
	Id      string        `json:"id"`
	Object  string        `json:"object"`
	Created int64         `json:"created"`
	Model   string        `json:"model"`
	Choices []*Completion `json:"choices"`
	Usage   Usage         `json:"usage"`
}

type CompletionService interface {
	Create(ctx context.Context, req *CompletionCreateRequest) (chan *CompletionCreateResponse, error)
}

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

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

type Edit struct {
	Text  string `json:"text"`
	Index int64  `json:"index"`
}

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

type EditCreateResponse struct {
	Object  string  `json:"object"`
	Created int64   `json:"created"`
	Choices []*Edit `json:"choices"`
	Usage   Usage   `json:"usage"`
}

type EditService interface {
	Create(ctx context.Context, request *EditCreateRequest) (*EditCreateResponse, error)
}

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

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

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

type EmbeddingCreateResponse struct {
	Object string          `json:"object"`
	Data   []*Embedding    `json:"data"`
	Model  string          `json:"model"`
	Usage  *EmbeddingUsage `json:"usage"`
}

type EmbeddingService interface {
	Create(ctx context.Context, req *EmbeddingCreateRequest) (*EmbeddingCreateResponse, error)
}

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

type EmbeddingUsage struct {
	PromptTokens int64 `json:"prompt_tokens"`
	TotalTokens  int64 `json:"total_tokens"`
}

type Event struct {
	Id    string
	Event string
	Data  string
	Retry time.Duration
	Err   error
}

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

type EventSource chan Event

type File struct {
	Id        string `json:"id"`
	Object    string `json:"object"`
	Bytes     int64  `json:"bytes"`
	CreatedAt int64  `json:"created_at"`
	Filename  string `json:"filename"`
	Purpose   string `json:"purpose"`
}

type FileDeleteResponse struct {
	Id      string `json:"id"`
	Object  string `json:"object"`
	Deleted bool   `json:"deleted"`
}

type FileListResponse struct {
	Data []*File `json:"data"`
}

type FileService interface {
	List(ctx context.Context) (*FileListResponse, error)
	Upload(ctx context.Context, req *FileUploadRequest) (*File, error)
	Delete(ctx context.Context, fileId string) (*FileDeleteResponse, error)
	Retrieve(ctx context.Context, fileId string) (*File, error)
	RetrieveContent(ctx context.Context, fileId string) ([]byte, error)
}

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

type FileUploadRequest struct {
	File    string `json:"file"`
	Purpose string `json:"purpose"`
}

type FineTune struct {
	Id              string           `json:"id"`
	Object          string           `json:"object"`
	Model           string           `json:"model"`
	CreateAt        int64            `json:"create_at"`
	Events          []*FineTuneEvent `json:"events"`
	FineTunedModel  string           `json:"fine_tuned_model"`
	Hyperparams     Hyperparams      `json:"hyperparams"`
	OrganizationId  string           `json:"organization_id"`
	ResultFiles     []*File          `json:"result_files"`
	Status          string           `json:"status"`
	ValidationFiles []*File          `json:"validation_files"`
	TrainingFiles   []*File          `json:"training_files"`
	UpdatedAt       int64            `json:"updated_at"`
}

type FineTuneCreateRequest struct {
	// TrainingFile is the ID to the training file
	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        int64     `json:"classification_n_classes,omitempty"`
	ClassificationPositiveClasses string    `json:"classification_positive_classes,omitempty"`
	ClassificationBetas           []float64 `json:"classification_betas,omitempty"`
	Suffix                        string    `json:"suffix,omitempty"`
}

type FineTuneEvent struct {
	Object   string `json:"object"`
	CreateAt int64  `json:"create_at"`
	Level    string `json:"level"`
	Message  string `json:"message"`
}

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

type FineTuneService interface {
	Create(ctx context.Context, req *FineTuneCreateRequest) (*FineTune, error)
	List(ctx context.Context) (*FineTuneListResponse, error)
	Retrieve(ctx context.Context, id string) (*FineTune, error)
	Cancel(ctx context.Context, id string) (*FineTune, error)
	ListEvents(ctx context.Context, id string, stream ...bool) (chan *EventListResponse, error)
	DeleteModel(ctx context.Context, model string) (*ModelDeleteResponse, error)
}

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

type FormBuilder interface {
	CreateFormFile(name string, filename string) error
	CreateFormField(name string, value string) error
	FormDataContentType() string
	io.Closer
}

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

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

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

type FunctionCall struct {
	Name      string `json:"name"`
	Arguments []byte `json:"arguments,omitempty"`
}

type FunctionCallString string

type Hyperparams struct {
	BatchSize              int64   `json:"batch_size"`
	LearningRateMultiplier float64 `json:"learning_rate_multiplier"`
	NEpochs                int64   `json:"n_epochs"`
	PromptLossWeight       float64 `json:"prompt_loss_weight"`
}

type IFunctionCall interface {
	Call()
}

type Image struct {
	Url string `json:"url"`
}

type ImageAttributes struct {
	N              int    `json:"n,omitempty"`
	Size           string `json:"size,omitempty"`
	ResponseFormat string `json:"response_format,omitempty"`
	User           string `json:"user,omitempty"`
}

type ImageCreateRequest struct {
	Prompt string `json:"prompt"`
	ImageAttributes
}

type ImageEditRequest struct {
	// Image is the path to the image file
	Image string `json:"image"`
	// Mask is the path to the mask file
	Mask   string `json:"mask,omitempty"`
	Prompt string `json:"prompt"`
	ImageAttributes
}

type ImageResponse struct {
	Created int64   `json:"created"`
	Data    []Image `json:"data"`
}

type ImageService interface {
	Create(ctx context.Context, req *ImageCreateRequest) (*ImageResponse, error)
	Edit(ctx context.Context, req *ImageEditRequest) (*ImageResponse, error)
	Variation(ctx context.Context, req *ImageVariationRequest) (*ImageResponse, error)
}

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

type ImageVariationRequest struct {
	Image string `json:"image"`
	ImageAttributes
}

type Message struct {
	Role         string       `json:"role"` // user,assistant,system,function
	Content      string       `json:"content,omitempty"`
	Name         string       `json:"name,omitempty"`
	FunctionCall FunctionCall `json:"function_call,omitempty"`
}

type Model struct {
	Id         string   `json:"id"`
	Object     string   `json:"object"`
	OwnedBy    string   `json:"owned_by"`
	Permission []string `json:"permission"`
}

type ModelDeleteResponse struct {
	Id      string `json:"id"`
	Object  string `json:"object"`
	Deleted bool   `json:"deleted"`
}

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

type ModelService interface {
	List(ctx context.Context) (*ModelResponse, error)
	Retrieve(ctx context.Context, model string) (*Model, error)
}

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

type Moderation struct {
	Categories     ModerationCategory      `json:"categories"`
	CategoryScores ModerationCategoryScore `json:"category_scores"`
	Flagged        bool                    `json:"flagged"`
}

type ModerationCategory 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"`
}

type ModerationCategoryScore 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"`
}

type ModerationCreateRequest struct {
	Input string `json:"input"`
	Model string `json:"model,omitempty"`
}

type ModerationCreateResponse struct {
	Id      string        `json:"id"`
	Model   string        `json:"model"`
	Results []*Moderation `json:"results"`
}

type ModerationService interface {
	Create(ctx context.Context, req *ModerationCreateRequest) (*ModerationCreateResponse, error)
}

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

type Option func(*Client)

type Parameter struct {
	Type       string               `json:"type"` // object only
	Properties map[string]*Property `json:"properties,omitempty"`
	Required   []string             `json:"required,omitempty"`
}

type Property struct {
	Type        string               `json:"type"`
	Description string               `json:"description,omitempty"`
	Properties  map[string]*Property `json:"properties,omitempty"` // if type is object, this field will be set
}

type Proxy struct {
	Url      string
	Username string
	Password string
	TlsCfg   *tls.Config
}

type TranscriptionsRequest struct {
	// File is the path to the audio file
	File           string  `json:"file"`
	Model          string  `json:"model"`
	Prompt         string  `json:"prompt,omitempty"`
	ResponseFormat string  `json:"response_format,omitempty"`
	Temperature    float64 `json:"temperature,omitempty"`
	Language       string  `json:"language,omitempty"`
}

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

type TranslationsRequest struct {
	File           string  `json:"file"`
	Model          string  `json:"model"`
	Prompt         string  `json:"prompt,omitempty"`
	ResponseFormat string  `json:"response_format,omitempty"`
	Temperature    float64 `json:"temperature,omitempty"`
}

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

type Usage struct {
	PromptTokens     int64 `json:"prompt_tokens"`
	CompletionTokens int64 `json:"completion_tokens"`
	TotalTokens      int64 `json:"total_tokens"`
}