tgbot_misaka_5882f7/openai/chat.go

73 lines
2.6 KiB
Go
Raw Normal View History

2023-03-08 03:28:26 +09:00
package openai
const ChatAPIPath = "/v1/chat/completions"
type ChatRole string
const (
ChatRoleSystem ChatRole = "system"
ChatRoleAssistant ChatRole = "assistant"
ChatRoleUser ChatRole = "user"
)
type ChatMessage struct {
Role ChatRole `json:"role"`
Content string `json:"content"`
}
type ChatRequest struct {
Model string `json:"model"`
Messages []ChatMessage `json:"messages"`
2023-03-21 14:35:25 +09:00
Temperature *float64 `json:"temperature,omitempty"` // What sampling temperature to use, between 0 and 2.
TopP *float64 `json:"top_p,omitempty"` // Nucleus sampling. Specify this or temperature but not both.
N int `json:"n,omitempty"` // How many chat completion choices to generate for each input message.
Stream bool `json:"stream,omitempty"` // If set, partial message deltas will be sent as data-only server-sent events as they become available.
Stop []string `json:"stop,omitempty"` // Up to 4 sequences where the API will stop generating further tokens.
2023-03-08 03:28:26 +09:00
MaxTokens int `json:"max_tokens,omitempty"`
2023-03-21 14:35:25 +09:00
PresencePenalty *float64 `json:"presence_penalty,omitempty"` // Number between -2.0 and 2.0.
FrequencyPenalty *float64 `json:"frequency_penalty,omitempty"` // Number between -2.0 and 2.0.
LogitBias map[string]float64 `json:"logit_bias,omitempty"` // Modify the likelihood of specified tokens appearing in the completion.
User string `json:"user,omitempty"` // A unique identifier representing your end-user, which can help OpenAI to monitor and detect abuse.
2023-03-08 03:28:26 +09:00
}
type ChatResponseChoice struct {
Message ChatMessage `json:"message"`
FinishReason string `json:"finish_reason"`
Index int `json:"index"`
}
type ChatResponse struct {
ID string `json:"id"`
Object string `json:"object"`
Created int `json:"created"`
Model string `json:"model"`
Usage map[string]int `json:"usage"`
Choices []ChatResponseChoice `json:"choices"`
}
2023-03-19 15:45:20 +09:00
type ChatResponseStream struct {
ID string
Object string
Created int
Model string
Stream chan string
Done chan struct{}
2023-03-19 15:45:20 +09:00
Err error
}
type ChatResponseStreamChunk struct {
ID string
Object string
Created int
Model string
Choices []ChatResponseStreamChoice
}
type ChatResponseStreamChoice struct {
Index int `json:"index"`
FinishReason string `json:"finish_reason"`
Delta struct {
Content string `json:"content"`
} `json:"delta"`
}