models

type AnthropicConfig struct {
	BaseUrl string `json:"base_url"`
	ApiKey  string `json:"api_key"`
	Version string `json:"version"`
}

type AnthropicCountTokens struct {
	BaseURL string `json:"base_url"`
	APIKey  string `json:"api_key"`
	Version string `json:"version"`
}

type AuthKey struct {
	gorm.Model
	Name       string // 项目名称
	Key        string
	Status     *bool      // 是否启用
	IOLog      *bool      // 是否记录IO
	AllowAll   *bool      // 是否允许所有模型
	Models     []string   `gorm:"serializer:json"` // 允许的模型列表
	ExpiresAt  *time.Time // nil=永不过期，有值=具体过期时间
	UsageCount int64      // 使用次数统计
	LastUsedAt *time.Time // 最后使用时间
}

type Block struct {
	ID uint `gorm:"primarykey"`
	// Digest 是 SHA-256 截断到 128 位的内容标识。唯一索引既是去重的依据，
	// 也是并发写入的安全网：两个事务同时插入同一个块时，后来的那个撞索引，
	// 回查即可，不会写出两份。
	Digest  []byte `gorm:"uniqueIndex;size:16"`
	Size    int
	GroupID uint `gorm:"index"` // 所在的组
	Off     int  // 在组内的偏移
}

type BlockGroup struct {
	ID     uint `gorm:"primarykey"`
	Data   []byte
	RawLen int
}

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

type BodyBytes []byte

type ChatIO struct {
	gorm.Model
	LogId uint
	// Input 是请求体。它是三列里唯一走**块表**的（见 docs/db-compression-phase0.md
	// 与 models/body.go），因为只有它带着 97% 的跨行重复。
	//
	// `type:text` 与 OutputUnion 同理：钉死列类型，AutoMigrate 才不会重建 7 GB 的表。
	// 钉的是 TEXT，但帧按 BLOB 落进去——SQLite 的列亲和性只转换 TEXT↔数字，
	// 不碰 BLOB，所以 `typeof(input)` 仍然分得清明文与帧。
	Input BodyBytes `gorm:"type:text"`
	// SkipCompress 只对**这一次写入**生效：压不动或撞锁时退回明文存储。
	//
	// 它不是数据库列（gorm:"-"），也不参与读路径——读只认帧头。存在的理由是
	// "压缩是优化，IO 记录是观测"：线上并发写把库锁住时，入块表失败被当成整条
	// 记录失败，结果成功的聊天请求被标成 error；撞锁时退回明文重写一次，
	// 代价是多占几百 KB，换的是这条 IO 记录还在、请求状态不撒谎。
	SkipCompress bool `gorm:"-" json:"-"`
	OutputUnion
}

type ChatLog struct {
	ID        uint      `gorm:"primarykey"`
	CreatedAt time.Time `` /* 145-byte string literal not displayed */
	UpdatedAt time.Time
	// DeletedAt 保留这一列以兼容历史库，但**故意不建索引**：chat_logs 的每一条
	// 删除路径都是 Unscoped 硬删除（定时清理、按 id 清理），这一列恒为 NULL。
	// 那条索引命中整张表，却因为"看起来选择性很高"永远排在查询计划器首选，
	// 把按时间窗取数的那类查询从索引范围扫描逼成全表扫 + 临时 B 树排序。
	// 存量库上的旧索引由 DropChatLogsDeletedAtIndex 在启动时摘掉。
	DeletedAt gorm.DeletedAt

	Name          string `gorm:"index;index:idx_chat_logs_name_created,priority:1"`
	TraceID       string `gorm:"index"`
	ProviderModel string `gorm:"index"`
	ProviderName  string `gorm:"index"`
	Status        string `gorm:"index;index:idx_chat_logs_status_created,priority:1"` // error or success
	Style         string // 类型（客户端协议）
	// UpstreamStyle 实际使用的上游协议。直连时与 Style 相同；不同即说明这次转发经过了
	// 一层协议翻译（见 service/bridge.go）。
	UpstreamStyle string `gorm:"index" json:"upstream_style,omitempty"`
	// BridgeNotes 这次协议翻译"改了什么"，逗号分隔的短码（见 bridge.Note）。
	// 只作排查用：丢了哪个参数、补了什么默认值，客户端从响应里看不出来，这里留个线索。
	BridgeNotes string `json:"bridge_notes,omitempty"`
	UserAgent   string `gorm:"index"` // 用户代理
	RemoteIP    string // 访问ip
	AuthKeyID   uint   `gorm:"index;index:idx_chat_logs_key_created,priority:1"` // 使用的AuthKey ID
	SessionID   string `gorm:"index"`                                            // 请求体中的session_id
	ChatIO      bool   // 是否开启IO记录
	// Backup 本次响应由备用关联提供：主关联在本请求内全部失败出局后降级接管。
	// 非指针 bool 是后加列，AutoMigrate 不给存量行填默认值，NULL 由 init.go 回填 false。
	Backup bool

	Error          string        // if status is error, this field will be set
	Retry          int           // 重试次数
	ProxyTime      time.Duration // 代理耗时
	FirstChunkTime time.Duration // 首个chunk耗时
	ChunkTime      time.Duration // chunk耗时
	Tps            float64
	Size           int // 响应大小 字节
	Usage
	InputPrice     float64 `json:"input_price"`
	CacheReadPrice float64 `json:"cache_read_price"`
	OutputPrice    float64 `json:"output_price"`
	Currency       string  `json:"currency"`
	// PeakPeriod 记录本条请求命中的计费时段名，空表示按基础价计费。
	// 只作展示与排查用——价格本身已按实际生效值快照到上面三个字段，
	// 因此成本计算不依赖本字段，改了时段配置也不会影响历史成本。
	PeakPeriod string `json:"peak_period,omitempty"`
}

type Config struct {
	gorm.Model
	Key   string // 配置类型
	Value string // 配置内容
}

type HolidayDay struct {
	Date string `json:"date"`
	Name string `json:"name"`
	// IsOffDay 为 true 表示放假（休息日），false 表示调休上班（工作日）。
	IsOffDay bool `json:"isOffDay"`
}

type LogCleanupPolicy struct {
	Enabled       bool `json:"enabled"`
	RetentionDays int  `json:"retention_days"`
}

type LogCleanupRecord struct {
	gorm.Model
	RetentionDays int    `json:"RetentionDays"`
	DeletedCount  int64  `json:"DeletedCount"`
	DurationMs    int64  `json:"DurationMs"`
	Source        string `json:"Source"`
	Type          string `json:"Type"`
}

type LogCompressBackup struct {
	Path  string `json:"path"`
	Size  int64  `json:"size"`
	MTime string `json:"mtime"`
	At    string `json:"at"`
	// Source 是**探出来的**结论，不是操作员说的：
	//   manual  —— 探到备份，且不比库小
	//   stale   —— 探到备份，但比库小（多半是迁移前的旧备份，盖回去会丢数据）
	//   missing —— 路径上什么都没有
	//   forced  —— 没探到，但操作员明确确认了要继续
	Source string `json:"source"`
}

type LogCompressPolicy struct {
	// Enabled 管的是**后台自动推进**。手动触发（POST .../run）不看它——
	// 与日志清理的语义保持一致。
	Enabled bool `json:"enabled"`
	// BatchRows / BatchBytes 双重封顶。只按行数封顶会让"一批"的字节数随
	// body 大小失控（这份语料平均 470 KiB/行，64 行就是 30 MiB），
	// 而 PutBatch 要求整批原文同时在内存里。
	BatchRows  int   `json:"batch_rows"`
	BatchBytes int64 `json:"batch_bytes"`
	// QuiesceSec 是冷静期：updated_at 比它新的行不碰。它挡的是"正在被写的那一行"
	// ——Create 之后、补响应体的 Updates 之前，那一行是半成品。
	QuiesceSec int `json:"quiesce_sec"`
	// BatchIntervalMs 是批与批之间的**主动停顿**，0 表示不停。
	//
	// 它不改变任何一批的行为，改的是**迁移占用写锁的时间比例**。一批持锁约
	// 185 ms（真机 12,483 行 / 36s / 批 64 行折算），默认 0 时这批接着那批，
	// 迁移期间库有八成时间在写事务里——请求不会报错（busy_timeout 罩得住），
	// 但每次日志 INSERT 都要排队。
	//
	// 所以这是个**占用率旋钮**，不是性能旋钮：调大它迁移更慢、但库更闲。
	// "迁移该跑多凶"取决于这台机器同时在干什么，是运维判断而不是技术判断，
	// 所以它可配。
	BatchIntervalMs int `json:"batch_interval_ms"`
}

type LogCompressState struct {
	// Status: idle / running / paused / done / failed。
	// 启动时读到 running 就是"上次被中断了"，直接续跑（水位在库里）。
	Status string `json:"status"`
	// LastID 是水位：id <= LastID 的行已经扫过。
	LastID uint `json:"last_id"`
	// MaxID 是本轮起始时的 id 快照。迁移期间新产生的行由写路径直接以新形态
	// 落库，不归这一轮管（否则水位会被不断推着往前跑，永远追不上）。
	MaxID     uint  `json:"max_id"`
	TotalRows int64 `json:"total_rows"` // 快照时的候选行数，进度条分母
	Scanned   int64 `json:"scanned"`
	Packed    int64 `json:"packed"`  // 真正改写了形态的行数
	Skipped   int64 `json:"skipped"` // 扫到但无需改动的行数
	// BytesBefore / BytesAfter 只统计**扫过的行**在这一列上的字节数，
	// 所以两者之差恰好是 input 列被省掉的字节——不含组表，组表要另算。
	BytesBefore int64 `json:"bytes_before"`
	BytesAfter  int64 `json:"bytes_after"`
	// BytesTotal 是**全表**原文在这一列上的合计，只在"从水位 0 起跑"那一次量。
	//
	// 它存在的唯一理由是当压缩比的分子。用 BytesBefore 当分子是错的：
	// 那个数只覆盖**扫过的行**，而分母（真正落库）是全表——半程时分子是全表
	// 的一半、分母是全表，算出来的比值恰好是真值的一半，一个看起来很专业的
	// 错误数字。两个数必须盖住同一批行。
	//
	// 它同时是"实时"的来源：从起跑那一刻起分子就固定了，分母随着行改形态
	// 一路缩，比值肉眼可见地往上爬；而 BytesBefore 要等跑完才等于它。
	//
	// 续跑时**保留**不重算：水位之后还有明文行不代表全表都还是明文，
	// 已经压过的行的原文已经不在了，量不回来。所以它的定义是"起跑时那一量"。
	BytesTotal int64  `json:"bytes_total"`
	Attempts   int    `json:"attempts"` // 连续失败次数，成功一批就清零
	LastError  string `json:"last_error"`
	StartedAt  string `json:"started_at"`
	FinishedAt string `json:"finished_at"`
}

type LogReclaimPolicy struct {
	// Enabled 打开后，服务会按 CheckIntervalSec 定期看一眼，够条件就自动跑一趟。
	// 手动点「回收」不看它（与迁移、日志清理的语义一致）。
	Enabled bool `json:"enabled"`
	// MinBytes 是"值得跑一趟"的门槛。没有它，一个刚回收过、只删了几行的库
	// 会在每个周期被拉起来占一次写锁，换回来几 MiB——收益与打扰完全不成比例。
	MinBytes int64 `json:"min_bytes"`
	// CheckIntervalSec 是多久看一次。看一次很便宜（一次 freelist 计数），
	// 所以它管的是"发现有空闲页之后多久动手"，不是性能旋钮。
	CheckIntervalSec int `json:"check_interval_sec"`
}

type LogReclaimState struct {
	// Status: idle / running / done / failed。`done` 里再分跑完没跑完，见 StopReason。
	Status string `json:"status"`
	// Kind 是这一次**做的是什么**：
	//   reclaim —— 增量回收：一条语句放一页，慢慢把洞收掉
	//   vacuum  —— 重整：整库重写一遍，把 freelist 一次还干净
	//   ""      —— 加这个字段之前写下的记录，一律当 reclaim（那时只有那一条路）
	//
	// 与 Source 正交，两者都需要：Source 说"谁发起"，Kind 说"做了什么"。
	// 同样是 startup 发起的，一次转换（VACUUM）和它之后可能接着跑的增量回收
	// 是两件事，界面上要给两套说法；手动那条路上更明显——「立即重整」和
	// 「开始回收」在记录里必须分得开。
	Kind string `json:"kind"`
	// Source 是这一次是谁发起的：
	//   startup   —— 启动期转换/VACUUM（在监听端口之前跑）
	//   manual    —— 用户在控制台点的
	//   scheduled —— 定时回收（见 LogReclaimPolicy）自己起来的
	Source string `json:"source"`
	// Continuous 记这一次是不是"持续到放完"。
	//
	// 它与 Rounds 一起构成回执：`Rounds=1` 的 done/budget 是"点了一下、跑满一段"，
	// 而 `Rounds=37` 才是"一路放到底"。不记这两项，两种完全不同的运维动作
	// 在记录里长得一模一样。
	Continuous bool `json:"continuous"`
	// Rounds 是跑了几轮。单轮模式恒为 1；持续模式下一轮一批（见
	// service.reclaimContinuousPause 的"一批一松手"），所以它同时是
	// "打了几批"的另一种写法。
	Rounds int `json:"rounds"`
	// PageSize 是页大小，用来把页数折成字节（界面上要说"放掉了 5.4 GiB"）。
	PageSize int64 `json:"page_size"`
	// FreedPages / FreedBytes 是这一次放掉的页数与字节数。
	FreedPages int64 `json:"freed_pages"`
	FreedBytes int64 `json:"freed_bytes"`
	// FileSizeBefore / FileSizeAfter 是文件大小的收尾。**回收的意义全在这两个数上**
	// ——freelist 少了不等于文件小了（auto_vacuum=0 时文件一字节都不会缩）。
	FileSizeBefore int64 `json:"file_size_before"`
	FileSizeAfter  int64 `json:"file_size_after"`
	FreelistBefore int64 `json:"freelist_before"`
	FreelistAfter  int64 `json:"freelist_after"`
	// Calls 是这一次打了几条 `PRAGMA incremental_vacuum`。它是**页数的同义词**：
	// 生产驱动忽略参数、每条恰好放一页（见 service.reclaimBatchMax）。
	Calls int `json:"calls"`
	// BatchSize 是这一趟**最后收敛到的**批量（一个事务里连打几条语句）。
	//
	// 它是诊断读数而不是配置：批量由 service.nextBatchSize 按每批实测耗时自己
	// 走，起点与边界见那边的 reclaimBatch*。记下来是因为"这台机器上一批能收到
	// 多少"直接决定"收完这个洞要几轮"，而它随库的形态走，事先算不出来。
	BatchSize  int    `json:"batch_size"`
	DurationMs int64  `json:"duration_ms"`
	StartedAt  string `json:"started_at"`
	FinishedAt string `json:"finished_at"`
	// StopReason 是收尾时的判定。
	// 增量回收（manual / scheduled）：
	//   empty          —— freelist 空了，这一趟把能放的全放了
	//   budget         —— 到点了（本轮时间预算用完），还剩着没放，可以再点一次
	//   stopped        —— 用户按了「停止」（只在持续模式里可能出现）
	//   busy           —— 持续模式的轮间松手之后，维护锁被别的任务抢走了，让给它
	//   no_auto_vacuum —— 库的 auto_vacuum 不是 INCREMENTAL，这句 PRAGMA 是空操作
	//   stalled        —— 放了一批 freelist 却没少（引擎行为反常），主动收工并记 LastError
	// 重整（vacuum，见 Kind）：
	//   vacuumed           —— 整库重写了一遍
	//   converted          —— 顺带把 auto_vacuum 转成了 INCREMENTAL（转换本身就是一次 VACUUM）
	//   insufficient_space —— 磁盘不够，**跳过**（不是失败：启动期服务照常起）
	//
	// 三条对 manual 与 startup 都成立，只有 `insufficient_space` 例外：手动那次
	// 走的是 StartVacuum 的**预检**，磁盘不够时压根不会开始（见 ErrVacuumNoSpace），
	// 不会留下一条"跑了但跳过"的记录。
	//
	// 两者共有：failed —— 出错。
	StopReason string `json:"stop_reason"`
	LastError  string `json:"last_error"`
}

type Model struct {
	gorm.Model
	Name         string
	Remark       string
	MaxRetry     int    // 重试次数限制
	TimeOut      int    // 超时时间 单位秒
	Strategy     string // 负载均衡策略 默认 lottery
	Breaker      *bool  // 是否开启熔断
	DisplayOrder int    // 模型展示顺序，值越大越靠前
	// PreferDirect 优先匹配相同协议：候选池里先挑说客户端协议的提供商，一个都没有
	// 才退回可转换的那些。关掉则整池按权重摇（见 service.poolFor）。
	// nil 表示没配过（迁移上来的老行），按**开**处理——转换功能上线前候选池只有
	// 本协议的，这样老模型的行为不会因为多出一列而改变。
	PreferDirect *bool
}

type ModelAutofillPolicy struct {
	// Enabled 是总开关：关闭后关联弹窗不再自动预填。
	Enabled bool `json:"enabled"`
	// Overwrite 决定预填是否改写已有值。默认 false：只补空。
	//
	// 默认 false 是调研的直接结论：聚合源存在可测缺口（models.dev 的
	// structured_output 只有 55.8%，LiteLLM 的 supports_vision 只覆盖 42%
	// 的对话条目），而覆盖用户**已经确认过**的值风险高于收益——把手工设成
	// true 的 ToolCall 改写成 false，会让这个上游从工具调用请求的候选池里
	// 直接消失（§3.5.2）。
	Overwrite bool `json:"overwrite"`
	// AllowDeprecated 是否给已废弃的模型建议。默认 false。
	// models.dev 的 8,389 个模型里有 261 个 deprecated、71 个 beta。
	AllowDeprecated bool `json:"allow_deprecated"`
	// Sources 是按序尝试的来源，留空用默认 ["models.dev", "litellm"]。
	Sources []string `json:"sources"`
}

type ModelWithProvider struct {
	gorm.Model
	ModelID          uint
	ProviderModel    string
	ProviderID       uint
	ToolCall         *bool // 能否接受带有工具调用的请求
	StructuredOutput *bool // 能否接受带有结构化输出的请求
	Image            *bool // 能否接受带有图片的请求(视觉)
	WithHeader       *bool // 是否透传header
	Status           *bool // 是否启用
	// Backup 是否为备用关联：不参与日常轮询，仅当主关联在本请求内全部不可用后接管。
	// nil（迁移上来的老行）按 false 处理，见 models/init.go 的回填。
	Backup          *bool
	CustomerHeaders map[string]string `gorm:"serializer:json"` // 自定义headers
	ExtraBody       map[string]any    `gorm:"serializer:json"` // 额外请求体参数
	Weight          int
	InputPrice      *float64
	CacheReadPrice  *float64
	OutputPrice     *float64
	Currency        string
	// Peak 这一条关联的峰谷计费条款，nil 表示没配（按基础价计费）。
	//
	// 与三档单价同处一行是刻意的：乘数就是乘在它们上面的，
	// 拆到别处就会出现"价格改了、条款还指着旧的那套"的可能。
	Peak *PeakTerms `gorm:"serializer:json"`
}

type OutputUnion struct {
	OfString      string   `gorm:"type:text;serializer:compress"`
	OfStringArray []string `gorm:"type:text;serializer:compress"`
}

type PeakCalendar struct {
	// Timezone 用于判定"此刻是本地的几点/星期几"。为空时按 server 本地时区。
	// 必须显式配置的原因：上游的峰谷窗口按供应商所在时区定义，
	// 而部署机可能是任意时区，二者不一致会让时段整体错位。
	Timezone string `json:"timezone"`
	// Weekdays 工作日定义（0=周日 … 6=周六）。为空时默认周一至周五。
	Weekdays []int `json:"weekdays,omitempty"`
	// DateOverrides 按日期覆盖工作日判定，键为 "2006-01-02"，值为 "work" 或 "rest"。
	//
	// 这一项是法定节假日与调休的落点：中国的调休制度会让某个周六成为工作日，
	// 单靠星期几无法表达，必须能按具体日期覆盖。
	//
	// 刻意不加 omitempty：空 map 也要落库为 {}，否则配置文件里看不到这个字段，
	// 手工编辑时不知道有它可填。
	DateOverrides map[string]string `json:"dateOverrides"`
	// HolidaySyncedAt 最近一次同步节假日数据的 unix 秒，0 表示从未同步。
	HolidaySyncedAt int64 `json:"holidaySyncedAt,omitempty"`
	// HolidaySource 最近一次同步的数据来源，供界面展示来源与可信度。
	HolidaySource string `json:"holidaySource,omitempty"`
}

type PeakPeriod struct {
	Name string `json:"name"`
	// Start/End 为一天中的 "HH:MM"，允许 "24:00"。Start > End 表示跨零点。
	Start string `json:"start"`
	End   string `json:"end"`
	// Multiplier 为相对基础价的乘数，1 表示不加价不减价。
	Multiplier float64 `json:"multiplier"`
	// Days 限定星期几（0=周日 … 6=周六）。为空表示不限制。
	Days []int `json:"days,omitempty"`
	// Workday 限定是否只在工作日/休息日生效。nil 表示不限制。
	Workday *bool `json:"workday,omitempty"`
}

type PeakTerms struct {
	Enabled bool `json:"enabled"`
	// Periods 时段列表，顺序即优先级。
	Periods []PeakPeriod `json:"periods"`
}

type PromptTokensDetails struct {
	CachedTokens int64 `json:"cached_tokens"`
	AudioTokens  int64 `json:"audio_tokens"`
}

type Provider struct {
	gorm.Model
	Name         string
	Type         string
	Config       string
	Console      string // 控制台地址
	Proxy        string // HTTP 代理地址
	ErrorMatcher string // 响应体错误识别规则，多行或分号分隔 sample
}

type ReqMeta struct {
	UserAgent string // 用户代理
	RemoteIP  string // 访问ip
	Header    http.Header
}

type SelfHealPolicy struct {
	Enabled bool `json:"enabled"`
	// MaxDrops 是一家上游最多删几个参数（多发自几次请求）。<=0 视为未配置，
	// 消费侧归一到 service.SelfHealMaxDrops。
	MaxDrops int `json:"max_drops"`
}

type Usage struct {
	PromptTokens        int64               `json:"prompt_tokens"`
	CompletionTokens    int64               `json:"completion_tokens"`
	TotalTokens         int64               `json:"total_tokens"`
	PromptTokensDetails PromptTokensDetails `json:"prompt_tokens_details" gorm:"serializer:json"`
}