chatgpt

type ChatCompletionRequest struct {
	// (Required)
	// ID of the model to use.
	Model ChatGPTModel `json:"model"`

	// Required
	// The messages to generate chat completions for
	Messages []ChatMessage `json:"messages"`

	// (Optional - default: 1)
	// What sampling temperature to use, between 0 and 2. Higher values like 0.8 will make the output more random, while lower values like 0.2 will make it more focused and deterministic.
	// We generally recommend altering this or top_p but not both.
	Temperature float64 `json:"temperature,omitempty"`

	// (Optional - default: 1)
	// An alternative to sampling with temperature, called nucleus sampling, where the model considers the results of the tokens with top_p probability mass. So 0.1 means only the tokens comprising the top 10% probability mass are considered.
	// We generally recommend altering this or temperature but not both.
	Top_P float64 `json:"top_p,omitempty"`

	// (Optional - default: 1)
	// How many chat completion choices to generate for each input message.
	N int `json:"n,omitempty"`

	// (Optional - default: infinite)
	// The maximum number of tokens allowed for the generated answer. By default,
	// the number of tokens the model can return will be (4096 - prompt tokens).
	MaxTokens int `json:"max_tokens,omitempty"`

	// (Optional - default: 0)
	// Number between -2.0 and 2.0. Positive values penalize new tokens based on whether they appear in the text so far,
	// increasing the model's likelihood to talk about new topics.
	PresencePenalty float64 `json:"presence_penalty,omitempty"`

	// (Optional - default: 0)
	// Number between -2.0 and 2.0. Positive values penalize new tokens based on their existing frequency in the text so far,
	// decreasing the model's likelihood to repeat the same line verbatim.
	FrequencyPenalty float64 `json:"frequency_penalty,omitempty"`

	// (Optional)
	// A unique identifier representing your end-user, which can help OpenAI to monitor and detect abuse
	User string `json:"user,omitempty"`
}

type ChatGPTModel string

type ChatGPTModelRole string

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

type ChatResponse struct {
	ID        string               `json:"id"`
	Object    string               `json:"object"`
	CreatedAt int64                `json:"created_at"`
	Choices   []ChatResponseChoice `json:"choices"`
	Usage     ChatResponseUsage    `json:"usage"`
}

type ChatResponseChoice struct {
	Index        int         `json:"index"`
	Message      ChatMessage `json:"message"`
	FinishReason string      `json:"finish_reason"`
}

type ChatResponseUsage struct {
	Prompt_Tokens     int `json:"prompt_tokens"`
	Completion_Tokens int `json:"completion_tokens"`
	Total_Tokens      int `json:"total_tokens"`
}

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

type Config struct {
	// Base URL for API requests.
	BaseURL string

	// API Key (Required)
	APIKey string

	// Organization ID (Optional)
	OrganizationID string
}

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

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       FilePurpose `json:"purpose"`
	Status        FileStatus  `json:"status"`         // Deprecated
	StatusDetails string      `json:"status_details"` // Deprecated
}

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

type FilePurpose string

type FileStatus string

type FineTuningEvent struct {
	Object    string `json:"object"`
	ID        string `json:"id"`
	CreatedAt int    `json:"created_at"`
	Level     string `json:"level"`
	Message   string `json:"message"`
	Data      any    `json:"data,omitempty"`
	Type      string `json:"type"`
}

type FineTuningEventsList struct {
	Object  string            `json:"object"`
	Data    []FineTuningEvent `json:"data"`
	HasMore bool              `json:"has_more"`
}

type FineTuningJob struct {
	ID        string `json:"id"`
	CreatedAt int    `json:"created_at"`
	Error     struct {
		Code    string `json:"code"`
		Message string `json:"message"`
		Param   string `json:"param,omitempty"`
	} `json:"error,omitempty"`
	FineTunedModel  string `json:"fine_tuned_model,omitempty"`
	FinishedAt      int    `json:"finished_at,omitempty"`
	Hyperparameters struct {
		NEpochs int `json:"n_epochs"`
	} `json:"hyperparameters,omitempty"`
	Model          string              `json:"model,omitempty"`
	Object         string              `json:"object"`
	OrganizationID string              `json:"organization_id"`
	ResultFiles    []string            `json:"result_files"`
	Status         FineTuningJobStatus `json:"status"`
	TrainedTokens  int                 `json:"trained_tokens,omitempty"`
	TrainingFile   string              `json:"training_file"`
	ValidationFile string              `json:"validation_file,omitempty"`
}

type FineTuningJobStatus string

type FineTuningList struct {
	Object  string          `json:"object"`
	Data    []FineTuningJob `json:"data"`
	HasMore bool            `json:"has_more"`
}

type FineTuningRequest struct {
	Model           ChatGPTModel `json:"model"`
	TrainingFile    string       `json:"training_file"`
	Hyperparameters struct {
		NEpochs int `json:"n_epochs,omitempty"` // Optional
	} `json:"hyperparameters,omitempty"` // Optional
	Suffix         string `json:"suffix,omitempty"`          // Optional
	ValidationFile string `json:"validation_file,omitempty"` // Optional
}

type FineTuningResponse struct {
	Object         string `json:"object"`
	ID             string `json:"id"`
	Model          string `json:"model"`
	CreatedAt      int    `json:"created_at"`
	FineTunedModel any    `json:"fine_tuned_model"`
	OrganizationID string `json:"organization_id"`
	ResultFiles    []any  `json:"result_files"`
	Status         string `json:"status"`
	ValidationFile any    `json:"validation_file"`
	TrainingFile   string `json:"training_file"`
}

type ListOptions struct {
	After *string // Identifier for the last event from the previous pagination request.
	Limit *int    // Number of events to retrieve. Defaults to 20.
}