common

type AdminLogin struct {
	AccountLogin  bool `bson:"account_login"  json:"account_login"`  // 账密登录
	EmailLogin    bool `bson:"email_login"    json:"email_login"`    // 邮箱登录
	EmailRetrieve bool `bson:"email_retrieve" json:"email_retrieve"` // 找回密码
	SessionExpire int  `bson:"session_expire" json:"session_expire"` // 会话过期, 单位: 秒
}

type Announcement struct {
	Content string `bson:"content,omitempty"  json:"content,omitempty"`  // 公告内容
	JumpUrl string `bson:"jump_url,omitempty" json:"jump_url,omitempty"` // 跳转URL
}

type AudioQuota struct {
	BillingMethod   int     `bson:"billing_method,omitempty"   json:"billing_method,omitempty"`         // 计费方式[1:倍率, 2:固定额度]
	PromptRatio     float64 `bson:"prompt_ratio,omitempty"     json:"prompt_ratio,omitempty"     d:"1"` // 提示倍率(提问倍率)
	CompletionRatio float64 `bson:"completion_ratio,omitempty" json:"completion_ratio,omitempty" d:"1"` // 补全倍率(回答倍率)
	FixedQuota      int     `bson:"fixed_quota"                json:"fixed_quota"`                      // 固定额度
}

type AutoDisabledError struct {
	Open   bool     `bson:"open"   json:"open"`   // 开关
	Errors []string `bson:"errors" json:"errors"` // 错误
}

type AutoEnableError struct {
	Open         bool          `bson:"open"          json:"open"`          // 开关
	EnableErrors []EnableError `bson:"enable_errors" json:"enable_errors"` // 启用错误
}

type Base struct {
	ErrRetry                int   `bson:"err_retry"                   json:"err_retry"`                   // 错误重试次数
	ModelKeyErrDisable      int64 `bson:"model_key_err_disable"       json:"model_key_err_disable"`       // 模型密钥禁用次数
	ModelAgentErrDisable    int64 `bson:"model_agent_err_disable"     json:"model_agent_err_disable"`     // 模型代理禁用次数
	ModelAgentKeyErrDisable int64 `bson:"model_agent_key_err_disable" json:"model_agent_key_err_disable"` // 模型代理密钥禁用次数
}

type Carousel struct {
	ImageUrl string `bson:"image_url,omitempty" json:"image_url,omitempty"` // 图像URL
	JumpUrl  string `bson:"jump_url,omitempty"  json:"jump_url,omitempty"`  // 跳转URL
}

type Core struct {
	SecretKeyPrefix string `bson:"secret_key_prefix" json:"secret_key_prefix"` // 密钥前缀
	ErrorPrefix     string `bson:"error_prefix"      json:"error_prefix"`      // 错误码前缀
	ChannelPrefix   string `bson:"channel_prefix"    json:"channel_prefix"`    // 通道前缀
}

type Debug struct {
	Open bool `bson:"open" json:"open"` // 开关
}

type Document struct {
	Title   string `bson:"title,omitempty"    json:"title,omitempty"`    // 文档标题
	JumpUrl string `bson:"jump_url,omitempty" json:"jump_url,omitempty"` // 跳转URL
}

type Email struct {
	Open     bool   `bson:"open"      json:"open"`      // 开关
	Host     string `bson:"host"      json:"host"`      // smtp.xxx.com
	Port     int    `bson:"port"      json:"port"`      // 端口号
	UserName string `bson:"user_name" json:"user_name"` // 登录账号
	Password string `bson:"password"  json:"password"`  // 登录密码
	FromName string `bson:"from_name" json:"from_name"` // 发送人名称
}

type EnableError struct {
	Cron       string        `bson:"cron"        json:"cron"`        // CRON表达式
	EnableTime time.Duration `bson:"enable_time" json:"enable_time"` // 启用时间, 单位: 秒
	Error      string        `bson:"error"       json:"error"`       // 错误
}

type FallbackConfig struct {
	ModelAgent     string `bson:"model_agent,omitempty"      json:"model_agent,omitempty"`      // 后备模型代理
	ModelAgentName string `bson:"model_agent_name,omitempty" json:"model_agent_name,omitempty"` // 后备模型代理名称
	Model          string `bson:"model,omitempty"            json:"model,omitempty"`            // 后备模型
	ModelName      string `bson:"model_name,omitempty"       json:"model_name,omitempty"`       // 后备模型名称
}

type ForwardConfig struct {
	ForwardRule       int      `bson:"forward_rule,omitempty"   json:"forward_rule,omitempty"`        // 转发规则[1:全部转发, 2:按关键字, 3:内容长度]
	MatchRule         []int    `bson:"match_rule,omitempty"     json:"match_rule,omitempty"`          // 转发规则为2时的匹配规则[1:智能匹配, 2:正则匹配]
	TargetModel       string   `bson:"target_model,omitempty"   json:"target_model,omitempty"`        // 转发规则为1和3时的目标模型
	DecisionModel     string   `bson:"decision_model,omitempty" json:"decision_model,omitempty"`      // 转发规则为2时并且匹配规则为1时的判定模型
	Keywords          []string `bson:"keywords,omitempty"       json:"keywords,omitempty"`            // 转发规则为2时的关键字
	TargetModels      []string `bson:"target_models,omitempty"  json:"target_models,omitempty"`       // 转发规则为2时的目标模型
	ContentLength     int      `bson:"content_length,omitempty" json:"content_length,omitempty"`      // 转发规则为3时的内容长度
	TargetModelName   string   `bson:"-"                        json:"target_model_name,omitempty"`   // 转发规则为1和3时的目标模型名称
	DecisionModelName string   `bson:"-"                        json:"decision_model_name,omitempty"` // 转发规则为2时并且匹配规则为1时的判定模型名称
	TargetModelNames  []string `bson:"-"                        json:"target_model_names,omitempty"`  // 转发规则为2时的目标模型名称
}

type Http struct {
	Timeout  time.Duration `bson:"timeout"   json:"timeout"`   // 超时时间
	ProxyUrl string        `bson:"proxy_url" json:"proxy_url"` // 代理地址
}

type ImageData struct {
	URL           string `bson:"url,omitempty"`
	B64JSON       string `bson:"b64_json,omitempty"`
	RevisedPrompt string `bson:"revised_prompt,omitempty"`
}

type ImageQuota struct {
	Width      int    `bson:"width,omitempty"      json:"width,omitempty"`      // 宽度
	Height     int    `bson:"height,omitempty"     json:"height,omitempty"`     // 高度
	Mode       string `bson:"mode,omitempty"       json:"mode,omitempty"`       // 模式[low, high, auto]
	FixedQuota int    `bson:"fixed_quota"          json:"fixed_quota"`          // 固定额度
	IsDefault  bool   `bson:"is_default,omitempty" json:"is_default,omitempty"` // 是否默认选项
}

type Log struct {
	Open         bool          `bson:"open"          json:"open"`          // 开关
	ChatRecords  []string      `bson:"chat_records"  json:"chat_records"`  // 聊天日志记录内容
	ChatReserve  time.Duration `bson:"chat_reserve"  json:"chat_reserve"`  // 聊天日志保留天数
	ImageReserve time.Duration `bson:"image_reserve" json:"image_reserve"` // 绘图日志保留天数
	AudioReserve time.Duration `bson:"audio_reserve" json:"audio_reserve"` // 音频日志保留天数
	Status       []int         `bson:"status"        json:"status"`        // 删除日志状态
	Cron         string        `bson:"cron"          json:"cron"`          // CRON表达式
}

type Message struct {
	Role         string               `bson:"role,omitempty"          json:"role,omitempty"`    // 角色
	Content      string               `bson:"content,omitempty"       json:"content,omitempty"` // 内容
	Refusal      string               `bson:"refusal,omitempty"       json:"refusal,omitempty"`
	Name         string               `bson:"name,omitempty"          json:"name,omitempty"`
	FunctionCall *openai.FunctionCall `bson:"function_call,omitempty" json:"function_call,omitempty"`
	ToolCalls    []openai.ToolCall    `bson:"tool_calls,omitempty"    json:"tool_calls,omitempty"`
	ToolCallId   string               `bson:"tool_call_id,omitempty"  json:"tool_call_id,omitempty"`
	Audio        *openai.Audio        `bson:"audio,omitempty"         json:"audio,omitempty"`
}

type Midjourney struct {
	Open            bool   `bson:"open"              json:"open"` // 开关
	CdnUrl          string `bson:"cdn_url"           json:"cdn_url"`
	ApiBaseUrl      string `bson:"api_base_url"      json:"api_base_url"`
	ApiSecret       string `bson:"api_secret"        json:"api_secret"`
	ApiSecretHeader string `bson:"api_secret_header" json:"api_secret_header"`
	CdnOriginalUrl  string `bson:"cdn_original_url"  json:"cdn_original_url"`
}

type MidjourneyQuota struct {
	Name       string `bson:"name,omitempty"        json:"name,omitempty"`   // 名称
	Action     string `bson:"action,omitempty"      json:"action,omitempty"` // 动作[IMAGINE, UPSCALE, VARIATION, ZOOM, PAN, DESCRIBE, BLEND, SHORTEN, SWAP_FACE]
	Path       string `bson:"path,omitempty"        json:"path,omitempty"`   // 路径
	FixedQuota int    `bson:"fixed_quota"           json:"fixed_quota"`      // 固定额度
}

type ModelStat struct {
	ModelId        string `bson:"model_id,omitempty"        json:"model_id,omitempty"`        // 模型ID
	Model          string `bson:"model,omitempty"           json:"model,omitempty"`           // 模型
	Total          int    `bson:"total,omitempty"           json:"total,omitempty"`           // 总数
	Tokens         int    `bson:"tokens,omitempty"          json:"tokens,omitempty"`          // 令牌数
	Abnormal       int    `bson:"abnormal,omitempty"        json:"abnormal,omitempty"`        // 异常数
	AbnormalTokens int    `bson:"abnormal_tokens,omitempty" json:"abnormal_tokens,omitempty"` // 异常令牌数
}

type MultimodalAudioQuota struct {
	TextQuota  TextQuota  `bson:"text_quota,omitempty"  json:"text_quota,omitempty"`  // 文本额度
	AudioQuota AudioQuota `bson:"audio_quota,omitempty" json:"audio_quota,omitempty"` // 音频额度
	FixedQuota int        `bson:"fixed_quota"           json:"fixed_quota"`           // 固定额度
}

type MultimodalQuota struct {
	BillingRule  int           `bson:"billing_rule,omitempty" json:"billing_rule,omitempty"` // 计费规则[1:按官方, 2:按系统]
	TextQuota    TextQuota     `bson:"text_quota,omitempty"   json:"text_quota,omitempty"`   // 文本额度
	ImageQuotas  []ImageQuota  `bson:"image_quotas,omitempty" json:"image_quotas,omitempty"` // 图像额度
	SearchQuota  int           `bson:"search_quota"           json:"search_quota"`           // 搜索额度
	SearchQuotas []SearchQuota `bson:"search_quotas"          json:"search_quotas"`          // 搜索额度
}

type NotRetryError struct {
	Open   bool     `bson:"open"   json:"open"`   // 开关
	Errors []string `bson:"errors" json:"errors"` // 错误
}

type NotShieldError struct {
	Open   bool     `bson:"open"   json:"open"`   // 开关
	Errors []string `bson:"errors" json:"errors"` // 错误
}

type Notice struct {
	Open        bool          `bson:"open"         json:"open"`         // 开关
	Cron        string        `bson:"cron"         json:"cron"`         // CRON表达式
	LockMinutes time.Duration `bson:"lock_minutes" json:"lock_minutes"` // 锁定时长, 单位: 分钟
}

type PresetConfig struct {
	IsSupportSystemRole bool   `bson:"is_support_system_role" json:"is_support_system_role,omitempty"` // 是否支持system角色
	SystemRolePrompt    string `bson:"system_role_prompt"     json:"system_role_prompt,omitempty"`     // system角色预设提示词
	MinTokens           int    `bson:"min_tokens"             json:"min_tokens,omitempty"`             // max_tokens取值的最小值
	MaxTokens           int    `bson:"max_tokens"             json:"max_tokens,omitempty"`             // max_tokens取值的最大值
}

type QuotaWarning struct {
	Open             bool          `bson:"open"              json:"open"`              // 额度预警开关
	Threshold        int           `bson:"threshold"         json:"threshold"`         // 额度预警阈值, 单位: $
	ExhaustionNotice bool          `bson:"exhaustion_notice" json:"exhaustion_notice"` // 额度耗尽通知开关
	ExpireWarning    bool          `bson:"expire_warning"    json:"expire_warning"`    // 额度过期预警开关
	ExpireThreshold  time.Duration `bson:"expire_threshold"  json:"expire_threshold"`  // 额度过期预警阈值, 单位: 天
	ExpireNotice     bool          `bson:"expire_notice"     json:"expire_notice"`     // 额度过期通知开关
}

type RealtimeQuota struct {
	TextQuota  TextQuota  `bson:"text_quota,omitempty"  json:"text_quota,omitempty"`  // 文本额度
	AudioQuota AudioQuota `bson:"audio_quota,omitempty" json:"audio_quota,omitempty"` // 音频额度
	FixedQuota int        `bson:"fixed_quota"           json:"fixed_quota"`           // 固定额度
}

type Retry struct {
	IsRetry    bool   `bson:"is_retry,omitempty"    json:"is_retry,omitempty"`    // 是否重试
	RetryCount int    `bson:"retry_count,omitempty" json:"retry_count,omitempty"` // 重试次数
	ErrMsg     string `bson:"err_msg,omitempty"     json:"err_msg,omitempty"`     // 错误信息
}

type SearchQuota struct {
	SearchContextSize string `bson:"search_context_size,omitempty" json:"search_context_size,omitempty"` // 搜索上下文大小[high, medium, low]
	FixedQuota        int    `bson:"fixed_quota"                   json:"fixed_quota"`                   // 固定额度
	IsDefault         bool   `bson:"is_default,omitempty"          json:"is_default,omitempty"`          // 是否默认选项
}

type Statistics struct {
	Open        bool          `bson:"open"         json:"open"`         // 开关
	Cron        string        `bson:"cron"         json:"cron"`         // CRON表达式
	Limit       int64         `bson:"limit"        json:"limit"`        // 查询条数
	LockMinutes time.Duration `bson:"lock_minutes" json:"lock_minutes"` // 锁定时长, 单位: 分钟
}

type TextQuota struct {
	BillingMethod   int     `bson:"billing_method,omitempty"   json:"billing_method,omitempty"`         // 计费方式[1:倍率, 2:固定额度]
	PromptRatio     float64 `bson:"prompt_ratio,omitempty"     json:"prompt_ratio,omitempty"     d:"1"` // 提示倍率(提问倍率)
	CompletionRatio float64 `bson:"completion_ratio,omitempty" json:"completion_ratio,omitempty" d:"1"` // 补全倍率(回答倍率)
	FixedQuota      int     `bson:"fixed_quota"                json:"fixed_quota"`                      // 固定额度
}

type UserLoginRegister struct {
	AccountLogin  bool `bson:"account_login"  json:"account_login"`  // 账密登录
	EmailLogin    bool `bson:"email_login"    json:"email_login"`    // 邮箱登录
	EmailRegister bool `bson:"email_register" json:"email_register"` // 邮箱注册
	EmailRetrieve bool `bson:"email_retrieve" json:"email_retrieve"` // 找回密码
	SessionExpire int  `bson:"session_expire" json:"session_expire"` // 会话过期, 单位: 秒
}

type UserShieldError struct {
	Open   bool     `bson:"open"   json:"open"`   // 开关
	Errors []string `bson:"errors" json:"errors"` // 错误
}