together

type API struct {
	APIKey    string
	BaseURL   string
	UserAgent string

	Client *retryablehttp.Client
	Debug  bool
	// contains filtered or unexported fields
}

type Args struct {
	Model       string  `json:"model"`
	Prompt      string  `json:"prompt"`
	Temperature float64 `json:"temperature"`
	TopP        float64 `json:"top_p"`
	TopK        int     `json:"top_k"`
	MaxTokens   int     `json:"max_tokens"`
}

type ChatCompletionsRequest struct {
	Model             string               `json:"model"`
	Messages          []Message            `json:"messages"`
	Stream            bool                 `json:"stream"`
	MaxTokens         int32                `json:"max_tokens"`
	Stop              []string             `json:"stop"`
	Temperature       float64              `json:"temperature"`
	TopP              float64              `json:"top_p"`
	TopK              int32                `json:"top_k"`
	RepetitionPenalty float64              `json:"repetition_penalty"`
	Logprobs          int32                `json:"logprobs"`
	Echo              bool                 `json:"echo"`
	N                 int32                `json:"n"`
	SafetyModel       string               `json:"safety_model"`
	ResponseFormat    ResponseFormatObject `json:"response_format"`
	Tools             []Tool               `json:"tools"`
	ToolChoice        []ToolChoiceObject   `json:"tool_choice"`
	FrequencyPenalty  float64              `json:"frequency_penalty"`
	PresencePenalty   float64              `json:"presence_penalty"`
	MinP              float64              `json:"min_p"`
}

type ChatCompletionsResponse struct {
	Id      string      `json:"id"`
	Choices []Message   `json:"choices"`
	Usage   UsageObject `json:"usage"`
	Created int         `json:"created"`
	Model   string      `json:"model"`
	Object  string      `json:"object"`
}

type Choice struct {
	FinishReason string `json:"finish_reason"`
	Index        int    `json:"index"`
	Text         string `json:"text"`
}

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

type CompletionsRequest struct {
	Model             string   `json:"model"`
	Prompt            string   `json:"prompt"`
	MaxTokens         int32    `json:"max_tokens"`
	Stream            bool     `json:"stream"`
	Stop              []string `json:"stop"`
	Temperature       float64  `json:"temperature"`
	TopP              float64  `json:"top_p"`
	TopK              int32    `json:"top_k"`
	RepetitionPenalty float64  `json:"repetition_penalty"`
	Logprobs          int32    `json:"logprobs"`
	Echo              bool     `json:"echo"`
	N                 int32    `json:"n"`
	SafetyModel       string   `json:"safety_model"`
}

type CompletionsResponse struct {
	Id      string         `json:"id"`
	Choices []ChoiceObject `json:"choices"`
	Usage   UsageObject    `json:"usage"`
	Created int            `json:"created"`
	Model   string         `json:"model"`
	Object  string         `json:"object"`
}

type EmbeddingsRequest struct {
	Model string `json:"model"`
	Input string `json:"input"`
}

type EmbeddingsResponse struct {
	Status     string   `json:"status"`
	Prompt     []string `json:"prompt"`
	Model      string   `json:"model"`
	ModelOwner string   `json:"model_owner"`
	Tags       Tags     `json:"tags"`
	NumReturns int      `json:"num_returns"`
	Args       Args     `json:"args"`
	Subjobs    []Subjob `json:"subjobs"`
	Output     Output   `json:"output"`
}

type FineTuningResponse struct {
	Status     string   `json:"status"`
	Prompt     []string `json:"prompt"`
	Model      string   `json:"model"`
	ModelOwner string   `json:"model_owner"`
	Tags       Tags     `json:"tags"` // API Documentation states this is an object, but does not define it or provide an example.
	NumReturns int      `json:"num_returns"`
	Args       Args     `json:"args"`
	Subjobs    []Subjob `json:"subjobs"` // API Documentation states this is an array, but does not define it or provide an example.
	Output     Output   `json:"output"`
}

type FunctionObject struct {
	Description string          `json:"description"`
	Name        string          `json:"name"`
	Parameters  json.RawMessage `json:"parameters"` // maybe should be *json.RawMessage ?
}

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

type Output struct {
	Choices        []Choice `json:"choices"`
	RawComputeTime float64  `json:"raw_compute_time"`
	ResultType     string   `json:"result_type"`
}

type ResponseFormatObject struct {
	Type   string          `json:"type"`
	Schema json.RawMessage `json:"schema"` // maybe should be *json.RawMessage ?
}

type Subjob struct{} // TODO: define Subjob.

type Tags struct{} // TODO: define Tags.

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

type ToolChoiceObject struct {
	Type     string         `json:"type"`
	Function FunctionObject `json:"function"`
}

type UsageObject struct {
	PromptTokens     int `json:"prompt_tokens"`
	CompletionTokens int `json:"completion_tokens"`
	TotalTokens      int `json:"total_tokens"`
}