chatgpt

type ChatCompChoice struct {
	Message      ChatCompMsg `json:"message,omitempty"`
	FinishReason string      `json:"finish_reason,omitempty"`
	Index        uint        `json:"index,omitempty"`
}

type ChatCompMsg struct {
	// enum: user, system, assistant
	Role    string `json:"role"`
	Content string `json:"content"`
}

type ChatCompReq struct {
	Model       string        `json:"model"`
	Temperature float32       `json:"temperature"`
	Messages    []ChatCompMsg `json:"messages"`
}

type ChatCompRes struct {
	Id      string `json:"id,omitempty"`
	Object  string `json:"object,omitempty"`
	Created int64  `json:"created,omitempty"`
	Model   string `json:"model,omitempty"`
	Usage   struct {
		PromptTokens     uint32 `json:"prompt_tokens,omitempty"`
		CompletionTokens uint32 `json:"completion_tokens,omitempty"`
		TotalTokens      uint32 `json:"total_tokens,omitempty"`
	} `json:"usage,omitempty"`
	Choices []ChatCompChoice `json:"choices,omitempty"`
}

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

type Config struct {
	Url     string `mapstructure:"url"`
	API_Key string `mapstructure:"api_key"`
	ORG_ID  string `mapstructure:"org_id"`

	Proxy         string `mapstructure:"proxy"`
	TlsSkipVerify bool   `mapstructure:"tls_skip_verify"`
}

type ImgGenReq struct {
	Prompt string `json:"prompt"`
	N      uint32 `json:"n,omitempty"`
	Size   string `json:"size,omitempty"`
}

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

type Model struct {
	Id         string       `json:"id,omitempty"`
	Object     string       `json:"object,omitempty"`
	Created    int64        `json:"created,omitempty"`
	OwnedBy    string       `json:"owned_by,omitempty"`
	Root       string       `json:"root,omitempty"`
	Permission []Permission `json:"permission,omitempty"`
}

type ModelsRes struct {
	Oobject string  `json:"object,omitempty"`
	Data    []Model `json:"data,omitempty"`
}

type Permission struct {
	Id                  string `json:"id,omitempty"`
	Object              string `json:"object,omitempty"`
	Created             int64  `json:"created,omitempty"`
	Organization        string `json:"organization,omitempty"`
	AllowCreateEngine   bool   `json:"allow_create_engine,omitempty"`
	AllowSampling       bool   `json:"allow_sampling,omitempty"`
	AllowLogprobs       bool   `json:"allow_logprobs,omitempty"`
	AllowSearch_indices bool   `json:"allow_search_indices,omitempty"`
	AllowView           bool   `json:"allow_view,omitempty"`
	AllowFineTuning     bool   `json:"allow_fine_tuning,omitempty"`
	IsBlocking          bool   `json:"is_blocking,omitempty"`
}