chat

type ApiType string

type Client struct{}

type Config struct {
	BaseUrl           string   `yaml:"base_url"`
	ApiVersion        string   `yaml:"api_version"`
	AppToken          string   `yaml:"app_token"`
	Prompt            string   `yaml:"prompt"`
	Temperature       float64  `yaml:"temperature"`
	PresencePenalty   float64  `yaml:"presence_penalty"`
	FrequencyPenalty  float64  `yaml:"frequency_penalty"`
	DefaultPermission []string `yaml:"default_permission"`
}

type DrawCommand struct {
	plugin.ImageReplyCommand
}

type GptCommand struct {
	plugin.TextReplyMessageBaseImplementation
}

type GptCompletionsRequest struct {
	Model            GptModel     `json:"model"`
	Messages         []GptMessage `json:"messages"`
	Temperature      float64      `json:"temperature"`
	PresencePenalty  float64      `json:"presence_penalty"`
	FrequencyPenalty float64      `json:"frequency_penalty"`
	TopP             float64      `json:"top_p"`
	Stream           bool         `json:"stream"`
}

type GptCompletionsResponse struct {
	ID      string            `json:"id"`
	Object  string            `json:"object"`
	Created int64             `json:"created"`
	Model   string            `json:"model"`
	Choices []GptReplyChoices `json:"choices"`
	Usage   GptUsage          `json:"usage"`
}

type GptConfigOptions struct {
	Model            GptModel `json:"model"`
	Temperature      float64  `json:"temperature"`
	PresencePenalty  float64  `json:"presence_penalty"`
	FrequencyPenalty float64  `json:"frequency_penalty"`
}

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

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

type GptMessageRole string

type GptModel string

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

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

type ImageGenerationRequest struct {
	Prompt string `json:"prompt"`
	Number int    `json:"n"`
	Size   string `json:"size"`
}

type ImageGenerationResponse struct {
	Created uint64                        `json:"created"`
	Data    []ImageGenerationResponseData `json:"data"`
}

type ImageGenerationResponseData struct {
	Url        string `json:"url"`
	Base64Json any    `json:"b64_json"`
}

type ImageSize string

type ManagementCommand struct {
	plugin.TextReplyMessageBaseImplementation
}

type Permission struct {
	UserID      string `gorm:"column:user_id;primaryKey;type:varchar(20)"`
	UserName    string `gorm:"column:user_name;type:varchar(255)"`
	Permissions uint64 `gorm:"column:permissions;index"`
}

type PermissionData uint64

type Plugin struct{}