lanyuanxiaoyao
395887667d
实现统一模型 ID 格式 (provider_id/model_name)，支持跨协议模型标识和 Smart Passthrough。 核心变更： - 新增 pkg/modelid 包：解析、格式化、校验统一模型 ID - 数据库迁移：models 表使用 UUID 主键 + UNIQUE(provider_id, model_name) 约束 - Repository 层：FindByProviderAndModelName、ListEnabled 方法 - Service 层：联合唯一校验、provider ID 字符集校验 - Conversion 层：ExtractModelName、RewriteRequestModelName/RewriteResponseModelName 方法 - Handler 层：统一模型 ID 路由、Smart Passthrough、Models API 本地聚合 - 新增 error-responses、unified-model-id 规范 测试覆盖： - 单元测试：modelid、conversion、handler、service、repository - 集成测试：统一模型 ID 路由、Smart Passthrough 保真性、跨协议转换 - 迁移测试：UUID 主键、UNIQUE 约束、级联删除 OpenSpec： - 归档 unified-model-id 变更到 archive/2026-04-21-unified-model-id - 同步 11 个 delta specs 到 main specs - 新增 error-responses、unified-model-id 规范文件
59 lines
1.9 KiB
Go
59 lines
1.9 KiB
Go
package config
|
||
|
||
import (
|
||
"time"
|
||
)
|
||
|
||
// Provider 供应商模型
|
||
type Provider struct {
|
||
ID string `gorm:"primaryKey" json:"id"`
|
||
Name string `gorm:"not null" json:"name"`
|
||
APIKey string `gorm:"not null" json:"api_key"`
|
||
BaseURL string `gorm:"not null" json:"base_url"`
|
||
Protocol string `gorm:"column:protocol;default:'openai'" json:"protocol"`
|
||
Enabled bool `gorm:"default:true" json:"enabled"`
|
||
CreatedAt time.Time `json:"created_at"`
|
||
UpdatedAt time.Time `json:"updated_at"`
|
||
Models []Model `gorm:"foreignKey:ProviderID;constraint:OnDelete:CASCADE" json:"models,omitempty"`
|
||
}
|
||
|
||
// Model 模型配置（id 为 UUID 自动生成，UNIQUE(provider_id, model_name)）
|
||
type Model struct {
|
||
ID string `gorm:"primaryKey" json:"id"`
|
||
ProviderID string `gorm:"not null;index;uniqueIndex:idx_provider_model" json:"provider_id"`
|
||
ModelName string `gorm:"not null;index;uniqueIndex:idx_provider_model" json:"model_name"`
|
||
Enabled bool `gorm:"default:true" json:"enabled"`
|
||
CreatedAt time.Time `json:"created_at"`
|
||
}
|
||
|
||
// UsageStats 用量统计
|
||
type UsageStats struct {
|
||
ID uint `gorm:"primaryKey;autoIncrement" json:"id"`
|
||
ProviderID string `gorm:"not null;index" json:"provider_id"`
|
||
ModelName string `gorm:"not null;index" json:"model_name"`
|
||
RequestCount int `gorm:"default:0" json:"request_count"`
|
||
Date time.Time `gorm:"type:date;not null;uniqueIndex:idx_provider_model_date" json:"date"`
|
||
}
|
||
|
||
// TableName 指定表名
|
||
func (Provider) TableName() string {
|
||
return "providers"
|
||
}
|
||
|
||
func (Model) TableName() string {
|
||
return "models"
|
||
}
|
||
|
||
func (UsageStats) TableName() string {
|
||
return "usage_stats"
|
||
}
|
||
|
||
// MaskAPIKey 掩码 API Key（仅显示最后 4 个字符）
|
||
func (p *Provider) MaskAPIKey() {
|
||
if len(p.APIKey) > 4 {
|
||
p.APIKey = "***" + p.APIKey[len(p.APIKey)-4:]
|
||
} else {
|
||
p.APIKey = "***"
|
||
}
|
||
}
|